feat: winfixbuf support (kawre#137)

* feat: add `winfixbuf` support

* feat(utils): `with_version`

* refactor: `cmd.menu()`

* feat(ui-utils): new utility

* fix(yank): check for `0.10`

Author: kawre

lua/leetcode-ui/popup/info.lua

@@ -134,12 +134,12 @@ function Info:mount()
     local ui_utils = require("leetcode-ui.utils")
     local winhighlight = "Normal:NormalSB,FloatBorder:FloatBorder"
 
-    ui_utils.set_win_opts(self.winid, {
+    ui_utils.win_set_opts(self.winid, {
         winhighlight = winhighlight,
         wrap = true,
     })
 
-    ui_utils.set_win_opts(self.border.winid, {
+    ui_utils.win_set_opts(self.border.winid, {
         winhighlight = winhighlight,
     })
 

lua/leetcode-ui/question.lua

@@ -5,6 +5,7 @@ local Object = require("nui.object")
 
 local api_question = require("leetcode.api.question")
 local utils = require("leetcode.utils")
+local ui_utils = require("leetcode-ui.utils")
 local config = require("leetcode.config")
 local log = require("leetcode.logger")
 
@@ -88,14 +89,15 @@ function Question:create_buffer()
     vim.cmd("$tabe " .. path)
     self.bufnr = vim.api.nvim_get_current_buf()
     self.winid = vim.api.nvim_get_current_win()
+    ui_utils.win_set_winfixbuf(self.winid)
 
     self:open_buffer(existed)
 end
 
 ---@param existed boolean
 function Question:open_buffer(existed)
-    vim.api.nvim_win_set_buf(self.winid, self.bufnr)
-    vim.api.nvim_set_option_value("buflisted", true, { buf = self.bufnr })
+    ui_utils.buf_set_opts(self.bufnr, { buflisted = true })
+    ui_utils.win_set_buf(self.winid, self.bufnr, true)
 
     local i = self:fold_range()
     if i then

lua/leetcode-ui/renderer/init.lua

@@ -37,7 +37,7 @@ function Renderer:draw(component)
     self._.line_idx = 1
 
     local c_ok, c = pcall(vim.api.nvim_win_get_cursor, self.winid)
-    self:modifiable(function()
+    self:with_modifiable(function()
         vim.api.nvim_buf_set_lines(self.bufnr, 0, -1, false, {})
         vim.api.nvim_buf_clear_namespace(self.bufnr, -1, 0, -1)
 
@@ -51,7 +51,7 @@ end
 ---@private
 ---
 ---@param fn function
-function Renderer:modifiable(fn)
+function Renderer:with_modifiable(fn)
     local bufnr = self.bufnr
     if not (bufnr and vim.api.nvim_buf_is_valid(bufnr)) then
         return
@@ -112,7 +112,7 @@ function Renderer:clear()
     self._.line_idx = 1
     self._.buttons = {}
     self:clear_keymaps()
-    self:modifiable(function()
+    self:with_modifiable(function()
         vim.api.nvim_buf_set_lines(self.bufnr, 0, -1, false, {})
     end)
 end

lua/leetcode-ui/renderer/menu.lua

@@ -1,7 +1,8 @@
 local log = require("leetcode.logger")
 local cookie = require("leetcode.cache.cookie")
 local config = require("leetcode.config")
-local utils = require("leetcode-ui.utils")
+local ui_utils = require("leetcode-ui.utils")
+local utils = require("leetcode.utils")
 local Renderer = require("leetcode-ui.renderer")
 local api = vim.api
 
@@ -123,7 +124,7 @@ end
 function Menu:apply_options()
     api.nvim_buf_set_name(self.bufnr, "")
 
-    utils.set_buf_opts(self.bufnr, {
+    ui_utils.buf_set_opts(self.bufnr, {
         modifiable = false,
         buflisted = false,
         matchpairs = "",
@@ -132,7 +133,7 @@ function Menu:apply_options()
         filetype = config.name,
         synmaxcol = 0,
     })
-    utils.set_win_opts(self.winid, {
+    ui_utils.win_set_opts(self.winid, {
         wrap = false,
         colorcolumn = "",
         foldlevel = 999,
@@ -145,6 +146,7 @@ function Menu:apply_options()
         spell = false,
         signcolumn = "no",
     })
+    ui_utils.win_set_winfixbuf(self.winid)
 end
 
 function Menu:unmount()
@@ -173,7 +175,7 @@ function Menu:remount()
         api.nvim_buf_delete(self.bufnr, { force = true })
     end
 
-    vim.cmd("$tabnew")
+    vim.cmd("0tabnew")
     self.bufnr = api.nvim_get_current_buf()
     self.winid = api.nvim_get_current_win()
 

lua/leetcode-ui/split/description.lua

@@ -34,7 +34,7 @@ function Description:mount()
     self:populate()
 
     local ui_utils = require("leetcode-ui.utils")
-    ui_utils.set_buf_opts(self.bufnr, {
+    ui_utils.buf_set_opts(self.bufnr, {
         modifiable = false,
         buflisted = false,
         matchpairs = "",
@@ -43,7 +43,7 @@ function Description:mount()
         filetype = config.name,
         synmaxcol = 0,
     })
-    ui_utils.set_win_opts(self.winid, {
+    ui_utils.win_set_opts(self.winid, {
         winhighlight = "Normal:NormalFloat,FloatBorder:FloatBorder",
         wrap = not img_sup,
         colorcolumn = "",
@@ -58,6 +58,8 @@ function Description:mount()
         signcolumn = "no",
         linebreak = true,
     })
+    ui_utils.win_set_winfixbuf(self.winid)
+
     if not img_ok and config.user.image_support then
         log.error("image.nvim not found but `image_support` is enabled")
     end

lua/leetcode-ui/utils.lua

@@ -103,7 +103,7 @@ function utils.get_padding(lines, layout)
     return padding
 end
 
-function utils.set_buf_opts(bufnr, options)
+function utils.buf_set_opts(bufnr, options)
     if not vim.api.nvim_buf_is_valid(bufnr) then
         return
     end
@@ -116,7 +116,7 @@ function utils.set_buf_opts(bufnr, options)
     end
 end
 
-function utils.set_win_opts(winid, options)
+function utils.win_set_opts(winid, options)
     if not vim.api.nvim_win_is_valid(winid) then
         return
     end
@@ -130,6 +130,35 @@ function utils.set_win_opts(winid, options)
     end
 end
 
+---@param winid number
+function utils.win_set_winfixbuf(winid)
+    local u = require("leetcode.utils")
+    u.with_version("0.10.0", function()
+        utils.win_set_opts(winid, { winfixbuf = true })
+    end)
+end
+
+---@param winid number
+---@param bufnr number
+---@param force? boolean
+function utils.win_set_buf(winid, bufnr, force)
+    local u = require("leetcode.utils")
+
+    u.with_version("0.10.0", function()
+        local wfb = vim.api.nvim_win_get_option(winid, "winfixbuf")
+
+        if not wfb then
+            vim.api.nvim_win_set_buf(winid, bufnr)
+        elseif force then
+            utils.win_set_opts(winid, { winfixbuf = false })
+            vim.api.nvim_win_set_buf(winid, bufnr)
+            utils.win_set_opts(winid, { winfixbuf = true })
+        end
+    end, function()
+        vim.api.nvim_win_set_buf(winid, bufnr)
+    end)
+end
+
 function utils.is_instance(instance, class)
     return type(instance) == "table" and O.is_instance(instance, class)
 end

lua/leetcode/command/init.lua

@@ -177,10 +177,11 @@ end
 function cmd.menu()
     local winid, bufnr = _Lc_state.menu.winid, _Lc_state.menu.bufnr
     local ok, tabp = pcall(api.nvim_win_get_tabpage, winid)
+    local ui = require("leetcode-ui.utils")
 
     if ok then
         api.nvim_set_current_tabpage(tabp)
-        api.nvim_win_set_buf(winid, bufnr)
+        ui.win_set_buf(winid, bufnr)
     else
         _Lc_state.menu:remount()
     end
@@ -198,7 +199,9 @@ function cmd.yank()
         and (q.winid and api.nvim_win_is_valid(q.winid))
     then
         api.nvim_set_current_win(q.winid)
-        api.nvim_set_current_buf(q.bufnr)
+        utils.with_version("0.10.0", nil, function()
+            api.nvim_set_current_buf(q.bufnr)
+        end)
 
         local start_i, end_i, lines = q:range()
         vim.cmd(("%d,%dyank"):format(start_i or 1, end_i or #lines))
@@ -358,6 +361,7 @@ end
 
 function cmd.restore()
     local utils = require("leetcode.utils")
+    local ui = require("leetcode-ui.utils")
     local q = utils.curr_question()
     if not q then
         return
@@ -367,7 +371,7 @@ function cmd.restore()
         (q.winid and api.nvim_win_is_valid(q.winid))
         and (q.bufnr and api.nvim_buf_is_valid(q.bufnr))
     then
-        api.nvim_win_set_buf(q.winid, q.bufnr)
+        ui.win_set_buf(q.winid, q.bufnr)
     end
 
     q.description:show()
@@ -377,7 +381,7 @@ function cmd.restore()
         (winid and api.nvim_win_is_valid(winid)) --
         and (bufnr and api.nvim_buf_is_valid(bufnr))
     then
-        api.nvim_win_set_buf(winid, bufnr)
+        ui.win_set_buf(q.winid, q.bufnr)
     end
 end
 

lua/leetcode/utils.lua

@@ -138,4 +138,10 @@ function utils.norm_ins(str)
     return ins:sub(2, #ins - 1)
 end
 
+function utils.with_version(v, with, without)
+    with = with or function() end
+    without = without or function() end
+    return (vim.fn.has("nvim-" .. v) == 1 and with or without)()
+end
+
 return utils