396 lines
11 KiB
Lua
396 lines
11 KiB
Lua
local api = vim.api
|
|
local conf = require('Trans').conf
|
|
local hover = conf.hover
|
|
local buffer = require('Trans.buffer')()
|
|
local error_msg = conf.icon.notfound .. ' 没有找到相关的翻译'
|
|
|
|
local node = require('Trans.node')
|
|
local it, t, f = node.item, node.text, node.format
|
|
|
|
local function handle_result(result)
|
|
local icon = conf.icon
|
|
local notfound = icon.notfound
|
|
local indent = ' '
|
|
|
|
local word = result.title.word
|
|
if hover.auto_play then
|
|
string.play(word:isEn() and word or result.definition)
|
|
end
|
|
|
|
local addtitle = function(title)
|
|
buffer:addline {
|
|
it('', 'TransTitleRound'),
|
|
it(title, 'TransTitle'),
|
|
it('', 'TransTitleRound'),
|
|
}
|
|
end
|
|
|
|
local process = {
|
|
title = function(title)
|
|
local oxford = title.oxford
|
|
local collins = title.collins
|
|
local phonetic = title.phonetic
|
|
|
|
if not phonetic and not collins and not oxford then
|
|
buffer:addline(it(word, 'TransWord'))
|
|
|
|
else
|
|
buffer:addline(f {
|
|
width = hover.width,
|
|
text = t {
|
|
it(word, 'TransWord'),
|
|
t {
|
|
it('['),
|
|
it((phonetic and phonetic ~= '') and phonetic or notfound, 'TransPhonetic'),
|
|
it(']')
|
|
},
|
|
it(collins and icon.star:rep(collins) or notfound, 'TransCollins'),
|
|
it(oxford == 1 and icon.yes or icon.no)
|
|
},
|
|
})
|
|
end
|
|
end,
|
|
|
|
tag = function(tag)
|
|
addtitle('标签')
|
|
local tag_map = {
|
|
zk = '中考',
|
|
gk = '高考',
|
|
ky = '考研',
|
|
cet4 = '四级',
|
|
cet6 = '六级',
|
|
ielts = '雅思',
|
|
toefl = '托福',
|
|
gre = 'gre ',
|
|
}
|
|
|
|
local tags = {}
|
|
local size = 0
|
|
local interval = ' '
|
|
for _tag in vim.gsplit(tag, ' ', true) do
|
|
size = size + 1
|
|
tags[size] = tag_map[_tag]
|
|
end
|
|
|
|
|
|
for i = 1, size, 3 do
|
|
buffer:addline(
|
|
it(
|
|
indent .. tags[i] ..
|
|
(tags[i + 1] and interval .. tags[i + 1] ..
|
|
(tags[i + 2] and interval .. tags[i + 2] or '') or ''),
|
|
'TransTag'
|
|
)
|
|
)
|
|
end
|
|
|
|
buffer:addline('')
|
|
end,
|
|
|
|
pos = function(pos)
|
|
addtitle('词性')
|
|
local pos_map = {
|
|
a = '代词pron ',
|
|
c = '连接词conj ',
|
|
i = '介词prep ',
|
|
j = '形容词adj ',
|
|
m = '数词num ',
|
|
n = '名词n ',
|
|
p = '代词pron ',
|
|
r = '副词adv ',
|
|
u = '感叹词int ',
|
|
v = '动词v ',
|
|
x = '否定标记not ',
|
|
t = '不定式标记infm ',
|
|
d = '限定词determiner ',
|
|
}
|
|
|
|
local s = '%s %2s%%'
|
|
for _pos in vim.gsplit(pos, '/', true) do
|
|
buffer:addline(
|
|
it(indent .. s:format(pos_map[_pos:sub(1, 1)], _pos:sub(3)), 'TransPos')
|
|
)
|
|
end
|
|
|
|
buffer:addline('')
|
|
end,
|
|
|
|
exchange = function(exchange)
|
|
addtitle('词形变化')
|
|
local exchange_map = {
|
|
['p'] = '过去式 ',
|
|
['d'] = '过去分词 ',
|
|
['i'] = '现在分词 ',
|
|
['r'] = '比较级 ',
|
|
['t'] = '最高级 ',
|
|
['s'] = '复数 ',
|
|
['0'] = '原型 ',
|
|
['1'] = '类别 ',
|
|
['3'] = '第三人称单数',
|
|
['f'] = '第三人称单数',
|
|
}
|
|
local interval = ' '
|
|
for exc in vim.gsplit(exchange, '/', true) do
|
|
buffer:addline(
|
|
it(indent .. exchange_map[exc:sub(1, 1)] .. interval .. exc:sub(3), 'TransExchange')
|
|
)
|
|
end
|
|
|
|
buffer:addline('')
|
|
end,
|
|
|
|
translation = function(translation)
|
|
addtitle('中文翻译')
|
|
|
|
for trs in vim.gsplit(translation, '\n', true) do
|
|
buffer:addline(
|
|
it(indent .. trs, 'TransTranslation')
|
|
)
|
|
end
|
|
|
|
buffer:addline('')
|
|
end,
|
|
|
|
definition = function(definition)
|
|
addtitle('英文注释')
|
|
|
|
for def in vim.gsplit(definition, '\n', true) do
|
|
def = def:gsub('^%s+', '', 1) -- TODO :判断是否需要分割空格
|
|
buffer:addline(
|
|
it(indent .. def, 'TransDefinition')
|
|
)
|
|
end
|
|
|
|
buffer:addline('')
|
|
end,
|
|
}
|
|
|
|
buffer:set('modifiable', true)
|
|
for _, field in ipairs(conf.order) do
|
|
local value = result[field]
|
|
if value and value ~= '' then
|
|
process[field](value)
|
|
end
|
|
end
|
|
buffer:set('modifiable', false)
|
|
end
|
|
|
|
local function open_window(opts)
|
|
opts = opts or {}
|
|
|
|
local col = opts.col or 1
|
|
local row = opts.row or 1
|
|
local width = opts.width or hover.width
|
|
local height = opts.height or hover.height
|
|
local relative = opts.relative or 'cursor'
|
|
|
|
return require('Trans.window') {
|
|
col = col,
|
|
row = row,
|
|
buf = buffer,
|
|
relative = relative,
|
|
width = width,
|
|
height = height,
|
|
title = hover.title,
|
|
border = hover.border,
|
|
animation = hover.animation,
|
|
ns = require('Trans').ns,
|
|
}
|
|
end
|
|
|
|
local function handle_keymap(win, word)
|
|
local keymap = hover.keymap
|
|
local cur_buf = api.nvim_get_current_buf()
|
|
local del = vim.keymap.del
|
|
local function try_del_keymap()
|
|
for _, key in pairs(keymap) do
|
|
pcall(del, 'n', key)
|
|
end
|
|
end
|
|
|
|
local lock = false
|
|
local cmd_id
|
|
local next
|
|
local action = {
|
|
pageup = function()
|
|
buffer:normal('gg')
|
|
end,
|
|
|
|
pagedown = function()
|
|
buffer:normal('G')
|
|
end,
|
|
|
|
pin = function()
|
|
if lock then
|
|
error('请先关闭窗口')
|
|
else
|
|
lock = true
|
|
end
|
|
pcall(api.nvim_del_autocmd, cmd_id)
|
|
local width, height = win.width, win.height
|
|
local col = vim.o.columns - width - 3
|
|
local buf = buffer.bufnr
|
|
local run = win:try_close()
|
|
run(function()
|
|
local w, r = open_window {
|
|
width = width,
|
|
height = height,
|
|
relative = 'editor',
|
|
col = col,
|
|
}
|
|
|
|
next = w.winid
|
|
win = w
|
|
r(function()
|
|
w:set('wrap', true)
|
|
end)
|
|
|
|
del('n', keymap.pin)
|
|
api.nvim_create_autocmd('BufWipeOut', {
|
|
callback = function(opt)
|
|
if opt.buf == buf or opt.buf == cur_buf then
|
|
lock = false
|
|
api.nvim_del_autocmd(opt.id)
|
|
end
|
|
end
|
|
})
|
|
end)
|
|
end,
|
|
|
|
close = function()
|
|
pcall(api.nvim_del_autocmd, cmd_id)
|
|
local run = win:try_close()
|
|
run(function()
|
|
buffer:delete()
|
|
end)
|
|
try_del_keymap()
|
|
end,
|
|
|
|
toggle_entry = function()
|
|
if lock and win:is_valid() then
|
|
local prev = api.nvim_get_current_win()
|
|
api.nvim_set_current_win(next)
|
|
next = prev
|
|
else
|
|
del('n', keymap.toggle_entry)
|
|
end
|
|
end,
|
|
|
|
play = function()
|
|
if word then
|
|
word:play()
|
|
end
|
|
end,
|
|
}
|
|
local set = vim.keymap.set
|
|
for act, key in pairs(hover.keymap) do
|
|
set('n', key, action[act])
|
|
end
|
|
|
|
if hover.auto_close_events then
|
|
cmd_id = api.nvim_create_autocmd(
|
|
hover.auto_close_events, {
|
|
buffer = 0,
|
|
callback = action.close,
|
|
})
|
|
end
|
|
end
|
|
|
|
local function online_query(win, word)
|
|
local lists = {
|
|
remove = table.remove
|
|
}
|
|
local engines = conf.engines
|
|
local size = #engines
|
|
local icon = conf.icon
|
|
local error_line = it(error_msg, 'TransFailed')
|
|
if size == 0 then
|
|
buffer:addline(error_line)
|
|
return
|
|
end
|
|
|
|
for i = 1, size do
|
|
lists[size] = require('Trans.query.' .. engines[i])(word)
|
|
end
|
|
local cell = icon.cell
|
|
local timeout = hover.timeout
|
|
local spinner = require('Trans.ui.spinner')[hover.spinner]
|
|
local range = #spinner
|
|
local interval = math.floor(timeout / (win.width - spinner[1]:width()))
|
|
local win_width = win.width
|
|
|
|
local s = '%s %s'
|
|
local width, height = hover.width, hover.height
|
|
local function waitting_result(this, times)
|
|
for i = 1, size do
|
|
local res = lists[i][1]
|
|
if res then
|
|
buffer:wipe()
|
|
win:set_width(width)
|
|
handle_result(res)
|
|
height = math.min(height, buffer:height(width))
|
|
|
|
win:expand {
|
|
field = 'height',
|
|
target = height,
|
|
}
|
|
this.run = false
|
|
return
|
|
elseif res == false then
|
|
lists:remove(i)
|
|
size = size - 1
|
|
end
|
|
end
|
|
|
|
if size == 0 or times == win_width then
|
|
buffer:addline(error_line, 1)
|
|
this.run = false
|
|
else
|
|
buffer:addline(it(s:format(spinner[times % range + 1], cell:rep(times)), 'MoreMsg'), 1)
|
|
end
|
|
end
|
|
|
|
buffer:set('modifiable', true)
|
|
local run = require('Trans.util.display') {
|
|
times = win_width,
|
|
interval = interval,
|
|
frame = waitting_result,
|
|
}
|
|
|
|
run(function()
|
|
buffer:set('modifiable', false)
|
|
end)
|
|
end
|
|
|
|
---处理不同hover模式的窗口
|
|
---@param word string 待查询的单词
|
|
return function(word)
|
|
buffer:init()
|
|
local result = require('Trans.query.offline')(word)
|
|
|
|
if result then
|
|
handle_result(result)
|
|
local width = hover.width
|
|
local win, run = open_window {
|
|
width = width,
|
|
height = math.min(buffer:height(width), hover.height)
|
|
}
|
|
run(function()
|
|
win:set('wrap', true)
|
|
handle_keymap(win, word)
|
|
end)
|
|
|
|
else
|
|
local win, run = open_window {
|
|
width = error_msg:width(),
|
|
height = 1,
|
|
}
|
|
|
|
run(function()
|
|
win:set('wrap', true)
|
|
handle_keymap(win, word)
|
|
online_query(win, word)
|
|
end)
|
|
end
|
|
end
|