From ba08913b1ee1db535d1fe9696fd3e94d410ad548 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B0=8F=E5=B0=9B?= <m19357513004@163.com>
Date: Tue, 25 Apr 2023 13:48:52 +0800
Subject: [PATCH] feat: get_lines

---
 lua/Trans/core/util.lua | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/lua/Trans/core/util.lua b/lua/Trans/core/util.lua
index 2b84603..0450d33 100644
--- a/lua/Trans/core/util.lua
+++ b/lua/Trans/core/util.lua
@@ -32,10 +32,30 @@ function M.get_select()
         local e = #lines
         lines[1] = lines[1]:sub(s_col)
         lines[e] = line:sub(1, e_col)
-        return table.concat(lines)
+        return table.concat(lines, '\n')
     end
 end
 
+---Get selected text
+---@return string
+function M.get_lines()
+  local _start = vim.fn.getpos 'v'
+  local _end = vim.fn.getpos '.'
+
+  if _start[2] > _end[2] then
+    _start, _end = _end, _start
+  end
+
+  local s_row, e_row = _start[2], _end[2]
+
+  if s_row == e_row then
+    return vim.fn.getline(s_row)
+  else
+    local lines = vim.fn.getline(s_row, e_row)
+    return table.concat(lines, "\n")
+  end
+end
+
 ---Get Text which need to be translated
 ---@param mode string
 ---@return string
@@ -52,8 +72,8 @@ function M.get_str(mode)
             return fn.input '需要翻译的字符串: '
         end,
         V = function()
-            print 'TODO'
-            return ''
+            api.nvim_input '<Esc>'
+            return M.get_lines()
         end,
     })[mode]():match '^%s*(.-)%s*$'
 end