package cn.iocoder.yudao.module.wxcp.service.push;

import cn.hutool.core.util.StrUtil;
import cn.iocoder.yudao.module.wxcp.controller.admin.vo.WxCpPushResultVO;
import cn.iocoder.yudao.module.wxcp.enums.activity.WxCpPushScopeEnum;
import cn.iocoder.yudao.module.wxcp.framework.config.WxCpProperties;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import me.chanjar.weixin.cp.api.WxCpService;
import me.chanjar.weixin.common.bean.result.WxMediaUploadResult;
import me.chanjar.weixin.cp.bean.article.NewArticle;
import me.chanjar.weixin.cp.bean.message.WxCpMessage;
import me.chanjar.weixin.cp.bean.message.WxCpMessageSendResult;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

import java.util.Objects;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.wxcp.enums.ErrorCodeConstants.ACTIVITY_PUSH_TARGET_EMPTY;

/**
 * 企业微信消息推送 Service 实现类
 *
 * @author 超级管理员
 */
@Service
@Validated
@Slf4j
public class WxCpMessagePushServiceImpl implements WxCpMessagePushService {

    @Resource
    private WxCpService wxCpService;
    @Resource
    private WxCpProperties wxCpProperties;

    @Override
    public WxCpPushResultVO pushCard(String title, String description, String url,
                                     String btnTxt, Integer scope, String target) {
        // textcard 类型的 url 必填，留空企业微信会返回 41010
        if (StrUtil.isBlank(url)) {
            throw new IllegalArgumentException("推送卡片消息时 url 不能为空");
        }
        WxCpMessage message = WxCpMessage.TEXTCARD()
                .agentId(agentId())
                .title(title)
                .description(description)
                .url(url)
                .btnTxt(StrUtil.blankToDefault(btnTxt, "查看详情"))
                .build();
        return doSend(message, scope, target, "card");
    }

    @Override
    public WxCpPushResultVO pushText(String content, Integer scope, String target) {
        WxCpMessage message = WxCpMessage.TEXT()
                .agentId(agentId())
                .content(content)
                .build();
        return doSend(message, scope, target, "text");
    }

    @Override
    public WxCpPushResultVO pushNews(String title, String description, String url, String picUrl,
                                     Integer scope, String target) {
        if (StrUtil.isBlank(url)) {
            throw new IllegalArgumentException("推送图文消息时 url 不能为空");
        }
        NewArticle article = NewArticle.builder()
                .title(title)
                .description(StrUtil.blankToDefault(description, ""))
                .url(url)
                .picUrl(picUrl)
                .build();
        WxCpMessage message = WxCpMessage.NEWS()
                .agentId(agentId())
                .addArticle(article)
                .build();
        return doSend(message, scope, target, "news");
    }

    @Override
    public WxCpPushResultVO pushFile(java.io.File file, String fileName, String textContent,
                                     Integer scope, String target) {
        if (file == null || !file.exists()) {
            return new WxCpPushResultVO(false, null, "文件不存在，无法推送", null, null);
        }
        try {
            // 1. 上传文件到企业微信，换取 mediaId
            WxMediaUploadResult upload = wxCpService.getMediaService().upload("file", file);
            if (upload == null || StrUtil.isBlank(upload.getMediaId())) {
                return new WxCpPushResultVO(false, null, "文件上传企业微信失败", null, null);
            }
            // 2. 先发送说明文本，再发送文件；文本失败不阻塞文件发送
            if (StrUtil.isNotBlank(textContent)) {
                pushText(textContent, scope, target);
            }
            // 3. 发送文件消息
            WxCpMessage message = WxCpMessage.FILE()
                    .agentId(agentId())
                    .mediaId(upload.getMediaId())
                    .build();
            WxCpPushResultVO result = doSend(message, scope, target, "file");
            log.info("[pushFile][文件推送完成 fileName={} success={}]", fileName, result.getSuccess());
            return result;
        } catch (Exception e) {
            log.error("[pushFile][文件上传或推送异常 fileName={}]", fileName, e);
            return new WxCpPushResultVO(false, null, "文件推送异常：" + e.getMessage(), null, null);
        }
    }

    @Override
    public WxCpPushResultVO pushImage(java.io.File file, Integer scope, String target) {
        if (file == null || !file.exists()) {
            return new WxCpPushResultVO(false, null, "图片文件不存在，无法推送", null, null);
        }
        try {
            // 企业微信图片消息要求先用 image 类型上传换取 mediaId
            WxMediaUploadResult upload = wxCpService.getMediaService().upload("image", file);
            if (upload == null || StrUtil.isBlank(upload.getMediaId())) {
                return new WxCpPushResultVO(false, null, "图片上传企业微信失败", null, null);
            }
            WxCpMessage message = WxCpMessage.IMAGE()
                    .agentId(agentId())
                    .mediaId(upload.getMediaId())
                    .build();
            return doSend(message, scope, target, "image");
        } catch (Exception e) {
            log.error("[pushImage][图片上传或推送异常 file={}]", file.getName(), e);
            return new WxCpPushResultVO(false, null, "图片推送异常：" + e.getMessage(), null, null);
        }
    }

    /**
     * 统一处理接收者解析、发送与错误包装
     */
    private WxCpPushResultVO doSend(WxCpMessage message, Integer scope, String target, String type) {
        Recipients recipients = resolveRecipients(scope, target);
        message.setToUser(recipients.toUser);
        message.setToParty(recipients.toParty);

        WxCpMessageSendResult result;
        try {
            result = wxCpService.getMessageService().send(message);
        } catch (Exception e) {
            // 网络/配置异常不向上抛，转成可读结果，避免前端拿到 500 不知道原因
            log.error("[doSend][企业微信消息推送异常 type={} toUser={} toParty={}]",
                    type, recipients.toUser, recipients.toParty, e);
            return new WxCpPushResultVO(false, null, "推送异常：" + e.getMessage(),
                    recipients.toUser, recipients.toParty);
        }

        if (result.getErrCode() != null && result.getErrCode() == 0) {
            log.info("[doSend][企业微信消息推送成功 type={} toUser={} toParty={}]",
                    type, recipients.toUser, recipients.toParty);
            return new WxCpPushResultVO(true, 0, "推送成功", recipients.toUser, recipients.toParty);
        }
        log.warn("[doSend][企业微信消息推送失败 type={} errCode={} errMsg={}]",
                type, result.getErrCode(), result.getErrMsg());
        return new WxCpPushResultVO(false, result.getErrCode(),
                "企业微信返回：" + result.getErrMsg(), recipients.toUser, recipients.toParty);
    }

    /**
     * 解析推送范围与目标，转成企业微信要求的 | 分隔格式
     */
    private Recipients resolveRecipients(Integer scope, String target) {
        int actualScope = scope != null ? scope : WxCpPushScopeEnum.ALL.getScope();
        if (Objects.equals(actualScope, WxCpPushScopeEnum.ALL.getScope())) {
            return new Recipients("@all", null);
        }
        if (StrUtil.isBlank(target)) {
            throw exception(ACTIVITY_PUSH_TARGET_EMPTY);
        }
        // 同时容忍中英文逗号与竖线，避免配置时手误
        String normalized = target.replace("，", ",").replace(",", "|").replace(" ", "");
        if (Objects.equals(actualScope, WxCpPushScopeEnum.PARTY.getScope())) {
            return new Recipients(null, normalized);
        }
        return new Recipients(normalized, null);
    }

    private Integer agentId() {
        return StrUtil.isBlank(wxCpProperties.getAgentId())
                ? null : Integer.valueOf(wxCpProperties.getAgentId());
    }

    private record Recipients(String toUser, String toParty) {
    }

}