From 76a33393fce9400f820d77feb2bc5a786198d9a2 Mon Sep 17 00:00:00 2001
From: JuanZoran <1430359574@qq.com>
Date: Wed, 21 Dec 2022 16:18:36 +0800
Subject: [PATCH] feat: suppose translate in visual mode

---
 lua/Trans/display.lua | 48 ++++++++++++++++++++++++++++++++-----------
 lua/Trans/setup.lua   |  6 +++++-
 2 files changed, 41 insertions(+), 13 deletions(-)

diff --git a/lua/Trans/display.lua b/lua/Trans/display.lua
index e2b26e7..f6f739d 100644
--- a/lua/Trans/display.lua
+++ b/lua/Trans/display.lua
@@ -40,7 +40,7 @@ local function get_title(text, query_res)
 
     pos_info.title = {}
     pos_info.title.word = #query_res.word
-    pos_info.title.phonetic = #query_res.phonetic
+    pos_info.title.phonetic = query_res.phonetic and #query_res.phonetic or 3
     pos_info.title.line = line
     line = line + 1
 end
@@ -220,9 +220,11 @@ local function hl_zh()
 end
 
 local function hl_en()
-    api.nvim_buf_add_highlight(buf, -1, hl.ref, pos_info.en.line, 0, -1)
-    for i = 1, pos_info.en.content, 1 do
-        api.nvim_buf_add_highlight(buf, -1, hl.en, pos_info.en.line + i, 0, -1)
+    if pos_info.en then
+        api.nvim_buf_add_highlight(buf, -1, hl.ref, pos_info.en.line, 0, -1)
+        for i = 1, pos_info.en.content, 1 do
+            api.nvim_buf_add_highlight(buf, -1, hl.en, pos_info.en.line + i, 0, -1)
+        end
     end
 end
 
@@ -246,21 +248,43 @@ local function clear_tmp_info()
     line = 0
 end
 
-function M.query_cursor()
-    local word = vim.fn.expand('<cword>')
+local function get_visual_selection()
+    local s_start = vim.fn.getpos("'<")
+    local s_end = vim.fn.getpos("'>")
+    assert(s_end[2] == s_start[2])
+    local lin = vim.api.nvim_buf_get_lines(0, s_start[2] - 1, s_end[2], false)[1]
+    local word = string.sub(lin, s_start[3], s_end[3])
+    return word
+end
+
+
+function M.query(mode)
+    assert(buf > 0)
+    local word = ''
+    if mode == 'n' then
+        word = vim.fn.expand('<cword>')
+    elseif mode == 'v' then
+        word = get_visual_selection()
+    else
+        print(mode, 'is invalid')
+        assert(false)
+    end
+
     local res = require("Trans.database").query(word)
     local width, height = set_text(res)
     show_win(width, height)
-    set_hl()
-    clear_tmp_info()
+    if res then
+        set_hl()
+        clear_tmp_info()
+    end
 end
 
-function M.query()
-    -- TODO:
+function M.query_cursor()
+    M.query('n')
 end
 
-function M.toggle()
-    -- TODO: wrap some function
+function M.query_select()
+    M.query('v')
 end
 
 function M.close_win()
diff --git a/lua/Trans/setup.lua b/lua/Trans/setup.lua
index 302ae5c..5d4b544 100644
--- a/lua/Trans/setup.lua
+++ b/lua/Trans/setup.lua
@@ -3,6 +3,7 @@ local db = require("Trans").db
 
 
 vim.api.nvim_create_user_command('TranslateCurosorWord', require("Trans.display").query_cursor, {})
+vim.api.nvim_create_user_command('TranslateSelectWord', require("Trans.display").query_select, {})
 
 
 local group = vim.api.nvim_create_augroup("Trans", { clear = true })
@@ -20,11 +21,14 @@ vim.api.nvim_create_autocmd('VimLeave', {
 local auto_close = require("Trans.conf").auto_close
 if auto_close then
     vim.api.nvim_create_autocmd(
-        { 'InsertEnter', 'CursorMoved', 'BufLeave',  }, {
+        { 'InsertEnter', 'CursorMoved', 'BufLeave', }, {
         group = group,
         pattern = '*',
         callback = require('Trans.display').close_win
     })
 end
 
+
+vim.keymap.set('n', 'mm', '<cmd>TranslateCurosorWord<cr>')
+vim.keymap.set('v', 'mm', '<Esc><cmd>TranslateSelectWord<cr>')
 require("Trans.highlight").set_hl()