barrett/cp.nvim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 13

fix input display

Browse source
This commit is contained in:
Barrett Ruth 2025-10-23 20:03:17 -04:00
parent dc6f2fd5b6
commit 59f5066327
3 changed files with 24 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lua/cp/commands/init.lua
View file

@ -57,10 +57,13 @@ local function parse_command(args)
if test_arg then
local test_index = tonumber(test_arg)
if not test_index then
return { type = 'error', message = 'Test index must be a number' }
return {
type = 'error',
message = ("Test index '%s' is not a number"):format(test_index),
}
end
if test_index < 1 or test_index ~= math.floor(test_index) then
return { type = 'error', message = 'Test index must be >= 1' }
return { type = 'error', message = ("'%s' is not a valid test index"):format(test_index) }
end
return { type = 'action', action = 'run', test_index = test_index }
else

2
lua/cp/setup.lua
View file

@ -66,7 +66,7 @@ local function start_tests(platform, contest_id, problems)
local test_cases = cache.get_test_cases(platform, contest_id, state.get_problem_id())
local input_lines = {}
for _, tc in ipairs(test_cases) do
for _, line in ipairs(vim.split(tc.input, '\n', { plain = true, trimempty = false })) do
for _, line in ipairs(vim.split(tc.input, '\n')) do
table.insert(input_lines, line)
end
end

28
lua/cp/ui/panel.lua
View file

@ -85,9 +85,9 @@ function M.toggle_interactive(interactor_cmd)
if
not contest_data
or not contest_data.index_map
or not contest_data.problems[contest_data.index_map[problem_id]].interactive
or contest_data.problems[contest_data.index_map[problem_id]].interactive
then
logger.log('This problem is not interactive. Use :CP {run,panel}.', vim.log.levels.ERROR)
logger.log('This problem is interactive. Use :CP {run,panel}.', vim.log.levels.ERROR)
return
end
@ -275,9 +275,18 @@ function M.ensure_io_view()
utils.update_buffer_content(input_buf, {})
utils.update_buffer_content(output_buf, {})
vim.api.nvim_set_current_win(solution_win)
local test_cases = cache.get_test_cases(platform, contest_id, problem_id)
if test_cases and #test_cases > 0 then
local input_lines = {}
for _, tc in ipairs(test_cases) do
for _, line in ipairs(vim.split(tc.input, '\n')) do
table.insert(input_lines, line)
end
end
utils.update_buffer_content(input_buf, input_lines, nil, nil)
end
populate_input()
vim.api.nvim_set_current_win(solution_win)
end
function M.run_io_view(test_index)
@ -295,9 +304,9 @@ function M.run_io_view(test_index)
if
not contest_data
or not contest_data.index_map
or not contest_data.problems[contest_data.index_map[problem_id]].interactive
or contest_data.problems[contest_data.index_map[problem_id]].interactive
then
logger.log('This problem is not interactive. Use :CP {run,panel}.', vim.log.levels.ERROR)
logger.log('This problem is interactive. Use :CP {run,panel}.', vim.log.levels.ERROR)
return
end
@ -320,7 +329,7 @@ function M.run_io_view(test_index)
test_index,
#test_state.test_cases
),
vim.log.levels.ERROR
vim.log.levels.WARN
)
return
end
@ -368,7 +377,7 @@ function M.run_io_view(test_index)
local input_lines = {}
for _, idx in ipairs(test_indices) do
local tc = test_state.test_cases[idx]
for _, line in ipairs(vim.split(tc.input, '\n', { plain = true, trimempty = false })) do
for _, line in ipairs(vim.split(tc.input, '\n')) do
table.insert(input_lines, line)
end
end
@ -439,7 +448,6 @@ function M.toggle_panel(panel_opts)
end
state.set_active_panel(nil)
M.ensure_io_view()
populate_input()
return
end
@ -452,7 +460,7 @@ function M.toggle_panel(panel_opts)
if not platform or not contest_id then
logger.log(
'No platform/contest/problem configured. Use :CP <platform> <contest> [...] first.',
'No platform/contest configured. Use :CP <platform> <contest> [...] first.',
vim.log.levels.ERROR
)
return