Amethyst Custom Layout Simple Square Grid

grid.js

function layout() {
    return {
        name: "Grid",
        getFrameAssignments: (windows, screenFrame) => {
            const numWindows = windows.length;
            const numRows = Math.floor(Math.sqrt(numWindows));
            const numCols = Math.ceil(Math.sqrt(numWindows));

            const cellWidth = screenFrame.width / numCols;
            const cellHeight = screenFrame.height / numRows;

            const frames = windows.map((window, index) => {
                const row = Math.floor(index / numCols);
                const col = index % numCols;
                const frame = {
                    x: screenFrame.x + (cellWidth * col),
                    y: screenFrame.y + (cellHeight * row),
                    width: cellWidth,
                    height: cellHeight
                };
                return { [window.id]: frame };
            });

            return frames.reduce((frames, frame) => ({ ...frames, ...frame }), {});
        }
    };
}