64 lines
1.6 KiB
Lua
64 lines
1.6 KiB
Lua
M = require('Trans').metatable('util')
|
|
|
|
local fn, api = vim.fn, vim.api
|
|
|
|
function M.get_select()
|
|
local _start = fn.getpos("v")
|
|
local _end = fn.getpos('.')
|
|
|
|
if _start[2] > _end[2] or (_start[3] > _end[3] and _start[2] == _end[2]) then
|
|
_start, _end = _end, _start
|
|
end
|
|
local s_row, s_col = _start[2], _start[3]
|
|
local e_row, e_col = _end[2], _end[3]
|
|
|
|
-- print(s_row, e_row, s_col, e_col)
|
|
---@type string
|
|
---@diagnostic disable-next-line: assign-type-mismatch
|
|
local line = fn.getline(e_row)
|
|
local uidx = vim.str_utfindex(line, math.min(#line, e_col))
|
|
---@diagnostic disable-next-line: param-type-mismatch
|
|
e_col = vim.str_byteindex(line, uidx)
|
|
|
|
|
|
if s_row == e_row then
|
|
return line:sub(s_col, e_col)
|
|
else
|
|
local lines = fn.getline(s_row, e_row)
|
|
local e = #lines
|
|
lines[1] = lines[1]:sub(s_col)
|
|
lines[e] = line:sub(1, e_col)
|
|
return table.concat(lines)
|
|
end
|
|
end
|
|
|
|
---Get Text which need to be translated
|
|
---@param mode string 'n' | 'v' | 'i'
|
|
---@return string
|
|
function M.get_str(mode)
|
|
return ({
|
|
normal = function()
|
|
return fn.expand('<cword>')
|
|
end,
|
|
visual = function()
|
|
api.nvim_input('<ESC>')
|
|
return M.get_select()
|
|
end,
|
|
input = function()
|
|
return fn.input('请输入需要查询的单词:')
|
|
end,
|
|
})[mode]()
|
|
end
|
|
|
|
function M.is_English(str)
|
|
local char = { str:byte(1, -1) }
|
|
for i = 1, #str do
|
|
if char[i] > 128 then
|
|
return false
|
|
end
|
|
end
|
|
return true
|
|
end
|
|
|
|
return M
|