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

Merge pull request #9 from JuanZoran/experimental

Browse Source 
增加判断linux下直接使用festival
This commit is contained in:
Zoran 2023-01-24 19:03:32 +08:00 committed by GitHub
commit 4ba795b67c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 350 additions and 223 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
README.md
View File

@ -5,6 +5,7 @@
- [特点](#特点)
- [屏幕截图](#屏幕截图)
- [安装](#安装)
- [Festival配置](#festival配置)
- [配置](#配置)
- [快捷键绑定](#快捷键绑定)
- [高亮组](#高亮组)
@ -104,11 +105,54 @@ use {
> 后续会增加 `healthcheck` 进行检查
- **`auto_play`** 使用步骤:
> linux 只需要安装`festival`
> sudo apt-get install festival festvox-kallpc16k
> ***如果你想要设置音色，发音可以访问:*** [Festival官方](https://www.cstr.ed.ac.uk/projects/festival/morevoices.html)
> 可以选择英音、美音、男声、女声
> 其他操作系统
- 需要确保安装了`nodejs`
- 进入插件的`tts`目录运行`npm install`
> 如果`install.sh`运行正常则自动安装，如果安装失败，请尝试手动安装
- linux 用户需要额外安装以下依赖:
> sudo apt-get install festival festvox-kallpc16k
## Festival配置
> 仅针对`linux`用户说明
- 配置文件
- 全局配置: `/usr/share/festival/siteinit.scm`
- 用户配置: `~/.festivalrc`
- 更改声音
- 在festival的voices文件内建立自己的文件夹
> 一般其默认配置目录在`/usr/share/festival/voices`
示例:
> `sudo mkdir /usr/share/festival/voices/my_voices`
- 下载想要的voices文件并解压
> 正常均需要
- 试听[在这里](https://www.cstr.ed.ac.uk/projects/festival/morevoices.html))
- 下载[在这里](http://festvox.org/packed/festival/2.5/voices/))
> 假设下载的文件在`Downloads`文件夹, 下载的文件为:`festvox_cmu_us_aew_cg.tar.gz`
示例:
> `cd ~/Downloads && tar -xf festvox_cmu_us_aew_cg.tar.gz`
- 将音频文件拷贝到festival文件夹
示例:
> `sudo cp -r festival/lib/voices/us/cmu_us_aew_cg/ /usr/share/festival/voices/my_voices/`
- 在配置文件中设置默认的声音
示例:
> 加入`(set! voice_default voice_cmu_indic_hin_ab_cg)`到配置文件
- 安装完成
- 相关说明网站
> 正常均需要
- [wiki](https://archlinux.org/packages/community/any/festival-us/) 查看更多详细配置
- [官方网站](http://festvox.org/dbs/index.html)
- [用户手册](http://www.festvox.org/docs/manual-2.4.0/festival_toc.html)
## 配置
```lua
@ -286,5 +330,5 @@ vim.keymap.set('n', 'mi', '<Cmd>TranslateInput<CR>')
- [ ] 历史查询结果保存
- [ ] 在线多引擎异步查询
- [ ] 快捷键定义
- [ ] 自动读音
- [x] 自动读音
- [ ] `句子翻译` | `中翻英` 的支持

9
install.sh
View File

@ -14,5 +14,12 @@ wget https://github.com/skywind3000/ECDICT-ultimate/releases/download/1.0.0/ecdi
unzip /tmp/dict.zip -d $HOME/.vim/dict
rm -rf /tmp/dict.zip
cd ./tts/ && npm install
uNames=`uname -s`
osName=${uNames: 0: 4}
if [ "$osName" != "Linux" ];then
cd ./tts/ && npm install
fi

183
lua/Trans/content.lua
View File

@ -1,4 +1,38 @@
local api = vim.api
local item_meta = {
load_hl = function(self, content, line, col)
if self.hl then
content:newhl {
name = self.hl,
line = line,
_start = col,
_end = col + #self.text,
}
end
end
}
local text_meta = {
load_hl = function(self, content, line, col)
for _, item in ipairs(self.items) do
item:load_hl(content, line, col)
col = col + #item.text
end
end,
}
local format_meta = {
load_hl = function(self, content, line, col)
local space = self.space
for _, item in ipairs(self.items) do
item:load_hl(content, line, col)
col = col + #item.text + space
end
end
}
local content = {
newline = function(self, value)
if not self.modifiable then
@ -16,24 +50,6 @@ local content = {
self.highlights[self.hl_size] = opt
end,
center_line = function(self, text, highlight)
vim.validate {
text = { text, 's' }
}
local space = math.floor((self.window.width - text:width()) / 2)
local interval = (' '):rep(space)
self:newline(interval .. text)
if highlight then
self:newhl {
name = highlight,
line = self.size - 1,
_start = space,
_end = space + #text,
}
end
end,
wipe = function(self)
local clear = require('table.clear')
clear(self.lines)
@ -51,9 +67,12 @@ local content = {
self.window:bufset('modifiable', true)
local window = self.window
api.nvim_buf_set_lines(window.bufnr, offset, offset + 1, true, self.lines)
--- NOTE : 使用-1 则需要按顺序设置
api.nvim_buf_set_lines(window.bufnr, offset, -1, true, self.lines)
for _, hl in ipairs(self.highlights) do
local hl
for i = 1, self.hl_size do
hl = self.highlights[i]
api.nvim_buf_add_highlight(window.bufnr, window.hl, hl.name, offset + hl.line, hl._start, hl._end)
end
self.window:bufset('modifiable', false)
@ -61,7 +80,7 @@ local content = {
actual_height = function(self, wrap)
wrap = wrap or self.window:option('wrap')
if wrap then
if wrap then
local height = 0
local width = self.window.width
local lines = self.lines
@ -75,78 +94,70 @@ local content = {
end
end,
addline = function(self, newline, highlight)
self:newline(newline)
if highlight then
self:newhl {
name = highlight,
line = self.size - 1,
_start = 0,
_end = -1,
}
end
item_wrap = function(text, hl)
return setmetatable({
text = text,
hl = hl,
}, { __index = item_meta })
end,
items_wrap = function(self)
local items = {}
local size = 0
text_wrap = function(...)
local items = { ... }
local strs = {}
for i, item in ipairs(items) do
strs[i] = item.text
end
return setmetatable({
text = table.concat(strs),
items = items,
}, { __index = text_meta })
end,
format = function(self, ...)
local nodes = { ... }
local size = #nodes
assert(size > 1, 'check items size')
local width = 0
local strs = {}
for i, node in ipairs(nodes) do
local str = node.text
strs[i] = str
width = width + str:width()
end
return {
add_item = function(item, highlight)
size = size + 1
items[size] = { item, highlight }
width = width + item:width()
end,
load = function()
assert(size > 1, 'no item need be loaded')
local space = math.floor((self.window.width - width) / (size - 1))
assert(space > 0, 'try to expand window width')
local interval = (' '):rep(space)
local line = ''
local function load_item(idx)
local item = items[idx]
if item[2] then
self:newhl {
name = item[2],
line = self.size, -- NOTE : 此时还没插入新行, size ==> 行号(zero index)
_start = #line,
_end = #line + #item[1],
}
end
line = line .. item[1]
end
load_item(1)
for i = 2, size do
line = line .. interval
load_item(i)
end
self:newline(line)
end
}
local space = math.floor(((self.window.width - width) / (size - 1)))
assert(space > 0, 'try to expand window size')
local interval = (' '):rep(space)
return setmetatable({
text = table.concat(strs, interval),
items = nodes,
space = space,
}, { __index = format_meta })
end,
line_wrap = function(self)
self:newline('')
local index = self.size
return function(text, highlight)
if highlight then
local _start = #self.lines[index]
local _end = _start + #text
self:newhl {
name = highlight,
line = index - 1,
_start = _start,
_end = _end,
}
end
self.lines[index] = self.lines[index] .. text
center = function(self, item)
local space = bit.rshift(self.window.width - item.text:width(), 1)
item.text = (' '):rep(space) .. item.text
local load_hl = item.load_hl
item.load_hl = function (this, content, line, col)
load_hl(this, content, line, col + space)
end
return item
end,
addline = function(self, ...)
local strs = {}
local col = 0
for i, node in ipairs { ... } do
local str = node.text
strs[i] = str
node:load_hl(self, self.size, col)
col = col + #str
end
self:newline(table.concat(strs))
end
}

23
lua/Trans/init.lua
View File

@ -7,8 +7,8 @@ M.conf = {
v = 'hover',
},
hover = {
width = 36,
height = 26,
width = 37,
height = 27,
border = 'rounded',
title = {
{ '', 'TransTitleRound' },
@ -54,30 +54,33 @@ M.conf = {
open = 'fold',
close = 'fold',
interval = 10,
},
tag = {
wait = '#519aba',
fail = '#e46876',
success = '#10b981',
},
engine = {
'本地',
}
},
order = {
-- offline = {
order = { -- only work on hover mode
'title',
'tag',
'pos',
'exchange',
'translation',
'definition',
-- },
-- online = {
-- -- TODO
-- },
},
icon = {
star = '',
-- notfound = '❔',
notfound = '',
yes = '',
no = ''
-- star = '⭐',
-- notfound = '❔',
-- yes = '✔️',
-- no = '❌'
-- star = '⭐',
},
db_path = '$HOME/.vim/dict/ultimate.db',

80
lua/Trans/view/float.lua
View File

@ -1,12 +1,52 @@
local conf = require('Trans').conf
local m_window
local m_result
local m_content
local engine_map = {
['本地'] = 'offline',
['百度'] = 'baidu',
['有道'] = 'youdao',
}
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'
local github = 'https://github.com/JuanZoran/Trans.nvim'
-- TODO :config this
title:center_line(github, '@text.uri')
local item = title.item_wrap
title:addline(
title:center(item(github, '@text.uri'))
)
local text = title.text_wrap
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(
text(
item('', round),
item(format:format(engine_ch, i), engine_us),
item('', round)
)
)
end
end
local action = {
@ -16,24 +56,46 @@ local action = {
}
local handle = {
title = function()
-- TODO :
end,
}
return function(word)
-- TODO :online query
local float = require('Trans').conf.float
m_result = require('Trans.query.offline')(word)
local float = conf.float
local engine_ch = '本地'
local engine_us = engine_map[engine_ch]
local opt = {
m_result = require('Trans.query.' .. engine_us)(word)
local opt = {
relative = 'editor',
width = float.width,
height = float.height,
border = float.border,
title = float.title,
row = math.floor((vim.o.lines - float.height) / 2),
col = math.floor((vim.o.columns - float.width) / 2),
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)
m_window = require('Trans.window')(true, opt)
m_window.animation = float.animation
set_title()
m_content = m_window.contents[2]
if m_result then
set_tag_hl(engine_us, 'success')
for _, proc in pairs(handle) do
proc()
end
else
set_tag_hl(engine_us, 'fail')
end
m_window:draw()
m_window:open()
m_window:bufset('bufhidden', 'wipe')

217
lua/Trans/view/hover.lua
View File

@ -1,92 +1,63 @@
local api = vim.api
local conf = require('Trans').conf
local icon = conf.icon
local m_window
local m_result
local m_content
-- content utility
local text
local item
local m_indent = ' '
local title = function(str)
local wrapper = m_content:line_wrap()
-- wrapper('', 'TransTitleRound')
wrapper('', 'TransTitleRound')
wrapper(str, 'TransTitle')
wrapper('', 'TransTitleRound')
-- wrapper('', 'TransTitleRound')
m_content:addline(
text(
item('', 'TransTitleRound'),
item(str, 'TransTitle'),
item('', 'TransTitleRound')
)
)
end
local tag_map = {
zk = '中考',
gk = '高考',
ky = '考研',
cet4 = '四级',
cet6 = '六级',
ielts = '雅思',
toefl = '托福',
gre = 'gre ',
}
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 exchange_map = {
['p'] = '过去式 ',
['d'] = '过去分词 ',
['i'] = '现在分词 ',
['r'] = '比较级 ',
['t'] = '最高级 ',
['s'] = '复数 ',
['0'] = '原型 ',
['1'] = '类别 ',
['3'] = '第三人称单数',
['f'] = '第三人称单数',
}
local exist = function(str)
return str and str ~= ''
end
local process = {
title = function()
local line = m_content:items_wrap()
line.add_item(
m_result.word,
'TransWord'
)
local icon = conf.icon
line.add_item(
'[' .. (exist(m_result.phonetic) and m_result.phonetic or icon.notfound) .. ']',
'TransPhonetic'
m_content:addline(
m_content:format(
item(m_result.word, 'TransWord'),
text(
item('['),
item(exist(m_result.phonetic) and m_result.phonetic or icon.notfound, 'TransPhonetic'),
item(']')
),
item(m_result.collins and icon.star:rep(m_result.collins) or icon.notfound, 'TransCollins'),
item(m_result.oxford == 1 and icon.yes or icon.no)
)
)
line.add_item(
(exist(m_result.collins) and icon.star:rep(m_result.collins) or icon.notfound),
'TransCollins'
)
line.add_item(
(m_result.oxford == 1 and icon.yes or icon.no)
)
line.load()
end,
tag = function()
if exist(m_result.tag) then
title('标签')
local tag_map = {
zk = '中考',
gk = '高考',
ky = '考研',
cet4 = '四级',
cet6 = '六级',
ielts = '雅思',
toefl = '托福',
gre = 'gre ',
}
local tags = {}
local size = 0
local interval = ' '
@ -95,45 +66,75 @@ local process = {
tags[size] = tag_map[tag]
end
for i = 1, size, 3 do
m_content:addline(
m_indent .. tags[i] .. interval .. (tags[i + 1] or '') .. interval .. (tags[i + 2] or ''),
'TransTag'
item(
m_indent ..
tags[i] ..
(tags[i + 1] and interval .. tags[i + 1] ..
(tags[i + 2] and interval .. tags[i + 2] or '') or ''),
'TransTag'
)
)
end
m_content:addline('')
m_content:newline('')
end
end,
pos = function()
if exist(m_result.pos) then
title('词性')
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 ',
}
for pos in vim.gsplit(m_result.pos, '/', true) do
m_content:addline(
m_indent .. pos_map[pos:sub(1, 1)] .. pos:sub(3) .. '%',
'TransPos'
item(m_indent .. pos_map[pos:sub(1, 1)] .. pos:sub(3) .. '%', 'TransPos')
)
end
m_content:addline('')
m_content:newline('')
end
end,
exchange = function()
if exist(m_result.exchange) then
title('词形变化')
local exchange_map = {
['p'] = '过去式 ',
['d'] = '过去分词 ',
['i'] = '现在分词 ',
['r'] = '比较级 ',
['t'] = '最高级 ',
['s'] = '复数 ',
['0'] = '原型 ',
['1'] = '类别 ',
['3'] = '第三人称单数',
['f'] = '第三人称单数',
}
local interval = ' '
for exc in vim.gsplit(m_result.exchange, '/', true) do
m_content:addline(
m_indent .. exchange_map[exc:sub(1, 1)] .. interval .. exc:sub(3),
'TransExchange'
item(m_indent .. exchange_map[exc:sub(1, 1)] .. interval .. exc:sub(3), 'TransExchange')
)
end
m_content:addline('')
m_content:newline('')
end
end,
@ -142,12 +143,11 @@ local process = {
for trs in vim.gsplit(m_result.translation, '\n', true) do
m_content:addline(
m_indent .. trs,
'TransTranslation'
item(m_indent .. trs, 'TransTranslation')
)
end
m_content:addline('')
m_content:newline('')
end,
definition = function()
@ -157,37 +157,35 @@ local process = {
for def in vim.gsplit(m_result.definition, '\n', true) do
def = def:gsub('^%s+', '', 1) -- TODO :判断是否需要分割空格
m_content:addline(
m_indent .. def,
'TransDefinition'
item(m_indent .. def, 'TransDefinition')
)
end
m_content:addline('')
m_content:newline('')
end
end,
failed = function()
m_content:addline(
icon.notfound .. m_indent .. '没有找到相关的翻译',
'TransFailed'
item(conf.icon.notfound .. m_indent .. '没有找到相关的翻译', 'TransFailed')
)
m_window:set_width(m_content.lines[1]:width())
end,
}
local cmd_id
local pin = false
local try_del_keymap = function()
for _, key in pairs(conf.hover.keymap) do
pcall(vim.keymap.del, 'n', key, { buffer = true })
end
end
local action
local next
local _word
local cmd_id
local pin
local next
local action
action = {
pageup = function()
m_window:normal('gg')
@ -255,9 +253,11 @@ action = {
end
end,
play = function()
play = vim.fn.has('linux') == 1 and function()
vim.fn.jobstart('echo ' .. m_result.word .. ' | festival --tts')
end or function()
local file = debug.getinfo(1, "S").source:sub(2):match('(.*)lua/') .. 'tts/say.js'
vim.fn.jobstart('node ' .. file .. ' ' .. _word)
vim.fn.jobstart('node ' .. file .. ' ' .. m_result.word)
end,
}
@ -266,15 +266,11 @@ return function(word)
vim.validate {
word = { word, 's' },
}
_word = word
-- 目前只处理了本地数据库的查询
m_result = require('Trans.query.offline')(word)
local hover = conf.hover
if hover.auto_play then
action.play()
end
local opt = {
m_result = require('Trans.query.offline')(word) -- 目前只处理了本地数据库的查询
local hover = conf.hover
m_window = require("Trans.window")(false, {
relative = 'cursor',
width = hover.width,
height = hover.height,
@ -282,19 +278,28 @@ return function(word)
border = hover.border,
col = 2,
row = 2,
}
})
m_window = require("Trans.window")(false, opt)
m_window.animation = hover.animation
m_content = m_window.contents[1]
if not text then
text = m_content.text_wrap
item = m_content.item_wrap
end
if m_result then
if hover.auto_play then action.play() end
for _, field in ipairs(conf.order) do
process[field]()
end
for act, key in pairs(hover.keymap) do
vim.keymap.set('n', key, action[act], { buffer = true, silent = true })
end
else
process.failed()
m_window:set_width(m_content.lines[1]:width())
end
m_window:draw()
@ -308,8 +313,6 @@ return function(word)
m_window:set('wrap', true)
end)
-- Auto Close
cmd_id = api.nvim_create_autocmd(
hover.auto_close_events, {
@ -321,10 +324,4 @@ return function(word)
api.nvim_del_autocmd(cmd_id)
end,
})
if m_result then
for act, key in pairs(hover.keymap) do
vim.keymap.set('n', key, action[act], { buffer = true, silent = true })
end
end
end

7
lua/Trans/window.lua
View File

@ -179,7 +179,7 @@ local window = {
self.winid = -1
end
end
end,
end,
reopen = function(self, entry, opt, callback)
check_busy()
@ -192,6 +192,10 @@ local window = {
self.winid = api.nvim_open_win(self.bufnr, entry, self.config)
self:open(callback)
end,
set_hl = function(self, name, hl)
api.nvim_set_hl(self.hl, name, hl)
end
}
@ -258,6 +262,7 @@ return function(entry, option)
win:bufset('filetype', 'Trans')
win:bufset('buftype', 'nofile')
api.nvim_win_set_hl_ns(win.winid, win.hl)
---@diagnostic disable-next-line: return-type-mismatch
return win
end

4
tts/say.js
View File

@ -1,6 +1,4 @@
const say = require('say')
word = process.argv
// console.log(word)
say.speak(word.slice(2))
say.speak(process.argv.slice(2))