refactor: leetcode.open

Author: kawre

README.md

@@ -442,7 +442,7 @@ https://github.com/kawre/leetcode.nvim/assets/69250723/b7be8b95-5e2c-4153-8845-4
   ```lua
   {
       "kawre/leetcode.nvim",
-      lazy = leet_arg ~= vim.fn.argv()[1],
+      lazy = leet_arg ~= vim.fn.argv(0, -1),
       opts = { arg = leet_arg },
   }
   ```

lua/leetcode-plugins/non_standalone/init.lua

@@ -5,7 +5,7 @@ non_standalone.opts = {
     lazy = false,
 }
 
-function non_standalone.load() --
+function non_standalone.load()
     require("leetcode-plugins.non_standalone.leetcode")
 end
 

lua/leetcode-plugins/non_standalone/leetcode.lua

@@ -1,42 +1,14 @@
+---@diagnostic disable: invisible, duplicate-set-field
+
 local leetcode = require("leetcode")
 local config = require("leetcode.config")
 
-local standalone = true
+local is_standalone = true
 local prev_cwd = nil
 
----@param on_vimenter boolean
-leetcode.should_skip = function(on_vimenter)
-    if on_vimenter then
-        if vim.fn.argc() ~= 1 then
-            return true
-        end
-
-        local usr_arg, arg = vim.fn.argv()[1], config.user.arg
-        if usr_arg ~= arg then
-            return true
-        end
-
-        local lines = vim.api.nvim_buf_get_lines(0, 0, -1, true)
-        if #lines > 1 or (#lines == 1 and lines[1]:len() > 0) then
-            local log = require("leetcode.logger")
-            log.warn(("Failed to initialize: `%s` is not an empty buffer"):format(usr_arg))
-            return true
-        end
-    else
-        for _, buf_id in pairs(vim.api.nvim_list_bufs()) do
-            local bufinfo = vim.fn.getbufinfo(buf_id)[1]
-            if bufinfo and (bufinfo.listed == 1 and #bufinfo.windows > 0) then
-                return false, true
-            end
-        end
-    end
-
-    return false
-end
-
 ---@param on_vimenter boolean
 leetcode.start = function(on_vimenter)
-    local skip, buflisted = leetcode.should_skip(on_vimenter)
+    local skip, standalone = leetcode.should_skip(on_vimenter)
     if skip then
         return false
     end
@@ -49,14 +21,14 @@ leetcode.start = function(on_vimenter)
     theme.setup()
 
     if not on_vimenter then
-        if buflisted then
+        if not standalone then
             prev_cwd = vim.fn.getcwd()
             vim.cmd.tabe()
         else
             vim.cmd.enew()
         end
 
-        standalone = not buflisted
+        is_standalone = standalone ---@diagnostic disable-line: cast-local-type
     end
 
     vim.api.nvim_set_current_dir(config.storage.home:absolute())
@@ -71,7 +43,7 @@ leetcode.start = function(on_vimenter)
 end
 
 leetcode.stop = vim.schedule_wrap(function()
-    if standalone then
+    if is_standalone then
         return vim.cmd("qa!")
     end
 

lua/leetcode.lua

@@ -3,38 +3,81 @@ local config = require("leetcode.config")
 ---@class lc.LeetCode
 local leetcode = {}
 
+---@private
+local function log_failed_to_init()
+    local log = require("leetcode.logger")
+    log.warn("Failed to initialize: `neovim` contains listed buffers")
+end
+
+local function log_buf_not_empty(bufname)
+    local log = require("leetcode.logger")
+    if bufname and bufname ~= "" then
+        log.warn(("Failed to initialize: `%s` is not an empty buffer"):format(bufname))
+    else
+        log.warn("Failed to initialize: not an empty buffer")
+    end
+end
+
+local function buf_is_empty(buf)
+    local lines = vim.api.nvim_buf_get_lines(buf, 0, -1, true)
+    return not (#lines > 1 or (#lines == 1 and lines[1]:len() > 0))
+end
+
 ---@param on_vimenter boolean
 ---
----@return boolean
+---@return boolean, boolean? (skip, standalone)
 function leetcode.should_skip(on_vimenter)
     if on_vimenter then
-        if vim.fn.argc() ~= 1 then
+        if vim.fn.argc(-1) ~= 1 then
             return true
         end
 
-        local usr_arg, arg = config.user.arg, vim.fn.argv()[1]
+        local usr_arg, arg = config.user.arg, vim.fn.argv(0, -1)
         if usr_arg ~= arg then
             return true
         end
 
-        local lines = vim.api.nvim_buf_get_lines(0, 0, -1, true)
-        if #lines > 1 or (#lines == 1 and lines[1]:len() > 0) then
-            local log = require("leetcode.logger")
-            log.warn(("Failed to initialize: `%s` is not an empty buffer"):format(usr_arg))
+        if not buf_is_empty(0) then
+            log_buf_not_empty(usr_arg)
             return true
         end
+
+        return false, true
     else
-        for _, buf_id in pairs(vim.api.nvim_list_bufs()) do
-            local bufinfo = vim.fn.getbufinfo(buf_id)[1]
-            if bufinfo and (bufinfo.listed == 1 and #bufinfo.windows > 0) then
-                local log = require("leetcode.logger")
-                log.warn("Failed to initialize: `neovim` contains listed buffers")
+        local listed_bufs = vim.tbl_filter(function(info)
+            return info.listed == 1
+        end, vim.fn.getbufinfo())
+
+        if #listed_bufs == 0 then
+            return false, true
+        elseif vim.fn.argc(-1) == 0 and #listed_bufs == 1 then
+            local buf = listed_bufs[1]
+
+            if vim.api.nvim_get_current_buf() ~= buf.bufnr then
+                if config.plugins.non_standalone then
+                    return false, false
+                else
+                    log_failed_to_init()
+                    return true
+                end
+            end
+
+            vim.schedule(function()
+                if buf.changed == 1 then
+                    vim.api.nvim_buf_delete(buf.bufnr, { force = true })
+                end
+            end)
+
+            return false, true
+        elseif #listed_bufs >= 1 then
+            if config.plugins.non_standalone then
+                return false, false
+            else
+                log_failed_to_init()
                 return true
             end
         end
     end
-
-    return false
 end
 
 function leetcode.setup_cmds()
@@ -43,7 +86,8 @@ end
 
 ---@param on_vimenter boolean
 function leetcode.start(on_vimenter)
-    if leetcode.should_skip(on_vimenter) then
+    local skip = leetcode.should_skip(on_vimenter)
+    if skip then
         return false
     end
 

lua/leetcode/config/init.lua

@@ -21,6 +21,7 @@ local config = {
     version = "1.0.1",
     storage = {}, ---@type table<string, Path>
     theme = {}, ---@type lc.highlights
+    plugins = {},
 
     translator = false,
 
@@ -98,6 +99,7 @@ function config.validate()
 end
 
 function config.load_plugins()
+    config.plugins = {}
     local plugins = {}
 
     if config.user.cn.enabled then
@@ -119,6 +121,7 @@ function config.load_plugins()
             else
                 table.insert(lazy_plugs, plug.load)
             end
+            config.plugins[plugin] = true
         else
             table.insert(lazy_plugs, function()
                 local log = require("leetcode.logger")