45 lines
1.1 KiB
Lua
45 lines
1.1 KiB
Lua
local M = {}
|
|
|
|
local cache = require('cp.cache')
|
|
local logger = require('cp.log')
|
|
local state = require('cp.state')
|
|
|
|
function M.restore_from_current_file()
|
|
local current_file = vim.fn.expand('%:p')
|
|
if current_file == '' then
|
|
logger.log('No file is currently open.', vim.log.levels.ERROR)
|
|
return false
|
|
end
|
|
|
|
cache.load()
|
|
local file_state = cache.get_file_state(current_file)
|
|
if not file_state then
|
|
logger.log(
|
|
'No cached state found for current file. Use :CP <platform> <contest> <problem> [...] first.',
|
|
vim.log.levels.ERROR
|
|
)
|
|
return false
|
|
end
|
|
|
|
logger.log(
|
|
('Restoring from cached state: %s %s %s'):format(
|
|
file_state.platform,
|
|
file_state.contest_id,
|
|
file_state.problem_id
|
|
)
|
|
)
|
|
|
|
local setup = require('cp.setup')
|
|
if not setup.set_platform(file_state.platform) then
|
|
return false
|
|
end
|
|
|
|
state.set_contest_id(file_state.contest_id)
|
|
state.set_problem_id(file_state.problem_id)
|
|
|
|
setup.setup_problem(file_state.contest_id, file_state.problem_id, file_state.language)
|
|
|
|
return true
|
|
end
|
|
|
|
return M
|