--[[

     Licensed under GNU General Public License v2
      * (c) 2022,      Sean Corbett
      * (c) 2018,      Eugene Pakhomov
      * (c) 2016,      Henrik Antonsson
      * (c) 2015,      Joerg Jaspert
      * (c) 2014,      projektile
      * (c) 2013,      Luca CPZ
      * (c) 2010-2012, Peter Hofmann

--]]

local floor, max, min, mouse, mousegrabber, screen = math.floor, math.max, math.min, mouse, mousegrabber, screen

local multicenterwork = {
    name       = "multicenterwork",
    horizontal = { name = "multicenterworkh" }
}

local function arrange(p, layout)
    local t   = p.tag or screen[p.screen].selected_tag
    local wa  = p.workarea
    local cls = p.clients

    if #cls == 0 then return end

    -- Main column, fixed width and height
    local mwfact          = t.master_width_factor
    local nmast           = t.master_count
    local mainhei         = floor(wa.height * mwfact)
    local mainwid         = floor(wa.width * mwfact)
    local maxmasthei      = floor(wa.height * 0.7)
    local maxmastwid      = floor(wa.width * 0.6)
    local maxslahei      = floor(wa.height * 0.5)
    local maxslawid      = floor(wa.width * 0.5)
    local slavewid        = wa.width - mainwid
    local slaveLwid       = floor(slavewid / 2)
    local slaveRwid       = slavewid - slaveLwid
    local slavehei        = wa.height - mainhei
    local slaveThei       = floor(slavehei / 2)
    local slaveBhei       = slavehei - slaveThei
    local nbrFirstSlaves  = floor((#cls-nmast+1) / 2)
    local nbrSecondSlaves = floor((#cls-nmast) / 2)

    local slaveFirstDim, slaveSecondDim = 0, 0

    if layout.name == "multicenterwork" then -- vertical
        if nbrFirstSlaves  > 0 then slaveFirstDim  = floor(wa.height / nbrFirstSlaves) end
        if nbrSecondSlaves > 0 then slaveSecondDim = floor(wa.height / nbrSecondSlaves) end

        if nmast > 0 then
            for i = 1, nmast do
                local c, g = cls[i], {}

                g.height = min(floor(wa.height / nmast), maxmasthei)
                g.width  = mainwid

                g.x = wa.x + slaveLwid
                g.y = wa.y + (i - 1) * floor(wa.height / nmast) + floor((wa.height / nmast - g.height) / 2)

                g.width  = max(g.width, 1)
                g.height = max(g.height, 1)

                p.geometries[c] = g
            end
        end

    else -- horizontal
        if nbrFirstSlaves  > 0 then slaveFirstDim  = floor(wa.width / nbrFirstSlaves) end
        if nbrSecondSlaves > 0 then slaveSecondDim = floor(wa.width / nbrSecondSlaves) end

        if nmast > 0 then
            for i = 1, nmast do
                local c, g = cls[i], {}

                g.height  = mainhei
                g.width = min(floor(wa.width / nmast), maxmastwid)

                g.x = wa.x + (i - 1) * floor(wa.width / nmast) + floor((wa.width / nmast - g.width) / 2)
                g.y = wa.y + slaveThei

                g.width  = max(g.width, 1)
                g.height = max(g.height, 1)

                p.geometries[c] = g
            end
        end

    end

    -- Auxiliary clients
    if #cls <= nmast then return end
    for i = nmast + 1, #cls do
        local c, g = cls[i], {}
        local idxChecker, dimToAssign
        local sli = i - nmast

        local rowIndex = floor((sli+1)/2)

        if layout.name == "multicenterwork" then -- horizontal
            if sli % 2 == 1 then -- left slave
                g.x     = wa.x
                g.y     = wa.y + (rowIndex - 1) * slaveFirstDim
                g.width = slaveLwid

                idxChecker, dimToAssign = nbrFirstSlaves, slaveFirstDim
            else -- right slave
                g.x     = wa.x + slaveLwid + mainwid
                g.y     = wa.y + (rowIndex - 1) * slaveSecondDim
                g.width = slaveRwid

                idxChecker, dimToAssign = nbrSecondSlaves, slaveSecondDim
            end

            -- if last slave in row, use remaining space for it
            if rowIndex == idxChecker then
                g.height = wa.y + wa.height - g.y
            else
                g.height = dimToAssign
            end

            -- center-up oversize clients
            if g.height > maxslahei then
                g.y = g.y + (g.height - maxslahei) / 2
                g.height = maxslahei
            end

        else -- vertical
            if sli % 2 == 1 then -- top slave
                g.x      = wa.x + (rowIndex - 1) * slaveFirstDim
                g.y      = wa.y
                g.height = slaveThei

                idxChecker, dimToAssign = nbrFirstSlaves, slaveFirstDim
            else -- bottom slave
                g.x      = wa.x + (rowIndex - 1) * slaveSecondDim
                g.y      = wa.y + slaveThei + mainhei
                g.height = slaveBhei

                idxChecker, dimToAssign = nbrSecondSlaves, slaveSecondDim
            end

            -- if last slave in row, use remaining space for it
            if rowIndex == idxChecker then
                g.width = wa.x + wa.width - g.x
            else
                g.width = dimToAssign
            end

            -- center-up oversize clients
            if g.width > maxslawid then
                g.x = g.x + (g.width - maxslawid) / 2
                g.width = maxslawid
            end
        end

        g.width  = max(g.width, 1)
        g.height = max(g.height, 1)

        p.geometries[c] = g
    end
end

local function mouse_resize_handler(c, corner, x, y, orientation)
    local wa     = c.screen.workarea
    local mwfact = c.screen.selected_tag.master_width_factor
    local g      = c:geometry()
    local offset = 0
    local cursor = "cross"

    local corner_coords

    if orientation == 'vertical' then
        if g.height + 15 >= wa.height then
            offset = g.height * .5
            cursor = "sb_h_double_arrow"
        elseif not (g.y + g.height + 15 > wa.y + wa.height) then
            offset = g.height
        end
        corner_coords = { x = wa.x + wa.width * (1 - mwfact) / 2, y = g.y + offset }
    else
        if g.width + 15 >= wa.width then
            offset = g.width * .5
            cursor = "sb_v_double_arrow"
        elseif not (g.x + g.width + 15 > wa.x + wa.width) then
            offset = g.width
        end
        corner_coords = { y = wa.y + wa.height * (1 - mwfact) / 2, x = g.x + offset }
    end

    mouse.coords(corner_coords)

    local prev_coords = {}

    mousegrabber.run(function(_mouse)
        if not c.valid then return false end
        for _, v in ipairs(_mouse.buttons) do
            if v then
                prev_coords = { x = _mouse.x, y = _mouse.y }
                local new_mwfact
                if orientation == 'vertical' then
                    new_mwfact = 1 - (_mouse.x - wa.x) / wa.width * 2
                else
                    new_mwfact = 1 - (_mouse.y - wa.y) / wa.height * 2
                end
                c.screen.selected_tag.master_width_factor = math.min(math.max(new_mwfact, 0.01), 0.99)
                return true
            end
        end
        return prev_coords.x == _mouse.x and prev_coords.y == _mouse.y
    end, cursor)
end

function multicenterwork.arrange(p)
    return arrange(p, multicenterwork)
end

function multicenterwork.horizontal.arrange(p)
    return arrange(p, multicenterwork.horizontal)
end

function multicenterwork.mouse_resize_handler(c, corner, x, y)
    return mouse_resize_handler(c, corner, x, y, 'vertical')
end

function multicenterwork.horizontal.mouse_resize_handler(c, corner, x, y)
    return mouse_resize_handler(c, corner, x, y, 'horizontal')
end

return multicenterwork