yhlxj/netflix-service/src/main/java/com/cyksj/service/recharge/impl/AfDianRechargeServiceImpl.java

package com.cyksj.service.recharge.impl;

import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.cyksj.common.exception.BusinessRuntimeException;
import com.cyksj.common.task.GlobalThreadPoolTaskExecutor;
import com.cyksj.common.util.Jsons;
import com.cyksj.mapper.*;
import com.cyksj.mapper.manage.cms.CmsUserMapper;
import com.cyksj.model.entity.*;
import com.cyksj.model.response.recharge.AfDianCardKeyResp;
import com.cyksj.model.response.recharge.AfDianTaskStatusResp;
import com.cyksj.redis.RedisService;
import com.cyksj.server.recharge.dto.TaskResult;
import com.cyksj.server.recharge.dto.TokenParseResult;
import com.cyksj.service.chatgpt.GptUserRechargeRecordService;
import com.cyksj.service.recharge.AfDianRechargeService;
import com.cyksj.service.recharge.GptRechargeCardKeyChannelService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.util.Date;

/**
 * 项目名: yhlxj11111111
 * 文件名: AfDianRechargeServiceImpl
 * 创建者: JavaZou
 * 创建时间:2026/1/26 18:16
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class AfDianRechargeServiceImpl implements AfDianRechargeService {
	private final GptRechargeCardKeyChannelService gptRechargeCardKeyChannelService;

	private final CmsUserMapper cmsUserMapper;

	private final GroupsRelationMapper relationMapper;

	private final GroupsMapper groupsMapper;

	private final GoodsDonSkuMapper skuMapper;

	private final RedisService redisService;

	private final OrderDonMapper orderDonMapper;

	private final GptRechargeCardKeyMapper gptRechargeCardKeyMapper;

	private final GptUserRechargeRecordService gptUserRechargeRecordService;

	private static final GlobalThreadPoolTaskExecutor TASK_EXECUTOR = GlobalThreadPoolTaskExecutor.getInstance();

	@Override
	public AfDianCardKeyResp afDianValidateCardKey(String cardKey) throws Exception {
		String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(cardKey);
		String url = BASE_URL + "/cdks/public/check";

		JSONObject params = new JSONObject();
		params.putOpt("code", cardKey);
		HttpResponse response = HttpRequest.post(url)
				.body(params.toString())
				.execute();
		AfDianCardKeyResp afDianCardKeyResp = Jsons.parseObject(response.body(), AfDianCardKeyResp.class);
		return afDianCardKeyResp;
	}

	@Override
	public String confirmRecharge(GroupsRelation relation, String gptCardKey, CmsUser cmsUser) {
		Long relationId = relation.getId();
		//校验卡密是否可用
		AfDianCardKeyResp afDianCardKeyResp = null;
		try {
			afDianCardKeyResp = afDianValidateCardKey(gptCardKey);
		} catch (Exception e) {
			throw BusinessRuntimeException.getInstance("系统繁忙，请重试");
		}
		if (afDianCardKeyResp.getUsed()) {
			log.error("卡密:{}已被使用", gptCardKey);
			throw BusinessRuntimeException.getInstance("系统繁忙，请重试");
		}
		String accessToken = relation.getGptToken();
		String rechargeAccount = relation.getAccount();
		//提交任务
		String taskId = null;
		try {
			taskId = submitTask(gptCardKey, accessToken);
		} catch (Exception e) {
			setRechargeRelationStatusAutoError(relation);
			if (cmsUser == null) {
				throw BusinessRuntimeException.getInstance("充值失败，请联系客服开通");
			}
			throw BusinessRuntimeException.getInstance("充值失败");
		}
		String operator = null;
		if (cmsUser != null) {
			operator = cmsUser.getNickname();
		}
		String password = relation.getPassword();
		relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
				//记录代充账号
				.set(GroupsRelation::getAccount, rechargeAccount)
				//记录操作人员
				.set(operator != null, GroupsRelation::getOperator, operator)
				//密码
				.set(StrUtil.isNotBlank(password), GroupsRelation::getPassword, password)
				.set(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharging)
				.set(GroupsRelation::getCardKey, gptCardKey)
				.set(GroupsRelation::getGptToken, accessToken)
				.set(GroupsRelation::getGptTaskId, taskId)
				.eq(GroupsRelation::getId, relationId));
		return taskId;
	}

	private String submitTask(String cardKey, String accessToken) {
		log.info("开始提交afdian任务 - 卡密: {}", cardKey);
		try {
			String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(cardKey);
			String url = BASE_URL + "/stocks/public/outstock";
			JSONObject requestBody = new JSONObject();
			requestBody.put("cdk", cardKey);
			requestBody.put("user", accessToken);

			HttpResponse response = HttpRequest.post(url)
					.setConnectionTimeout(10000)
					.body(requestBody.toString())
					.execute();
			String body = response.body();
			log.info("任务提交响应: {}", body);

			return body;
		} catch (Exception e) {
			log.error("提交任务失败: {}", e.getMessage(), e);
			throw new RuntimeException("提交任务失败: " + e.getMessage(), e);
		}
	}

	@Override
	public TaskResult getTaskResult(GroupsRelation relation) {
		try {
			String taskId = relation.getGptTaskId();
			String cardKey = relation.getCardKey();
			String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(cardKey);
			String url = BASE_URL + "/stocks/public/outstock/" + taskId;
			HttpResponse response = HttpRequest.get(url).setConnectionTimeout(10000).execute();

			String body = response.body();
			log.info("任务查询响应: {}", body);

			AfDianTaskStatusResp taskStatusResp = Jsons.parseObject(body, AfDianTaskStatusResp.class);
			//任务不存在
			if (StrUtil.isEmpty(taskStatusResp.getTask_id())) {
				return null;
			}
			Boolean pending = taskStatusResp.getPending();

			Boolean status = taskStatusResp.getSuccess();
			String resultResult = taskStatusResp.getMessage();
			//更新代充状态
			resetGroupsRelationRechargeStatus(relation, pending, status, resultResult);

			TaskResult taskResult = new TaskResult();
			if (status && !pending) {
				taskResult.setStatus("completed");
			} else taskResult.setStatus("failed");
			return taskResult;
		} catch (Exception e) {
			log.error("获取任务结果失败: {}", e.getMessage(), e);
			throw new RuntimeException("获取任务结果失败: " + e.getMessage(), e);
		}
	}

	public static void main(String[] args) throws Exception {
		String url = "https://cz.afdian.org/api" + "/stocks/public/outstock/" + "9fae51b5-344c-46e5-839d-c470c32f1abb";
		HttpResponse response = HttpRequest.get(url).setConnectionTimeout(10000).execute();

		String body = response.body();
		System.out.println(body);
		AfDianTaskStatusResp taskStatusResp = Jsons.parseObject(body, AfDianTaskStatusResp.class);
		System.out.println(Jsons.toJson(taskStatusResp));
	}

	/**
	 * 设置代充自动充值异常
	 */
	public void setRechargeRelationStatusAutoError(GroupsRelation relation) {
		TASK_EXECUTOR.execute(()->{
			relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class)
					.set(GroupsRelation::getAccount, relation.getAccount())
					.set(GroupsRelation::getGptToken, relation.getGptToken())
					.set(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharge_error)
					.eq(GroupsRelation::getId, relation.getId()));
		});
	}

	public TokenParseResult parseToken(String accessToken) {
		log.info("开始验证凭证: {}", accessToken.substring(0, Math.min(20, accessToken.length())) + "...");
		try {
			String BASE_URL = gptRechargeCardKeyChannelService.getCardKeyDomain(null);
			String url = BASE_URL + "/parse-token";
			JSONObject requestBody = new JSONObject();
			requestBody.put("access_token", accessToken);

			HttpResponse response = HttpRequest.post(url)
					.header("accept", "application/json, text/plain, */*")
					.header("accept-language", "zh-CN,zh;q=0.9,en;q=0.8")
					.header("content-type", "application/json")
					.header("origin", "https://www.ow520.com")
					.header("referer", "https://www.ow520.com/")
					.header("user-agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36")
					.body(requestBody.toString())
					.execute();
			String body = response.body();
			log.info("凭证验证响应: {}", body);

			JSONObject jsonResponse = JSONUtil.parseObj(body);
			TokenParseResult result = new TokenParseResult(
					jsonResponse.getStr("message"),
					jsonResponse.getBool("success")
			);
			log.info("凭证验证结果: {}", result);
			return result;
		} catch (Exception e) {
			log.error("验证凭证失败: {}", e.getMessage(), e);
			throw new RuntimeException("验证凭证失败: " + e.getMessage(), e);
		}
	}

	/**
	 * 更新代充状态
	 */
	private void resetGroupsRelationRechargeStatus(GroupsRelation relation, Boolean pending, Boolean status, String result) {
		GroupsRelation.RechargeStatus rechargeStatus;
		if (status && !pending) {
			rechargeStatus = GroupsRelation.RechargeStatus.complete;
		} else {
			rechargeStatus = GroupsRelation.RechargeStatus.recharge_error;
		}
		String taskId = relation.getGptTaskId();
		if (relation != null && relation.getRechargeStatus() == GroupsRelation.RechargeStatus.recharging) {
			Integer rechargeRemainNum = relation.getRechargeRemainNum();
			String cardKey = relation.getCardKey();
			GptRechargeCardKey gptRechargeCardKey = gptRechargeCardKeyMapper.selectOne(Wrappers.lambdaQuery(GptRechargeCardKey.class).eq(GptRechargeCardKey::getCardKey, cardKey).last("limit 1"));
			if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
				if (gptRechargeCardKey != null) {
					gptRechargeCardKeyMapper.update(null, Wrappers.lambdaUpdate(GptRechargeCardKey.class).set(GptRechargeCardKey::getStatus, Boolean.FALSE).set(GptRechargeCardKey::getOrderId, null).set(GptRechargeCardKey::getOrderNo, null).eq(GptRechargeCardKey::getId, gptRechargeCardKey.getId()).eq(GptRechargeCardKey::getStatus, Boolean.TRUE));
				}
			}
			if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
				if (!redisService.setNx(RedisService.key.GPT_RECHARGE_RECOVER_KEY.getName() + taskId, taskId, RedisService.key.GPT_RECHARGE_RECOVER_KEY.getTimeout())) {
					return;
				}
			}
			Date startTime = relation.getStartTime();
			DateTime expiryTime = null;
			if (rechargeStatus == GroupsRelation.RechargeStatus.complete && startTime == null) {
				GroupsTrips groupsTrips = groupsMapper.selectById(relation.getGroupsId());
				if (groupsTrips != null) {
					startTime = DateTime.now();
					GoodsDonSku sku = skuMapper.selectById(groupsTrips.getSkuId());
					if (sku.getDays() != null && sku.getDays() > 0) {
						expiryTime = DateUtil.offsetDay(startTime, sku.getDays());
					} else {
						expiryTime = DateUtil.offsetMonth(startTime, sku.getMonths());
					}
				}
			}
			String rechargeErrorInfo = null;
			if (rechargeStatus == GroupsRelation.RechargeStatus.recharge_error) {
				rechargeErrorInfo = result;
			}
			relationMapper.update(null, Wrappers.lambdaUpdate(GroupsRelation.class).set(rechargeStatus == GroupsRelation.RechargeStatus.recharge_error, GroupsRelation::getRechargeRemainNum, rechargeRemainNum + 1).set(GroupsRelation::getRechargeStatus, rechargeStatus).set(expiryTime != null, GroupsRelation::getStartTime, startTime).set(expiryTime != null, GroupsRelation::getExpiryTime, expiryTime).set(StrUtil.isNotEmpty(rechargeErrorInfo), GroupsRelation::getRechargeErrorInfo, rechargeErrorInfo).eq(GroupsRelation::getId, relation.getId()).le(GroupsRelation::getRechargeRemainNum, relation.getRechargeNum()).eq(GroupsRelation::getRechargeStatus, GroupsRelation.RechargeStatus.recharging));
			//记录代充记录
			if (rechargeStatus == GroupsRelation.RechargeStatus.complete) {
				String remark = "自动代充";
				if (StrUtil.isNotBlank(relation.getOperator())) {
					remark = "后台自动代充";
				}
				gptUserRechargeRecordService.recordGptUserRechargeNum(relation.getUserId(), relation.getId(), relation.getAccount(), -1, relation.getOperator(), remark);
				//修改对应订单状态
				if (gptRechargeCardKey != null) {
					Long orderId = gptRechargeCardKey.getOrderId();
					OrderDon orderDon = orderDonMapper.selectById(orderId);
					if (orderDon != null && orderDon.getStatus() == OrderDon.Status.hasPayment) {
						orderDon.setStatus(OrderDon.Status.complete);
						orderDonMapper.updateById(orderDon);
					}
				}
			}
		}
	}
}