style: code format
This commit is contained in:
JuanZoran
@ -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,
|
||||
|
||||
@ -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
|
||||
|
||||
Reference in New Issue
Block a user