64 lines
1.6 KiB
Lua
64 lines
1.6 KiB
Lua
|
local M = {}
|
||
|
|
|
||
|
|
local fn, api = vim.fn, vim.api
|
||
|
|
|
||
|
|
M.get_select = function()
|
||
|
|
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
|
||
|
|
M.get_str = function(mode)
|
||
|
|
if mode == 'n' then
|
||
|
|
return fn.expand('<cword>')
|
||
|
|
elseif mode == 'v' then
|
||
|
|
api.nvim_input('<ESC>')
|
||
|
|
return M.get_select()
|
||
|
|
elseif mode == 'i' then
|
||
|
|
-- TODO Use Telescope with fuzzy finder
|
||
|
|
---@diagnostic disable-next-line: param-type-mismatch
|
||
|
|
return fn.input('请输入需要查询的单词:')
|
||
|
|
else
|
||
|
|
error('invalid mode: ' .. mode)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
M.is_English = function(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
|