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

feat: add solution to window state

Browse source
This commit is contained in:
Barrett Ruth 2025-10-23 10:07:22 -04:00
parent 018d801121
commit 347be72774
2 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
lua/cp/setup.lua
View file

@ -172,6 +172,7 @@ function M.setup_problem(problem_id, language)
vim.api.nvim_buf_set_name(prov.bufnr, source_file)
vim.bo[prov.bufnr].swapfile = true
vim.cmd(string.format('silent keepalt noautocmd write! %s', vim.fn.fnameescape(source_file)))
state.set_solution_win(vim.api.nvim_get_current_win())
if config.hooks and config.hooks.setup_code and not vim.b[prov.bufnr].cp_setup_done then
local ok = pcall(config.hooks.setup_code, state)
if ok then
@ -194,6 +195,7 @@ function M.setup_problem(problem_id, language)
vim.cmd.only({ mods = { silent = true } })
vim.cmd.e(source_file)
local bufnr = vim.api.nvim_get_current_buf()
state.set_solution_win(vim.api.nvim_get_current_win())
if config.hooks and config.hooks.setup_code and not vim.b[bufnr].cp_setup_done then
local ok = pcall(config.hooks.setup_code, state)
if ok then

14
lua/cp/state.lua
View file

@ -35,6 +35,7 @@ local state = {
saved_session = nil,
active_panel = nil,
provisional = nil,
solution_win = nil,
}
---@return string|nil
@ -153,6 +154,19 @@ function M.set_provisional(p)
state.provisional = p
end
---@return integer?
function M.get_solution_win()
if state.solution_win and vim.api.nvim_win_is_valid(state.solution_win) then
return state.solution_win
end
return vim.api.nvim_get_current_win()
end
---@param win integer?
function M.set_solution_win(win)
state.solution_win = win
end
M._state = state
return M