local workspace_map = {
    "Home",
    "Dev",
    "Art",
    "Gaming",
    "Writing",
}

local monitor_binds = {}

local function setup_monitors()
    -- Clear old binds
    for _, bind in ipairs(monitor_binds) do
        hl.unbind(bind)
    end
    monitor_binds = {}

    local monitors = hl.get_monitors()
    local monitor_count = #monitors

    -- Move active window to a workspace with SUPER + SHIFT + [1-5]
    for i = 1, 5 do
        for j = monitor_count, 1, -1 do
            local monitor = monitors[j]
            -- local workspace_id = "name:" .. workspace_map[i] .. "_" .. j
            local workspace_id = (i - 1) * monitor_count + j
            local workspace_name = workspace_map[i] .. "_" .. j
            hl.workspace_rule({
                workspace = workspace_id,
                default_name = workspace_name,
                monitor = monitor.name,
                default = i == 1,
            })

            table.insert(
                monitor_binds,
                hl.bind(
                    "SUPER + code:1" .. (i - 1),
                    hl.dsp.focus({ workspace = "name:" .. workspace_name }),
                    { description = "Switch to workspace " .. workspace_map[i] }
                )
            )
        end

        table.insert(
            monitor_binds,
            hl.bind("SUPER + SHIFT + code:1" .. (i - 1), function()
                local active_workspace = hl.get_active_workspace()
                local active_workspace_name = active_workspace.name
                local monitor_suffix = active_workspace_name:match("_(%d+)$") or "1"
                local new_workspace = "name:" .. workspace_map[i] .. "_" .. monitor_suffix
                -- local new_workspace = (i - 1) * monitor_count + active_workspace.id
                hl.notification.create({ text = new_workspace, duration = 2000 })
                hl.dispatch(hl.dsp.window.move({ workspace = new_workspace, follow = false }))
            end, { description = "Move window to workspace " .. workspace_map[i] })
        )
    end
end

hl.on("monitor.added", setup_monitors)
hl.on("monitor.removed", setup_monitors)
hl.on("config.reloaded", setup_monitors)