136 lines
4 KiB
Lua
136 lines
4 KiB
Lua
---@class LanguageConfig
|
|
---@field compile? string[] Compile command template
|
|
---@field run string[] Run command template
|
|
---@field debug? string[] Debug command template
|
|
---@field executable? string Executable name
|
|
---@field version? number Language version
|
|
---@field extension string File extension
|
|
|
|
---@class PartialLanguageConfig
|
|
---@field compile? string[] Compile command template
|
|
---@field run? string[] Run command template
|
|
---@field debug? string[] Debug command template
|
|
---@field executable? string Executable name
|
|
---@field version? number Language version
|
|
---@field extension? string File extension
|
|
|
|
---@class ContestConfig
|
|
---@field cpp LanguageConfig
|
|
---@field python LanguageConfig
|
|
---@field default_language string
|
|
---@field timeout_ms number
|
|
|
|
---@class PartialContestConfig
|
|
---@field cpp? PartialLanguageConfig
|
|
---@field python? PartialLanguageConfig
|
|
---@field default_language? string
|
|
---@field timeout_ms? number
|
|
|
|
---@class Hooks
|
|
---@field before_run? fun(ctx: ProblemContext)
|
|
---@field before_debug? fun(ctx: ProblemContext)
|
|
---@field setup_code? fun(ctx: ProblemContext)
|
|
|
|
---@class cp.Config
|
|
---@field contests table<string, ContestConfig>
|
|
---@field snippets table[]
|
|
---@field hooks Hooks
|
|
---@field debug boolean
|
|
---@field tile? fun(source_buf: number, input_buf: number, output_buf: number)
|
|
---@field filename? fun(contest: string, contest_id: string, problem_id?: string, config: cp.Config, language?: string): string
|
|
|
|
---@class cp.UserConfig
|
|
---@field contests? table<string, PartialContestConfig>
|
|
---@field snippets? table[]
|
|
---@field hooks? Hooks
|
|
---@field debug? boolean
|
|
---@field tile? fun(source_buf: number, input_buf: number, output_buf: number)
|
|
---@field filename? fun(contest: string, contest_id: string, problem_id?: string, config: cp.Config, language?: string): string
|
|
|
|
local M = {}
|
|
local languages = require("cp.languages")
|
|
|
|
---@type cp.Config
|
|
M.defaults = {
|
|
contests = {},
|
|
snippets = {},
|
|
hooks = {
|
|
before_run = nil,
|
|
before_debug = nil,
|
|
setup_code = nil,
|
|
},
|
|
debug = false,
|
|
tile = nil,
|
|
filename = nil,
|
|
}
|
|
|
|
---@param user_config cp.UserConfig|nil
|
|
---@return cp.Config
|
|
function M.setup(user_config)
|
|
vim.validate({
|
|
user_config = { user_config, { "table", "nil" }, true },
|
|
})
|
|
|
|
if user_config then
|
|
vim.validate({
|
|
contests = { user_config.contests, { "table", "nil" }, true },
|
|
snippets = { user_config.snippets, { "table", "nil" }, true },
|
|
hooks = { user_config.hooks, { "table", "nil" }, true },
|
|
debug = { user_config.debug, { "boolean", "nil" }, true },
|
|
tile = { user_config.tile, { "function", "nil" }, true },
|
|
filename = { user_config.filename, { "function", "nil" }, true },
|
|
})
|
|
|
|
if user_config.hooks then
|
|
vim.validate({
|
|
before_run = { user_config.hooks.before_run, { "function", "nil" }, true },
|
|
before_debug = { user_config.hooks.before_debug, { "function", "nil" }, true },
|
|
setup_code = { user_config.hooks.setup_code, { "function", "nil" }, true },
|
|
})
|
|
end
|
|
|
|
if user_config.contests then
|
|
for contest_name, contest_config in pairs(user_config.contests) do
|
|
for lang_name, lang_config in pairs(contest_config) do
|
|
if type(lang_config) == "table" and lang_config.extension then
|
|
if
|
|
not vim.tbl_contains(vim.tbl_keys(languages.filetype_to_language), lang_config.extension)
|
|
then
|
|
error(
|
|
("Invalid extension '%s' for language '%s' in contest '%s'. Valid extensions: %s"):format(
|
|
lang_config.extension,
|
|
lang_name,
|
|
contest_name,
|
|
table.concat(vim.tbl_keys(languages.filetype_to_language), ", ")
|
|
)
|
|
)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local config = vim.tbl_deep_extend("force", M.defaults, user_config or {})
|
|
return config
|
|
end
|
|
|
|
---@param contest_id string
|
|
---@param problem_id? string
|
|
---@return string
|
|
local function default_filename(contest_id, problem_id)
|
|
vim.validate({
|
|
contest_id = { contest_id, "string" },
|
|
problem_id = { problem_id, { "string", "nil" }, true },
|
|
})
|
|
|
|
if problem_id then
|
|
return problem_id:lower()
|
|
else
|
|
return contest_id:lower()
|
|
end
|
|
end
|
|
|
|
M.default_filename = default_filename
|
|
|
|
return M
|