Trans.nvim/lua/Trans/view/float.lua
2023-01-31 12:14:37 +08:00

108 lines
2.3 KiB
Lua

local conf = require('Trans').conf
local m_window
local m_result
local m_content
local engine_map = {
['本地'] = 'offline',
['百度'] = 'baidu',
['有道'] = 'youdao',
}
local node = require("Trans.node")
local t = node.text
local it = node.item
local function set_tag_hl(name, status)
local hl = conf.float.tag[status]
m_window:set_hl(name, {
fg = '#000000',
bg = hl,
})
m_window:set_hl(name .. 'round', {
fg = hl,
})
end
local function set_title()
local title = m_window.contents[1]
local github = ' https://github.com/JuanZoran/Trans.nvim'
title:addline(
title:center(it(github, '@text.uri'))
)
local format = '%s(%d)'
for i, engine_ch in ipairs(conf.float.engine) do
local engine_us = engine_map[engine_ch]
set_tag_hl(engine_us, 'wait')
local round = engine_us .. 'round'
title:addline(
t(
it('', round),
it(format:format(engine_ch, i), engine_us),
it('', round)
)
)
title:newline('')
end
end
local action = {
quit = function()
m_window:try_close()
end,
}
local function process()
-- TODO :
end
return function(word)
-- TODO :online query
local float = conf.float
local engine_ch = '本地'
local engine_us = engine_map[engine_ch]
vim.notify('[注意]: float窗口目前还待开发, 如果需要input查询功能, 请将窗口改成hover',
vim.log.WARN)
m_result = require('Trans.query.' .. engine_us)(word)
local opt = {
relative = 'editor',
width = float.width,
height = float.height,
border = float.border,
title = float.title,
animation = float.animation,
row = bit.rshift((vim.o.lines - float.height), 1),
col = bit.rshift((vim.o.columns - float.width), 1),
zindex = 50,
}
m_window = require('Trans.window')(true, opt)
set_title()
m_content = m_window.contents[2]
if m_result then
set_tag_hl(engine_us, 'success')
process()
else
set_tag_hl(engine_us, 'fail')
end
m_window:open()
m_window:bufset('bufhidden', 'wipe')
for act, key in pairs(float.keymap) do
m_window:map(key, action[act])
end
end