Subversion Repositories SmartDukaan

import com.spice.profitmandi.dao.model.LastMonthCreditedIncomeModel;

import java.time.LocalDate;

import java.time.YearMonth;

import java.time.format.DateTimeFormatter;

    @Autowired

    SchemeInOutRepository schemeInOutRepository;

    @Autowired

    OfferPayoutRepository offerPayoutRepository;

    @Autowired

    PartnerTypeChangeService partnerTypeChangeService;

    @Override

    public Map<String, Object> computeIncomeMap(int fofoId, int yearMonth) {

        LocalDateTime startOfMonth = LocalDate.now().minusMonths(yearMonth).withDayOfMonth(1).atStartOfDay();

        LocalDateTime endOfMonth = startOfMonth.plusMonths(1);

        YearMonth monthYear = YearMonth.now();

        // 3) Fetch brand-wise data (same as original)

        List<LastMonthCreditedIncomeModel> lastMonthPendingIncomeModels = schemeInOutRepository

                .selectLastMonthPendingIncomeByFofoId(fofoId, startOfMonth, endOfMonth);

        List<LastMonthCreditedIncomeModel> lastMonthPurchaseInMargins = schemeInOutRepository

                .selectLastMonthPurchaseInMarginByFofoId(fofoId, startOfMonth, endOfMonth);

        List<LastMonthCreditedIncomeModel> lastMonthFrontEndIncomes = schemeInOutRepository

                .selectFrontIncomeByBrand(fofoId, startOfMonth, endOfMonth);

        List<LastMonthCreditedIncomeModel> lastMonthSaleMargins = schemeInOutRepository

                .selectLastMonthCreditedIncomeByFofoId(fofoId, startOfMonth, endOfMonth);

        List<OfferPayoutImeiIncomeModel> offerPayoutImeiIncomeModels = offerPayoutRepository

                .getTotalPayoutsByPartnerPeriod(

                        YearMonth.of(startOfMonth.getYear(), startOfMonth.getMonth()),

                        fofoId, null, null);

        // 4) Convert to maps for brand-wise access

        Map<String, LastMonthCreditedIncomeModel> lastMonthPendingIncomeMap =

                lastMonthPendingIncomeModels.stream()

                        .collect(Collectors.toMap(x -> x.getBrand(), x -> x));

        Map<String, LastMonthCreditedIncomeModel> lastMonthPurchaseInMarginMap =

                lastMonthPurchaseInMargins.stream()

                        .collect(Collectors.toMap(x -> x.getBrand(), x -> x));

        Map<String, LastMonthCreditedIncomeModel> lastMonthFrontEndIncomeMap =

                lastMonthFrontEndIncomes.stream()

                        .collect(Collectors.toMap(x -> x.getBrand(), x -> x));

        Map<String, LastMonthCreditedIncomeModel> lastMonthSaleMarginMap =

                lastMonthSaleMargins.stream()

                        .collect(Collectors.toMap(x -> x.getBrand(), x -> x));

        Map<String, Double> additionalPurchasePayout = offerPayoutImeiIncomeModels.stream()

                .collect(Collectors.groupingBy(OfferPayoutImeiIncomeModel::getBrand,

                        Collectors.summingDouble(OfferPayoutImeiIncomeModel::getPurchasePayout)));

        Map<String, Double> additionSalePayout = offerPayoutImeiIncomeModels.stream()

                .collect(Collectors.groupingBy(OfferPayoutImeiIncomeModel::getBrand,

                        Collectors.summingDouble(OfferPayoutImeiIncomeModel::getSalePayout)));

        // 5) Get all unique brands

        Set<String> brandSet = new HashSet<>();

        brandSet.addAll(lastMonthPurchaseInMarginMap.keySet());

        brandSet.addAll(lastMonthSaleMarginMap.keySet());

        brandSet.addAll(lastMonthPendingIncomeMap.keySet());

        brandSet.addAll(additionalPurchasePayout.keySet());

        brandSet.addAll(additionSalePayout.keySet());

        brandSet.addAll(lastMonthFrontEndIncomeMap.keySet());

        // 6) Calculate total amount per brand (same logic as original)

        Map<String, Float> totalAmountMap = new HashMap<>();

        brandSet.forEach(brand -> {

            float total =

                    (lastMonthSaleMarginMap.get(brand) == null ? 0 : lastMonthSaleMarginMap.get(brand).getAmount()) +

                            (lastMonthPurchaseInMarginMap.get(brand) == null ? 0 : lastMonthPurchaseInMarginMap.get(brand).getAmount()) +

                            (lastMonthPendingIncomeMap.get(brand) == null ? 0 : lastMonthPendingIncomeMap.get(brand).getAmount()) +

                            (additionalPurchasePayout.get(brand) == null ? 0 : additionalPurchasePayout.get(brand).floatValue()) +

                            (additionSalePayout.get(brand) == null ? 0 : additionSalePayout.get(brand).floatValue()) +

                            (lastMonthFrontEndIncomeMap.get(brand) == null ? 0 : lastMonthFrontEndIncomeMap.get(brand).getAmount());

            totalAmountMap.put(brand, total);

        });

        // 7) Calculate aggregated totals

        float totalIncome = totalAmountMap.values().stream()

                .reduce(0f, Float::sum);

        float pendingIncome = lastMonthPendingIncomeMap.values().stream()

                .map(LastMonthCreditedIncomeModel::getAmount)

                .reduce(0f, Float::sum);

        float creditedIncome = totalIncome - pendingIncome;

        // 8) Build month labels

        Map<Integer, String> monthValueMap = new HashMap<>();

        DateTimeFormatter fmt = DateTimeFormatter.ofPattern("MMM''uu");

        for (int i = 0; i <= 5; i++) {

            LocalDateTime m = LocalDateTime.now()

                    .withDayOfMonth(1)

                    .minusMonths(i);

            monthValueMap.put(i, m.format(fmt));

        }

        // 9) Assemble result with all brand-wise data

        Map<String, Object> result = new HashMap<>();

        result.put("totalIncome", totalIncome);

        result.put("creditedIncome", creditedIncome);

        result.put("pendingIncome", pendingIncome);

        result.put("monthIndex", yearMonth);

        result.put("monthValueMap", monthValueMap);

        // Include brand-wise data for detailed analysis

        result.put("brandSet", brandSet);

        result.put("totalAmountMap", totalAmountMap);

        result.put("lastMonthPurchaseInMarginMap", lastMonthPurchaseInMarginMap);

        result.put("lastMonthSaleMarginMap", lastMonthSaleMarginMap);

        result.put("lastMonthPendingIncomeMap", lastMonthPendingIncomeMap);

        result.put("additionalPurchasePayoutMap", additionalPurchasePayout);

        result.put("additionalSalePayoutMap", additionSalePayout);

        result.put("lastMonthFrontEndIncomeMap", lastMonthFrontEndIncomeMap);

        return result;

    }