-- This module is responsible for the assignment of reviewers -- and assignees in Gitlab, those who must review an MR. local u = require("gitlab.utils") local job = require("gitlab.job") local List = require("gitlab.utils.list") local state = require("gitlab.state") local M = {} M.add_assignee = function() M.add_popup("assignee") end M.delete_assignee = function() M.delete_popup("assignee") end M.add_reviewer = function() M.add_popup("reviewer") end M.delete_reviewer = function() M.delete_popup("reviewer") end M.add_popup = function(type) local plural = type .. "s" local current = state.INFO[plural] local eligible = M.filter_eligible(state.PROJECT_MEMBERS, current) vim.ui.select(eligible, { prompt = "Choose " .. type .. " to add", format_item = function(user) return user.username .. " (" .. user.name .. ")" end, }, function(choice) if not choice then return end local current_ids = u.extract(current, "id") table.insert(current_ids, choice.id) local body = { ids = current_ids } job.run_job("/mr/" .. type, "PUT", body, function(data) u.notify(data.message, vim.log.levels.INFO) state.INFO[plural] = data[plural] end) end) end M.delete_popup = function(type) local plural = type .. "s" local current = state.INFO[plural] vim.ui.select(current, { prompt = "Choose " .. type .. " to delete", format_item = function(user) return user.username .. " (" .. user.name .. ")" end, }, function(choice) if not choice then return end local ids = u.extract(M.filter_eligible(current, { choice }), "id") local body = { ids = ids } job.run_job("/mr/" .. type, "PUT", body, function(data) u.notify(data.message, vim.log.levels.INFO) state.INFO[plural] = data[plural] end) end) end M.filter_eligible = function(current, to_remove) local ids = u.extract(to_remove, "id") return List.new(current):filter(function(member) if not u.contains(ids, member.id) then return true end end) end return M