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 M = {}

local 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
            local cp = require('cp')
            cp.handle_command({ fargs = { 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 M.pick(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 M