require "sudoku.sudoku"
require "ui.logic"

local function iterBoardLine(start, step)
    local step = step or 1
    local i = start
    local n = 0
    return function()
        i = i + step
        n = n + 1
        if n > 9 then return end
        if i > 9 then i = 1 end
        if i < 1 then i = 9 end
        return i
    end
end

function handleRight(shift)
    if shift then
        for x in iterBoardLine(cursor.x) do
            if uiBoard[1][cursor.y][x] == 0 then
                cursor.x = x
                break
            end
        end
    else
        cursor.x = cursor.x + 1
        if cursor.x > 9 then cursor.x = 1 end
    end
end

function handleLeft(shift)
    if shift then
        for x in iterBoardLine(cursor.x, -1) do
            if uiBoard[1][cursor.y][x] == 0 then
                cursor.x = x
                break
            end
        end
    else
        cursor.x = cursor.x - 1
        if cursor.x < 1 then cursor.x = 9 end
    end
end

function handleDown(shift)
    if shift then
        for y in iterBoardLine(cursor.y) do
            if uiBoard[1][y][cursor.x] == 0 then
                cursor.y = y
                break
            end
        end
    else
        cursor.y = cursor.y + 1
        if cursor.y > 9 then cursor.y = 1 end
    end
end

function handleUp(shift)
    if shift then
        for y in iterBoardLine(cursor.y, -1) do
            if uiBoard[1][y][cursor.x] == 0 then
                cursor.y = y
                break
            end
        end
    else
        cursor.y = cursor.y - 1
        if cursor.y < 1 then cursor.y = 9 end
    end
end

function love.keypressed(key, scancode)
    if inputBox.focused then
        if key == "backspace" then
            inputBox:delete()
        end
        return 
    end
    local shift = love.keyboard.isDown('rshift') or love.keyboard.isDown('lshift')
    if key == "escape" or key == "q" then
        love.event.quit()
    elseif key == "right" or key == "l" then
        handleRight(shift)
    elseif key == "left" or key == "h" then
        handleLeft(shift)
    elseif key == "down" or key == "j" then
        handleDown(shift)
    elseif key == "up" or key == "k" then
        handleUp(shift)
    elseif key == "s" then
        displaySearchSpace()
    elseif key == "c" then
        checkBoard()
    elseif key == "f2" then
        saveCurrentBoard()
    elseif key == "f3" then
        loadLastBoard()
    elseif uiBoard[1][cursor.y][cursor.x] == 0 then
        local n = tonumber(key) or -1
        if n > 0 and n < 10 then
            if shift or cursor.editMode then
                uiBoard[2][cursor.y][cursor.x] = 0
                cursor.editMode = true
                smallNumbersVal[cursor.x][cursor.y][n] = not smallNumbersVal[cursor.x][cursor.y][n]
            else
                -- for i = 1, 9 do
                --     smallNumbersVal[cursor.x][cursor.y][i] = false
                -- end
                uiBoard[2][cursor.y][cursor.x] = n
                uiBoard[3][cursor.y][cursor.x] = n
            end
        elseif key == "delete" or key == "0" then
            if uiBoard[2][cursor.y][cursor.x] ~= 0 then
                uiBoard[2][cursor.y][cursor.x] = 0
            else
                if uiBoard[1][cursor.y][cursor.x] == 0 then
                    for i = 1, 9 do
                        smallNumbersVal[cursor.x][cursor.y][i] = false
                    end
                end
            end
        elseif key == "tab" then
            if cursor.editMode == false then
                uiBoard[3][cursor.y][cursor.x] = uiBoard[2][cursor.y][cursor.x]
                uiBoard[2][cursor.y][cursor.x] = 0
                cursor.editMode = true
            else
                uiBoard[2][cursor.y][cursor.x] = uiBoard[3][cursor.y][cursor.x]
                cursor.editMode = false
            end
        elseif key == "e" then
            inputBox.enabled = true
            inputBox.focused = true
        end
    end
end