cp.nvim/lua/cp/test_render.lua

---@class StatusInfo
---@field text string
---@field highlight_group string

local M = {}

local exit_code_names = {
  [128] = 'SIGHUP',
  [129] = 'SIGINT',
  [130] = 'SIGQUIT',
  [131] = 'SIGILL',
  [132] = 'SIGTRAP',
  [133] = 'SIGABRT',
  [134] = 'SIGBUS',
  [135] = 'SIGFPE',
  [136] = 'SIGKILL',
  [137] = 'SIGUSR1',
  [138] = 'SIGSEGV',
  [139] = 'SIGUSR2',
  [140] = 'SIGPIPE',
  [141] = 'SIGALRM',
  [142] = 'SIGTERM',
  [143] = 'SIGCHLD',
}

---Convert test status to CP terminology with colors
---@param test_case TestCase
---@return StatusInfo
function M.get_status_info(test_case)
  if test_case.status == 'pass' then
    return { text = 'AC', highlight_group = 'CpTestAC' }
  elseif test_case.status == 'fail' then
    if test_case.timed_out then
      return { text = 'TLE', highlight_group = 'CpTestError' }
    elseif test_case.code and test_case.code >= 128 then
      return { text = 'RTE', highlight_group = 'CpTestError' }
    else
      return { text = 'WA', highlight_group = 'CpTestError' }
    end
  elseif test_case.status == 'timeout' then
    return { text = 'TLE', highlight_group = 'CpTestError' }
  elseif test_case.status == 'running' then
    return { text = '...', highlight_group = 'CpTestPending' }
  else
    return { text = '', highlight_group = 'CpTestPending' }
  end
end

---Render test cases as a clean table
---@param test_state TestPanelState
---@return string[], table[] lines and highlight positions
function M.render_test_list(test_state)
  local lines = {}
  local highlights = {}

  local header = ' #  │ Status │ Time │ Exit Code'
  local separator =
    '────┼────────┼──────┼───────────'
  table.insert(lines, header)
  table.insert(lines, separator)

  for i, test_case in ipairs(test_state.test_cases) do
    local is_current = i == test_state.current_index
    local prefix = is_current and '>' or ' '
    local status_info = M.get_status_info(test_case)

    local num_col = string.format('%s%-2d', prefix, i)
    local status_col = string.format(' %-6s', status_info.text)
    local time_col = test_case.time_ms and string.format('%4.0fms', test_case.time_ms) or '  —  '
    local exit_col = ' — '
    if test_case.code then
      local signal_name = exit_code_names[test_case.code]
      if signal_name then
        exit_col = string.format(' %d (%s)', test_case.code, signal_name)
      else
        exit_col = string.format(' %d', test_case.code)
      end
    end

    local line = string.format('%s │%s │ %s │%s', num_col, status_col, time_col, exit_col)
    table.insert(lines, line)

    if status_info.text ~= '' then
      local status_start = #num_col + 3
      local status_end = status_start + #status_info.text
      table.insert(highlights, {
        line = #lines - 1,
        col_start = status_start,
        col_end = status_end,
        highlight_group = status_info.highlight_group,
      })
    end

    if is_current and test_case.input and test_case.input ~= '' then
      for _, input_line in
        ipairs(vim.split(test_case.input, '\n', { plain = true, trimempty = false }))
      do
        table.insert(lines, input_line)
      end
    end

    local separator =
      '─────────────────────────────────────────────────────────────'
    table.insert(lines, separator)
  end

  return lines, highlights
end

---Create status bar content for diff pane
---@param test_case TestCase?
---@return string
function M.render_status_bar(test_case)
  if not test_case then
    return ''
  end

  local parts = {}

  if test_case.time_ms then
    table.insert(parts, string.format('%.0fms', test_case.time_ms))
  end

  if test_case.code then
    table.insert(parts, string.format('Exit: %d', test_case.code))
  end

  return table.concat(parts, ' │ ')
end

---Get highlight groups needed for test rendering
---@return table<string, table>
function M.get_highlight_groups()
  return {
    CpTestAC = { fg = '#10b981', bold = true },
    CpTestError = { fg = '#ef4444', bold = true },
    CpTestPending = { fg = '#6b7280' },
    CpDiffRemoved = { fg = '#ef4444', bg = '#1f1f1f' },
    CpDiffAdded = { fg = '#10b981', bg = '#1f1f1f' },
  }
end

---Setup highlight groups
function M.setup_highlights()
  local groups = M.get_highlight_groups()
  for group_name, opts in pairs(groups) do
    vim.api.nvim_set_hl(0, group_name, opts)
  end
end

return M