feat: implement queryCursor function and add some todos

lua/Trans/conf.lua

@@ -1,25 +1,30 @@
 return {
     display = {
-        style = 'default',
-        phnoetic = true,
+        style = 'minimal',
+        max_height = 50,
+        max_width = 50,
+        -- phnoetic = true,
         collins_star = true,
+        pos = true,
         tag = true,
         oxford = true,
-        history = false,
+        -- history = false,
+        exchange = true,
+        Trans_en = true,
+        Trans_zh = true,
     },
-    default_keymap = true,
-    map = {
-
-    },
-
     view = {
         -- TODO: style: buffer | cursor | window
         -- style = 'buffer',
         -- buffer_pos = 'bottom', -- only works when view.style == 'buffer'
     },
+    db_path = '/home/zoran/project/neovim/ecdict-ultimate-sqlite/ultimate.db', -- FIXME: change the path
 
-    db_path = '/home/zoran/project/neovim/ecdict-sqlite-28/stardict.db', -- FIXME: change the path
-
+    icon = {
+        star = '⭐',
+        isOxford = '✔',
+        notOxford = ''
+    }
     -- TODO: async process
     -- async = false,
 

lua/Trans/database.lua

@@ -14,8 +14,6 @@ function M.query(arg)
     else
         vim.notify('query argument error!')
     end
-
-    vim.pretty_print(res)
     return res[1]
 end
 

lua/Trans/display.lua

@@ -1,5 +1,14 @@
 -- local conf = require("Trans").conf
 local M = {}
+
+local display = require("Trans.conf").display
+local icon = require("Trans.conf").icon
+
+
+local buf = vim.api.nvim_create_buf(false, true)
+vim.api.nvim_buf_set_option(buf, 'filetype', 'Trans')
+
+
 -- {
 --     audio = "",
 --     bnc = 5222,
@@ -13,16 +22,120 @@ local M = {}
 --     tag = "",
 --     translation = "vt. 与...性交, 欺骗, 诅咒\nvi. 性交\nn. 性交, 些微, 杂种\ninterj. 他妈的, 混帐",
 --     word = "fuck"
---  } 
---
+--  }
+
+local function show_win(width, height)
+    vim.api.nvim_open_win(buf, true, {
+        relative = 'cursor',
+        title = 'Trans',
+        title_pos = 'center',
+        style = display.style,
+        row = 0, col = 0, width = width > display.max_width and display.max_width or width,
+        height = height > display.max_height and display.max_height or height,
+        border = 'rounded',
+        focusable = false,
+    })
+    -- vim.api.nvim_win_set_option(win, 'warp', true)
 
-function M.query_cursor()
-    -- TODO: 
 end
 
+-- @return string array
+local function get_text(query_res)
+    local text = {
+        -- NOTE: word + phonetic + collins_star
+        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 '')
+    }
+
+    -- NOTE: tag
+    if display.tag and query_res.tag:len() > 0 then
+        local tag = query_res.tag:gsub('zk', '中考'):gsub('gk', '高考'):gsub('ky', '考研'):gsub('cet4', '四级'):
+            gsub('cet6', '六级'):
+            gsub('ielts', '雅思'):gsub('toefl', '托福'):gsub('gre', 'GRE')
+        table.insert(text, '标签:')
+        table.insert(text, '    ' .. tag)
+    end
+    table.insert(text, '')
+
+    -- NOTE: pos 词性
+    if display.pos and query_res.pos:len() > 0 then
+        table.insert(text, '词性:')
+        -- TODO: figure out pos sense
+        table.insert(text, '    ' .. query_res.pos)
+    end
+
+    -- NOTE: exchange
+    if display.exchange and query_res.exchange:len() > 0 then
+        -- local list = vim.gsplit(query_res.exchange, [[\]])
+        table.insert(text, '词形变化:')
+        local exchange_map = {
+            p = '过去式',
+            d = '过去分词',
+            i = '现在分词',
+            r = '形容词比较级',
+            t = '形容词最高级',
+            s = '名词复数形式',
+            O = '词干',
+            ['3'] = '第三人称单数',
+        }
+        for v in vim.gsplit(query_res.exchange, [[/]]) do
+            table.insert(text, string.format('    %s:  %s', exchange_map[v:sub(1, 1)], v:sub(3)))
+            -- FIXME: 中文字符与字母位宽不一致, 暂时无法对齐
+        end
+    end
+    table.insert(text, '')
+
+    -- NOTE: 中文翻译
+    if display.Trans_zh and query_res.translation:len() > 0 then
+
+        table.insert(text, '中文翻译:')
+        for v in vim.gsplit(query_res.translation, '\n') do
+            -- table.insert(text, '    ' .. v)
+            table.insert(text, '    ' .. v)
+        end
+    end
+    table.insert(text, '')
+
+    -- NOTE: 英文翻译
+    if display.Trans_en and query_res.definition:len() > 0 then
+        table.insert(text, '英文翻译:')
+        for v in vim.gsplit(query_res.definition, '\n') do
+            table.insert(text, '    ' .. v)
+        end
+    end
+    table.insert(text, '')
+
+    return text
+end
+
+local function set_text(query_res)
+    local text = query_res and get_text(query_res) or { '没有找到相关定义' }
+
+    vim.api.nvim_buf_set_lines(buf, 0, -1, false, text)
+    local width = 0
+    for _, v in ipairs(text) do
+        if v:len() > width then
+            width = v:len()
+        end
+    end
+    return width, #text
+end
+
+function M.query_cursor()
+    vim.api.nvim_buf_set_option(buf, 'modifiable', true)
+    local word = vim.fn.expand('<cword>')
+    local res = require("Trans.database").query(word)
+
+    vim.pretty_print(res)
+
+    local width, height = set_text(res)
+    show_win(width, height)
+    vim.api.nvim_buf_set_option(buf, 'modifiable', false)
+end
 
 function M.query()
-    -- TODO: 
+    -- TODO:
 end
 
 return M

lua/Trans/setup.lua

@@ -1,5 +1,5 @@
 local db = require("Trans").db
-local conf = require("Trans").conf
+-- local conf = require("Trans").conf
 
 local group = vim.api.nvim_create_augroup('closedb', { clear = true })
 vim.api.nvim_create_autocmd('VimLeave', {
@@ -12,11 +12,4 @@ vim.api.nvim_create_autocmd('VimLeave', {
     end,
 })
 
-vim.api.nvim_create_user_command('TranslateCurosorWord', require("Trans").query_cursor, {
-
-})
-
-
-if conf.default_keymap then
-    --- TODO: keymap
-end
+vim.api.nvim_create_user_command('TranslateCurosorWord', require("Trans").query_cursor, {})