feat: add query from input word cmd

lua/Trans/display.lua

@@ -13,6 +13,7 @@ local win = 0
 local line = 0
 local pos_info = {}
 
+local handler = {}
 api.nvim_buf_set_option(buf, 'filetype', 'Trans')
 
 local function show_win(width, height)
@@ -33,7 +34,7 @@ local function show_win(width, height)
 end
 
 -- NOTE: title
-local function get_title(text, query_res)
+handler.title =  function (text, query_res)
     local title = string.format('%s    [%s]    ', query_res.word, query_res.phonetic) ..
         (display.oxford and (query_res.oxford == 1 and icon.isOxford .. '    ' or icon.notOxford .. '    ') or '') ..
         ((display.collins_star and query_res.collins) and string.rep(icon.star, query_res.collins) or '')
@@ -47,11 +48,11 @@ local function get_title(text, query_res)
 end
 
 -- NOTE: tag
-local function get_tag(text, query_res)
+handler.tag = function(text, query_res)
     if query_res.tag and #query_res.tag > 0 then
         local tag = query_res.tag:gsub('zk', '中考'):gsub('gk', '高考'):gsub('ky', '考研'):gsub('cet4', '四级'):
-            gsub('cet6', '六级'):
-            gsub('ielts', '雅思'):gsub('toefl', '托福'):gsub('gre', 'GRE')
+            gsub('cet6', '六级'):gsub('ielts', '雅思'):gsub('toefl', '托福'):gsub('gre', 'GRE')
+
         table.insert(text, '标签:')
         table.insert(text, '    ' .. tag)
         table.insert(text, '')
@@ -62,7 +63,7 @@ local function get_tag(text, query_res)
 end
 
 -- NOTE: pos 词性
-local function get_pos(text, query_res)
+handler.pos = function(text, query_res)
     if query_res.pos and #query_res.pos > 0 then
         table.insert(text, '词性:')
 
@@ -82,7 +83,7 @@ local function get_pos(text, query_res)
 end
 
 -- NOTE: exchange
-local function get_exchange(text, query_res)
+handler.exchange = function(text, query_res)
     if query_res.exchange and #query_res.exchange > 0 then
         table.insert(text, '词形变化:')
 
@@ -113,7 +114,7 @@ local function get_exchange(text, query_res)
 end
 
 -- NOTE: 中文翻译
-local function get_zh(text, query_res)
+handler.zh = function(text, query_res)
     if query_res.translation then
         table.insert(text, '中文翻译:')
 
@@ -132,7 +133,7 @@ local function get_zh(text, query_res)
 end
 
 -- NOTE: 英文翻译
-local function get_en(text, query_res)
+handler.en = function(text, query_res)
     if query_res.definition and #query_res.definition > 0 then
         table.insert(text, '英文翻译:')
 
@@ -150,16 +151,6 @@ local function get_en(text, query_res)
     end
 end
 
-local handler = {
-    title = get_title,
-    tag = get_tag,
-    pos = get_pos,
-    exchange = get_exchange,
-    zh = get_zh,
-    en = get_en,
-}
-
-
 -- @return string array
 local function get_text(query_res)
     local text = {}
@@ -169,6 +160,7 @@ local function get_text(query_res)
     return text
 end
 
+
 local function set_text(query_res)
     local text = query_res and get_text(query_res) or { '没有找到相关定义' }
 
@@ -184,7 +176,6 @@ end
 
 local hl_handler = {}
 
-
 hl_handler.title = function()
     api.nvim_buf_add_highlight(buf, -1, hl.word, pos_info.title.line, 0, pos_info.title.word)
     api.nvim_buf_add_highlight(buf, -1, hl.phonetic, pos_info.title.line, pos_info.title.word + 5,
@@ -239,6 +230,7 @@ local function set_hl()
     end
 end
 
+
 local function clear_tmp_info()
     pos_info = {}
     line = 0
@@ -260,9 +252,12 @@ function M.query(mode)
         word = vim.fn.expand('<cword>')
     elseif mode == 'v' then
         word = get_visual_selection()
+    elseif mode == 'I' then
+        vim.ui.input({prompt = '请输入您要输查询的单词: '}, function (input)
+            word = input
+        end)
     else
-        print(mode, 'is invalid')
-        assert(false)
+        error('mode argument is invalid')
     end
 
     local res = require("Trans.database").query(word)
@@ -291,6 +286,10 @@ function M.query_select()
     M.query('v')
 end
 
+function M.query_input()
+    M.query('I')
+end
+
 function M.close_win()
     if win > 0 then
         api.nvim_win_close(win, true)

lua/Trans/highlight.lua

@@ -13,13 +13,13 @@ M.hlgroup = {
 
 function M.set_hl()
     local set_hl = vim.api.nvim_set_hl
-    set_hl(0, M.hlgroup.word, { fg = '#98c379', bold = true })
+    set_hl(0, M.hlgroup.word, { fg = '#7ee787', bold = true })
     set_hl(0, M.hlgroup.phonetic, { fg = '#8b949e' })
     set_hl(0, M.hlgroup.ref, { fg = '#75beff', bold = true })
     set_hl(0, M.hlgroup.tag, { fg = '#e5c07b' })
     set_hl(0, M.hlgroup.pos, { link = M.hlgroup.tag })
     set_hl(0, M.hlgroup.exchange, { link = M.hlgroup.tag })
-    set_hl(0, M.hlgroup.zh, { fg = '#7ee787' })
+    set_hl(0, M.hlgroup.zh, { link = M.hlgroup.word })
     set_hl(0, M.hlgroup.en, { fg = '#bc8cff' })
 end
 

lua/Trans/setup.lua

@@ -4,6 +4,7 @@ local db = require("Trans").db
 
 vim.api.nvim_create_user_command('TranslateCursorWord', require("Trans.display").query_cursor, {})
 vim.api.nvim_create_user_command('TranslateSelectWord', require("Trans.display").query_select, {})
+vim.api.nvim_create_user_command('TranslateInputWord', require("Trans.display").query_input, {})
 
 
 local group = vim.api.nvim_create_augroup("Trans", { clear = true })