Merge pull request kawre#41 from Eric-Song-Nop/inject_code

feat: custom injection code snippets

Author: kawre

README.md

@@ -145,6 +145,8 @@ To see full configuration types see [template.lua](./lua/leetcode/config/templat
         focus_testcases = "H", ---@type string
         focus_result = "L", ---@type string
     },
+
+    injector = {} ---@type table<lc.lang, lc.inject>
 }
 ```
 
@@ -198,6 +200,22 @@ Whether to log [leetcode.nvim] status notifications
 logging = true
 ```
 
+### injector
+
+Inject code before or after your solution, injected code won't be submitted or run.
+
+```lua
+injector = { ---@type table<lc.lang, lc.inject>
+    ["cpp"] = {
+        before = { "#include <bits/stdc++.h>", "using namespace std;" },
+        after = "int main() {}",
+    },
+    ["java"] = {
+        before = "import java.util.*;",
+    },
+}
+```
+
 ### hooks
 
 List of functions that get executed on specified event
@@ -207,7 +225,7 @@ hooks = {
     ---@type fun()[]
     LeetEnter = {},
 
-    ---@type fun(question: { lang: string })[]
+    ---@type fun(question: lc.ui.Question)[]
     LeetQuestionNew = {},
 },
 ```

README.zh.md

@@ -99,6 +99,8 @@ https://github.com/kawre/leetcode.nvim/assets/69250723/aee6584c-e099-4409-b114-1
     ---@type boolean
     logging = true,
 
+    injector = {} ---@type table<lc.lang, lc.inject>
+
     cache = {
         update_interval = 60 * 60 * 24 * 7, ---@type integer 7 days
     },
@@ -205,6 +207,22 @@ directory = vim.fn.stdpath("data") .. "/leetcode/"
 logging = true
 ```
 
+### injector
+
+在你的答案前后注入额外代码，注入的代码不会被提交或测试。
+
+```lua
+injector = { ---@type table<lc.lang, lc.inject>
+    ["cpp"] = {
+        before = { "#include <bits/stdc++.h>", "using namespace std;" },
+        after = "int main() {}",
+    },
+    ["java"] = {
+        before = "import java.util.*;",
+    },
+}
+```
+
 ### hooks
 
 在指定事件上执行的函数列表
@@ -214,7 +232,7 @@ hooks = {
     ---@type fun()[]
     LeetEnter = {},
 
-    ---@type fun(question: { lang: string })[]
+    ---@type fun(question: lc.ui.Question)[]
     LeetQuestionNew = {},
 },
 ```

lua/leetcode-ui/question.lua

@@ -25,7 +25,10 @@ function Question:get_snippet()
 
     local lang = utils.get_lang(self.lang)
     snip.code = (snip.code or ""):gsub("\r\n", "\n")
-    return ("%s @leet start\n%s\n%s @leet end"):format(lang.comment, snip.code, lang.comment)
+
+    return self:injector(
+        ("%s @leet start\n%s\n%s @leet end"):format(lang.comment, snip.code, lang.comment)
+    )
 end
 
 ---@private
@@ -37,6 +40,37 @@ function Question:create_file()
     if not self.file:exists() then self.file:write(self:get_snippet(), "w") end
 end
 
+---@private
+---@param code string
+function Question:injector(code)
+    local injector = config.user.injector
+
+    local inject = injector[self.lang]
+    if not inject or vim.tbl_isempty(inject) then return code end
+
+    ---@param inj? string|string[]
+    ---@param before boolean
+    local function norm_inject(inj, before)
+        local res
+
+        if type(inj) == "table" then
+            res = table.concat(inj, "\n")
+        elseif type(inj) == "string" then
+            res = inj
+        end
+
+        if res and res ~= "" then
+            return before and (res .. "\n\n") or ("\n\n" .. res)
+        else
+            return ""
+        end
+    end
+
+    return norm_inject(inject.before, true) --
+        .. code
+        .. norm_inject(inject.after, false)
+end
+
 Question.unmount = vim.schedule_wrap(function(self)
     self.info:unmount()
     self.console:unmount()

lua/leetcode/config/template.lua

@@ -37,6 +37,8 @@
 
 ---@alias lc.direction "col" | "row"
 
+---@alias lc.inject { before?: string|string[], after?: string|string[] }
+
 ---@class lc.UserConfig
 local M = {
     ---@type string
@@ -57,6 +59,8 @@ local M = {
     ---@type boolean
     logging = true,
 
+    injector = {}, ---@type table<lc.lang, lc.inject>
+
     cache = {
         update_interval = 60 * 60 * 24 * 7, ---@type integer 7 days
     },
@@ -98,9 +102,6 @@ local M = {
         LeetQuestionNew = {},
     },
 
-    ---@type boolean
-    image_support = false, -- setting this to `true` will disable question description wrap
-
     keys = {
         toggle = { "q", "<Esc>" }, ---@type string|string[]
         confirm = { "<CR>" }, ---@type string|string[]
@@ -110,6 +111,9 @@ local M = {
         focus_testcases = "H", ---@type string
         focus_result = "L", ---@type string
     },
+
+    ---@type boolean
+    image_support = false, -- setting this to `true` will disable question description wrap
 }
 
 return M