-- This module is responsible for the MR pipline
-- This lets the user see the current status of the pipeline
-- and retrigger the pipeline from within the editor
local Popup    = require("nui.popup")
local state    = require("gitlab.state")
local job      = require("gitlab.job")
local u        = require("gitlab.utils")
local M        = {}

-- The function will render the Pipeline state in a popup
M.open         = function()
  local pipeline = state.INFO.pipeline

  if type(pipeline) ~= "table" or (type(pipeline) == "table" and u.table_size(pipeline) == 0) then
    vim.notify("Pipeline not found", vim.log.levels.WARN)
    return
  end

  local body = { pipeline_id = state.INFO.pipeline.id }
  job.run_job("/pipeline", "GET", body, function(data)
    local pipeline_jobs = u.reverse(type(data.Jobs) == "table" and data.Jobs or {})

    local width = string.len(pipeline.web_url) + 10
    local height = 6 + #pipeline_jobs + 3

    local pipeline_popup = Popup(u.create_popup_state("Loading Pipeline...", width, height))
    pipeline_popup:mount()

    local bufnr = vim.api.nvim_get_current_buf()
    vim.opt_local.wrap = false

    local lines = {}

    u.switch_can_edit_buf(bufnr, true)
    table.insert(lines, string.format("Status: %s (%s)", state.settings.pipeline[pipeline.status], pipeline.status))
    table.insert(lines, "")
    table.insert(lines, string.format("Last Run: %s", u.format_date(pipeline.created_at)))
    table.insert(lines, string.format("Url: %s", pipeline.web_url))
    table.insert(lines, string.format("Triggered By: %s", pipeline.source))

    table.insert(lines, "")
    table.insert(lines, "Jobs:")
    for _, pipeline_job in ipairs(pipeline_jobs) do
      table.insert(lines,
        string.format("%s (%s) %s", state.settings.pipeline[pipeline_job.status], pipeline_job.status, pipeline_job.name))
    end

    vim.schedule(function()
      vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, lines)
      M.color_status(pipeline.status, bufnr, lines[1], 1)

      for i, pipeline_job in ipairs(pipeline_jobs) do
        M.color_status(pipeline_job.status, bufnr, lines[7 + i], 7 + i)
      end

      pipeline_popup.border:set_text("top", "Pipeline Status", "center")
      state.set_popup_keymaps(pipeline_popup, M.retrigger)
      u.switch_can_edit_buf(bufnr, false)
    end)
  end)
end

M.retrigger    = function()
  local body = { pipeline_id = state.INFO.pipeline.id }
  if state.INFO.pipeline.status ~= 'failed' then
    vim.notify("Pipeline is not in a failed state!", vim.log.levels.WARN)
    return
  end

  job.run_job("/pipeline", "POST", body, function(data)
    vim.notify("Pipeline re-triggered!", vim.log.levels.INFO)
    state.INFO.pipeline = data.Pipeline
  end)
end

M.color_status = function(status, bufnr, status_line, linnr)
  local ns_id = vim.api.nvim_create_namespace("GitlabNamespace")
  vim.cmd(string.format("highlight default StatusHighlight guifg=%s", state.settings.pipeline[status]))

  local status_to_color_map = {
    created = 'DiagnosticWarn',
    pending = 'DiagnosticWarn',
    preparing = 'DiagnosticWarn',
    scheduled = 'DiagnosticWarn',
    running = 'DiagnosticWarn',
    canceled = 'DiagnosticWarn',
    skipped = 'DiagnosticWarn',
    failed = 'DiagnosticError',
    success = 'DiagnosticOK',
  }

  vim.api.nvim_buf_set_extmark(bufnr, ns_id, linnr - 1, 0,
    { end_row = linnr - 1, end_col = string.len(status_line), hl_group = status_to_color_map[status] })
end

return M