local curl = require("plenary.curl") local state = require("gitlab.state") local notify = require("notify") local discussions = require("gitlab.discussions") local summary = require("gitlab.summary") local keymaps = require("gitlab.keymaps") local comment = require("gitlab.comment") local job = require("gitlab.job") local u = require("gitlab.utils") -- Root Module Scope local M = {} M.summary = summary.summary M.approve = job.approve M.revoke = job.revoke M.create_comment = comment.create_comment M.list_discussions = discussions.list_discussions M.edit_comment = comment.edit_comment M.delete_comment = comment.delete_comment M.reply = discussions.reply -- Builds the Go binary, initializes the plugin, fetches MR info M.build = function(args) if args == nil then args = {} end local command = string.format("cd %s && make", state.BIN_PATH) local installCode = os.execute(command .. "> /dev/null") if installCode ~= 0 then notify("Could not install gitlab.nvim!", "error") return else M.setup(args, true) end end M.setup = function(args, build_only) local file_path = M.current_file_path() local parent_dir = vim.fn.fnamemodify(file_path, ":h:h:h") state.BIN_PATH = parent_dir state.BIN = parent_dir .. "/bin" if args == nil then args = {} end if args.dev == true then M.build(args) end local binExists = io.open(state.BIN, "r") if not binExists or args.dev == true then local command = string.format("cd %s && make", state.BIN_PATH) local installCode = os.execute(command .. "> /dev/null") if installCode ~= 0 then require("notify")("Could not install gitlab.nvim! Do you have Go installed?", "error") return end end local binary_exists = vim.loop.fs_stat(state.BIN) if binary_exists == nil then return -- Ensure build function completes before initializing plugin end if build_only then return end local config_file_path = vim.fn.getcwd() .. "/.gitlab.nvim" local config_file_content = u.read_file(config_file_path) if config_file_content == nil then return end args.project_id = config_file_content if args.project_id == nil then args.project_id = u.read_file(state.BIN_PATH .. "/.gitlab/project_id") error("No project ID provided!") end state.PROJECT_ID = args.project_id if args.base_branch ~= nil then state.BASE_BRANCH = args.base_branch end local error_message = "Failed to set up gitlab.nvim, could not get project information." if u.is_gitlab_repo() then state.PORT = args.port or 21036 vim.fn.jobstart(state.BIN .. " " .. state.PROJECT_ID .. " " .. state.PORT, { on_stdout = function(job_id) if job_id <= 0 then notify(error_message, "error") return else local response_ok, response = pcall(curl.get, "localhost:" .. state.PORT .. "/info", { timeout = 750 }) if response == nil or not response_ok then notify(error_message, "error") return end local body = response.body local parsed_ok, data = pcall(vim.json.decode, body) if parsed_ok ~= true then notify(error_message, "error") return end state.INFO = data keymaps.set_keymap_keys(args.keymaps) keymaps.set_keymaps() end end }) end end M.current_file_path = function() local path = debug.getinfo(1, 'S').source:sub(2) return vim.fn.fnamemodify(path, ':p') end return M