package org.ruoyi.system.service.impl;

import cn.dev33.satoken.exception.NotLoginException;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.lang.TypeReference;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import okhttp3.Request;
import org.apache.commons.lang3.math.NumberUtils;
import org.ruoyi.common.core.exception.base.BaseException;
import org.ruoyi.common.core.service.ConfigService;
import org.ruoyi.common.mybatis.core.page.PageQuery;
import org.ruoyi.common.mybatis.core.page.TableDataInfo;
import org.ruoyi.common.wechat.itchat4j.utils.enums.ResultEnum;
import org.ruoyi.system.cofing.OkHttpConfig;
import org.ruoyi.system.domain.Cover;
import org.ruoyi.system.domain.CoverPromptAudio;
import org.ruoyi.system.domain.model.ApiResult;
import org.ruoyi.system.domain.vo.cover.*;
import org.ruoyi.system.mapper.CoverMapper;
import org.ruoyi.system.mapper.CoverPromptAudioMapper;
import org.ruoyi.system.service.ICoverService;
import org.ruoyi.system.util.WeChatScanHttpUtil;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import static org.ruoyi.common.satoken.utils.LoginHelper.getLoginUser;

/**
 * 翻唱Service业务层处理
 *
 * @author NSL
 * @since 2024-12-25
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class CoverServiceImpl implements ICoverService {

    private final CoverMapper coverMapper;
    private final CoverPromptAudioMapper coverPromptAudioMapper;
    private final ConfigService configService;
    private final IChatCostService chatCostService;
    private final WeChatScanHttpUtil weChatScanHttpUtil;
    private final OkHttpConfig okHttpConfig;

    @Override
    public List<MusicVo> searchMusic(String musicName) {
        Map<String, String> queryMap = new HashMap<>();
        queryMap.put("keyword", musicName);
        String json = JSONUtil.toJsonStr(queryMap);
        Request postRequest = weChatScanHttpUtil.createPostRequest(getKey("search_music_url"), json);
        String body = weChatScanHttpUtil.executeRequest(postRequest);
        ApiResult apiResult = JSONUtil.toBean(body, ApiResult.class);
        Object data = apiResult.getData();
        Map<String, List<MusicVo>> userMap = JSONUtil.toBean(JSONUtil.toJsonStr(data), new TypeReference<>() {}, true);
        return userMap.get("musicList");
    }

    @Override
    public void callback(CoverCallbackVo coverCallbackVo) {
        Cover cover = BeanUtil.copyProperties(coverCallbackVo, Cover.class);
        coverMapper.insertOrUpdate(cover);
    }

    @Override
    public void saveCoverTask(CoverParamVo coverParamVo) {
        Long userId = null;
        try {
            userId = getLoginUser().getUserId();
        } catch (Exception e) {
            throw new NotLoginException(NotLoginException.NOT_TOKEN_MESSAGE, NotLoginException.NOT_TOKEN, NotLoginException.NOT_TOKEN);
        }
        List<String> promptAudioUrl = coverParamVo.getPromptAudioUrl();
        Cover cover = BeanUtil.copyProperties(coverParamVo, Cover.class);
        cover.setUserId(userId);
        coverMapper.insertOrUpdate(cover);
        if (CollUtil.isEmpty(promptAudioUrl)) {
            // 查询用户最近一次翻唱记录
            List<CoverPromptAudioVo> coverPromptAudioVoList = coverPromptAudioMapper.selectLatestVoByUserId(userId);
            if (CollUtil.isEmpty(coverPromptAudioVoList)) {
                throw new BaseException("未查询到翻唱记录，请上传参考音频！");
            }
                List<String> promptAudioUrlList = coverPromptAudioVoList.stream().map(CoverPromptAudioVo::getPromptAudioUrl).collect(Collectors.toList());
            coverParamVo.setPromptAudioUrl(promptAudioUrlList);
        } else {
            // 保存用户翻唱记录
            for (String audioUrl : promptAudioUrl) {
                CoverPromptAudio coverPromptAudio = new CoverPromptAudio();
                coverPromptAudio.setPromptAudioUrl(audioUrl);
                coverPromptAudio.setUserId(userId);
                coverPromptAudio.setCoverId(cover.getId());
                coverPromptAudioMapper.insertOrUpdate(coverPromptAudio);
            }
        }
        coverParamVo.setUserId(String.valueOf(userId));
        coverParamVo.setOrderId(String.valueOf(cover.getId()));
        coverParamVo.setUserGender("");
        log.info("发起翻唱任务,入参{}", JSONUtil.toJsonStr(coverParamVo));
        // 调用翻唱API
        Request postRequest = weChatScanHttpUtil.createPostRequest(getKey("cover_url"), JSONUtil.toJsonStr(coverParamVo));
        String body = weChatScanHttpUtil.executeRequest(postRequest);
        ApiResult apiResult = JSONUtil.toBean(body, ApiResult.class);
        if (ResultEnum.SUCCESS.getCode().equals(String.valueOf(apiResult.getCode()))) {
            // 扣费
            chatCostService.taskDeduct("cover", "token", NumberUtils.toDouble(okHttpConfig.getGenerate(), 2.5));
        }
        log.info("调用翻唱结果,出参{}", body);
    }

    @Override
    public TableDataInfo<CoverVo> searchCoverRecord(PageQuery pageQuery) {
        LambdaQueryWrapper<Cover> queryWrapper = Wrappers.lambdaQuery();
        queryWrapper.eq(Cover::getUserId, getLoginUser().getUserId());
        queryWrapper.orderByDesc(Cover::getCreateTime);
        Page<CoverVo> result = coverMapper.selectVoPage(pageQuery.build(), queryWrapper);
        return TableDataInfo.build(result);
    }

    public String getKey(String key) {
        return configService.getConfigValue("cover", key);
    }

}