package org.dromara.ai.service;

import org.dromara.ai.domain.bo.AiUserQuotaBo;
import org.dromara.ai.domain.vo.AiUserQuotaVo;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;

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

/**
 * AI用户配额Service接口
 *
 * @author pengles
 */
public interface AiUserQuotaService {

    /**
     * 查询AI用户配额
     *
     * @param id 主键
     * @return 结果
     */
    AiUserQuotaVo queryById(Long id);

    /**
     * 查询AI用户配额列表
     *
     * @param bo         查询条件
     * @param pageQuery  分页查询条件
     * @return 分页结果
     */
    TableDataInfo<AiUserQuotaVo> queryPageList(AiUserQuotaBo bo, PageQuery pageQuery);

    /**
     * 查询AI用户配额列表
     *
     * @param bo 查询条件
     * @return 列表
     */
    List<AiUserQuotaVo> queryList(AiUserQuotaBo bo);

    /**
     * 新增AI用户配额
     *
     * @param bo AI用户配额
     * @return 是否成功
     */
    Boolean insertByBo(AiUserQuotaBo bo);

    /**
     * 修改AI用户配额
     *
     * @param bo AI用户配额
     * @return 是否成功
     */
    Boolean updateByBo(AiUserQuotaBo bo);

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

    /**
     * 根据用户ID查询配额
     *
     * @param userId 用户ID
     * @return 配额信息
     */
    AiUserQuotaVo queryByUserId(Long userId);

    /**
     * 重置用户配额（按日/按月）
     *
     * @param userId 用户ID
     * @param resetType 重置类型(DAILY/MONTHLY)
     * @return 是否成功
     */
    Boolean resetQuota(Long userId, String resetType);

    /**
     * 增加用户配额使用量
     *
     * @param userId 用户ID
     * @param dailyAmount 日使用量
     * @param monthlyAmount 月使用量
     * @return 是否成功
     */
    Boolean increaseUsage(Long userId, Long dailyAmount, Long monthlyAmount);

}
