package cc.mrbird.febs.ai.service.impl;

import cc.mrbird.febs.ai.entity.AiMemberProductUnlock;
import cc.mrbird.febs.ai.entity.AiProduct;
import cc.mrbird.febs.ai.entity.AiProductDependency;
import cc.mrbird.febs.ai.entity.AiProductPoint;
import cc.mrbird.febs.ai.mapper.AiProductMapper;
import cc.mrbird.febs.ai.req.product.ApiProductInfoDto;
import cc.mrbird.febs.ai.req.product.ApiProductPageDto;
import cc.mrbird.febs.ai.req.product.ApiTranslateDto;
import cc.mrbird.febs.ai.res.product.ApiProductInfoVo;
import cc.mrbird.febs.ai.res.product.ApiProductVo;
import cc.mrbird.febs.ai.res.productPoint.ApiProductPointVo;
import cc.mrbird.febs.ai.service.*;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.exception.FebsException;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.aigc.generation.GenerationParam;
import com.alibaba.dashscope.aigc.generation.GenerationResult;
import com.alibaba.dashscope.aigc.generation.TranslationOptions;
import com.alibaba.dashscope.common.Message;
import com.alibaba.dashscope.common.Role;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * AI产品表 Service实现类
 *
 * @author yourname
 * @date 2025-07-29
 */
@Slf4j
@Service
@RequiredArgsConstructor
public class AiProductServiceImpl extends ServiceImpl<AiProductMapper, AiProduct> implements AiProductService {

    private final AiProductMapper aiProductMapper;
    private final AiMemberRoleService aiMemberRoleService;
    private final AiProductCategoryService aiProductCategoryService;
    private final AiProductPointService aiProductPointService;
    private final AiProductDependencyService aiProductDependencyService;
    private final AiMemberProductUnlockService aiMemberProductUnlockService;

    @Override
    public AiProduct getById(String id) {
        return aiProductMapper.selectById( id);
    }

    @Override
    public Page<ApiProductVo> getPageListByQuery(Page<ApiProductVo> page, ApiProductPageDto dto) {
        return aiProductMapper.selectPageListByQuery(page, dto);
    }

    @Override
    public FebsResponse productList(ApiProductPageDto dto) {

        if(StrUtil.isEmpty(dto.getMemberRoleId())){
            String memberRoleId = aiMemberRoleService.getDefaultMemberRoleId();
            dto.setMemberRoleId(memberRoleId);
        }

        if(StrUtil.isEmpty(dto.getCategoryId())){
            String categoryId = aiProductCategoryService.getDefaultProductCategoryId();
            dto.setCategoryId(categoryId);
        }
        // 创建分页对象，传入当前页和每页大小
        Page<ApiProductVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
        Page<ApiProductVo> pageListByQuery = this.getPageListByQuery(page, dto);
        List<ApiProductVo> records = pageListByQuery.getRecords();
        if (CollUtil.isNotEmpty( records)){

            //stream流操作records，返回一个id的集合
            Set<String> productIds = records.stream().map(ApiProductVo::getId).collect(Collectors.toSet());

            List<AiProductDependency> aiProductDependencies = aiProductDependencyService.selectListByProductIds(productIds);
            if (CollUtil.isNotEmpty( aiProductDependencies)){
                //stream流操作aiProductDependencies，返回一个Map<targetProductId,AiProductDependency>
                Map<String, AiProductDependency> targetProductIdAiProductDependencyMap = aiProductDependencies.stream().collect(Collectors.toMap(AiProductDependency::getTargetProductId, aiProductDependency -> aiProductDependency));
                for (ApiProductVo record : records){
                    if (targetProductIdAiProductDependencyMap.containsKey(record.getId())){
                        record.setState(0);
                    }
                }
            }

            if (StrUtil.isNotEmpty(dto.getMemberId())){
                List<AiMemberProductUnlock> aiMemberProductUnlocks = aiMemberProductUnlockService.selectListByMemberId(dto.getMemberId());
                if (CollUtil.isNotEmpty( aiMemberProductUnlocks)){
                    //stream流操作aiMemberProductUnlocks，返回一个Map<targetProductId,AiMemberProductUnlock>
                    Map<String, AiMemberProductUnlock> aiMemberProductUnlocksMap = aiMemberProductUnlocks.stream().collect(Collectors.toMap(AiMemberProductUnlock::getProductId, aiMemberProductUnlock -> aiMemberProductUnlock));
                    for (ApiProductVo record : records){
                        if (aiMemberProductUnlocksMap.containsKey(record.getId())){
                            record.setState(1);
                        }
                    }
                }
            }
        }
        return new FebsResponse().success().data(pageListByQuery);
    }

    @Override
    public FebsResponse productInfo(ApiProductInfoDto dto) {

        ApiProductInfoVo vo = new ApiProductInfoVo();
        String id = dto.getId();
        AiProduct entity = this.getById(id);
        if (ObjectUtil.isNull( entity)){
            throw new FebsException("产品不存在");
        }
        vo.setId(entity.getId());
        vo.setProductCategoryId(entity.getProductCategoryId());
        vo.setName(entity.getName());
        vo.setScene(entity.getScene());
        vo.setTarget(entity.getTarget());
        vo.setBackImg(entity.getBackImg());
        vo.setIconImg(entity.getIconImg());
        vo.setDescription(entity.getDescription());
        vo.setQuestionCount(entity.getQuestionCount());

        vo.setProductPointList(aiProductPointService.listByProductId(id));
        return new FebsResponse().success().data(vo);
    }

    @Override
    public FebsResponse pointInfoList(ApiProductInfoDto dto) {
        String id = dto.getId();
        AiProduct entity = this.getById(id);
        List<ApiProductPointVo> apiProductPointVos = aiProductPointService.listByProductId(id);
        return new FebsResponse().success().data(apiProductPointVos);
    }

    @Override
    public List<AiProduct> getProductListByQuery(LambdaQueryWrapper<AiProduct> productQuery) {
        return aiProductMapper.selectList(productQuery);
    }

    @Override
    public FebsResponse translate(ApiTranslateDto dto) {
        long startTime = System.currentTimeMillis();
        
        HashMap<Object, Object> map = new HashMap<>();
        Generation gen = new Generation();
        Message userMsg = Message.builder()
                .role(Role.USER.getValue())
                .content(dto.getContent())
                .build();
        TranslationOptions options = TranslationOptions.builder()
                .sourceLang(dto.getSourceLang())
                .targetLang(dto.getTargetLang())
                .build();
        GenerationParam param = GenerationParam.builder()
                // 若没有配置环境变量，请用阿里云百炼API Key将下行替换为：.apiKey("sk-xxx")
                .apiKey("sk-babdcf8799144134915cee2683794b2f")
                .model("qwen-mt-plus")
                .messages(Collections.singletonList(userMsg))
                .resultFormat(GenerationParam.ResultFormat.MESSAGE)
                .translationOptions(options)
                .build();
        try {
            GenerationResult result = gen.call(param);
            String content = result.getOutput().getChoices().get(0).getMessage().getContent();
            map.put("content", dto.getContent());
            map.put("translate", content);
            // 计算翻译耗时
            long endTime = System.currentTimeMillis();
            long duration = endTime - startTime;
            map.put("durationMs", duration + "ms");
        } catch (NoApiKeyException e) {
            e.printStackTrace();
        } catch (InputRequiredException e) {
            e.printStackTrace();
        }
        return new FebsResponse().success().data(map);
    }

}