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

feat: start lang refactor

Browse source
This commit is contained in:
Barrett Ruth 2025-10-24 01:11:19 -04:00
parent ce12ab0e1a
commit bd30fb626c
7 changed files with 229 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

42
lua/cp/setup.lua
View file

@ -8,6 +8,18 @@ local logger = require('cp.log')
local scraper = require('cp.scraper')
local state = require('cp.state')
---Get the language of the current file from cache
---@return string|nil
local function get_current_file_language()
local current_file = vim.fn.expand('%:p')
if current_file == '' then
return nil
end
cache.load()
local file_state = cache.get_file_state(current_file)
return file_state and file_state.language or nil
end
---@class TestCaseLite
---@field input string
---@field expected string
@ -86,6 +98,14 @@ function M.setup_contest(platform, contest_id, problem_id, language)
state.set_platform(platform)
state.set_contest_id(contest_id)
if language then
local lang_result = config_module.get_language_for_platform(platform, language)
if not lang_result.valid then
logger.log(lang_result.error, vim.log.levels.ERROR)
return
end
end
local is_new_contest = old_platform ~= platform and old_contest_id ~= contest_id
cache.load()
@ -173,6 +193,15 @@ function M.setup_problem(problem_id, language)
local config = config_module.get_config()
local lang = language
or (config.platforms[platform] and config.platforms[platform].default_language)
if language then
local lang_result = config_module.get_language_for_platform(platform, language)
if not lang_result.valid then
logger.log(lang_result.error, vim.log.levels.ERROR)
return
end
end
local source_file = state.get_source_file(lang)
if not source_file then
return
@ -235,7 +264,8 @@ function M.setup_problem(problem_id, language)
end
---@param direction integer
function M.navigate_problem(direction)
---@param language? string
function M.navigate_problem(direction, language)
if direction == 0 then
return
end
@ -274,7 +304,15 @@ function M.navigate_problem(direction)
require('cp.ui.views').disable()
end
M.setup_contest(platform, contest_id, problems[new_index].id)
local lang = language or get_current_file_language()
if lang then
local lang_result = config_module.get_language_for_platform(platform, lang)
if not lang_result.valid then
lang = nil
end
end
M.setup_contest(platform, contest_id, problems[new_index].id, lang)
end
return M