feat: add TranslateInput command

lua/Trans/core/translate.lua

@@ -8,6 +8,7 @@ local function init_opts(opts)
         v = 'visual',
     })[vim.api.nvim_get_mode().mode]
 
+
     opts.str = util.get_str(opts.mode)
     return opts
 end

lua/Trans/core/util.lua

@@ -1,5 +1,3 @@
-local Trans = require('Trans')
-
 local fn, api = vim.fn, vim.api
 
 ---@class TransUtil
@@ -42,16 +40,18 @@ end
 ---@param mode TransMode
 ---@return string
 function M.get_str(mode)
-    if mode == 'n' or mode == 'normal' then
+    return ({
-        return fn.expand('<cword>')
+        normal = function()
-    elseif mode == 'v' or mode == 'visual' then
+            return fn.expand('<cword>')
-        api.nvim_input('<ESC>')
+        end,
-        return M.get_select()
+        visual = function()
-    elseif mode == 'input' then
+            api.nvim_input('<Esc>')
-        return fn.expand('<cword>')
+            return M.get_select()
-    else
+        end,
-        error('Unsupported mode' .. mode)
+        input = function()
-    end
+            return fn.input('需要翻译的字符串: ')
+        end,
+    })[mode]():match('^%s*(.-)%s*$')
 end
 
 ---Puase coroutine for {ms} milliseconds
@@ -166,7 +166,7 @@ end
 ---@param opts { winid: integer, height: integer }?
 ---@return string[]
 function M.visible_lines(opts)
-    opts = opts or {}
+    opts                        = opts or {}
 
     -- INFO : don't calculate the height of statusline and cmdheight or winbar?
     local winid                 = opts.winid or 0
@@ -188,7 +188,6 @@ function M.is_word(str)
     return str:match('%w+') == str
 end
 
-
 ---@class Trans
 ---@field util TransUtil
 return M

plugin/Trans.lua

@@ -26,11 +26,16 @@ local command = api.nvim_create_user_command
 
 command("Translate", function()
     Trans.translate()
-end, { desc = "  单词翻译" })
+end, { desc = "  Translate cursor word" })
+
+
+command("TranslateInput", function()
+    Trans.translate({ mode = 'input' })
+end, { desc = "  Translate input word" })
 
 command("TransPlay", function()
     local str = Trans.util.get_str(api.nvim_get_mode().mode)
     if str and str ~= "" and Trans.util.is_English(str) then
         str:play()
     end
-end, { desc = " 自动发音" })
+end, { desc = " auto play" })