gold-java/src/main/java/com/example/demo/controller/cash/CashRefundController.java

package com.example.demo.controller.cash;

import com.example.demo.Util.JWTUtil;
import com.example.demo.Util.LanguageTranslationUtil;
import com.example.demo.config.RateLimitUtil;
import com.example.demo.domain.entity.Admin;
import com.example.demo.domain.vo.cash.CashRecordDTO;
import com.example.demo.domain.vo.cash.CashRecordDone;
import com.example.demo.domain.vo.cash.CashRecordRefund;
import com.example.demo.domain.vo.coin.Page;
import com.example.demo.domain.vo.coin.RechargeUser;
import com.example.demo.domain.vo.coin.Result;
import com.example.demo.service.cash.RefundService;
import com.example.demo.service.coin.MarketService;
import com.example.demo.service.coin.TranslationService;

import jakarta.annotation.Resource;
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;
import java.util.Objects;

/**
 * @program: GOLD
 * @ClassName RefundController
 * @description:
 * @author: huangqizhen
 * @create: 2025−09-26 14:15
 * @Version 1.0
 **/
@RestController
@RequestMapping("/Money")
@RequiredArgsConstructor
@Slf4j
@CrossOrigin
public class CashRefundController {
    @Autowired
    private RefundService refundService;
    @Autowired
    MarketService marketService;

    @Autowired
    private LanguageTranslationUtil languageTranslationUtil;

    @Autowired
    private TranslationService translationService;

    /**
     * 当地财务负责人退款记录
     */
    @PostMapping("/select")
    public Result select(@RequestBody Page page, @RequestHeader(defaultValue = "zh_CN") String lang) throws Exception {
        try {
            // 解析语言代码
            String languageCode = parseLanguageCode(lang);

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

            // 获取当前请求对象
            HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
            String token = request.getHeader("token");

            // 解析 token 获取用户信息
            Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
            List<String> userMarkets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));
            List<String> markets = marketService.getMarketIds(userMarkets);

            // 校验分页参数
            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);
            }

            // 获取传入的市场列表
            List<String> requestedMarkets = page.getCashRecordDTO() != null ? page.getCashRecordDTO().getMarkets() : null;

            // 权限校验逻辑
            if (markets.contains("9") || markets.contains("9999")) {
                // 特权市场：9 或 9999，跳过权限校验，直接放行传入的 markets
                // 如果业务需要，也可以在这里做空值处理
                if (page.getCashRecordDTO() != null) {
                    // 保持 requestedMarkets 不变，原样接受
                    // 可选：如果 requestedMarkets 为 null，可设为默认值或保持 null
                }
            } else {
                // 普通用户：必须校验权限
                if (requestedMarkets == null || requestedMarkets.isEmpty()) {
                    page.getCashRecordDTO().setMarkets(markets);
                }
                if (!markets.containsAll(requestedMarkets)) {
                    String errorMsg = languageTranslationUtil.translate("无权限！请求的市场不在授权范围内。", lang);
                    return Result.error(errorMsg);
                }
                // 校验通过，保持 requestedMarkets 不变
            }

            Result result = Result.success(refundService.financeSelect(page.getPageNum(), page.getPageSize(), page.getCashRecordDTO()));

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

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

    /**
     * 添加退款现金记录
     */
    @PostMapping("/add")
    public Result add(@RequestBody CashRecordRefund cashRecordRefund, @RequestHeader(defaultValue = "zh_CN") String lang) throws Exception {
        try {
            // 解析语言代码
            String languageCode = parseLanguageCode(lang);

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

            cashRecordRefund.setStatus(10);
            int resultCode = refundService.add(cashRecordRefund, lang);
            String successMsg = languageTranslationUtil.translate("提交成功", lang);
            return Result.success(successMsg);
        } catch (Exception e) {
            String errorMsg = languageTranslationUtil.translate(e.getMessage(), lang);
            return Result.error(errorMsg);
        }
    }

    /**
     * 执行人查看退款现金记录
     */
    @PostMapping("/exSelect")
    public Result executor(@RequestBody Page page, @RequestHeader(defaultValue = "zh_CN") String lang) throws Exception {
        try {
            // 解析语言代码
            String languageCode = parseLanguageCode(lang);

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

            HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
            String token = request.getHeader("token");

            // 解析 token 获取用户信息
            Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
            List<String> userMarkets = Arrays.asList(StringUtils.split(admin.getMarkets(), ","));
            List<String> markets = marketService.getMarketIds(userMarkets);

            // 校验分页参数
            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);
            }

            Result result = Result.success(refundService.exSelect(page.getPageNum(), page.getPageSize(), page.getCashRecordDTO()));

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

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

    /**
     * 查询客服提交现金记录
     */
    @PostMapping("/selecta")
    public Result selecta(@RequestBody Page page, @RequestHeader(defaultValue = "zh_CN") String lang) {
        try {
            // 解析语言代码
            String languageCode = parseLanguageCode(lang);

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

            // 校验分页参数
            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);
            }
            // 获取传入的市场列表
            List<String> requestedMarkets = page.getCashRecordDTO() != null ? page.getCashRecordDTO().getMarkets() : null;

            Result result = Result.success(refundService.select(page.getPageNum(), page.getPageSize(), page.getCashRecordDTO()));

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

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

    @PostMapping("/update")
    public Result update(@RequestBody CashRecordDone cashRecordDone, @RequestHeader(defaultValue = "zh_CN") String lang)throws  Exception {
        try {
            if (cashRecordDone.getStatus() == null) {
                String errorMsg = languageTranslationUtil.translate("状态为空", lang);
                return Result.error(errorMsg);
            }
            if (cashRecordDone.getStatus() == 10) {
                int result = refundService.withdraw(cashRecordDone);
                String msg = result > 0 ? languageTranslationUtil.translate("操作成功", lang) : languageTranslationUtil.translate("操作失败", lang);
                return Result.success(msg);
            }
            else if (cashRecordDone.getStatus() == 11) {
                try {
                    int result = refundService.update(cashRecordDone, lang);
                    String msg = result > 0 ? languageTranslationUtil.translate("操作成功", lang) : languageTranslationUtil.translate("操作失败", lang);
                    return Result.success(msg);
                } catch (Exception e) {
                    String errorMsg = languageTranslationUtil.translate(e.getMessage(), lang);
                    return Result.error(errorMsg);
                }
            }

            else {
                String errorMsg = languageTranslationUtil.translate("该订单状态无法支持此操作", lang);
                return Result.error(errorMsg);
            }
        } catch (Exception e) {
            String errorMsg = languageTranslationUtil.translate("操作失败", lang);
            return Result.error(errorMsg + ": " + e.getMessage());
        }
    }

    @PostMapping("/review")
    public Result review(@RequestBody CashRecordDone cashRecordDone, @RequestHeader(defaultValue = "zh_CN") String lang){
        try {
            int result = refundService.review(cashRecordDone, lang);
            String msg = result > 0 ? languageTranslationUtil.translate("操作成功", lang) : languageTranslationUtil.translate("操作失败", lang);
            return Result.success(msg);
        } catch (Exception e) {
            String errorMsg = languageTranslationUtil.translate(e.getMessage(), lang);
            return Result.error(errorMsg);
        }
    }

    @PostMapping("/finalReview")
    public Result finalReview(@RequestBody CashRecordDone cashRecordDone, HttpServletRequest request, @RequestHeader(defaultValue = "zh_CN") String lang) {
        {
            // --------------- 限流逻辑开始 ---------------
            String limitKey = null;
            try {
                // 1. 优先用「用户ID」作为限流标识（从token解析，精准限流）
                HttpServletRequest req = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
                String token = req.getHeader("token");
                Admin admin = (Admin) JWTUtil.getUserDetailsList(String.valueOf(token), Admin.class);
                if (admin != null && admin.getId() != null) {
                    limitKey = "finalReview_" + admin.getId(); // 格式：接口名_用户ID
                }
            } catch (Exception e) {
                // token解析失败（用户未登录），降级用「IP地址」限流
                limitKey = "finalReview_" + getIpAddress(request); // 格式：接口名_IP
            }

            // 2. 校验限流：3秒内同一key不允许重复请求
            if (Objects.isNull(limitKey) || !RateLimitUtil.isAllowed(limitKey)) {
                String errorMsg = languageTranslationUtil.translate("3秒内只能请求一次，请稍后再试", lang);
                return Result.error(errorMsg); // 限流提示
            }
            // --------------- 限流逻辑结束 ---------------

            try {
                // 原有业务逻辑：执行最终审核
                int result = refundService.finalreview(cashRecordDone, lang);
                String msg = result > 0 ? languageTranslationUtil.translate("操作成功", lang) : languageTranslationUtil.translate("审核失败", lang);
                return Result.success(msg);
            } catch (Exception e) {
                // 接口执行失败时，移除限流标识（允许用户重新尝试）
                RateLimitUtil.removeKey(limitKey);
                String errorMsg = languageTranslationUtil.translate("审核失败：", lang) + e.getMessage();
                return Result.error(errorMsg);
            }
        }
    }

    @PostMapping("/executor")
    public Result executor(@RequestBody CashRecordDone cashRecordDone, @RequestHeader(defaultValue = "zh_CN") String lang) throws Exception {
        try {
            int result = refundService.executor(cashRecordDone);
            String msg = result > 0 ? languageTranslationUtil.translate("操作成功", lang) : languageTranslationUtil.translate("操作失败", lang);
            return Result.success(msg);
        }
        catch (Exception e) {
            String errorMsg = languageTranslationUtil.translate(e.getMessage(), lang);
            return Result.error(errorMsg);
        }
    }

    /**
     * 新增线上退款订单
     */
    @PostMapping("/addOnline")
    public Result addOnline(@RequestBody CashRecordRefund cashRecordRefund, @RequestHeader(defaultValue = "zh_CN") String lang){
        try {
            // 解析语言代码
            String languageCode = parseLanguageCode(lang);

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

            cashRecordRefund.setStatus(20);
            int resultCode = refundService.add(cashRecordRefund,lang);
            String successMsg = languageTranslationUtil.translate("提交成功", lang);
            return Result.success(successMsg);
        } catch (Exception e) {
            String errorMsg = languageTranslationUtil.translate(e.getMessage(), lang);
            return Result.error(errorMsg);
        }
    }

    @PostMapping("/export")
    public Result export(@RequestBody Page page, @RequestHeader(defaultValue = "zh_CN") String lang) throws Exception {
        try {
            // 解析语言代码
            String languageCode = parseLanguageCode(lang);

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

            // 校验分页参数
            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);
            }

            Result result = Result.success(refundService.financeSelect(page.getPageNum(), page.getPageSize(), page.getCashRecordDTO()));

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

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

    @PostMapping("/ceshi")
    public Result ceshi(@RequestHeader(defaultValue = "zh_CN") String lang) {
        String msg = languageTranslationUtil.translate("测试消息", lang);
        return Result.success(msg);
    }


    @PostMapping("/funds")
    public Result funds(@RequestBody Page page){
        refundService.funds(page.getPageNum(), page.getPageSize(), page.getFundsDTO());
        return Result.success(refundService.funds(page.getPageNum(), page.getPageSize(), page.getFundsDTO()));
    }
    /**
     * 辅助方法：获取用户真实IP（处理反向代理/负载均衡场景）
     */
    private String getIpAddress(HttpServletRequest request) {
        String ip = request.getHeader("x-forwarded-for");
        if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("Proxy-Client-IP");
        }
        if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("WL-Proxy-Client-IP");
        }
        if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getRemoteAddr();
        }
        // 多IP场景（如多层代理），取第一个非unknown的IP
        if (ip != null && ip.contains(",")) {
            ip = ip.split(",")[0].trim();
        }
        return ip;
    }

    /**
     * 转换现金记录DTO的多语言字段
     */
    private void translateCashRecordDTOs(com.github.pagehelper.PageInfo<CashRecordDTO> pageInfo, String lang) {
        if (pageInfo != null && pageInfo.getList() != null) {
            for (CashRecordDTO dto : pageInfo.getList()) {
                // 翻译市场名称
                if (dto.getMarketName() != null) {
                    dto.setMarketName(languageTranslationUtil.translate(dto.getMarketName(), lang));
                }
                // 翻译商品名称
                if (dto.getGoodsName() != null) {
                    dto.setGoodsName(languageTranslationUtil.translate(dto.getGoodsName(), lang));
                }
                // 翻译支付方式
                if (dto.getPayType() != null) {
                    dto.setPayType(languageTranslationUtil.translate(dto.getPayType(), lang));
                }
                // 翻译支付币种
                if (dto.getPaymentCurrency() != null) {
                    dto.setPaymentCurrency(languageTranslationUtil.translate(dto.getPaymentCurrency(), lang));
                }
                // 翻译收款币种
                if (dto.getReceivedCurrency() != null) {
                    dto.setReceivedCurrency(languageTranslationUtil.translate(dto.getReceivedCurrency(), 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 convertTranslatedFieldsToChinese(CashRecordDTO cashRecordDTO, String languageCode) {
        if (cashRecordDTO != null) {
            // 转换商品名称
            if (cashRecordDTO.getGoodsName() != null && !cashRecordDTO.getGoodsName().isEmpty()) {
                String chineseGoodsName = translationService.findChineseSimplifiedByTranslation(
                        cashRecordDTO.getGoodsName(), languageCode);
                cashRecordDTO.setGoodsName(chineseGoodsName);
            }

            // 转换支付方式
            if (cashRecordDTO.getPayType() != null && !cashRecordDTO.getPayType().isEmpty()) {
                String chinesePayType = translationService.findChineseSimplifiedByTranslation(
                        cashRecordDTO.getPayType(), languageCode);
                cashRecordDTO.setPayType(chinesePayType);
            }

            // 转换支付币种
            if (cashRecordDTO.getPaymentCurrency() != null && !cashRecordDTO.getPaymentCurrency().isEmpty()) {
                String chineseCurrency = translationService.findChineseSimplifiedByTranslation(
                        cashRecordDTO.getPaymentCurrency(), languageCode);
                cashRecordDTO.setPaymentCurrency(chineseCurrency);
            }
        }
    }

    /**
     * 将退款记录查询条件中的翻译字段转换为中文简体
     */
    private void convertTranslatedRefundFieldsToChinese(CashRecordRefund cashRecordRefund, String languageCode) {
        if (cashRecordRefund != null) {
            // 转换退款理由
            if (cashRecordRefund.getRefundReason() != null && !cashRecordRefund.getRefundReason().isEmpty()) {
                String chineseReason = translationService.findChineseSimplifiedByTranslation(
                        cashRecordRefund.getRefundReason(), languageCode);
                cashRecordRefund.setRefundReason(chineseReason);
            }

            // 转换退款备注
            if (cashRecordRefund.getRefundRemark() != null && !cashRecordRefund.getRefundRemark().isEmpty()) {
                String chineseRemark = translationService.findChineseSimplifiedByTranslation(
                        cashRecordRefund.getRefundRemark(), languageCode);
                cashRecordRefund.setRefundRemark(chineseRemark);
            }

            // 转换退款途径
            if (cashRecordRefund.getRefundChannels() != null && !cashRecordRefund.getRefundChannels().isEmpty()) {
                String chineseChannels = translationService.findChineseSimplifiedByTranslation(
                        cashRecordRefund.getRefundChannels(), languageCode);
                cashRecordRefund.setRefundChannels(chineseChannels);
            }

            // 转换退款币种
            if (cashRecordRefund.getRefundCurrency() != null && !cashRecordRefund.getRefundCurrency().isEmpty()) {
                String chineseCurrency = translationService.findChineseSimplifiedByTranslation(
                        cashRecordRefund.getRefundCurrency(), languageCode);
                cashRecordRefund.setRefundCurrency(chineseCurrency);
            }
        }
    }
}