58 lines
1.3 KiB
Lua
58 lines
1.3 KiB
Lua
local M = {}
|
|
|
|
local type_check = require("Trans.util.debug").type_check
|
|
local query = require("Trans.database").query
|
|
|
|
local function get_select()
|
|
local s_start = vim.fn.getpos("'<")
|
|
local s_end = vim.fn.getpos("'>")
|
|
if s_start[2] ~= s_start[2] then
|
|
error('TODO: multiline translate')
|
|
end
|
|
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
|
|
|
|
|
|
local function get_word(method)
|
|
if not method then
|
|
local mode = vim.api.nvim_get_mode()
|
|
if mode == 'n' then
|
|
return vim.fn.expand('<cword>')
|
|
elseif mode == 'v' then
|
|
return get_select()
|
|
else
|
|
error('invalid mode')
|
|
end
|
|
end
|
|
|
|
if method == 'input' then
|
|
return vim.fn.input('请输入您要查询的单词:') -- TODO Use Telescope with fuzzy finder
|
|
|
|
-- TODO : other method
|
|
else
|
|
error('invalid method')
|
|
end
|
|
end
|
|
|
|
M.get_query_res = function(method)
|
|
type_check {
|
|
method = { method, 'string' },
|
|
}
|
|
local word = ''
|
|
if method == 'cursor' then
|
|
word = vim.fn.expand('<cword>')
|
|
elseif method == 'select' then
|
|
word = get_select():match('%S+')
|
|
elseif method == 'input' then
|
|
else
|
|
error('unknown method')
|
|
end
|
|
|
|
return query(word)
|
|
end
|
|
|
|
|
|
return M
|