import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 2+1联动模式激励机制核心功能实现
 * 产品商业模式专家设计
 */
public class IncentiveSystem {
    
    // 用户身份枚举
    public enum UserRole {
        AGENT("代理"), 
        PARTNER("合伙人");
        
        private final String description;
        
        UserRole(String description) {
            this.description = description;
        }
    }
    
    // 奖励类型枚举
    public enum RewardType {
        DIRECT_PUSH("直推奖"),
        INDIRECT_PUSH("间推奖"),
        PEER_LEVEL("平级奖"),
        DIVIDEND("分红奖"),
        SUPPORT_FUND("帮扶基金");
        
        private final String description;
        
        RewardType(String description) {
            this.description = description;
        }
    }
    
    // 用户实体类
    public static class User {
        private String userId;
        private String username;
        private UserRole role;
        private User referrer; // 推荐人
        private List<User> directReferrals = new ArrayList<>(); // 直接推荐列表
        private LocalDateTime registerTime;
        private BigDecimal totalRewards = BigDecimal.ZERO;
        private BigDecimal lockedFunds = BigDecimal.ZERO; // 帮扶基金锁定金额
        private boolean isQualifiedPartner = false; // 是否合格合伙人
        
        // 构造函数、getter、setter等
        public User(String userId, String username, UserRole role) {
            this.userId = userId;
            this.username = username;
            this.role = role;
            this.registerTime = LocalDateTime.now();
        }
        
        // 检查是否可以晋升为合伙人
        public boolean canPromoteToPartner() {
            return this.role == UserRole.AGENT && 
                   this.directReferrals.size() >= 2 &&
                   this.directReferrals.stream()
                       .filter(user -> user.getRole() == UserRole.AGENT)
                       .count() >= 2;
        }
        
        // 晋升为合伙人
        public void promoteToPartner() {
            if (canPromoteToPartner()) {
                this.role = UserRole.PARTNER;
                this.isQualifiedPartner = true;
            }
        }
    }
    
    // 奖励记录
    public static class RewardRecord {
        private String recordId;
        private String userId;
        private RewardType rewardType;
        private BigDecimal amount;
        private String description;
        private LocalDateTime rewardTime;
        private boolean isLocked; // 是否被帮扶基金锁定
        
        public RewardRecord(String userId, RewardType rewardType, BigDecimal amount, String description) {
            this.recordId = UUID.randomUUID().toString();
            this.userId = userId;
            this.rewardType = rewardType;
            this.amount = amount;
            this.description = description;
            this.rewardTime = LocalDateTime.now();
        }
    }
    
    // 激励机制核心服务类
    public static class IncentiveService {
        private Map<String, User> userMap = new ConcurrentHashMap<>();
        private Map<String, List<RewardRecord>> rewardRecords = new ConcurrentHashMap<>();
        private BigDecimal dividendPool = BigDecimal.ZERO; // 分红奖金池
        
        // 奖励配置
        private final BigDecimal DIRECT_REWARD_RATE = new BigDecimal("100"); // 直推奖金额
        private final BigDecimal INDIRECT_REWARD_RATE = new BigDecimal("50");  // 间推奖金额
        private final BigDecimal PEER_REWARD_RATE = new BigDecimal("0.1");     // 平级奖比例
        private final BigDecimal LOCK_RATE = new BigDecimal("0.2");           // 帮扶基金锁定比例
        
        /**
         * 用户注册并处理直推奖励
         */
        public User registerUser(String userId, String username, String referrerId) {
            User referrer = userMap.get(referrerId);
            User newUser = new User(userId, username, UserRole.AGENT);
            newUser.setReferrer(referrer);
            
            userMap.put(userId, newUser);
            
            if (referrer != null) {
                referrer.getDirectReferrals().add(newUser);
                // 处理直推奖励
                processDirectReward(referrer, newUser);
                // 处理间推奖励
                processIndirectReward(referrer, newUser);
            }
            
            return newUser;
        }
        
        /**
         * 处理直推奖励
         */
        private void processDirectReward(User referrer, User newUser) {
            BigDecimal rewardAmount = DIRECT_REWARD_RATE;
            boolean shouldLock = !referrer.isQualifiedPartner();
            
            RewardRecord record = new RewardRecord(
                referrer.getUserId(), 
                RewardType.DIRECT_PUSH, 
                rewardAmount, 
                "推荐新用户 " + newUser.getUsername() + " 的直推奖励"
            );
            
            if (shouldLock) {
                BigDecimal lockedAmount = rewardAmount.multiply(LOCK_RATE);
                BigDecimal actualAmount = rewardAmount.subtract(lockedAmount);
                
                referrer.setTotalRewards(referrer.getTotalRewards().add(actualAmount));
                referrer.setLockedFunds(referrer.getLockedFunds().add(lockedAmount));
                
                record.setAmount(actualAmount);
                record.setLocked(true);
            } else {
                referrer.setTotalRewards(referrer.getTotalRewards().add(rewardAmount));
            }
            
            addRewardRecord(referrer.getUserId(), record);
        }
        
        /**
         * 处理间推奖励
         */
        private void processIndirectReward(User directReferrer, User newUser) {
            User indirectReferrer = directReferrer.getReferrer();
            if (indirectReferrer != null && indirectReferrer.getRole() == UserRole.PARTNER) {
                BigDecimal rewardAmount = INDIRECT_REWARD_RATE;
                
                RewardRecord record = new RewardRecord(
                    indirectReferrer.getUserId(), 
                    RewardType.INDIRECT_PUSH, 
                    rewardAmount, 
                    "团队下级 " + directReferrer.getUsername() + " 推荐新用户的间推奖励"
                );
                
                indirectReferrer.setTotalRewards(indirectReferrer.getTotalRewards().add(rewardAmount));
                addRewardRecord(indirectReferrer.getUserId(), record);
            }
        }
        
        /**
         * 处理用户晋升为合伙人
         */
        public void promoteToPartner(String userId) {
            User user = userMap.get(userId);
            if (user != null && user.canPromoteToPartner()) {
                user.promoteToPartner();
                
                // 处理平级奖励
                processPeerReward(user);
                
                // 释放帮扶基金
                releaseSupportFunds(user);
            }
        }
        
        /**
         * 处理平级奖励
         */
        private void processPeerReward(User newPartner) {
            User superior = newPartner.getReferrer();
            if (superior != null && superior.getRole() == UserRole.PARTNER) {
                // 计算平级奖励：新合伙人收益的10%
                BigDecimal peerReward = newPartner.getTotalRewards().multiply(PEER_REWARD_RATE);
                
                RewardRecord record = new RewardRecord(
                    superior.getUserId(), 
                    RewardType.PEER_LEVEL, 
                    peerReward, 
                    "下级 " + newPartner.getUsername() + " 晋升合伙人的平级奖励"
                );
                
                superior.setTotalRewards(superior.getTotalRewards().add(peerReward));
                addRewardRecord(superior.getUserId(), record);
            }
        }
        
        /**
         * 释放帮扶基金
         */
        private void releaseSupportFunds(User user) {
            if (user.isQualifiedPartner() && user.getLockedFunds().compareTo(BigDecimal.ZERO) > 0) {
                BigDecimal releasedAmount = user.getLockedFunds();
                user.setTotalRewards(user.getTotalRewards().add(releasedAmount));
                user.setLockedFunds(BigDecimal.ZERO);
                
                RewardRecord record = new RewardRecord(
                    user.getUserId(), 
                    RewardType.SUPPORT_FUND, 
                    releasedAmount, 
                    "晋升合伙人后帮扶基金释放"
                );
                
                addRewardRecord(user.getUserId(), record);
            }
        }
        
        /**
         * 处理分红奖励
         */
        public void distributeDividends(Map<String, BigDecimal> partnerPerformance) {
            BigDecimal totalPerformance = partnerPerformance.values().stream()
                .reduce(BigDecimal.ZERO, BigDecimal::add);
            
            if (totalPerformance.compareTo(BigDecimal.ZERO) > 0) {
                for (Map.Entry<String, BigDecimal> entry : partnerPerformance.entrySet()) {
                    String partnerId = entry.getKey();
                    BigDecimal performance = entry.getValue();
                    BigDecimal dividendRatio = performance.divide(totalPerformance, 4, BigDecimal.ROUND_HALF_UP);
                    BigDecimal dividendAmount = dividendPool.multiply(dividendRatio);
                    
                    User partner = userMap.get(partnerId);
                    if (partner != null && partner.getRole() == UserRole.PARTNER) {
                        partner.setTotalRewards(partner.getTotalRewards().add(dividendAmount));
                        
                        RewardRecord record = new RewardRecord(
                            partnerId, 
                            RewardType.DIVIDEND, 
                            dividendAmount, 
                            String.format("基于业绩 %.2f 的分红奖励", performance)
                        );
                        
                        addRewardRecord(partnerId, record);
                    }
                }
                
                // 清空分红池
                dividendPool = BigDecimal.ZERO;
            }
        }
        
        /**
         * 向分红池注入资金
         */
        public void contributeToDividendPool(BigDecimal amount) {
            dividendPool = dividendPool.add(amount);
        }
        
        /**
         * 获取用户奖励记录
         */
        public List<RewardRecord> getUserRewards(String userId) {
            return rewardRecords.getOrDefault(userId, new ArrayList<>());
        }
        
        /**
         * 获取用户团队统计
         */
        public TeamStats getTeamStats(String userId) {
            User user = userMap.get(userId);
            if (user == null) return new TeamStats();
            
            return calculateTeamStats(user, 0);
        }
        
        private TeamStats calculateTeamStats(User user, int depth) {
            TeamStats stats = new TeamStats();
            stats.directCount = user.getDirectReferrals().size();
            stats.totalCount = 1; // 包含自己
            
            for (User referral : user.getDirectReferrals()) {
                TeamStats childStats = calculateTeamStats(referral, depth + 1);
                stats.totalCount += childStats.totalCount;
                stats.partnerCount += (referral.getRole() == UserRole.PARTNER ? 1 : 0) + childStats.partnerCount;
            }
            
            return stats;
        }
        
        // 添加奖励记录
        private void addRewardRecord(String userId, RewardRecord record) {
            rewardRecords.computeIfAbsent(userId, k -> new ArrayList<>()).add(record);
        }
        
        // Getter方法
        public Map<String, User> getUserMap() { return userMap; }
        public BigDecimal getDividendPool() { return dividendPool; }
    }
    
    // 团队统计信息
    public static class TeamStats {
        public int directCount;    // 直接推荐人数
        public int totalCount;     // 团队总人数
        public int partnerCount;   // 团队合伙人数
        
        @Override
        public String toString() {
            return String.format("直接推荐: %d人, 团队总数: %d人, 合伙人: %d人", 
                directCount, totalCount, partnerCount);
        }
    }
    
    // 测试示例
    public static void main(String[] args) {
        IncentiveService incentiveService = new IncentiveService();
        
        // 1. 创建根用户（第一个合伙人）
        User rootUser = new User("U001", "创始人", UserRole.PARTNER);
        rootUser.setQualifiedPartner(true);
        incentiveService.getUserMap().put("U001", rootUser);
        
        // 2. 模拟用户注册和推荐关系
        System.out.println("=== 用户注册和奖励分配 ===");
        
        // 用户A被创始人推荐
        User userA = incentiveService.registerUser("U002", "用户A", "U001");
        System.out.println("用户A注册，创始人获得直推奖");
        
        // 用户B被创始人推荐
        User userB = incentiveService.registerUser("U003", "用户B", "U001");
        System.out.println("用户B注册，创始人获得直推奖");
        
        // 用户C被用户A推荐
        User userC = incentiveService.registerUser("U004", "用户C", "U002");
        System.out.println("用户C注册，用户A获得直推奖，创始人获得间推奖");
        
        // 3. 用户A晋升为合伙人
        System.out.println("\n=== 用户晋升合伙人 ===");
        incentiveService.promoteToPartner("U002");
        System.out.println("用户A晋升为合伙人，帮扶基金释放，创始人获得平级奖");
        
        // 4. 分红分配
        System.out.println("\n=== 分红奖励分配 ===");
        incentiveService.contributeToDividendPool(new BigDecimal("10000"));
        
        Map<String, BigDecimal> performance = new HashMap<>();
        performance.put("U001", new BigDecimal("50000"));
        performance.put("U002", new BigDecimal("30000"));
        
        incentiveService.distributeDividends(performance);
        System.out.println("分红奖励分配完成");
        
        // 5. 显示结果
        System.out.println("\n=== 最终结果 ===");
        displayUserInfo(incentiveService, "U001");
        displayUserInfo(incentiveService, "U002");
        displayUserInfo(incentiveService, "U003");
        displayUserInfo(incentiveService, "U004");
    }
    
    private static void displayUserInfo(IncentiveService service, String userId) {
        User user = service.getUserMap().get(userId);
        if (user != null) {
            TeamStats stats = service.getTeamStats(userId);
            System.out.printf("\n用户: %s (%s)%n", user.getUsername(), user.getRole().description);
            System.out.printf("总收益: %.2f, 锁定资金: %.2f%n", 
                user.getTotalRewards(), user.getLockedFunds());
            System.out.printf("团队统计: %s%n", stats);
            
            List<RewardRecord> records = service.getUserRewards(userId);
            if (!records.isEmpty()) {
                System.out.println("奖励记录:");
                records.forEach(record -> 
                    System.out.printf("  %s: %.2f (%s)%n", 
                        record.getRewardType().description, 
                        record.getAmount(),
                        record.getDescription()));
            }
        }
    }
}

// 补充缺失的getter/setter方法
class User {
    // ... 其他代码同上
    
    public String getUserId() { return userId; }
    public String getUsername() { return username; }
    public UserRole getRole() { return role; }
    public User getReferrer() { return referrer; }
    public List<User> getDirectReferrals() { return directReferrals; }
    public BigDecimal getTotalRewards() { return totalRewards; }
    public BigDecimal getLockedFunds() { return lockedFunds; }
    public boolean isQualifiedPartner() { return isQualifiedPartner; }
    
    public void setReferrer(User referrer) { this.referrer = referrer; }
    public void setTotalRewards(BigDecimal totalRewards) { this.totalRewards = totalRewards; }
    public void setLockedFunds(BigDecimal lockedFunds) { this.lockedFunds = lockedFunds; }
    public void setQualifiedPartner(boolean qualifiedPartner) { isQualifiedPartner = qualifiedPartner; }
}

class RewardRecord {
    // ... 其他代码同上
    
    public String getRecordId() { return recordId; }
    public String getUserId() { return userId; }
    public RewardType getRewardType() { return rewardType; }
    public BigDecimal getAmount() { return amount; }
    public String getDescription() { return description; }
    public LocalDateTime getRewardTime() { return rewardTime; }
    public boolean isLocked() { return isLocked; }
    
    public void setAmount(BigDecimal amount) { this.amount = amount; }
    public void setLocked(boolean locked) { isLocked = locked; }
}