gold-java/src/main/java/com/example/demo/controller/coin/RechargeController.java

package com.example.demo.controller.coin;

import com.example.demo.Util.JWTUtil;
import com.example.demo.Util.LanguageTranslationUtil;
import com.example.demo.config.interfac.Log;
import com.example.demo.domain.entity.Admin;
import com.example.demo.domain.vo.coin.*;
import com.example.demo.service.coin.RechargeService;
import com.example.demo.service.coin.TranslationService;
import com.github.pagehelper.PageInfo;
import jakarta.servlet.http.HttpServletRequest;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.util.Arrays;
import java.util.List;

/**
 * @program: gold-java
 * @ClassName RechargeController
 * @description:
 * @author: Double
 * @create: 2025−06-29 13:01
 * @Version 1.0
 **/

@RestController
@RequestMapping("/recharge")
@RequiredArgsConstructor
@Slf4j
@CrossOrigin
public class RechargeController {
    @Autowired
    private RechargeService rechargeService;
    @Autowired
    private LanguageTranslationUtil languageTranslationUtil;
    @Autowired
    private TranslationService translationService;
    //充值明细
    @Log("获取全部充值明细")
    @PostMapping("/selectAll")
    public Result selectAll(@RequestBody Page page, @RequestHeader(defaultValue = "zh_CN") String lang) {
        try {
            if (ObjectUtils.isEmpty(page.getPageNum())) {
                return Result.error("页码数为空！");
            }
            if (ObjectUtils.isEmpty(page.getPageSize())) {
                return Result.error("页大小为空！");
            } else {
                // 解析语言代码
                String languageCode = parseLanguageCode(lang);

                // 如果不是中文环境，将查询条件中的翻译文本转换为中文简体
                if (!"zh".equalsIgnoreCase(languageCode) && !"zh_cn".equalsIgnoreCase(languageCode)) {
                    convertRechargeTranslatedFieldsToChinese(page.getRechargeUser(), languageCode);
                }

                //解token权限
                HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
                String token = request.getHeader("token");
                Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
                if (admin != null) {
                    List<String> list = Arrays.asList(admin.getMarkets().split(","));
                    page.getRechargeUser().setMarkets(list);
                }
                else{
                    return Result.error("角色为空");
                }
                Result result = Result.success(rechargeService.selectAll(page.getPageNum(), page.getPageSize(),page.getRechargeUser()));
                // 对返回结果进行多语言转换
                if (result.getCode() == 200 && result.getData() instanceof PageInfo) {
                    PageInfo<RechargeUser> pageInfo = (PageInfo<RechargeUser>) result.getData();
                    translateRechargeUsers(pageInfo, lang);
                }
                return result;
            }
        } catch (Exception e) {
            return Result.error(e.getMessage());
        }
    }

    //充值明细筛选
    @Log("充值明细筛选")
    @PostMapping("/selectBy")
    public Result selectBy(@RequestBody Page page, @RequestHeader(defaultValue = "zh_CN") String lang) {
        try {
            if (ObjectUtils.isEmpty(page.getPageNum())) {
                return Result.error("页码数为空！");
            }
            if (ObjectUtils.isEmpty(page.getPageSize())) {
                return Result.error("页大小为空！");
            } else {
                // 解析语言代码
                String languageCode = parseLanguageCode(lang);

                // 如果不是中文环境，将查询条件中的翻译文本转换为中文简体
                if (!"zh".equalsIgnoreCase(languageCode) && !"zh_cn".equalsIgnoreCase(languageCode)) {
                    convertRechargeTranslatedFieldsToChinese(page.getRechargeUser(), languageCode);
                }

                //解token权限
                if(page.getRechargeUser().getMarkets()==null||page.getRechargeUser().getMarkets().isEmpty()){
                    HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
                    String token = request.getHeader("token");
                    Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
                    if (admin != null) {
                        List<String> markets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));;
                        page.getRechargeUser().setMarkets(markets);
                    }
                }
                Result result = Result.success(rechargeService.selectBy(page.getPageNum(), page.getPageSize(), page.getRechargeUser()));
                // 对返回结果进行多语言转换
                if (result.getCode() == 200 && result.getData() instanceof PageInfo) {
                    PageInfo<RechargeUser> pageInfo = (PageInfo<RechargeUser>) result.getData();
                    translateRechargeUsers(pageInfo, lang);
                }
                return result;
            }
        } catch (Exception e) {
            e.printStackTrace();
            return Result.error(e.getMessage());
        }

    }

    //充值金币统计
    @Log("充值金币合计数统计")
    @PostMapping("/statsGold")
    public Result statsGold(@RequestBody RechargeUser rechargeUser) {
        try {
            //解token权限
            if(rechargeUser.getMarkets()==null||rechargeUser.getMarkets().isEmpty()){
                HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
                String token = request.getHeader("token");
                Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
                List<String> markets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));
                rechargeUser.setMarkets(markets);
            }
            Gold gold = rechargeService.statsGold(rechargeUser);
            return Result.success(gold);
        } catch (Exception e) {
            return Result.error(e.getMessage());
        }
    }

    //充值金币增加
    @Log("新增金币充值")
    @PostMapping("/add")
    public Result add(@RequestBody RechargeUser rechargeUser) {
        try {
            return rechargeService.add(rechargeUser);
        } catch (Exception e) {
            return Result.error(e.getMessage());
        }
    }

    @Log("DC免费金币充值")
    @PostMapping("/addDC")
    public Result addDC(@RequestBody RechargeUser rechargeUser) {
        try {
            return rechargeService.addDC(rechargeUser);
        } catch (Exception e) {
            return Result.error(e.getMessage());
        }
    }

    public Result select(@RequestBody Page page, @RequestHeader(defaultValue = "zh_CN") String lang) {
        try {
            if (ObjectUtils.isEmpty(page.getPageNum())) {
                String errorMsg = languageTranslationUtil.translate("页码数为空！", lang);
                return Result.error(errorMsg);
            }
            if (ObjectUtils.isEmpty(page.getPageSize())) {
                String errorMsg = languageTranslationUtil.translate("页大小为空！", lang);
                return Result.error(errorMsg);
            } else {
                // 解析语言代码
                String languageCode = parseLanguageCode(lang);

                // 如果不是中文环境，将查询条件中的翻译文本转换为中文简体
                if (!"zh".equalsIgnoreCase(languageCode) && !"zh_cn".equalsIgnoreCase(languageCode)) {
                    convertRechargeTranslatedFieldsToChinese(page.getRechargeUser(), languageCode);
                }

                //解token权限
                Result result = Result.success(rechargeService.selectBy(page.getPageNum(), page.getPageSize(), page.getRechargeUser()));

                // 对返回结果进行多语言转换
                if (result.getCode() == 200 && result.getData() instanceof PageInfo) {
                    PageInfo<RechargeUser> pageInfo = (PageInfo<RechargeUser>) result.getData();
                    translateRechargeUsers(pageInfo, lang);
                }

                return result;
            }
        } catch (Exception e) {
            String errorMsg = languageTranslationUtil.translate("查询失败", lang);
            return Result.error(errorMsg + ": " + e.getMessage());
        }
    }


    /**
     * 转换充值用户信息的多语言字段
     */
    private void translateRechargeUsers(PageInfo<RechargeUser> pageInfo, String lang) {
        if (pageInfo != null && pageInfo.getList() != null) {
            for (RechargeUser user : pageInfo.getList()) {
                // 翻译市场名称
                if (user.getMarket() != null) {
                    user.setMarket(languageTranslationUtil.translate(user.getMarket(), lang));
                }
                // 翻译汇率名称
                if (user.getRateName() != null) {
                    user.setRateName(languageTranslationUtil.translate(user.getRateName(), lang));
                }
                // 翻译支付方式
                if (user.getPayModel() != null) {
                    user.setPayModel(languageTranslationUtil.translate(user.getPayModel(), lang));
                }
                // 翻译备注
                if (user.getRemark() != null) {
                    user.setRemark(languageTranslationUtil.translate(user.getRemark(), lang));
                }
                // 翻译是否退款
                if (user.getIsRefund() != null) {
                    String isRefundDesc = convertIsRefundToString(user.getIsRefund());
                    user.setIsRefundDesc(languageTranslationUtil.translate(isRefundDesc, lang));
                }
            }
        }
    }
    /**
     * 解析语言代码
     */
    private String parseLanguageCode(String langHeader) {
        if (langHeader == null || langHeader.isEmpty()) {
            return "zh";
        }
        // 处理类似 "en-US" 或 "zh-TW" 的情况
        if (langHeader.contains("-")) {
            String[] parts = langHeader.split("-");
            // 特殊处理中文繁体
            if ("zh".equalsIgnoreCase(parts[0]) && "TW".equalsIgnoreCase(parts[1])) {
                return "zh_TW";
            }
            return parts[0].toLowerCase();
        }
        return langHeader.toLowerCase();
    }

    /**
     * 将充值用户查询条件中的翻译字段转换为中文简体
     */
    private void convertRechargeTranslatedFieldsToChinese(RechargeUser rechargeUser, String languageCode) {
        if (rechargeUser != null) {
            // 转换市场名称
            if (rechargeUser.getMarket() != null && !rechargeUser.getMarket().isEmpty()) {
                String chineseMarket = translationService.findChineseSimplifiedByTranslation(
                        rechargeUser.getMarket(), languageCode);
                rechargeUser.setMarket(chineseMarket);
            }

            // 转换汇率名称
            if (rechargeUser.getRateName() != null && !rechargeUser.getRateName().isEmpty()) {
                String chineseRateName = translationService.findChineseSimplifiedByTranslation(
                        rechargeUser.getRateName(), languageCode);
                rechargeUser.setRateName(chineseRateName);
            }

            // 转换支付方式
            if (rechargeUser.getPayModel() != null && !rechargeUser.getPayModel().isEmpty()) {
                String chinesePayModel = translationService.findChineseSimplifiedByTranslation(
                        rechargeUser.getPayModel(), languageCode);
                rechargeUser.setPayModel(chinesePayModel);
            }

            // 转换备注
            if (rechargeUser.getRemark() != null && !rechargeUser.getRemark().isEmpty()) {
                String chineseRemark = translationService.findChineseSimplifiedByTranslation(
                        rechargeUser.getRemark(), languageCode);
                rechargeUser.setRemark(chineseRemark);
            }
        }
    }

    /**
     * 将是否退款数字转换为中文描述
     */
    private String convertIsRefundToString(Integer isRefund) {
        if (isRefund == null) return "";
        switch (isRefund) {
            case 0: return "未退款";
            case 1: return "已退款";
            default: return "未知状态";
        }
    }
}