-- 根据是否在用户词典，在结尾加上一个星号 *
-- 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