rime-ice/lua/is_in_user_dict.lua
2023-12-13 18:23:20 +08:00

29 lines
759 B
Lua

-- 根据是否在用户词典,在结尾加上一个星号 *
-- is_in_user_dict: true 输入过的内容
-- is_in_user_dict: false 或不写 未输入过的内容
local M = {}
function M.init(env)
local config = env.engine.schema.config
env.name_space = env.name_space:gsub('^*', '')
M.is_in_user_dict = config:get_bool(env.name_space)
end
function M.func(input, env)
for cand in input:iter() do
if M.is_in_user_dict then
if cand.type == "user_phrase" then
cand.comment = cand.comment .. '*'
end
else
if cand.type ~= "user_phrase" then
cand.comment = cand.comment .. '*'
end
end
yield(cand)
end
end
return M