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

fix: misc fixes to snippets and the state managemnt

Browse source
This commit is contained in:
Barrett Ruth 2025-09-15 18:50:17 -04:00
parent f53f4bcbe7
commit c16f6576e2
3 changed files with 37 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
doc/cp.txt
View file

@ -134,7 +134,7 @@ Optional configuration with lazy.nvim: >
• {filename}? (`function`) Custom filename generation function.
`function(contest, contest_id, problem_id, config, language)`
Should return full filename with extension.
(default: uses problem_id or contest_id)
(default: concats contest_id and problem id)
*cp.ContestConfig*

2
lua/cp/config.lua
View file

@ -137,7 +137,7 @@ local function default_filename(contest_id, problem_id)
})
if problem_id then
return problem_id:lower()
return (contest_id .. problem_id):lower()
else
return contest_id:lower()
end

45
lua/cp/init.lua
View file

@ -99,7 +99,11 @@ local function setup_problem(contest_id, problem_id, language)
if vim.api.nvim_buf_get_lines(0, 0, -1, true)[1] == "" then
local has_luasnip, luasnip = pcall(require, "luasnip")
if has_luasnip then
local prefixed_trigger = ("cp.nvim/%s.%s"):format(state.platform, language)
local constants = require("cp.constants")
local filetype = vim.api.nvim_get_option_value("filetype", { buf = 0 })
local language_name = constants.filetype_to_language[filetype]
local canonical_language = constants.canonical_filetypes[language_name] or language_name
local prefixed_trigger = ("cp.nvim/%s.%s"):format(state.platform, canonical_language)
vim.api.nvim_buf_set_lines(0, 0, -1, false, { prefixed_trigger })
vim.api.nvim_win_set_cursor(0, { 1, #prefixed_trigger })
@ -234,6 +238,30 @@ local function toggle_test_panel()
vim.bo.filetype = "cptest"
vim.bo.bufhidden = "wipe"
local function navigate_test(delta)
local test_state = test_module.get_test_panel_state()
local new_index = test_state.current_index + delta
if new_index >= 1 and new_index <= #test_state.test_cases then
test_state.current_index = new_index
toggle_test_panel()
toggle_test_panel()
end
end
local function run_current_test()
local ctx = problem.create_context(state.platform, state.contest_id, state.problem_id, config)
local contest_config = config.contests[state.platform]
local test_state = test_module.get_test_panel_state()
test_module.run_test_case(ctx, contest_config, test_state.current_index)
toggle_test_panel()
toggle_test_panel()
end
vim.keymap.set("n", "j", function() navigate_test(1) end, { buffer = test_buf, silent = true })
vim.keymap.set("n", "k", function() navigate_test(-1) end, { buffer = test_buf, silent = true })
vim.keymap.set("n", "<CR>", run_current_test, { buffer = test_buf, silent = true })
vim.keymap.set("n", "q", function() toggle_test_panel() end, { buffer = test_buf, silent = true })
local test_state = test_module.get_test_panel_state()
local test_lines = {}
@ -261,26 +289,23 @@ local function toggle_test_panel()
if current_test then
table.insert(test_lines, string.format("── Test %d ──", test_state.current_index))
table.insert(test_lines, "Input:")
for line in current_test.input:gmatch("[^\n]*") do
table.insert(test_lines, " " .. line)
for _, line in ipairs(vim.split(current_test.input, "\n", { plain = true, trimempty = true })) do
table.insert(test_lines, line)
end
table.insert(test_lines, "Expected:")
for line in current_test.expected:gmatch("[^\n]*") do
table.insert(test_lines, " " .. line)
for _, line in ipairs(vim.split(current_test.expected, "\n", { plain = true, trimempty = true })) do
table.insert(test_lines, line)
end
if current_test.actual then
table.insert(test_lines, "Actual:")
for line in current_test.actual:gmatch("[^\n]*") do
table.insert(test_lines, " " .. line)
for _, line in ipairs(vim.split(current_test.actual, "\n", { plain = true, trimempty = true })) do
table.insert(test_lines, line)
end
end
end
table.insert(test_lines, "")
table.insert(test_lines, "j/k: navigate <space>: toggle <enter>: run q: quit")
vim.api.nvim_buf_set_lines(test_buf, 0, -1, false, test_lines)
vim.bo.modifiable = false