Trans.nvim/lua/Trans/core/handler.lua

---@diagnostic disable: unused-local
local M = {}
local icon = require('Trans').conf.icon


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 function exist(res)
    return res and res ~= ''
end

local function expl(c, text)
    local wrapper = c:text_wrapper()
    -- wrapper('', 'TransTitleRound')
    wrapper('', 'TransTitleRound')
    wrapper(text, 'TransTitle')
    wrapper('', 'TransTitleRound')
    -- wrapper('', 'TransTitleRound')
end

local indent = '    '

M.hover = {
    title = function(result, content)
        local line = content:alloc_items()
        line.add_item(
            result.word,
            'TransWord'
        )

        line.add_item(
            '[' .. (exist(result.phonetic) and result.phonetic or icon.notfound) .. ']',
            'TransPhonetic'
        )

        line.add_item(
            (exist(result.collins) and icon.star:rep(result.collins) or icon.notfound),
            'TransCollins'
        )

        line.add_item(
            (result.oxford == 1 and icon.yes or icon.no)
        )

        line.load()
    end,

    tag = function(result, content)
        if exist(result.tag) then
            expl(content, '标签')

            local tags = {}
            local size = 0
            for tag in vim.gsplit(result.tag, ' ', true) do
                size = size + 1
                tags[size] = tag_map[tag]
            end

            for i = 1, size, 3 do
                content:addline(
                    indent .. tags[i] .. '    ' .. (tags[i + 1] or '') .. '    ' .. (tags[i + 2] or ''),
                    'TransTag'
                )
            end
            content:addline('')
        end
    end,

    pos = function(result, content)
        if exist(result.pos) then
            expl(content, '词性')

            for pos in vim.gsplit(result.pos, '/', true) do
                content:addline(
                    indent .. pos_map[pos:sub(1, 1)] .. pos:sub(3) .. '%',
                    'TransPos'
                )
            end

            content:addline('')
        end
    end,

    exchange = function(result, content)
        if exist(result.exchange) then
            expl(content, '词形变化')

            for exc in vim.gsplit(result.exchange, '/', true) do
                content:addline(
                    indent .. exchange_map[exc:sub(1, 1)] .. '    ' .. exc:sub(3),
                    'TransExchange'
                )
            end

            content:addline('')
        end
    end,

    translation = function(result, content)
        expl(content, '中文翻译')

        for trs in vim.gsplit(result.translation, '\n', true) do
            content:addline(
                indent .. trs,
                'TransTranslation'
            )
        end

        content:addline('')
    end,

    definition = function(result, content)
        if exist(result.definition) then
            expl(content, '英文注释')

            for def in vim.gsplit(result.definition, '\n', true) do
                def = def:gsub('^%s+', '', 1) -- TODO :判断是否需要分割空格
                content:addline(
                    indent .. def,
                    'TransDefinition'
                )
            end

            content:addline('')
        end
    end,

    failed = function(content)
        content:addline(
            icon.notfound .. indent .. '没有找到相关的翻译',
            'TransFailed'
        )
    end,
}

M.process = function(view, result)
    local conf = require('Trans').conf
    local content = require('Trans.core.content'):new(conf.window[view].width)
    if result then
        if view == 'hover' then
            vim.tbl_map(function(handle)
                M.hover[handle](result, content)
            end, conf.order)

        elseif view == 'float' then
            -- TODO :

        else
            error('unknown view ' .. view)
        end
    else
        M[view].failed(content)
    end
    return content
end



--- TODO :Content Handler for float view
M.float = {
    title = function(result, content)

    end,
    tag = function(result, content)

    end,
    pos = function(result, content)

    end,
    exchange = function(result, content)

    end,
    translation = function(result, content)

    end,
    definition = function(result, content)

    end,
    faild = function(result, content)

    end,
}


return M