rime-ice/lua/autocap_filter.lua

--[[
	#302@abcdefg233  #305@Mirtle

	自动大写英文词汇：
	- 部分规则不做转换
	- 输入首字母大写，候选词转换为首字母大写： Hello → Hello
	- 输入至少前 2 个字母大写，候选词转换为全部大写： HEllo → HELLO

    大写时无法动态调整词频
--]]
local function autocap_filter(input, env)
    local code = env.engine.context.input -- 输入码
    local codeLen = #code
    local codeAllUCase = false
    local codeUCase = false
    -- 不转换：
    if codeLen == 1 or       -- 码长为 1
        code:find("^[%l%p]") -- 输入码首位为小写字母或标点
    then                     -- 输入码不满足条件不判断候选项
        for cand in input:iter() do
            yield(cand)
        end
        return
    ---- 输入码全大写
    -- elseif code == code:upper() then
    --     codeAllUCase = true
    -- 输入码前 2 - n 位大写
    elseif code:find("^%u%u+.*") then
        codeAllUCase = true
    -- 输入码首位大写
    elseif code:find("^%u.*") then
        codeUCase = true
    end

    local pureCode = code:gsub("[%s%p]", "")     -- 删除标点和空格的输入码
    for cand in input:iter() do
        local text = cand.text                   -- 候选词
        local pureText = text:gsub("[%s%p]", "") -- 删除标点和空格的候选词
        -- 不转换：
        if
            text:find("[^%w%p%s]") or                 -- 候选词包含非字母和数字、非标点符号、非空格的字符
            text:find("%s") or                        -- 候选词中包含空格
            pureText:find("^" .. code) or             -- 输入码完全匹配候选词
            (cand.type ~= "completion" and            -- 单词与其对应的编码不一致
                pureCode:lower() ~= pureText:lower()) -- 例如 PS - Photoshop
        then
            yield(cand)
        -- 输入码前 2~10 位大写，候选词转换为全大写
        elseif codeAllUCase then
            text = text:upper()
            yield(Candidate(cand.type, 0, codeLen, text, cand.comment))
        -- 输入码首位大写，候选词转换为首位大写
        elseif codeUCase then
            text = text:gsub("^%a", string.upper)
            yield(Candidate(cand.type, 0, codeLen, text, cand.comment))
        else
            yield(cand)
        end
    end
end

return autocap_filter