160 lines
3.8 KiB
Lua
160 lines
3.8 KiB
Lua
---@class cp.State
|
|
---@field get_platform fun(): string?
|
|
---@field set_platform fun(platform: string)
|
|
---@field get_contest_id fun(): string?
|
|
---@field set_contest_id fun(contest_id: string)
|
|
---@field get_problem_id fun(): string?
|
|
---@field set_problem_id fun(problem_id: string)
|
|
---@field get_test_cases fun(): table[]?
|
|
---@field set_test_cases fun(test_cases: table[])
|
|
---@field is_run_panel_active fun(): boolean
|
|
---@field set_run_panel_active fun(active: boolean)
|
|
---@field get_saved_session fun(): table?
|
|
---@field set_saved_session fun(session: table)
|
|
---@field get_context fun(): {platform: string?, contest_id: string?, problem_id: string?}
|
|
---@field has_context fun(): boolean
|
|
---@field reset fun()
|
|
---@field get_base_name fun(): string?
|
|
---@field get_source_file fun(language?: string): string?
|
|
---@field get_binary_file fun(): string?
|
|
---@field get_input_file fun(): string?
|
|
---@field get_output_file fun(): string?
|
|
---@field get_expected_file fun(): string?
|
|
|
|
local M = {}
|
|
|
|
local state = {
|
|
platform = nil,
|
|
contest_id = nil,
|
|
problem_id = nil,
|
|
test_cases = nil,
|
|
run_panel_active = false,
|
|
saved_session = nil,
|
|
}
|
|
|
|
function M.get_platform()
|
|
return state.platform
|
|
end
|
|
|
|
function M.set_platform(platform)
|
|
state.platform = platform
|
|
end
|
|
|
|
function M.get_contest_id()
|
|
return state.contest_id
|
|
end
|
|
|
|
function M.set_contest_id(contest_id)
|
|
state.contest_id = contest_id
|
|
end
|
|
|
|
function M.get_problem_id()
|
|
return state.problem_id
|
|
end
|
|
|
|
function M.set_problem_id(problem_id)
|
|
state.problem_id = problem_id
|
|
end
|
|
|
|
function M.get_test_cases()
|
|
return state.test_cases
|
|
end
|
|
|
|
function M.set_test_cases(test_cases)
|
|
state.test_cases = test_cases
|
|
end
|
|
|
|
function M.is_run_panel_active()
|
|
return state.run_panel_active
|
|
end
|
|
|
|
function M.set_run_panel_active(active)
|
|
state.run_panel_active = active
|
|
end
|
|
|
|
function M.get_saved_session()
|
|
return state.saved_session
|
|
end
|
|
|
|
function M.set_saved_session(session)
|
|
state.saved_session = session
|
|
end
|
|
|
|
function M.get_context()
|
|
return {
|
|
platform = state.platform,
|
|
contest_id = state.contest_id,
|
|
problem_id = state.problem_id,
|
|
}
|
|
end
|
|
|
|
function M.get_base_name()
|
|
if not state.contest_id then
|
|
return nil
|
|
end
|
|
|
|
local config_module = require('cp.config')
|
|
local config = config_module.get_config()
|
|
|
|
if config.filename then
|
|
return config.filename(state.platform or '', state.contest_id, state.problem_id, config)
|
|
else
|
|
return config_module.default_filename(state.contest_id, state.problem_id)
|
|
end
|
|
end
|
|
|
|
function M.get_source_file(language)
|
|
local base_name = M.get_base_name()
|
|
if not base_name or not state.platform then
|
|
return nil
|
|
end
|
|
|
|
local config = require('cp.config').get_config()
|
|
local contest_config = config.contests[state.platform]
|
|
if not contest_config then
|
|
return nil
|
|
end
|
|
|
|
local target_language = language or contest_config.default_language
|
|
local language_config = contest_config[target_language]
|
|
if not language_config or not language_config.extension then
|
|
return nil
|
|
end
|
|
|
|
return base_name .. '.' .. language_config.extension
|
|
end
|
|
|
|
function M.get_binary_file()
|
|
local base_name = M.get_base_name()
|
|
return base_name and ('build/%s.run'):format(base_name) or nil
|
|
end
|
|
|
|
function M.get_input_file()
|
|
local base_name = M.get_base_name()
|
|
return base_name and ('io/%s.cpin'):format(base_name) or nil
|
|
end
|
|
|
|
function M.get_output_file()
|
|
local base_name = M.get_base_name()
|
|
return base_name and ('io/%s.cpout'):format(base_name) or nil
|
|
end
|
|
|
|
function M.get_expected_file()
|
|
local base_name = M.get_base_name()
|
|
return base_name and ('io/%s.expected'):format(base_name) or nil
|
|
end
|
|
|
|
function M.has_context()
|
|
return state.platform and state.contest_id
|
|
end
|
|
|
|
function M.reset()
|
|
state.platform = nil
|
|
state.contest_id = nil
|
|
state.problem_id = nil
|
|
state.test_cases = nil
|
|
state.run_panel_active = false
|
|
state.saved_session = nil
|
|
end
|
|
|
|
return M
|