-- https://github.com/mfussenegger/nvim-dap

local M = {}

M.env_ok = false

function M.check_env()
    local utils = require("utils")

    utils.assert_installed("python3")
    utils.assert_python3_module_installed("debugpy")
    M.env_ok = true
end

function M.start(config)
    local dap = require("dap")

    if not M.env_ok then
        M:check_env()
    end

    dap.adapters.python = {
        type = "executable",
        command = "python",
        args = { "-m", "debugpy.adapter", },
        cwd = vim.fn.getcwd(),
    }
    dap.run(config)
    -- List of events described at https://microsoft.github.io/debug-adapter-protocol/specification#Events
    -- Also see :h dap-extensions
    dap.listeners.after["event_initialized"]["nvim-dap.lua"] = function ()
        dap.set_exception_breakpoints({ "userUnhandled", })
    end
end

function M.launch(args)
    assert(type(args) == "table", "Args not specified or of wrong type")
    local config = {
        name = "Launch file",
        type = "python",
        request = "launch",
        program = "${file}",
        -- python = 'python';
        -- program = vim.fn.getcwd() .. '/.venv/bin/pytest';
        console = "integratedTerminal",
        args = args,
    }
    M.start(config)
end

function M.pytest(args)
    assert(type(args) == "table", "Args not specified or of wrong type")
    local config = {
        name = "pytest " .. table.concat(args, " "),
        type = "python",
        request = "launch",

        -- pythonPath = vim.fn.getcwd() .. '/.venv/bin/python',
        module = "pytest",
        -- python = 'python';
        -- program = vim.fn.getcwd() .. '/.venv/bin/pytest';
        args = args,
        console = "integratedTerminal",
        -- program = "${file}";
    }
    M.start(config)
end

function M.setup()
    local dap = require("dap")

    vim.keymap.set("n", "<F5>", dap.continue)
    vim.keymap.set("n", "<F10>", dap.step_over)
    vim.keymap.set("n", "<F11>", dap.step_into)
    vim.keymap.set("n", "<F12>", dap.step_out)

    --[[
        TODO: Add this after loading dap for integrating catppuccin:
        local sign = vim.fn.sign_define

        sign("DapBreakpoint", { text = "●", texthl = "DapBreakpoint", linehl = "", numhl = ""})
        sign("DapBreakpointCondition", { text = "●", texthl = "DapBreakpointCondition", linehl = "", numhl = ""})
        sign("DapLogPoint", { text = "◆", texthl = "DapLogPoint", linehl = "", numhl = ""})
    --]]
end

return M