28 lines
657 B
Lua
28 lines
657 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) or nil
|
||
|
end
|
||
|
|
||
|
local is_user = {
|
||
|
user_table = true,
|
||
|
user_phrase = true,
|
||
|
}
|
||
|
|
||
|
function M.func(input)
|
||
|
for cand in input:iter() do
|
||
|
if is_user[cand.type] == M.is_in_user_dict then
|
||
|
cand.comment = cand.comment .. '*'
|
||
|
end
|
||
|
yield(cand)
|
||
|
end
|
||
|
end
|
||
|
|
||
|
return M
|