tqcq/Trans.nvim
0
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Activity

style: code format

Browse Source
This commit is contained in:
JuanZoran 2023-03-18 13:53:09 +08:00
parent a6a5a33bff
commit 46c69fb758
8 changed files with 190 additions and 272 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lua/Trans/backend/baidu.lua
View File

@ -6,12 +6,12 @@
---@field disable boolean
local M = {
uri = 'https://fanyi-api.baidu.com/api/trans/vip/translate',
uri = "https://fanyi-api.baidu.com/api/trans/vip/translate",
salt = tostring(math.random(bit.lshift(1, 15))),
name = 'baidu',
name = "baidu",
}
local Trans = require('Trans')
local Trans = require("Trans")
---@class BaiduQuery
---@field q string
@ -21,7 +21,6 @@ local Trans = require('Trans')
---@field salt string
---@field sign string
---Get content for query
---@param data TransData
---@return BaiduQuery
@ -53,8 +52,8 @@ function M.query(data)
local handle = function(res)
local status, body = pcall(vim.json.decode, res.body)
if not status or not body then
data.trace = res
data.result.baidu = false
data.trace = res
return
end
@ -65,13 +64,12 @@ function M.query(data)
assert(#result == 1)
result = result[1]
data.result.baidu = {
['str'] = result.src,
[data.from == 'en' and 'translation' or 'definition'] = { result.dst },
["str"] = result.src,
[data.from == "en" and "translation" or "definition"] = { result.dst },
}
end
end
Trans.curl.get(M.uri, {
query = M.get_content(data),
callback = handle,

151
lua/Trans/backend/offline.lua
View File

@ -1,69 +1,65 @@
---@class Offline: TransBackend
local M = {
name = "offline",
no_wait = true,
name = 'offline',
}
local db = require 'sqlite.db'
vim.api.nvim_create_autocmd('VimLeavePre', {
local db = require("sqlite.db")
vim.api.nvim_create_autocmd("VimLeavePre", {
once = true,
callback = function()
if db:isopen() then
db:close()
end
end
if db:isopen() then db:close() end
end,
})
---@param data any
---@return any
---@overload fun(TransData): TransResult
function M.query(data)
if data.is_word == false or data.from == 'zh' then return end
if data.is_word == false or data.from == "zh" then
return
end
local path = require('Trans').conf.dir .. '/ultimate.db'
local path = require("Trans").conf.dir .. "/ultimate.db"
local dict = db:open(path)
local db_name = data.db_name or 'stardict'
local res = dict:select(db_name, {
where = { word = data.str, },
keys = M.query_field,
local dict = db:open(path)
local db_name = data.db_name or "stardict"
local res = dict:select(db_name, {
where = { word = data.str },
keys = M.query_field,
limit = 1,
})[1]
data.result.offline = res and M.formatter(res) or false
end
-- this is a awesome plugin
M.query_field = {
'word',
'phonetic',
'definition',
'translation',
'pos',
'collins',
'oxford',
'tag',
'exchange',
"word",
"phonetic",
"definition",
"translation",
"pos",
"collins",
"oxford",
"tag",
"exchange",
}
local function exist(str)
return str and str ~= ''
return str and str ~= ""
end
---@type (fun(res):any)[]
local formatter = {
title = function(res)
local title = {
local title = {
word = res.word,
oxford = res.oxford,
collins = res.collins,
phonetic = res.phonetic,
}
res.word = nil
res.oxford = nil
res.collins = nil
@ -71,87 +67,97 @@ local formatter = {
return title
end,
tag = function(res)
if not exist(res.tag) then return end
if not exist(res.tag) then
return
end
local tag_map = {
zk = '中考',
gk = '高考',
ky = '考研',
gre = 'gre ',
cet4 = '四级',
cet6 = '六级',
ielts = '雅思',
toefl = '托福',
zk = "中考",
gk = "高考",
ky = "考研",
gre = "gre ",
cet4 = "四级",
cet6 = "六级",
ielts = "雅思",
toefl = "托福",
}
local tag = {}
for i, _tag in ipairs(vim.split(res.tag, ' ', { plain = true })) do
for i, _tag in ipairs(vim.split(res.tag, " ", { plain = true })) do
tag[i] = tag_map[_tag]
end
return tag
end,
exchange = function(res)
if not exist(res.exchange) then return end
if not exist(res.exchange) then
return
end
local exchange_map = {
['0'] = '原型 ',
['1'] = '类别 ',
['p'] = '过去式 ',
['r'] = '比较级 ',
['t'] = '最高级 ',
['s'] = '复数 ',
['d'] = '过去分词 ',
['i'] = '现在分词 ',
['3'] = '第三人称单数',
['f'] = '第三人称单数',
["0"] = "原型 ",
["1"] = "类别 ",
["p"] = "过去式 ",
["r"] = "比较级 ",
["t"] = "最高级 ",
["s"] = "复数 ",
["d"] = "过去分词 ",
["i"] = "现在分词 ",
["3"] = "第三人称单数",
["f"] = "第三人称单数",
}
local exchange = {}
for _, _exchange in ipairs(vim.split(res.exchange, '/', { plain = true })) do
for _, _exchange in ipairs(vim.split(res.exchange, "/", { plain = true })) do
exchange[exchange_map[_exchange:sub(1, 1)]] = _exchange:sub(3)
end
return exchange
end,
pos = function(res)
if not exist(res.pos) then return end
if not exist(res.pos) then
return
end
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 ',
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 pos = {}
for _, _pos in ipairs(vim.split(res.pos, '/', { plain = true })) do
pos[pos_map[_pos:sub(1, 1)]] = ('%2s%%'):format(_pos:sub(3))
for _, _pos in ipairs(vim.split(res.pos, "/", { plain = true })) do
pos[pos_map[_pos:sub(1, 1)]] = ("%2s%%"):format(_pos:sub(3))
end
return pos
end,
translation = function(res)
if not exist(res.translation) then return end
if not exist(res.translation) then
return
end
local translation = {}
for i, _translation in ipairs(vim.split(res.translation, '\n', { plain = true })) do
for i, _translation in ipairs(vim.split(res.translation, "\n", { plain = true })) do
translation[i] = _translation
end
return translation
end,
definition = function(res)
if not exist(res.definition) then return end
if not exist(res.definition) then
return
end
local definition = {}
for i, _definition in ipairs(vim.split(res.definition, '\n', { plain = true })) do
for i, _definition in ipairs(vim.split(res.definition, "\n", { plain = true })) do
-- -- TODO :判断是否需要分割空格
definition[i] = _definition:gsub('^%s+', '', 1)
definition[i] = _definition:gsub("^%s+", "", 1)
end
return definition
@ -169,5 +175,4 @@ function M.formatter(res)
return res
end
return M

27
lua/Trans/frontend/hover/execute.lua
View File

@ -3,35 +3,33 @@ local api = vim.api
---@type table<string, fun(hover: TransHover)>
local strategy = {
pageup = function(hover)
hover.buffer:normal('gg')
hover.buffer:normal("gg")
end,
pagedown = function(hover)
hover.buffer:normal('G')
hover.buffer:normal("G")
end,
pin = function(hover)
if hover.pin then return end
if hover.pin then
return
end
hover.pin = true
local window = hover.window
local width, height = window:width(), window:height()
local col = vim.o.columns - width - 3
window:try_close()
window = hover:init_window({
width = width,
height = height,
relative = 'editor',
col = col,
})
window = hover:init_window {
col = col,
width = width,
height = height,
relative = "editor",
}
window:set('wrap', true)
window:set("wrap", true)
end,
close = function(hover)
hover:destroy()
end,
toggle_entry = function(hover)
if api.nvim_get_current_win() ~= hover.window.winid then
api.nvim_set_current_win(hover.window.winid)
@ -47,7 +45,6 @@ local strategy = {
end,
}
---@class TransHover
---@field execute fun(hover: TransHover, action: string)
return function(hover, action)

141
lua/Trans/frontend/hover/init.lua
View File

@ -1,4 +1,4 @@
local Trans = require('Trans')
local Trans = require("Trans")
-- FIXME :Adjust Window Size
@ -10,8 +10,8 @@ local Trans = require('Trans')
---@field destroy_funcs fun(hover:TransHover)[] @functions to be executed when hover window is closed
---@field opts TransHoverOpts @options for hover window
---@field pin boolean @whether hover window is pinned
local M = Trans.metatable('frontend.hover', {
ns = vim.api.nvim_create_namespace('TransHoverWin'),
local M = Trans.metatable("frontend.hover", {
ns = vim.api.nvim_create_namespace("TransHoverWin"),
queue = {},
})
M.__index = M
@ -55,9 +55,12 @@ function M:destroy()
func(self)
end
if self.window:is_valid() then self.window:try_close() end
if self.buffer:is_valid() then self.buffer:destroy() end
if self.window:is_valid() then
self.window:try_close()
end
if self.buffer:is_valid() then
self.buffer:destroy()
end
self.pin = false
end)()
end
@ -67,25 +70,27 @@ end
---|{width?: integer, height?: integer, col?: integer, row?: integer, relative?: string}
---@return unknown
function M:init_window(opts)
opts = opts or {}
local m_opts = self.opts
local option = {
ns = self.ns,
buffer = self.buffer,
opts = opts or {}
local m_opts = self.opts
local option = {
ns = self.ns,
buffer = self.buffer,
animation = m_opts.animation,
}
-- stylua: ignore start
local win_opts = {
col = opts.col or 1,
row = opts.row or 1,
title = m_opts.title,
relative = opts.relative or 'cursor',
width = opts.width or m_opts.width,
height = opts.height or m_opts.height,
relative = opts.relative or "cursor",
}
-- stylua: ignore end
if win_opts.title then
win_opts.title_pos = 'center'
win_opts.title_pos = "center"
end
option.win_opts = win_opts
@ -98,7 +103,7 @@ end
---@return string formatted text
---@return integer _ replaced count
function M:icon_format(format)
return format:gsub('{{(%w+)}}', self.opts.icon, 1)
return format:gsub("{{(%w+)}}", self.opts.icon, 1)
end
---Get Check function for waiting
@ -122,7 +127,7 @@ function M:wait()
return function()
cur = cur + 1
buffer[1] = spinner[cur % size + 1] .. (cell):rep(cur)
buffer:add_highlight(1, 'TransWaitting')
buffer:add_highlight(1, "TransWaitting")
pause(interval)
return cur < times
end
@ -137,7 +142,6 @@ function M:fallback()
}
end
local buffer = self.buffer
buffer:wipe()
@ -146,22 +150,23 @@ function M:fallback()
-- TODO :Center
buffer[1] = Trans.util.center(fallback_msg, opts.width)
buffer:add_highlight(1, 'TransFailed')
buffer:add_highlight(1, "TransFailed")
self:defer()
end
---Defer function when process done
function M:defer()
self.window:set('wrap', true)
self.buffer:set('modifiable', false)
self.window:set("wrap", true)
self.buffer:set("modifiable", false)
local auto_close_events = self.opts.auto_close_events
if auto_close_events then
vim.api.nvim_create_autocmd(auto_close_events, {
once = true,
callback = function()
if self.pin then return end
if self.pin then
return
end
self:destroy()
end,
})
@ -172,7 +177,9 @@ end
---@param data TransData
---@overload fun(result:TransResult)
function M:process(data)
if self.pin then return end
if self.pin then
return
end
local result, name = data:get_available_result()
if not result then
@ -182,7 +189,7 @@ function M:process(data)
-- vim.pretty_print(result)
local opts = self.opts
if opts.auto_play then
(data.from == 'en' and data.str or result.definition[1]):play()
(data.from == "en" and data.str or result.definition[1]):play()
end
-- local node = Trans.util.node
-- local it, t, f = node.item, node.text, node.format
@ -213,6 +220,7 @@ function M:process(data)
width = math.min(opts.width, display_size.width + opts.padding),
}
end
self:defer()
end
@ -225,90 +233,3 @@ end
---@class TransFrontend
---@field hover TransHover @hover frontend
return M
-- 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

58
lua/Trans/health.lua
View File

@ -1,89 +1,89 @@
local health, fn = vim.health, vim.fn
local ok = health.report_ok
local warn = health.report_warn
local error = health.report_error
local has = fn.has
local executable = fn.executable
local function check_neovim_version()
if has('nvim-0.9') == 1 then
ok [[You have [neovim-nightly] ]]
if has("nvim-0.9") == 1 then
ok([[You have [neovim-nightly] ]])
else
warn [[Trans Title requires Neovim 0.9 or newer
warn([[Trans Title requires Neovim 0.9 or newer
See neovim-nightly: [https://github.com/neovim/neovim/releases/tag/nightly]
]]
]])
end
end
local function check_plugin_dependencies()
local plugin_dependencies = {
-- 'plenary',
'sqlite',
"sqlite",
}
for _, dep in ipairs(plugin_dependencies) do
if pcall(require, dep) then
ok(string.format('Dependency [%s] is installed', dep))
ok(string.format("Dependency [%s] is installed", dep))
else
error(string.format('Dependency [%s] is not installed', dep))
error(string.format("Dependency [%s] is not installed", dep))
end
end
end
local function check_binary_dependencies()
local binary_dependencies = {
'curl',
'sqlite3',
"curl",
"sqlite3",
}
if has('linux') == 1 then
binary_dependencies[3] = 'festival'
elseif has('mac') == 1 then
binary_dependencies[3] = 'say'
if has("linux") == 1 then
binary_dependencies[3] = "festival"
elseif has("mac") == 1 then
binary_dependencies[3] = "say"
else
binary_dependencies[3] = 'node'
binary_dependencies[3] = "node"
end
for _, dep in ipairs(binary_dependencies) do
if executable(dep) == 1 then
ok(string.format('Binary dependency [%s] is installed', dep))
ok(string.format("Binary dependency [%s] is installed", dep))
else
error(string.format('Binary dependency [%s] is not installed', dep))
error(string.format("Binary dependency [%s] is not installed", dep))
end
end
end
local function check_database()
local db_path = require('Trans').conf.dir .. '/ultimate.db'
local db_path = require("Trans").conf.dir .. "/ultimate.db"
if fn.filereadable(db_path) == 1 then
ok [[ultimate database found ]]
ok([[ultimate database found ]])
else
error [[Stardict database not found
error([[Stardict database not found
[Manually]: Please check the doc in github: [https://github.com/JuanZoran/Trans.nvim]
[Automatically]: Try to run `:lua require "Trans".install()`
]]
]])
end
end
local function check_configure_file()
local path = fn.expand(require('Trans').conf.dir .. '/Trans.json')
local path = fn.expand(require("Trans").conf.dir .. "/Trans.json")
if not fn.filereadable(path) then
warn 'Backend configuration file[%s] not found'
warn("Backend configuration file[%s] not found")
end
local file = io.open(path, "r")
local valid = file and pcall(vim.json.decode, file:read("*a"))
if valid then
ok(string.format([[Backend configuration file [%s] found and valid ]], path))
else
error(string.format([[Backend configuration file [%s] invalid
error(string.format(
[[Backend configuration file [%s] invalid
Please check the doc in github: [https://github.com/JuanZoran/Trans.nvim]
]], path))
]],
path
))
end
end
@ -95,4 +95,4 @@ local function check()
check_binary_dependencies()
end
return { check = check, }
return { check = check }

18
lua/Trans/init.lua
View File

@ -5,35 +5,33 @@
local function metatable(folder_name, origin)
return setmetatable(origin or {}, {
__index = function(tbl, key)
local status, result = pcall(require, ('Trans.%s.%s'):format(folder_name, key))
local status, result = pcall(require, ("Trans.%s.%s"):format(folder_name, key))
if status then
tbl[key] = result
return result
end
end
end,
})
end
---@class string
---@field width function @Get string display width
---@field play function @Use tts to play string
---@class Trans
---@field style table @Style module
---@field cache table<string, TransData> @Cache for translated data object
---@field modes string[] @all modes name
local M = metatable('core', {
style = metatable("style"),
local M = metatable("core", {
cache = {},
style = metatable("style"),
modes = {
'normal',
'visual',
'input',
"normal",
"visual",
"input",
},
augroup = vim.api.nvim_create_augroup('Trans', { clear = true })
augroup = vim.api.nvim_create_augroup("Trans", { clear = true }),
})
M.metatable = metatable

24
lua/Trans/util/node.lua
View File

@ -9,15 +9,15 @@ local item_meta = {
local text_meta = {
render = function(self, buffer, line, col)
local items = self.items
local step = self.step or ''
local len = #step
local step = self.step or ""
local len = #step
for i = 1, self.size do
local item = items[i]
item:render(buffer, line, col)
col = col + #item[1] + len
end
end
end,
}
item_meta.__index = item_meta
@ -25,7 +25,6 @@ text_meta.__index = function(self, key)
return text_meta[key] or (key == 1 and table.concat(self.strs, self.step) or nil)
end
local function item(text, highlight)
return setmetatable({
[1] = text,
@ -42,8 +41,8 @@ local function text(items)
end
return setmetatable({
strs = strs,
size = size,
strs = strs,
size = size,
items = items,
}, text_meta)
end
@ -52,9 +51,9 @@ local function format(opts)
local str = opts.text
local size = str.size
local width = opts.width
local spin = opts.spin or ' '
local spin = opts.spin or " "
local wid = str[1]:width()
local wid = str[1]:width()
local space = math.max(math.floor((width - wid) / (size - 1)), 0)
if space > 0 then
str.step = spin:rep(space)
@ -62,7 +61,6 @@ local function format(opts)
return str
end
---@type table<string, function>
return {
item = item,
@ -70,9 +68,9 @@ return {
format = format,
conjunction = function(str)
return {
item('', 'TransTitleRound'),
item(str, 'TransTitle'),
item('', 'TransTitleRound'),
item("", "TransTitleRound"),
item(str, "TransTitle"),
item("", "TransTitleRound"),
}
end
end,
}

29
plugin/Trans.lua
View File

@ -1,35 +1,36 @@
local api, fn = vim.api, vim.fn
string.width = api.nvim_strwidth
--- INFO :Define string play method
if fn.has('linux') == 1 then
if fn.has("linux") == 1 then
string.play = function(self)
local cmd = ([[echo "%s" | festival --tts]]):format(self)
local cmd = ([[echo %q | festival --tts]]):format(self)
fn.jobstart(cmd)
end
elseif fn.has('mac') == 1 then
elseif fn.has("mac") == 1 then
string.play = function(self)
local cmd = ([[say "%s"]]):format(self)
local cmd = ([[say %q]]):format(self)
fn.jobstart(cmd)
end
else
string.play = function(self)
local separator = fn.has('unix') and '/' or '\\'
local file = debug.getinfo(1, "S").source:sub(2):match('(.*)lua') .. separator .. 'tts' .. separator .. 'say.js'
fn.jobstart('node ' .. file .. ' ' .. self)
local separator = fn.has("unix") and "/" or "\\"
local file = debug.getinfo(1, "S").source:sub(2):match("(.*)lua") .. separator .. "tts" .. separator .. "say.js"
fn.jobstart("node " .. file .. " " .. self)
end
end
--- INFO :Define plugin command
local Trans = require('Trans')
local Trans = require("Trans")
local command = api.nvim_create_user_command
command('Translate', function() Trans.translate() end, { desc = ' 单词翻译', })
command('TransPlay', function()
command("Translate", function()
Trans.translate()
end, { desc = " 单词翻译" })
command("TransPlay", function()
local str = Trans.util.get_str(api.nvim_get_mode().mode)
if str and str ~= '' and Trans.util.is_English(str) then
if str and str ~= "" and Trans.util.is_English(str) then
str:play()
end
end, { desc = ' 自动发音' })
end, { desc = " 自动发音" })