feat: store testcase snapshots

lua/leetcode-ui/group/cases.lua

@@ -59,11 +59,11 @@ function Cases:init(item, parent)
  self.console = parent
 
  local total = item.total_testcases ~= vim.NIL and item.total_testcases or 0
+ local testcases = parent.testcase:by_id(item.submission_id)
+
  for i = 1, total do
  self.cases[i] = Case({
- -- TODO: cache the testcases on submission,
- -- so it doesn't get out of sync
- input = self.console.testcase.testcases[i],
+ input = testcases[i],
  output = item.code_answer[i],
  expected = item.expected_code_answer[i],
  std_output = item.std_output_list[i],

lua/leetcode-ui/popup/console/testcase.lua

@@ -5,40 +5,53 @@ local t = require("leetcode.translator")
 
 ---@class lc.ui.Console.TestcasePopup : lc.ui.Console.Popup
 ---@field testcases table<string[]>
+---@field testcase_len integer
 ---@field extmarks integer[]
+---@field snapshots table<string, string[]>
 local Testcase = ConsolePopup:extend("LeetTestcasePopup")
 
 function Testcase:content()
- self.testcases = {}
+ local lines = vim.api.nvim_buf_get_lines(self.bufnr, 0, -1, false)
+ lines = vim.tbl_filter(function(line) return line ~= "" end, lines)
+ return table.concat(lines, "\n")
+end
+
+function Testcase:snapshot(id, data) --
+ if not data.test_case then return end
+ self.testcases[id] = data.test_case
+end
+
+---@return string[][]
+function Testcase:by_id(id) --
+ local testcases = {} ---@type string[][]
 
- local tbl = vim.api.nvim_buf_get_lines(self.bufnr, 0, -1, false)
- local str = table.concat(tbl, "\n")
+ local i, n = 0, self.testcase_len
+ for case in vim.gsplit(self.testcases[id] or "", "\n") do
+ local j = math.floor(i / n) + 1
 
- local testcases = {}
- for tcase in vim.gsplit(str, "\n\n") do
- local s = vim.split(tcase, "\n")
+ if not testcases[j] then testcases[j] = {} end
+ table.insert(testcases[j], case)
 
- table.insert(self.testcases, s)
- testcases = vim.list_extend(testcases, s)
+ i = i + 1
  end
 
  return testcases
 end
 
 function Testcase:populate()
- local tbl = {}
- for i, case in ipairs(self.console.question.q.testcase_list) do
- if i ~= 1 then table.insert(tbl, "") end
+ local lines = {}
 
- table.insert(self.testcases, vim.split(case, "\n"))
+ local t_list = self.console.question.q.testcase_list
+ self.testcase_len = #vim.split(t_list[1] or "", "\n")
 
+ for i, case in ipairs(t_list) do
+ if i ~= 1 then table.insert(lines, "") end
  for s in vim.gsplit(case, "\n", { trimempty = true }) do
- table.insert(tbl, s)
+ table.insert(lines, s)
  end
  end
 
- vim.api.nvim_buf_set_lines(self.bufnr, 0, -1, false, tbl)
-
+ vim.api.nvim_buf_set_lines(self.bufnr, 0, -1, false, lines)
  return self:draw_extmarks()
 end
 
@@ -140,6 +153,7 @@ function Testcase:mount()
 
  self.testcases = {}
  self.extmarks = {}
+ self.testcase_len = 0
 
  self:autocmds()
  self:populate()
@@ -172,6 +186,7 @@ function Testcase:init(parent)
 
  self.testcases = {}
  self.extmarks = {}
+ self.testcase_len = 0
 
  self:populate()
 end

lua/leetcode/api/interpreter.lua

@@ -83,8 +83,9 @@ function interpreter.run(submit, q, body, callback)
  end
  callback(nil, nil, err)
  else
- q.console.result:clear()
  local id = submit and res.submission_id or res.interpret_id
+ q.console.testcase:snapshot(id, res)
+ q.console.result:clear()
  interpreter.listener(id, callback)
  end
  end,

lua/leetcode/logger/spinner/init.lua

@@ -50,7 +50,7 @@ function spinner:spin()
 
  self.index = (self.index + 1) % #stype.frames
 
- local fps = 1000 / #stype.frames
+ local fps = math.floor(1000 / #stype.frames)
  vim.defer_fn(function() self:spin() end, fps)
 end
 
@@ -63,6 +63,9 @@ function spinner:set(msg, lvl, opts)
  if msg then self:update(msg) end
  lvl = lvl or vim.log.levels.INFO
 
+ local log = require("leetcode.logger")
+ log.debug(self.noti)
+
  opts = vim.tbl_deep_extend("force", {
  hide_from_history = true,
  title = config.name,
@@ -97,6 +100,10 @@ function spinner:stop(msg, success, opts)
 
  self.spinner = nil
  local lvl = vim.log.levels[success and "INFO" or "ERROR"]
+
+ local log = require("leetcode.logger")
+ log.debug({ noti = self.noti, msg = "spinner stop" })
+
  self:set(msg, lvl, opts)
 end
 

lua/leetcode/runner/init.lua

@@ -36,7 +36,7 @@ function Runner:handle(submit)
  lang = question.lang,
  typed_code = self.question:lines(submit),
  question_id = question.q.id,
- data_input = not submit and table.concat(question.console.testcase:content(), "\n") or nil,
+ data_input = not submit and question.console.testcase:content(),
  }
 
  local judge = Judge:init()