cp.nvim/lua/cp/race.lua

local M = {}

local cache = require('cp.cache')
local constants = require('cp.constants')
local logger = require('cp.log')
local scraper = require('cp.scraper')

local race_state = {
  timer = nil,
  platform = nil,
  contest_id = nil,
  contest_name = nil,
  language = nil,
  start_time = nil,
}

local function format_countdown(seconds)
  local d = math.floor(seconds / 86400)
  local h = math.floor((seconds % 86400) / 3600)
  local m = math.floor((seconds % 3600) / 60)
  local s = seconds % 60
  if d > 0 then
    return string.format('%dd%dh%dm%ds', d, h, m, s)
  elseif h > 0 then
    return string.format('%dh%dm%ds', h, m, s)
  elseif m > 0 then
    return string.format('%dm%ds', m, s)
  end
  return string.format('%ds', s)
end

function M.start(platform, contest_id, language)
  if not platform or not vim.tbl_contains(constants.PLATFORMS, platform) then
    logger.log('Invalid platform', { level = vim.log.levels.ERROR })
    return
  end
  if not contest_id or contest_id == '' then
    logger.log('Contest ID required', { level = vim.log.levels.ERROR })
    return
  end
  if race_state.timer then
    M.stop()
  end

  cache.load()

  local display = constants.PLATFORM_DISPLAY_NAMES[platform] or platform
  local cached_countdown = cache.get_supports_countdown(platform)
  if cached_countdown == false then
    logger.log(('%s does not support --race'):format(display), { level = vim.log.levels.ERROR })
    return
  end

  local start_time = cache.get_contest_start_time(platform, contest_id)

  if not start_time then
    logger.log(
      'Fetching contest list...',
      { level = vim.log.levels.INFO, override = true, sync = true }
    )
    local result = scraper.scrape_contest_list(platform)
    if result then
      local sc = result.supports_countdown
      if sc == false then
        cache.set_contest_summaries(platform, result.contests or {}, { supports_countdown = false })
        logger.log(
          ('%s does not support --race'):format(display),
          { level = vim.log.levels.ERROR }
        )
        return
      end
      if result.contests and #result.contests > 0 then
        cache.set_contest_summaries(platform, result.contests, { supports_countdown = sc })
        start_time = cache.get_contest_start_time(platform, contest_id)
      end
    end
  end

  if not start_time then
    logger.log(
      ('No start time found for %s contest "%s"'):format(
        constants.PLATFORM_DISPLAY_NAMES[platform] or platform,
        contest_id
      ),
      { level = vim.log.levels.ERROR }
    )
    return
  end

  local remaining = start_time - os.time()
  if remaining <= 0 then
    logger.log(
      'Contest has already started, setting up...',
      { level = vim.log.levels.INFO, override = true }
    )
    require('cp.setup').setup_contest(platform, contest_id, nil, language)
    return
  end

  race_state.platform = platform
  race_state.contest_id = contest_id
  race_state.contest_name = cache.get_contest_display_name(platform, contest_id) or contest_id
  race_state.language = language
  race_state.start_time = start_time

  local timer = vim.uv.new_timer()
  race_state.timer = timer
  timer:start(
    1000,
    1000,
    vim.schedule_wrap(function()
      local r = race_state.start_time - os.time()
      if r <= 0 then
        timer:stop()
        timer:close()
        race_state.timer = nil
        local p = race_state.platform
        local c = race_state.contest_id
        local l = race_state.language
        race_state.platform = nil
        race_state.contest_id = nil
        race_state.contest_name = nil
        race_state.language = nil
        race_state.start_time = nil
        logger.log('Contest started!', { level = vim.log.levels.INFO, override = true })
        require('cp.setup').setup_contest(p, c, nil, l)
      else
        vim.notify(
          ('[cp.nvim]: %s race "%s" starts in %s'):format(
            constants.PLATFORM_DISPLAY_NAMES[race_state.platform] or race_state.platform,
            race_state.contest_name,
            format_countdown(r)
          ),
          vim.log.levels.INFO
        )
      end
    end)
  )
end

function M.stop()
  local timer = race_state.timer
  if not timer then
    logger.log('No active race', { level = vim.log.levels.WARN })
    return
  end
  local display = constants.PLATFORM_DISPLAY_NAMES[race_state.platform] or race_state.platform
  local name = race_state.contest_name or race_state.contest_id
  timer:stop()
  timer:close()
  race_state.timer = nil
  race_state.platform = nil
  race_state.contest_id = nil
  race_state.contest_name = nil
  race_state.language = nil
  race_state.start_time = nil
  logger.log(('Cancelled %s race "%s"'):format(display, name), { level = vim.log.levels.INFO, override = true })
end

function M.status()
  if not race_state.timer or not race_state.start_time then
    return { active = false }
  end
  return {
    active = true,
    platform = race_state.platform,
    contest_id = race_state.contest_id,
    remaining_seconds = math.max(0, race_state.start_time - os.time()),
  }
end

return M