From 02fe6c510a7ea920c367c0e91df9b725b092eabb Mon Sep 17 00:00:00 2001
From: JuanZoran <1430359574@qq.com>
Date: Mon, 13 Mar 2023 22:39:40 +0800
Subject: [PATCH] fix: fix buffer line index error and online process need to
 be fixed

---
 lua/Trans/core/buffer.lua         | 22 ++++++++--------
 lua/Trans/frontend/hover/init.lua | 42 ++++++++++++++++++++++---------
 lua/Trans/frontend/hover/load.lua |  2 +-
 3 files changed, 43 insertions(+), 23 deletions(-)

diff --git a/lua/Trans/core/buffer.lua b/lua/Trans/core/buffer.lua
index c680b74..1193681 100644
--- a/lua/Trans/core/buffer.lua
+++ b/lua/Trans/core/buffer.lua
@@ -121,20 +121,23 @@ end
 
 ---Set line content
 ---@param nodes string|table|table[] string -> as line content | table -> as a node | table[] -> as node[]
----@param linenr number? line number should be set[one index] or let it be nil to append
-function buffer:setline(nodes, linenr)
+---@param one_index number? line number should be set[one index] or let it be nil to append
+function buffer:setline(nodes, one_index)
     self:set('modifiable', true)
-    linenr = linenr and linenr - 1 or self:line_count()
+    if not one_index then
+        local line_count = self:line_count()
+        one_index = (line_count == 1 and self[1] == '') and 1 or line_count + 1
+    end
 
+    local zero_index = one_index - 1
     if type(nodes) == 'string' then
-        api.nvim_buf_set_lines(self.bufnr, linenr, linenr, false, { nodes })
-
+        api.nvim_buf_set_lines(self.bufnr, zero_index, zero_index, false, { nodes })
     else
         if type(nodes[1]) == 'string' then
             -- FIXME :set [nodes] type as node
             ---@diagnostic disable-next-line: assign-type-mismatch
-            api.nvim_buf_set_lines(self.bufnr, linenr, linenr, false, { nodes[1] })
-            nodes:render(self, linenr, 0)
+            api.nvim_buf_set_lines(self.bufnr, zero_index, zero_index, false, { nodes[1] })
+            nodes:render(self, one_index, 0)
         else
             local strs = {}
             local num = #nodes
@@ -142,11 +145,11 @@ function buffer:setline(nodes, linenr)
                 strs[i] = nodes[i][1]
             end
 
-            api.nvim_buf_set_lines(self.bufnr, linenr, linenr, false, { table.concat(strs) })
+            api.nvim_buf_set_lines(self.bufnr, zero_index, zero_index, false, { table.concat(strs) })
             local col = 0
             for i = 1, num do
                 local node = nodes[i]
-                node:render(self, linenr, col)
+                node:render(self, one_index, col)
                 col = col + #node[1]
             end
         end
@@ -186,7 +189,6 @@ function buffer.new()
         extmarks = {},
     }, buffer)
 
-
     new_buf:set('modifiable', false)
     new_buf:set('filetype', 'Trans')
     new_buf:set('buftype', 'nofile')
diff --git a/lua/Trans/frontend/hover/init.lua b/lua/Trans/frontend/hover/init.lua
index 2d1a73c..3e49e7f 100644
--- a/lua/Trans/frontend/hover/init.lua
+++ b/lua/Trans/frontend/hover/init.lua
@@ -90,7 +90,6 @@ function M:init_window(opts)
     opts.animation  = m_opts.animation
 
 
-
     self.window = Trans.window.new(opts)
     return self.window
 end
@@ -120,27 +119,47 @@ function M:wait(tbl, name, timeout)
 
     local interval = math.floor(timeout / width)
     local pause = Trans.util.pause
+    local buffer = self.buffer
     for i = 1, width do
         if tbl[name] ~= nil then break end
-        self.buffer[1] = update_text(i)
-        self.buffer:add_highlight(1, 'MoreMsg')
+        buffer[1] = update_text(i)
+        buffer:add_highlight(1, 'MoreMsg')
         pause(interval)
     end
 
+    buffer[1] = ''
+
     -- TODO : End waitting animation
 end
 
-
 function M:process(_, result)
-    if not self.window then self:init_window() end
+    -- local node = Trans.util.node
+    -- local it, t, f = node.item, node.text, node.format
+    -- self.buffer:setline(it('hello', 'MoreMsg'))
+    local opts = self.opts
 
-    local node = Trans.util.node
-    local it, t, f = node.item, node.text, node.format
-    self.buffer:setline(it('hello', 'MoreMsg'))
+    for _, field in ipairs(opts.order) do
+        if result[field] then
+            self:load(result, field)
+        end
+    end
 
-    -- for _, field in ipairs(self.opts.order) do
-    --     self:load(result, field)
-    -- end
+    local win = self.window
+    if not win then
+        win = self:init_window()
+    elseif win:width() ~= opts.width then
+        win:expand {
+            field = 'width',
+            to = opts.width
+        }
+    elseif win:height() ~= opts.height then
+        win:expand {
+            field = 'height',
+            to = opts.height
+        }
+    end
+
+    win:set('wrap', true)
 end
 
 ---Check if hover window and buffer are valid
@@ -149,7 +168,6 @@ function M:is_available()
     return self.buffer:is_valid() and self.window:is_valid()
 end
 
-
 return M
 -- local function handle_keymap(win, word)
 --     local keymap = hover.keymap
diff --git a/lua/Trans/frontend/hover/load.lua b/lua/Trans/frontend/hover/load.lua
index 9119445..9b02611 100644
--- a/lua/Trans/frontend/hover/load.lua
+++ b/lua/Trans/frontend/hover/load.lua
@@ -32,7 +32,7 @@ local strategy = {
             hover.buffer:setline(it(word, 'TransWord'))
         else
             hover.buffer:setline(f {
-                width = hover.width,
+                width = hover.opts.width,
                 text = t {
                     it(word, 'TransWord'),
                     t {