Merge branch 'main' into feat/epsilon-float-comparison

lua/cp/cache.lua

@@ -375,6 +375,8 @@ function M.get_data_pretty()
   return vim.inspect(cache_data)
 end
 
-M._cache = cache_data
+function M.get_raw_cache()
+  return cache_data
+end
 
 return M

lua/cp/config.lua

@@ -7,6 +7,7 @@
 ---@class CpLanguage
 ---@field extension string
 ---@field commands CpLangCommands
+---@field template? string
 
 ---@class CpPlatformOverrides
 ---@field extension? string
@@ -216,6 +217,10 @@ local function validate_language(id, lang)
     commands = { lang.commands, { 'table' } },
   })
 
+  if lang.template ~= nil then
+    vim.validate({ template = { lang.template, 'string' } })
+  end
+
   if not lang.commands.run then
     error(('[cp.nvim] languages.%s.commands.run is required'):format(id))
   end

lua/cp/health.lua

@@ -13,7 +13,7 @@ local function check()
     vim.health.error('cp.nvim requires Neovim 0.10.0+')
   end
 
-  local uname = vim.loop.os_uname()
+  local uname = vim.uv.os_uname()
   if uname.sysname == 'Windows_NT' then
     vim.health.error('Windows is not supported')
   end

lua/cp/runner/run.lua

@@ -323,26 +323,35 @@ function M.run_all_test_cases(indices, debug, on_each, on_done)
     end
   end
 
-  local function run_next(pos)
-    if pos > #to_run then
-      logger.log(
-        ('Finished %s %d test cases.'):format(debug and 'debugging' or 'running', #to_run),
-        vim.log.levels.INFO,
-        true
-      )
-      on_done(panel_state.test_cases)
-      return
-    end
-
-    M.run_test_case(to_run[pos], debug, function()
-      if on_each then
-        on_each(pos, #to_run)
-      end
-      run_next(pos + 1)
-    end)
+  if #to_run == 0 then
+    logger.log(
+      ('Finished %s %d test cases.'):format(debug and 'debugging' or 'running', 0),
+      vim.log.levels.INFO,
+      true
+    )
+    on_done(panel_state.test_cases)
+    return
   end
 
-  run_next(1)
+  local total = #to_run
+  local remaining = total
+
+  for _, idx in ipairs(to_run) do
+    M.run_test_case(idx, debug, function()
+      if on_each then
+        on_each(idx, total)
+      end
+      remaining = remaining - 1
+      if remaining == 0 then
+        logger.log(
+          ('Finished %s %d test cases.'):format(debug and 'debugging' or 'running', total),
+          vim.log.levels.INFO,
+          true
+        )
+        on_done(panel_state.test_cases)
+      end
+    end)
+  end
 end
 
 ---@return PanelState

lua/cp/scraper.lua

@@ -57,7 +57,7 @@ local function run_scraper(platform, subcommand, args, opts)
   env.CONDA_PREFIX = ''
 
   if opts and opts.ndjson then
-    local uv = vim.loop
+    local uv = vim.uv
     local stdout = uv.new_pipe(false)
     local stderr = uv.new_pipe(false)
     local buf = ''

lua/cp/setup.lua

@@ -8,6 +8,21 @@ local logger = require('cp.log')
 local scraper = require('cp.scraper')
 local state = require('cp.state')
 
+local function apply_template(bufnr, lang_id, platform)
+  local config = config_module.get_config()
+  local eff = config.runtime.effective[platform] and config.runtime.effective[platform][lang_id]
+  if not eff or not eff.template then
+    return
+  end
+  local path = vim.fn.expand(eff.template)
+  if vim.fn.filereadable(path) ~= 1 then
+    logger.log(('[cp.nvim] template not readable: %s'):format(path), vim.log.levels.WARN)
+    return
+  end
+  local lines = vim.fn.readfile(path)
+  vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, lines)
+end
+
 ---Get the language of the current file from cache
 ---@return string?
 local function get_current_file_language()
@@ -179,7 +194,7 @@ function M.setup_contest(platform, contest_id, problem_id, language)
       contest_id = contest_id,
       language = lang,
       requested_problem_id = problem_id,
-      token = vim.loop.hrtime(),
+      token = vim.uv.hrtime(),
     })
 
     logger.log('Fetching contests problems...', vim.log.levels.INFO, true)
@@ -270,14 +285,17 @@ function M.setup_problem(problem_id, language)
           mods = { silent = true, noautocmd = true, keepalt = true },
         })
         state.set_solution_win(vim.api.nvim_get_current_win())
-        if config.hooks and config.hooks.setup_code and not vim.b[prov.bufnr].cp_setup_done then
-          local ok = pcall(config.hooks.setup_code, state)
-          if ok then
+        if not vim.b[prov.bufnr].cp_setup_done then
+          apply_template(prov.bufnr, lang, platform)
+          if config.hooks and config.hooks.setup_code then
+            local ok = pcall(config.hooks.setup_code, state)
+            if ok then
+              vim.b[prov.bufnr].cp_setup_done = true
+            end
+          else
+            helpers.clearcol(prov.bufnr)
             vim.b[prov.bufnr].cp_setup_done = true
           end
-        elseif not vim.b[prov.bufnr].cp_setup_done then
-          helpers.clearcol(prov.bufnr)
-          vim.b[prov.bufnr].cp_setup_done = true
         end
         cache.set_file_state(
           vim.fn.fnamemodify(source_file, ':p'),
@@ -300,14 +318,21 @@ function M.setup_problem(problem_id, language)
   local bufnr = vim.api.nvim_get_current_buf()
   state.set_solution_win(vim.api.nvim_get_current_win())
   require('cp.ui.views').ensure_io_view()
-  if config.hooks and config.hooks.setup_code and not vim.b[bufnr].cp_setup_done then
-    local ok = pcall(config.hooks.setup_code, state)
-    if ok then
+  if not vim.b[bufnr].cp_setup_done then
+    local is_new = vim.api.nvim_buf_line_count(bufnr) == 1
+      and vim.api.nvim_buf_get_lines(bufnr, 0, 1, false)[1] == ''
+    if is_new then
+      apply_template(bufnr, lang, platform)
+    end
+    if config.hooks and config.hooks.setup_code then
+      local ok = pcall(config.hooks.setup_code, state)
+      if ok then
+        vim.b[bufnr].cp_setup_done = true
+      end
+    else
+      helpers.clearcol(bufnr)
       vim.b[bufnr].cp_setup_done = true
     end
-  elseif not vim.b[bufnr].cp_setup_done then
-    helpers.clearcol(bufnr)
-    vim.b[bufnr].cp_setup_done = true
   end
   cache.set_file_state(
     vim.fn.expand('%:p'),

lua/cp/utils.lua

@@ -5,7 +5,7 @@ local logger = require('cp.log')
 local _nix_python = nil
 local _nix_discovered = false
 
-local uname = vim.loop.os_uname()
+local uname = vim.uv.os_uname()
 
 local _time_cached = false
 local _time_path = nil
@@ -336,7 +336,7 @@ function M.timeout_capability()
 end
 
 function M.cwd_executables()
-  local uv = vim.uv or vim.loop
+  local uv = vim.uv
   local req = uv.fs_scandir('.')
   if not req then
     return {}