120 lines
3.6 KiB
Lua
120 lines
3.6 KiB
Lua
---@class StatusInfo
|
|
---@field text string
|
|
---@field highlight_group string
|
|
|
|
local M = {}
|
|
|
|
---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 │ Input'
|
|
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 = test_case.code and string.format(' %-3d', test_case.code) or ' — '
|
|
local input_col = test_case.input and test_case.input:gsub('\n', ' ') or ''
|
|
|
|
local line = string.format(
|
|
'%s │%s │ %s │%s │ %s',
|
|
num_col,
|
|
status_col,
|
|
time_col,
|
|
exit_col,
|
|
input_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
|
|
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
|