From 13af39b83ebbe632b265d5b84c3a0225022d4359 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Thu, 5 Jan 2023 16:52:02 +0100
Subject: [PATCH] refactor: easier to pass window options for floats

---
 lua/lazy/util.lua        | 23 ++++++++----
 lua/lazy/view/float.lua  | 80 ++++++++++++++++++++--------------------
 lua/lazy/view/init.lua   |  6 ++-
 lua/lazy/view/render.lua |  2 +-
 4 files changed, 62 insertions(+), 49 deletions(-)

diff --git a/lua/lazy/util.lua b/lua/lazy/util.lua
index 793e832..d1b4257 100644
--- a/lua/lazy/util.lua
+++ b/lua/lazy/util.lua
@@ -5,18 +5,22 @@ function M.file_exists(file)
   return vim.loop.fs_stat(file) ~= nil
 end
 
----@param opts? LazyViewOptions
+---@param opts? LazyFloatOptions
 function M.float(opts)
-  opts = vim.tbl_deep_extend("force", {
-    win_opts = { zindex = 60, border = "none" },
-    margin = { top = 3, left = 2, right = 2 },
-  }, opts or {})
+  opts = opts or {}
+  if require("lazy.view").visible() then
+    opts = vim.tbl_deep_extend("force", {
+      zindex = 60,
+      border = "none",
+      margin = { top = 3, left = 2, right = 2 },
+    }, opts)
+  end
   return require("lazy.view.float")(opts)
 end
 
 function M.open(uri)
   if M.file_exists(uri) then
-    return M.float({ win_opts = { style = "" }, file = uri })
+    return M.float({ style = "", file = uri })
   end
   local Config = require("lazy.core.config")
   local cmd
@@ -89,8 +93,13 @@ function M.throttle(ms, fn)
   end
 end
 
+---@class LazyCmdOpts
+---@field cwd? string
+---@field env? table<string,string>
+---@field float? LazyFloatOptions
+
 ---@param cmd string[]
----@param opts? {cwd:string, filetype:string, terminal?:boolean, close_on_exit?:boolean, enter?:boolean, float?:LazyViewOptions}
+---@param opts? {cwd:string, filetype:string, terminal?:boolean, close_on_exit?:boolean, enter?:boolean, float?:LazyFloatOptions}
 function M.open_cmd(cmd, opts)
   opts = opts or {}
   local float = M.float(opts.float)
diff --git a/lua/lazy/view/float.lua b/lua/lazy/view/float.lua
index 5539f35..6b22052 100644
--- a/lua/lazy/view/float.lua
+++ b/lua/lazy/view/float.lua
@@ -1,21 +1,21 @@
 local Config = require("lazy.core.config")
 local ViewConfig = require("lazy.view.config")
 
----@class LazyViewOptions
+---@class LazyFloatOptions
 ---@field buf? number
 ---@field file? string
 ---@field margin? {top?:number, right?:number, bottom?:number, left?:number}
----@field win_opts LazyViewWinOpts
 ---@field size? {width:number, height:number}
-local defaults = {
-  win_opts = {},
-}
+---@field zindex? number
+---@field style? "" | "minimal"
+---@field border? "none" | "single" | "double" | "rounded" | "solid" | "shadow"
 
 ---@class LazyFloat
 ---@field buf number
 ---@field win number
----@field opts LazyViewOptions
----@overload fun(opts?:LazyViewOptions):LazyFloat
+---@field opts LazyFloatOptions
+---@field win_opts LazyWinOpts
+---@overload fun(opts?:LazyFloatOptions):LazyFloat
 local M = {}
 
 setmetatable(M, {
@@ -24,16 +24,33 @@ setmetatable(M, {
   end,
 })
 
----@param opts? LazyViewOptions
+---@param opts? LazyFloatOptions
 function M.new(opts)
   local self = setmetatable({}, { __index = M })
   return self:init(opts)
 end
 
----@param opts? LazyViewOptions
+---@param opts? LazyFloatOptions
 function M:init(opts)
-  self.opts = vim.tbl_deep_extend("force", defaults, opts or {})
-  self.opts.size = vim.tbl_extend("keep", self.opts.size or {}, Config.options.ui.size)
+  self.opts = vim.tbl_deep_extend("force", {
+    size = Config.options.ui.size,
+    style = "minimal",
+    border = Config.options.ui.border,
+    zindex = 50,
+  }, opts or {})
+
+  ---@class LazyWinOpts
+  ---@field width number
+  ---@field height number
+  ---@field row number
+  ---@field col number
+  self.win_opts = {
+    relative = "editor",
+    style = self.opts.style ~= "" and self.opts.style or nil,
+    border = self.opts.border,
+    zindex = self.opts.zindex,
+    noautocmd = true,
+  }
   self:mount()
   self:on_key(ViewConfig.keys.close, self.close)
   self:on({ "BufDelete", "BufLeave", "BufHidden" }, self.close, { once = true })
@@ -44,25 +61,25 @@ function M:layout()
   local function size(max, value)
     return value > 1 and math.min(value, max) or math.floor(max * value)
   end
-  self.opts.win_opts.width = size(vim.o.columns, self.opts.size.width)
-  self.opts.win_opts.height = size(vim.o.lines, self.opts.size.height)
-  self.opts.win_opts.row = math.floor((vim.o.lines - self.opts.win_opts.height) / 2)
-  self.opts.win_opts.col = math.floor((vim.o.columns - self.opts.win_opts.width) / 2)
+  self.win_opts.width = size(vim.o.columns, self.opts.size.width)
+  self.win_opts.height = size(vim.o.lines, self.opts.size.height)
+  self.win_opts.row = math.floor((vim.o.lines - self.win_opts.height) / 2)
+  self.win_opts.col = math.floor((vim.o.columns - self.win_opts.width) / 2)
 
   if self.opts.margin then
     if self.opts.margin.top then
-      self.opts.win_opts.height = self.opts.win_opts.height - self.opts.margin.top
-      self.opts.win_opts.row = self.opts.win_opts.row + self.opts.margin.top
+      self.win_opts.height = self.win_opts.height - self.opts.margin.top
+      self.win_opts.row = self.win_opts.row + self.opts.margin.top
     end
     if self.opts.margin.right then
-      self.opts.win_opts.width = self.opts.win_opts.width - self.opts.margin.right
+      self.win_opts.width = self.win_opts.width - self.opts.margin.right
     end
     if self.opts.margin.bottom then
-      self.opts.win_opts.height = self.opts.win_opts.height - self.opts.margin.bottom
+      self.win_opts.height = self.win_opts.height - self.opts.margin.bottom
     end
     if self.opts.margin.left then
-      self.opts.win_opts.width = self.opts.win_opts.width - self.opts.margin.left
-      self.opts.win_opts.col = self.opts.win_opts.col + self.opts.margin.left
+      self.win_opts.width = self.win_opts.width - self.opts.margin.left
+      self.win_opts.col = self.win_opts.col + self.opts.margin.left
     end
   end
 end
@@ -78,25 +95,8 @@ function M:mount()
     self.buf = vim.api.nvim_create_buf(false, false)
   end
 
-  ---@class LazyViewWinOpts
-  ---@field width number
-  ---@field height number
-  ---@field row number
-  ---@field col number
-  local win_opts = {
-    relative = "editor",
-    style = "minimal",
-    border = Config.options.ui.border,
-    noautocmd = true,
-    zindex = 50,
-  }
-  self.opts.win_opts = vim.tbl_extend("force", win_opts, self.opts.win_opts)
-  if self.opts.win_opts.style == "" then
-    self.opts.win_opts.style = nil
-  end
-
   self:layout()
-  self.win = vim.api.nvim_open_win(self.buf, true, self.opts.win_opts)
+  self.win = vim.api.nvim_open_win(self.buf, true, self.win_opts)
   self:focus()
 
   vim.bo[self.buf].buftype = "nofile"
@@ -118,7 +118,7 @@ function M:mount()
       local config = {}
       for _, key in ipairs({ "relative", "width", "height", "col", "row" }) do
         ---@diagnostic disable-next-line: no-unknown
-        config[key] = self.opts.win_opts[key]
+        config[key] = self.win_opts[key]
       end
       vim.api.nvim_win_set_config(self.win, config)
       vim.cmd([[do User LazyFloatResized]])
diff --git a/lua/lazy/view/init.lua b/lua/lazy/view/init.lua
index cd91c4c..0f4e0d2 100644
--- a/lua/lazy/view/init.lua
+++ b/lua/lazy/view/init.lua
@@ -25,13 +25,17 @@ local M = {}
 ---@type LazyView
 M.view = nil
 
+function M.visible()
+  return M.view and M.view.win and vim.api.nvim_win_is_valid(M.view.win)
+end
+
 ---@param mode? string
 function M.show(mode)
   if Config.headless then
     return
   end
 
-  M.view = (M.view and M.view.win and vim.api.nvim_win_is_valid(M.view.win)) and M.view or M.create()
+  M.view = M.visible() and M.view or M.create()
   if mode then
     M.view.state.mode = mode
   end
diff --git a/lua/lazy/view/render.lua b/lua/lazy/view/render.lua
index f5ce860..fd4b138 100644
--- a/lua/lazy/view/render.lua
+++ b/lua/lazy/view/render.lua
@@ -26,7 +26,7 @@ function M.new(view)
   local self = setmetatable({}, { __index = setmetatable(M, { __index = Text }) })
   self.view = view
   self.padding = 2
-  self.wrap = view.opts.win_opts.width
+  self.wrap = view.win_opts.width
   return self
 end