145 lines
4 KiB
Lua
145 lines
4 KiB
Lua
local finders = require('telescope.finders')
|
|
local pickers = require('telescope.pickers')
|
|
local conf = require('telescope.config').values
|
|
local action_state = require('telescope.actions.state')
|
|
local actions = require('telescope.actions')
|
|
|
|
local picker_utils = require('cp.pickers')
|
|
|
|
local contest_picker, problem_picker
|
|
|
|
function contest_picker(opts, platform)
|
|
local constants = require('cp.constants')
|
|
local platform_display_name = constants.PLATFORM_DISPLAY_NAMES[platform] or platform
|
|
local contests = picker_utils.get_contests_for_platform(platform)
|
|
|
|
if #contests == 0 then
|
|
vim.notify(
|
|
('No contests found for platform: %s'):format(platform_display_name),
|
|
vim.log.levels.WARN
|
|
)
|
|
return
|
|
end
|
|
|
|
pickers
|
|
.new(opts, {
|
|
prompt_title = ('Select Contest (%s)'):format(platform_display_name),
|
|
results_title = '<C-r> refresh',
|
|
finder = finders.new_table({
|
|
results = contests,
|
|
entry_maker = function(entry)
|
|
return {
|
|
value = entry,
|
|
display = entry.display_name,
|
|
ordinal = entry.display_name,
|
|
}
|
|
end,
|
|
}),
|
|
sorter = conf.generic_sorter(opts),
|
|
attach_mappings = function(prompt_bufnr, map)
|
|
actions.select_default:replace(function()
|
|
local selection = action_state.get_selected_entry()
|
|
actions.close(prompt_bufnr)
|
|
|
|
if selection then
|
|
problem_picker(opts, platform, selection.value.id)
|
|
end
|
|
end)
|
|
|
|
map('i', '<C-r>', function()
|
|
local cache = require('cp.cache')
|
|
cache.clear_contest_list(platform)
|
|
actions.close(prompt_bufnr)
|
|
contest_picker(opts, platform)
|
|
end)
|
|
|
|
return true
|
|
end,
|
|
})
|
|
:find()
|
|
end
|
|
|
|
function problem_picker(opts, platform, contest_id)
|
|
local constants = require('cp.constants')
|
|
local platform_display_name = constants.PLATFORM_DISPLAY_NAMES[platform] or platform
|
|
local problems = picker_utils.get_problems_for_contest(platform, contest_id)
|
|
|
|
if #problems == 0 then
|
|
vim.notify(
|
|
("Contest %s %s hasn't started yet or has no available problems"):format(
|
|
platform_display_name,
|
|
contest_id
|
|
),
|
|
vim.log.levels.WARN
|
|
)
|
|
contest_picker(opts, platform)
|
|
return
|
|
end
|
|
|
|
pickers
|
|
.new(opts, {
|
|
prompt_title = ('Select Problem (%s %s)'):format(platform_display_name, contest_id),
|
|
finder = finders.new_table({
|
|
results = problems,
|
|
entry_maker = function(entry)
|
|
return {
|
|
value = entry,
|
|
display = entry.display_name,
|
|
ordinal = entry.display_name,
|
|
}
|
|
end,
|
|
}),
|
|
sorter = conf.generic_sorter(opts),
|
|
attach_mappings = function(prompt_bufnr)
|
|
actions.select_default:replace(function()
|
|
local selection = action_state.get_selected_entry()
|
|
actions.close(prompt_bufnr)
|
|
|
|
if selection then
|
|
local cp = require('cp')
|
|
cp.handle_command({ fargs = { platform, contest_id, selection.value.id } })
|
|
end
|
|
end)
|
|
return true
|
|
end,
|
|
})
|
|
:find()
|
|
end
|
|
|
|
local function platform_picker(opts)
|
|
opts = opts or {}
|
|
|
|
local platforms = picker_utils.get_platforms()
|
|
|
|
pickers
|
|
.new(opts, {
|
|
prompt_title = 'Select Platform',
|
|
finder = finders.new_table({
|
|
results = platforms,
|
|
entry_maker = function(entry)
|
|
return {
|
|
value = entry,
|
|
display = entry.display_name,
|
|
ordinal = entry.display_name,
|
|
}
|
|
end,
|
|
}),
|
|
sorter = conf.generic_sorter(opts),
|
|
attach_mappings = function(prompt_bufnr)
|
|
actions.select_default:replace(function()
|
|
local selection = action_state.get_selected_entry()
|
|
actions.close(prompt_bufnr)
|
|
|
|
if selection then
|
|
contest_picker(opts, selection.value.id)
|
|
end
|
|
end)
|
|
return true
|
|
end,
|
|
})
|
|
:find()
|
|
end
|
|
|
|
return {
|
|
platform_picker = platform_picker,
|
|
}
|