package org.dromara.ai.service;

import org.dromara.ai.domain.bo.AiSensitiveWordBo;
import org.dromara.ai.domain.vo.AiSensitiveWordVo;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;

import java.util.Collection;
import java.util.List;
import java.util.Set;

/**
 * AI敏感词Service接口
 *
 * @author pengles
 */
public interface AiSensitiveWordService {

    /**
     * 查询AI敏感词
     *
     * @param id 主键
     * @return 结果
     */
    AiSensitiveWordVo queryById(Long id);

    /**
     * 查询AI敏感词列表
     *
     * @param bo         查询条件
     * @param pageQuery  分页查询条件
     * @return 分页结果
     */
    TableDataInfo<AiSensitiveWordVo> queryPageList(AiSensitiveWordBo bo, PageQuery pageQuery);

    /**
     * 查询AI敏感词列表
     *
     * @param bo 查询条件
     * @return 列表
     */
    List<AiSensitiveWordVo> queryList(AiSensitiveWordBo bo);

    /**
     * 新增AI敏感词
     *
     * @param bo AI敏感词
     * @return 是否成功
     */
    Boolean insertByBo(AiSensitiveWordBo bo);

    /**
     * 修改AI敏感词
     *
     * @param bo AI敏感词
     * @return 是否成功
     */
    Boolean updateByBo(AiSensitiveWordBo bo);

    /**
     * 批量删除AI敏感词
     *
     * @param ids     需要删除的ID集合
     * @param isValid 是否校验,true-删除前校验,false-不校验
     * @return 是否成功
     */
    Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid);

    /**
     * 启用/禁用敏感词
     *
     * @param id     敏感词ID
     * @param enabled 启用状态(0=禁用 1=启用)
     * @return 是否成功
     */
    Boolean changeEnabled(Long id, Integer enabled);

    /**
     * 根据词类型获取启用的敏感词集合
     *
     * @param wordType 词类型
     * @return 敏感词集合
     */
    Set<String> getEnabledWordsByType(String wordType);

    /**
     * 获取所有启用的敏感词集合（按类型分组）
     *
     * @return 敏感词集合Map，key为词类型，value为敏感词集合
     */
    java.util.Map<String, Set<String>> getAllEnabledWords();

}