Trans.nvim/lua/Trans/core/translate.lua

local Trans = require('Trans')
local util = Trans.util

local function new_data(opts)
    opts = opts or {}
    local mode = opts.method or ({
        n = 'normal',
        v = 'visual',
    })[vim.api.nvim_get_mode().mode]

    local str = util.get_str(mode)
    if str == '' then return end

    local strategy = Trans.conf.strategy[mode]
    local data = {
        str = str,
        mode = mode,
        frontend = strategy.frontend,
        backend = strategy.backend,
        result = {},
    }

    if util.is_English(str) then
        data.from = 'en'
        data.to = 'zh'
    else
        data.from = 'zh'
        data.to = 'en'
    end

    -- TODO : Check if the str is a word
    data.is_word = true

    return data
end

local function set_result(data)
    local backend_list = data.backend
    local backends = Trans.backend


    local frontend = Trans.frontend[data.frontend]

    local function do_query(name)
        local backend = backends[name]
        if backend.no_wait then
            backend.query(data)
            if type(data.result[name]) == 'table' then
                return
            end
        else
            backend.query(data)
            frontend.wait(data.result, name, backend.timeout)
        end
    end

    for _, name in ipairs(backend_list) do
        do_query(name)
    end
end

local function render_window(data)
    -- TODO
    print('begin to render window')
end

-- HACK : Core process logic
local function process(opts)
    Trans.translate = coroutine.wrap(process)

    local data = new_data(opts)
    if not data then return end

    set_result(data)
    if data.result == false then return end

    render_window(data)
end

return coroutine.wrap(process)