package com.xiaopeng.operation.service.impl;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.PhoneUtil;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xiaopeng.api.model.account.SysUser;
import com.xiaopeng.api.model.operation.XpWhitelist;
import com.xiaopeng.api.model.operation.XpWhitelistOperlog;
import com.xiaopeng.common.core.domain.R;
import com.xiaopeng.common.core.exception.ServiceException;
import com.xiaopeng.common.core.utils.SensitiveDataCryptoUtil;
import com.xiaopeng.common.core.utils.ip.IpUtils;
import com.xiaopeng.common.core.web.page.PageInfo;
import com.xiaopeng.operation.constant.I18nKeyConstant;
import com.xiaopeng.operation.dto.WhitelistAddDto;
import com.xiaopeng.operation.mapper.XpWhitelistMapper;
import com.xiaopeng.operation.mapper.XpWhitelistOperlogMapper;
import com.xiaopeng.operation.query.WhitelistPageListQuery;
import com.xiaopeng.operation.service.IXpWhitelistService;
import org.apache.shardingsphere.transaction.annotation.ShardingSphereTransactionType;
import org.apache.shardingsphere.transaction.core.TransactionType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * <p>
 * 白名单管理 服务实现类
 * </p>
 *
 * @author mitao
 * @since 2026-02-03
 */
@Service
public class XpWhitelistServiceImpl extends ServiceImpl<XpWhitelistMapper, XpWhitelist> implements IXpWhitelistService {

    @Autowired
    private XpWhitelistOperlogMapper xpWhitelistOperlogMapper;
    @Override
    public PageInfo<XpWhitelist> pageList(WhitelistPageListQuery query) {
        query.setAccountQueryValues(SensitiveDataCryptoUtil.queryValues(query.getAccount()));
        PageInfo<XpWhitelist> pageInfo = new PageInfo<>(query.getPageNum(),query.getPageSize());
        List<XpWhitelist> list = this.baseMapper.pageList(pageInfo,query);
        decryptWhitelistAccounts(list);
        pageInfo.setRecords(list);
        return pageInfo;
    }

    @Override
    public List<XpWhitelist> exportList(WhitelistPageListQuery query) {
        query.setAccountQueryValues(SensitiveDataCryptoUtil.queryValues(query.getAccount()));
        List<XpWhitelist> list = this.baseMapper.exportList(query);
        decryptWhitelistAccounts(list);
        return list;
    }

    @Override
    @Transactional
    @ShardingSphereTransactionType(TransactionType.BASE)
    public R addWhitelist(WhitelistAddDto dto, SysUser sysUser) {
        List<String> accounts = Arrays.stream(dto.getAccount().split(","))
                .map(String::trim)
                .filter(StringUtils::hasLength)
                .collect(Collectors.toList());
        if (CollUtil.isEmpty(accounts)) {
            return R.fail("所有帐号添加失败，请检查帐号格式或唯一性");
        }

        int sampleLimitPerReason = 20;
        int totalCount = accounts.size();
        Map<String, Long> reasonCountMap = new HashMap<>();
        Map<String, LinkedHashSet<String>> reasonSampleMap = new HashMap<>();

        Set<String> seenAccounts = new HashSet<>();
        List<String> candidates = new ArrayList<>();
        for (String account : accounts) {
            if (!PhoneUtil.isMobile(account)) {
                recordFail(reasonCountMap, reasonSampleMap, "帐号格式不正确", account, sampleLimitPerReason);
                continue;
            }
            if (!seenAccounts.add(account)) {
                recordFail(reasonCountMap, reasonSampleMap, "文件内重复", account, sampleLimitPerReason);
                continue;
            }
            candidates.add(account);
        }

        Set<String> existsAccounts = new HashSet<>();
        if (CollUtil.isNotEmpty(candidates)) {
            existsAccounts = this.lambdaQuery()
                    .select(XpWhitelist::getAccount)
                    .in(XpWhitelist::getAccount, buildQueryValues(candidates))
                    .list()
                    .stream()
                    .map(XpWhitelist::getAccount)
                    .map(SensitiveDataCryptoUtil::decryptIfPossible)
                    .filter(StringUtils::hasLength)
                    .collect(Collectors.toSet());
        }

        List<XpWhitelist> addList = new ArrayList<>();
        for (String account : candidates) {
            if (existsAccounts.contains(account)) {
                recordFail(reasonCountMap, reasonSampleMap, "帐号已存在", account, sampleLimitPerReason);
                continue;
            }
            XpWhitelist xpWhitelist = new XpWhitelist();

            xpWhitelist.setAccount(SensitiveDataCryptoUtil.encrypt(account));
            xpWhitelist.setStatus(true);
            xpWhitelist.setIsPriorityDispatch(dto.getIsPriorityDispatch());
            xpWhitelist.setWhitelistSource(1);
            xpWhitelist.setCreateBy(sysUser.getNickName());
            addList.add(xpWhitelist);
        }

        int successCount = addList.size();
        int failCount = totalCount - successCount;

        String message;
        if (failCount == 0) {
            message = "成功添加" + successCount + "条白名单记录";
        } else if (successCount == 0) {
            message = "所有帐号添加失败，请检查帐号格式或唯一性";
        } else {
            message = "成功加" + successCount + "条，" + failCount + "条添加失败，失败原因：" + buildReasonDetail(reasonCountMap, reasonSampleMap);
        }

        if (successCount == 0) {
            return R.fail(message);
        }

        this.saveBatch(addList);

        StringBuilder operationContent = new StringBuilder("新增账号:");
        StringBuilder accountBuilder = new StringBuilder();
        for (XpWhitelist item : addList) {
            accountBuilder.append("\"").append(item.getAccount()).append("\"").append(",");
        }
        if (accountBuilder.length() > 0) {
            operationContent.append(accountBuilder.substring(0, accountBuilder.length() - 1));
        }
        if (failCount > 0) {
            operationContent.append("；失败").append(failCount).append("条");
            operationContent.append("；原因").append(buildReasonSummary(reasonCountMap));
        }

        XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
        xpWhitelistOperlog.setOperationName(sysUser.getNickName());
        xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
        xpWhitelistOperlog.setOperationType(1);
        xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
        xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
        xpWhitelistOperlog.setOperationContent(truncate(operationContent.toString(), 200));
        xpWhitelistOperlogMapper.insert(xpWhitelistOperlog);

        return R.ok(null, message);
    }

    private static void recordFail(Map<String, Long> reasonCountMap,
                                   Map<String, LinkedHashSet<String>> reasonSampleMap,
                                   String reason,
                                   String sample,
                                   int sampleLimitPerReason) {
        reasonCountMap.merge(reason, 1L, Long::sum);
        if (!StringUtils.hasLength(sample) || sampleLimitPerReason <= 0) {
            return;
        }
        LinkedHashSet<String> samples = reasonSampleMap.computeIfAbsent(reason, k -> new LinkedHashSet<>());
        if (samples.size() >= sampleLimitPerReason) {
            return;
        }
        samples.add(sample);
    }

    private static String buildReasonSummary(Map<String, Long> reasonCountMap) {
        if (reasonCountMap == null || reasonCountMap.isEmpty()) {
            return "请检查帐号格式或唯一性";
        }
        return reasonCountMap.entrySet().stream()
                .sorted(Comparator.<Map.Entry<String, Long>>comparingLong(e -> e.getValue() == null ? 0L : e.getValue())
                        .reversed()
                        .thenComparing(Map.Entry::getKey))
                .map(e -> e.getKey() + "(" + (e.getValue() == null ? 0L : e.getValue()) + ")")
                .collect(Collectors.joining("，"));
    }

    private static String buildReasonDetail(Map<String, Long> reasonCountMap,
                                           Map<String, LinkedHashSet<String>> reasonSampleMap) {
        if (reasonCountMap == null || reasonCountMap.isEmpty()) {
            return "请检查帐号格式或唯一性";
        }
        Map<String, LinkedHashSet<String>> safeSamples = reasonSampleMap == null ? new HashMap<>() : reasonSampleMap;
        return reasonCountMap.entrySet().stream()
                .sorted(Comparator.<Map.Entry<String, Long>>comparingLong(e -> e.getValue() == null ? 0L : e.getValue())
                        .reversed()
                        .thenComparing(Map.Entry::getKey))
                .map(entry -> {
                    String reason = entry.getKey();
                    long count = entry.getValue() == null ? 0L : entry.getValue();
                    LinkedHashSet<String> samples = safeSamples.getOrDefault(reason, new LinkedHashSet<>());
                    if (samples.isEmpty()) {
                        return reason + "(" + count + ")";
                    }
                    long remaining = count - samples.size();
                    String suffix = remaining > 0 ? "等" + remaining + "个" : "";
                    String suffixPart = StringUtils.hasLength(suffix) ? ("," + suffix) : "";
                    return reason + "(" + count + ")[" + String.join(",", samples) + suffixPart + "]";
                })
                .collect(Collectors.joining("；"));
    }

    private static String truncate(String value, int maxLength) {
        if (value == null || maxLength <= 0) {
            return value;
        }
        if (value.length() <= maxLength) {
            return value;
        }
        return value.substring(0, maxLength);
    }

    @Override
    public R editWhitelist(XpWhitelist dto, SysUser sysUser) {
        if (!PhoneUtil.isMobile(dto.getAccount())) {
            throw new ServiceException("以下帐号格式不正确：", I18nKeyConstant.ACCOUNT_FORMAT_ERROR, new Object[]{dto.getAccount()});
        }
        XpWhitelist old = this.lambdaQuery()
                .in(XpWhitelist::getAccount, SensitiveDataCryptoUtil.queryValues(dto.getAccount()))
                .ne(XpWhitelist::getId, dto.getId())
                .last("limit 1").one();
        if (old != null) {
            return R.fail("该账号已存在");
        }
        XpWhitelist whitelist = this.getById(dto.getId());

        StringBuilder operationContent = new StringBuilder("编辑账号:");
        StringBuilder accountBuilder = new StringBuilder();
        if (!whitelist.getAccount().equals(dto.getAccount())) {
            accountBuilder.append("白名单账号" + "\"" + whitelist.getAccount() + "\"" + "为" + "\"" + dto.getAccount() + "\"");
        }
        if (StringUtils.hasLength(whitelist.getNickName()) || StringUtils.hasLength(dto.getNickName())) {
            if (!StringUtils.hasLength(whitelist.getNickName())) {
                if (!whitelist.getNickName().equals(dto.getNickName())) {
                    accountBuilder.append("用户名称" + "\"" + whitelist.getNickName() + "\"" + "为" + "\"" + dto.getNickName() + "\"");
                }
            } else {
                if (!whitelist.getNickName().equals(dto.getNickName())) {
                    accountBuilder.append("用户名称" + "\"" + "为" + "\"" + dto.getNickName() + "\"");
                }
            }
        }
        if (whitelist.getStatus() != dto.getStatus()) {
            accountBuilder.append("状态" + "\"" + (whitelist.getIsPriorityDispatch() ? "正常" : "冻结") + "\"" + "为" + "\"" + (dto.getIsPriorityDispatch() ? "正常" : "冻结") + "\"");
        }
        if (whitelist.getIsPriorityDispatch() != dto.getIsPriorityDispatch()) {
            accountBuilder.append("优先派单" + "\"" + (whitelist.getIsPriorityDispatch() ? "是" : "否") + "\"" + "为" + "\"" + (dto.getIsPriorityDispatch() ? "是" : "否") + "\"");
        }
        XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
        xpWhitelistOperlog.setOperationName(sysUser.getNickName());
        xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
        xpWhitelistOperlog.setOperationType(2);
        xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
        xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
        accountBuilder.append("\"").append(dto.getAccount()).append("\"");
        StringBuilder content = operationContent.append(accountBuilder);
        xpWhitelistOperlog.setOperationContent(content.toString());
        xpWhitelistOperlogMapper.insert(xpWhitelistOperlog);
        whitelist.setNickName(dto.getNickName());
        whitelist.setIsPriorityDispatch(dto.getIsPriorityDispatch());



        whitelist.setAccount(SensitiveDataCryptoUtil.encrypt(whitelist.getAccount()));
        whitelist.setStatus(dto.getStatus());
        this.updateById(whitelist);
        return R.ok();
    }

    @Override
    public R freezeUnfreeze(Long id, SysUser sysUser) {
        XpWhitelist whitelist = this.getById(id);
        StringBuilder operationContent = new StringBuilder("编辑账号:"+"\""+whitelist.getAccount()+"\"");
        StringBuilder accountBuilder = new StringBuilder();
        accountBuilder.append("状态" + "\"" + (whitelist.getIsPriorityDispatch() ? "正常" : "冻结") + "\"" + "为" + "\"" + (!whitelist.getIsPriorityDispatch() ? "正常" : "冻结") + "\"");
        XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
        xpWhitelistOperlog.setOperationName(sysUser.getNickName());
        xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
        xpWhitelistOperlog.setOperationType(2);
        xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
        xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
        StringBuilder content = operationContent.append(accountBuilder);
        xpWhitelistOperlog.setOperationContent(content.toString());
        xpWhitelistOperlogMapper.insert(xpWhitelistOperlog);
        whitelist.setStatus(!whitelist.getStatus());
        this.updateById(whitelist);
        return R.ok();
    }

    @Override
    public R deleteByIds(List<Long> ids, SysUser sysUser) {
        List<String> accounts = this.lambdaQuery().in(XpWhitelist::getId, ids).list()
                .stream().map(XpWhitelist::getAccount)
                .map(SensitiveDataCryptoUtil::decryptIfPossible)
                .collect(Collectors.toList());
        this.removeBatchByIds(ids);
        StringBuilder operationContent = new StringBuilder("删除账号:");
        StringBuilder accountBuilder = new StringBuilder();
        for (String account : accounts) {
            accountBuilder.append("\"" + account + "\""+",");
        }
        XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
        xpWhitelistOperlog.setOperationName(sysUser.getNickName());
        xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
        xpWhitelistOperlog.setOperationType(3);
        xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
        xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
        StringBuilder content = operationContent.append(accountBuilder.substring(0, accountBuilder.length() - 1));
        xpWhitelistOperlog.setOperationContent(content.toString());
        xpWhitelistOperlogMapper.insert(xpWhitelistOperlog);
        return R.ok();
    }

    @Override
    public R freezeByIds(List<Long> ids, SysUser sysUser) {
        List<XpWhitelist> list = this.lambdaQuery().in(XpWhitelist::getId, ids).list();
        for (XpWhitelist xpWhitelist : list) {
            xpWhitelist.setStatus(false);
        }
        this.updateBatchById(list);
        StringBuilder operationContent = new StringBuilder("共计冻结"+ids.size()+"条数据");
        XpWhitelistOperlog xpWhitelistOperlog = new XpWhitelistOperlog();
        xpWhitelistOperlog.setOperationName(sysUser.getNickName());
        xpWhitelistOperlog.setOperationAccount(sysUser.getPhoneNumber());
        xpWhitelistOperlog.setOperationType(6);
        xpWhitelistOperlog.setOperationIp(IpUtils.getIpAddr());
        xpWhitelistOperlog.setOperationAccountId(sysUser.getId());
        xpWhitelistOperlog.setOperationContent(operationContent.toString());
        xpWhitelistOperlogMapper.insert(xpWhitelistOperlog);
        return R.ok();
    }

    /**
     * 获取所有白名单
     * @return
     */
    @Override
    public List<String> getAllWhiteList() {
        return lambdaQuery().select(XpWhitelist::getAccount).list()
                .stream().map(XpWhitelist::getAccount)
                .map(SensitiveDataCryptoUtil::decryptIfPossible)
                .collect(Collectors.toList());
    }

    private static List<String> buildQueryValues(Iterable<String> plainValues) {
        Set<String> values = new LinkedHashSet<>();
        if (plainValues == null) {
            return new ArrayList<>();
        }
        for (String plainValue : plainValues) {
            if (StringUtils.hasLength(plainValue)) {
                values.addAll(SensitiveDataCryptoUtil.queryValues(plainValue));
            }
        }
        return new ArrayList<>(values);
    }

    @Override
    public XpWhitelist getById(Serializable id) {
        XpWhitelist whitelist = super.getById(id);
        decryptWhitelistAccount(whitelist);
        return whitelist;
    }

    @Override
    public boolean save(XpWhitelist entity) {
        encryptWhitelistAccount(entity);
        try {
            return super.save(entity);
        } finally {
            decryptWhitelistAccount(entity);
        }
    }

    @Override
    public boolean saveBatch(java.util.Collection<XpWhitelist> entityList) {
        if (entityList != null) {
            entityList.forEach(this::encryptWhitelistAccount);
        }
        try {
            return super.saveBatch(entityList);
        } finally {
            decryptWhitelistAccounts(entityList);
        }
    }

    @Override
    public boolean updateById(XpWhitelist entity) {
        encryptWhitelistAccount(entity);
        try {
            return super.updateById(entity);
        } finally {
            decryptWhitelistAccount(entity);
        }
    }

    @Override
    public boolean updateBatchById(java.util.Collection<XpWhitelist> entityList) {
        if (entityList != null) {
            entityList.forEach(this::encryptWhitelistAccount);
        }
        try {
            return super.updateBatchById(entityList);
        } finally {
            decryptWhitelistAccounts(entityList);
        }
    }

    @Override
    public List<XpWhitelist> list(Wrapper<XpWhitelist> queryWrapper) {
        List<XpWhitelist> list = super.list(queryWrapper);
        decryptWhitelistAccounts(list);
        return list;
    }

    @Override
    public List<XpWhitelist> list() {
        List<XpWhitelist> list = super.list();
        decryptWhitelistAccounts(list);
        return list;
    }

    private void encryptWhitelistAccount(XpWhitelist whitelist) {
        if (whitelist != null) {
            whitelist.setAccount(SensitiveDataCryptoUtil.encrypt(whitelist.getAccount()));
        }
    }

    private static void decryptWhitelistAccount(XpWhitelist whitelist) {
        if (whitelist != null) {
            whitelist.setAccount(SensitiveDataCryptoUtil.decryptIfPossible(whitelist.getAccount()));
        }
    }

    private static void decryptWhitelistAccounts(Iterable<XpWhitelist> list) {
        if (list == null) {
            return;
        }
        for (XpWhitelist item : list) {
            decryptWhitelistAccount(item);
        }
    }
}