gitlab.nvim/lua/gitlab/init.lua

local state                   = require("gitlab.state")
local discussions             = require("gitlab.discussions")
local summary                 = require("gitlab.summary")
local assignees_and_reviewers = require("gitlab.assignees_and_reviewers")
local keymaps                 = require("gitlab.keymaps")
local comment                 = require("gitlab.comment")
local job                     = require("gitlab.job")
local u                       = require("gitlab.utils")

local M                       = {}
M.args                        = nil

-- Builds the binary (if not built) and sets the plugin arguments
M.setup                       = function(args)
  if args == nil then args = {} end
  local file_path = u.current_file_path()
  local parent_dir = vim.fn.fnamemodify(file_path, ":h:h:h:h")
  state.BIN_PATH = parent_dir
  state.BIN = parent_dir .. "/bin"

  local binary_exists = vim.loop.fs_stat(state.BIN)
  if binary_exists == nil then M.build() end

  if not M.setPluginConfiguration(args) then return end -- Return if not a valid gitlab project
  M.args = args                                         -- The  ensureState function won't start without args
end

-- Function names prefixed with "ensure" will ensure the plugin's state
-- is initialized prior to running other calls. These functions run
-- API calls if the state isn't initialized, which will set state containing
-- information that's necessary for other API calls, like description,
-- author, reviewer, etc.

-- This plugin will start the Go server and will call the "info"
-- endpoint and set the state of the MR.
M.go_server_running           = false
M.ensureState                 = function(callback)
  return function()
    if not M.args then
      vim.notify("The gitlab.nvim state was not set. Do you have a .gitlab.nvim file configured?", vim.log.levels.ERROR)
      return
    end

    if M.go_server_running then
      callback()
      return
    end

    local command = state.BIN
        .. " "
        .. state.PROJECT_ID
        .. " "
        .. state.GITLAB_URL
        .. " "
        .. state.PORT
        .. " "
        .. state.AUTH_TOKEN
        .. " "
        .. state.LOG_PATH

    vim.fn.jobstart(command, {
      on_stdout = function(job_id)
        if job_id <= 0 then
          vim.notify("Could not start gitlab.nvim binary", vim.log.levels.ERROR)
          return
        else
          -- Once the Go binary has go_server_running, call the info endpoint to set global state
          keymaps.set_keymap_keys(M.args.keymaps)
          M.go_server_running = true
          job.run_job("info", "GET", nil, function(data)
            state.INFO = data.info
            callback()
          end)
        end
      end,
      on_stderr = function(_, errors)
        local err_msg = ''
        for _, err in ipairs(errors) do
          if err ~= "" and err ~= nil then
            err_msg = err_msg .. err .. "\n"
          end
        end
        vim.notify(err_msg, vim.log.levels.ERROR)
      end
    })
  end
end

M.ensureProjectMembers        = function(callback)
  return function()
    if type(state.PROJECT_MEMBERS) ~= "table" then
      job.run_job("members", "GET", nil, function(data)
        state.PROJECT_MEMBERS = data.ProjectMembers
        callback()
      end)
    else
      callback()
    end
  end
end

-- Builds the Go binary
M.build                       = function()
  local command = string.format("cd %s && make", state.BIN_PATH)
  local installCode = os.execute(command .. "> /dev/null")
  if installCode ~= 0 then
    vim.notify("Could not install gitlab.nvim!", vim.log.levels.ERROR)
    return false
  end
  return true
end

-- Initializes state for the project based on the arguments
-- provided in the `.gitlab.nvim` file per project, and the args provided in the setup function
M.setPluginConfiguration      = function(args)
  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 false
  end

  local file = assert(io.open(config_file_path, "r"))
  local property = {}
  for line in file:lines() do
    for key, value in string.gmatch(line, "(.-)=(.-)$") do
      property[key] = value
    end
  end

  local project_id = property["project_id"]
  local gitlab_url = property["gitlab_url"]
  local auth_token = property["auth_token"]

  state.PROJECT_ID = project_id
  state.AUTH_TOKEN = auth_token or os.getenv("GITLAB_TOKEN")
  state.GITLAB_URL = gitlab_url or "https://gitlab.com"

  if state.AUTH_TOKEN == nil then
    error("Missing authentication token for Gitlab")
  end

  if state.AUTH_TOKEN == nil then
    error("Missing authentication token for Gitlab")
  end

  if state.PROJECT_ID == nil then
    error("Missing project ID in .gitlab.nvim file.")
  end

  if type(tonumber(state.PROJECT_ID)) ~= "number" then
    error("The .gitlab.nvim project file's 'project_id' must be number")
  end

  -- Configuration for the plugin, such as port of server, layout, etc
  state.PORT = args.port or 21036
  state.LOG_PATH = args.log_path or (vim.fn.stdpath("cache") .. "/gitlab.nvim.log")
  state.DISCUSSION = {
    SPLIT = {
      relative = args.keymaps and args.keymaps.discussion_tree and args.keymaps.discussion_tree.relative or "editor",
      position = args.keymaps and args.keymaps.discussion_tree and args.keymaps.discussion_tree.position or "left",
      size = args.keymaps and args.keymaps.discussion_tree and args.keymaps.discussion_tree.size or "20%",
    }
  }

  state.SYMBOLS = {
    resolved = (args.symbols and args.symbols.resolved or '✓'),
    unresolved = (args.symbols and args.symbols.unresolved or '')
  }

  return true
end

-- Root Module Scope
M.summary                     = M.ensureState(summary.summary)
M.approve                     = M.ensureState(job.approve)
M.revoke                      = M.ensureState(job.revoke)
M.list_discussions            = M.ensureState(discussions.list_discussions)
M.create_comment              = M.ensureState(comment.create_comment)
M.edit_comment                = M.ensureState(comment.edit_comment)
M.delete_comment              = M.ensureState(comment.delete_comment)
M.toggle_resolved             = M.ensureState(comment.toggle_resolved)
M.reply                       = M.ensureState(discussions.reply)
M.add_reviewer                = M.ensureProjectMembers(M.ensureState(assignees_and_reviewers.add_reviewer))
M.delete_reviewer             = M.ensureProjectMembers(M.ensureState(assignees_and_reviewers.delete_reviewer))
M.add_assignee                = M.ensureProjectMembers(M.ensureState(assignees_and_reviewers.add_assignee))
M.delete_assignee             = M.ensureProjectMembers(M.ensureState(assignees_and_reviewers.delete_assignee))
M.state                       = state

return M