83 lines
2.0 KiB
Lua
83 lines
2.0 KiB
Lua
---@class Trans
|
|
---@field util TransUtil
|
|
local Trans = require('Trans')
|
|
|
|
local fn, api = vim.fn, vim.api
|
|
|
|
---@class TransUtil
|
|
local M = require('Trans').metatable('util')
|
|
|
|
|
|
---Get selected text
|
|
---@return string
|
|
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 TransMode
|
|
---@return string
|
|
function M.get_str(mode)
|
|
if mode == 'n' or mode == 'normal' then
|
|
return fn.expand('<cword>')
|
|
elseif mode == 'v' or mode == 'visual' then
|
|
api.nvim_input('<ESC>')
|
|
return M.get_select()
|
|
elseif mode == 'input' then
|
|
return fn.expand('<cword>')
|
|
else
|
|
error('Unsupported mode' .. mode)
|
|
end
|
|
end
|
|
|
|
---Puase coroutine for {ms} milliseconds
|
|
---@param ms integer
|
|
function M.pause(ms)
|
|
local co = coroutine.running()
|
|
vim.defer_fn(function()
|
|
coroutine.resume(co)
|
|
end, ms)
|
|
coroutine.yield()
|
|
end
|
|
|
|
---Detect whether the string is English
|
|
---@param str string
|
|
---@return boolean
|
|
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
|
|
|
|
Trans.util = M
|