feat: @leet tags

lua/leetcode-ui/question.lua

@@ -58,6 +58,10 @@ function Question:inject(before)
 
  local res
 
+ if type(inj) == "boolean" and inj == true and before then --
+ inj = config.imports[self.lang]
+ end
+
  if type(inj) == "table" then
  res = table.concat(inj, "\n")
  elseif type(inj) == "string" then
@@ -129,7 +133,7 @@ function Question:handle_mount()
  self.console = Console(self)
  self.info = Info(self)
 
- utils.exec_hooks("LeetQuestionNew", self)
+ utils.exec_hook("LeetQuestionNew", self)
 
  return self
 end
@@ -163,7 +167,7 @@ end
 ---@return integer, integer, string[]
 function Question:range(inclusive)
  local lines = vim.api.nvim_buf_get_lines(self.bufnr, 0, -1, false)
- local start_i, end_i = 1, #lines
+ local start_i, end_i
 
  for i, line in ipairs(lines) do
  if line:match("@leet start") then
@@ -180,6 +184,10 @@ end
 ---@return string
 function Question:lines(submit)
  local start_i, end_i, lines = self:range()
+
+ start_i = start_i or 1
+ end_i = end_i or #lines
+
  local prefix = not submit and ("\n"):rep(start_i - 1) or ""
  return prefix .. table.concat(lines, "\n", start_i, end_i)
 end
@@ -201,7 +209,7 @@ Question.change_lang = vim.schedule_wrap(function(self, lang)
 
  self.bufnr = new_bufnr
  if bufloaded == 0 then --
- utils.exec_hooks("LeetQuestionNew", self)
+ utils.exec_hook("LeetQuestionNew", self)
  end
  else
  log.error("Changing language failed")

lua/leetcode.lua

@@ -58,7 +58,7 @@ function leetcode.start(on_vimenter)
  Menu():mount()
 
  local utils = require("leetcode.utils")
- utils.exec_hooks("LeetEnter")
+ utils.exec_hook("LeetEnter")
 
  return true
 end

lua/leetcode/command/init.lua

@@ -178,8 +178,8 @@ function cmd.yank()
  api.nvim_set_current_win(q.winid)
  api.nvim_set_current_buf(q.bufnr)
 
- local start_i, end_i = q:range()
- vim.cmd(("%d,%dyank"):format(start_i, end_i))
+ local start_i, end_i, lines = q:range()
+ vim.cmd(("%d,%dyank"):format(start_i or 1, end_i or #lines))
  end
 end
 
@@ -341,17 +341,30 @@ function cmd.inject()
  local q = utils.curr_question()
  if not q then return end
 
- local start_i, end_i = q:range(true)
-
  if vim.api.nvim_buf_is_valid(q.bufnr) then
- local before = q:inject(true)
- if before then
- vim.api.nvim_buf_set_lines(q.bufnr, 0, start_i - 1, false, vim.split(before, "\n"))
+ local start_i, end_i = q:range(true)
+
+ if start_i == nil and end_i == nil then
+ log.error("`@leet start` and `@leet end` not found")
+ return
  end
 
- local after = q:inject(false)
- if after then
- vim.api.nvim_buf_set_lines(q.bufnr, end_i + 1, -1, false, vim.split(after, "\n"))
+ if start_i == nil then
+ log.error("`@leet start` not found")
+ else
+ local before = q:inject(true)
+ if before then
+ vim.api.nvim_buf_set_lines(q.bufnr, 0, start_i - 1, false, vim.split(before, "\n"))
+ end
+ end
+
+ if end_i == nil then
+ log.error("`@leet start` not found")
+ else
+ local after = q:inject(false)
+ if after then
+ vim.api.nvim_buf_set_lines(q.bufnr, end_i + 1, -1, false, vim.split(after, "\n"))
+ end
  end
  end
 end

lua/leetcode/utils.lua

@@ -72,7 +72,7 @@ function utils.get_lang_by_name(name)
 end
 
 ---@param event lc.hook
-function utils.exec_hooks(event, ...)
+function utils.exec_hook(event, ...)
  local fns = config.user.hooks[event]
  if not fns then log.error("unknown hook event: " .. event) end