WebSVN – SmartDukaan – Diff – /trunk/profitmandi-fofo/src/main/java/com/spice/profitmandi/web/controller/ReportsController.java

 package com.spice.profitmandi.web.controller;
 import com.spice.profitmandi.common.enumuration.ReporticoProject;
 import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
 import com.spice.profitmandi.common.model.CustomRetailer;
+import com.spice.profitmandi.common.model.ProfitMandiConstants;
 import com.spice.profitmandi.common.model.ReporticoUrlInfo;
 import com.spice.profitmandi.common.services.ReporticoService;
 import com.spice.profitmandi.common.util.FileUtil;
 import com.spice.profitmandi.common.util.FormattingUtils;
 import com.spice.profitmandi.dao.entity.auth.AuthUser;
+import com.spice.profitmandi.dao.entity.cs.Position;
 import com.spice.profitmandi.dao.entity.fofo.FofoStore;
 import com.spice.profitmandi.dao.entity.fofo.PendingOrderItem;
 import com.spice.profitmandi.dao.entity.transaction.UserWallet;
 import com.spice.profitmandi.dao.entity.transaction.UserWalletHistory;
 import com.spice.profitmandi.dao.model.*;
 import com.spice.profitmandi.dao.repository.auth.AuthRepository;
 import com.spice.profitmandi.dao.repository.cs.CsService;
+import com.spice.profitmandi.dao.repository.cs.PositionRepository;
 import com.spice.profitmandi.dao.repository.dtr.FofoStoreRepository;
 import com.spice.profitmandi.dao.repository.dtr.RoleRepository;
 import com.spice.profitmandi.dao.repository.fofo.FofoOrderRepository;
 import com.spice.profitmandi.dao.repository.fofo.PendingOrderItemRepository;
 import com.spice.profitmandi.dao.repository.fofo.PendingOrderService;
     private UserWalletHistoryRepository userWalletHistoryRepository;
     @Autowired
     private TransactionService transactionService;
     @Autowired
     private LoanStatementRepository loanStatementRepository;
+    @Autowired
+    PositionRepository positionRepository;
     @PostMapping(value = "/reports/{projectName}/{fileName}")
     public ResponseEntity<?> fetchReport(HttpServletRequest request, @PathVariable String fileName,
                                          @PathVariable ReporticoProject projectName, @RequestBody Map<String, String> paramsMap)
             throws ProfitMandiBusinessException, UnsupportedOperationException, IOException {
         boolean isAdmin = roleManager.isAdmin(fofoDetails.getRoleIds());
         LocalDateTime currentDate = LocalDate.now().atStartOfDay();
         LocalDateTime currentStartMonth = currentDate.minusDays(30).toLocalDate().atStartOfDay();
         List<CollectionSummary> collectionSummaryList = orderRepository.selectCollectionSummary(fofoDetails.getFofoId(), currentStartMonth, currentDate);
         Logger.info("CollectionSummaryList {}", collectionSummaryList);
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", currentDate.minusDays(30).toLocalDate());
         model.addAttribute("endDate", LocalDate.now());
         model.addAttribute("collectionSummaryList", collectionSummaryList);
         model.addAttribute("isAdmin", isAdmin);
         return "partner-collection-summary";
         boolean isAdmin = roleManager.isAdmin(fofoDetails.getRoleIds());
         if (isAdmin) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
+        }
         endDate = LocalDate.now();
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         LOGGER.info("q {}, starDate - {}, endDate - {}", fofoId, startDate, endDate);
         if (isAdmin) {
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         // List<List<?>> rows = new ArrayList<>();
         LOGGER.info("walletSummartList {}", fofoId);
         DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern("dd/MM/yyyy HH:mm");
         String partnerDetail = null;
 // Headers for the table data
+        rows.add(Arrays.asList(
+//                "Id","Business_timestamp",
-        rows.add(Arrays.asList("Id", "Amount", "Refundable_amount", "Reference", "Reference_type", "Running Balance", "Business_timestamp", "timestamp", "Description"));
+                "timestamp", "Reference_type", "Reference", "Amount", "Refundable_amount", "Running Balance", "Description"));
         for (WalletSummaryReportModel walletSummary : walletSummartList) {
             partnerDetail = walletSummary.getName() + "(" + walletSummary.getCode() + "-" + walletSummary.getPhone() + ")" + "-" + walletSummary.getEmail();
             rows.add(Arrays.asList(
-                    walletSummary.getId(),
+//                    walletSummary.getId(),
+//                    FormattingUtils.format(walletSummary.getBusinessTimestamp() != null ? walletSummary.getBusinessTimestamp() : walletSummary.getTimestamp()),
+                    walletSummary.getTimestamp(),
+                    walletSummary.getReferenceType(),
+                    walletSummary.getReference(),
                     walletSummary.getAmount(),
                     walletSummary.getRefundableAmount(),
-                    walletSummary.getReference(),
-                    walletSummary.getReferenceType(),
                     openingBalance + walletSummary.getAmount(),
-                    FormattingUtils.format(walletSummary.getBusinessTimestamp() != null ? walletSummary.getBusinessTimestamp() : walletSummary.getTimestamp()),
-                    walletSummary.getTimestamp(),
                     walletSummary.getDescription()
             ));
+        }
                 LOGGER.info("endDate - " + endDate.atTime(LocalTime.MAX));
+            }
         } else {
             walletSummartList = fofoOrderRepository.selectWalletSummaryReport(loginDetails.getFofoId(), startDate.atStartOfDay(), endDate.atTime(LocalTime.MAX));
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         LOGGER.info("walletSummartList {}", walletSummartList);
         model.addAttribute("walletSummartList", walletSummartList);
         model.addAttribute("openingBalance", (openingBalance + openingPendingAmount));
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         List<WalletSummaryReportModel> walletSummartList = new ArrayList<>();
         if (!isAdmin) {
             walletSummartList = fofoOrderRepository
                     .selectWalletSummaryReport(fofoDetails.getFofoId(), startDate, endDate);
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         LOGGER.info("walletSummartList {}", walletSummartList);
         model.addAttribute("startDate", startDate.toLocalDate());
         model.addAttribute("endDate", endDate.toLocalDate());
         model.addAttribute("walletSummartList", walletSummartList);
         model.addAttribute("isAdmin", isAdmin);
     @RequestMapping(value = "/pendingIndentReport", method = RequestMethod.GET)
     public String getPendingIndentReport(HttpServletRequest request, Model model) throws Exception {
         LoginDetails fofoDetails = cookiesProcessor.getCookiesObject(request);
         boolean isAdmin = roleManager.isAdmin(fofoDetails.getRoleIds());
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         LocalDateTime currentDate = LocalDate.now().atStartOfDay();
         LocalDateTime currentStartMonth = currentDate.minusMonths(2).toLocalDate().atStartOfDay();
         List<PendingIndentReportModel> pendingIndentReports = fofoOrderRepository
         boolean isAdmin = roleManager.isAdmin(fofoDetails.getRoleIds());
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         if (isAdmin) {
             if (fofoId == 0) {
         boolean isAdmin = roleManager.isAdmin(loginDetails.getRoleIds());
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         //LOGGER.info("schemePayoutReports {}", schemePayoutReports);
         List<SchemePayoutReportModel> schemePayoutReports = fofoOrderRepository
         boolean isAdmin = roleManager.isAdmin(loginDetails.getRoleIds());
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            List<Integer> fofoIds = pp.get(authUser.getId());
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         LOGGER.info("q {}, starDate - {}, endDate - {}", fofoId, startDate, endDate);
     @RequestMapping(value = "/offerPayoutReport", method = RequestMethod.GET)
     public String getOfferPayoutReport(HttpServletRequest request, Model model, @RequestParam(required = false) LocalDate startDate, @RequestParam(required = false) LocalDate endDate) throws Exception {
         LoginDetails fofoDetails = cookiesProcessor.getCookiesObject(request);
         boolean isAdmin = roleManager.isAdmin(fofoDetails.getRoleIds());
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
         model.addAttribute("startDate", startDate);
         boolean isAdmin = roleManager.isAdmin(loginDetails.getRoleIds());
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         LOGGER.info("q {}, starDate - {}, endDate - {}", fofoId, startDate, endDate);
         if (isAdmin) {
         boolean isAdmin = roleManager.isAdmin(fofoDetails.getRoleIds());
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailer(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         List<PriceDropReportModel> priceDropReports = orderRepository.selectPriceDropReport(fofoDetails.getFofoId(),
                 startDate.atStartOfDay(), endDate.atTime(LocalTime.MAX));
         boolean isAdmin = roleManager.isAdmin(fofoDetails.getRoleIds());
         if (startDate == null) {
             startDate = LocalDate.now().minusDays(30);
             endDate = LocalDate.now();
+        }
+        AuthUser authUser = authRepository.selectByEmailOrMobile(fofoDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailer(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("startDate", startDate);
         model.addAttribute("endDate", endDate);
         model.addAttribute("isAdmin", isAdmin);
         if (isAdmin) {
             if (fofoId == 0) {
                                    @RequestParam(required = false) LocalDate endDate)
             throws Exception {
         LoginDetails loginDetails = cookiesProcessor.getCookiesObject(request);
         boolean isAdmin = roleManager.isAdmin(loginDetails.getRoleIds());
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            Set<Integer> fofoIds = new HashSet<>(pp.get(authUser.getId()));
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(x -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(x);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
         model.addAttribute("isAdmin", isAdmin);
+        model.addAttribute("isRBM", isRBM);
         return "partner-account-statement";
+    }
     @RequestMapping(value = "/loan/loan-statement", method = RequestMethod.GET)
     public String loanStatement(HttpServletRequest request, Model model)
             throws Exception {
         LoginDetails loginDetails = cookiesProcessor.getCookiesObject(request);
         boolean isAdmin = roleManager.isAdmin(loginDetails.getRoleIds());
+        AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());
+        List<Position> positions = positionRepository.selectPositionByAuthId(authUser.getId());
+        boolean isRBM = positions.stream()
+                .anyMatch(position ->
+                        ProfitMandiConstants.TICKET_CATEGORY_RBM == position.getCategoryId());
+        Map<Integer, List<Integer>> pp = csService.getAuthUserIdPartnerIdMapping();
+        if (isRBM && pp.containsKey(authUser.getId())) {
+            List<Integer> fofoIds = pp.get(authUser.getId());
+            Map<Integer, CustomRetailer> customRetailersMap = fofoIds.stream().map(fofoId -> {
+                try {
+                    return retailerService.getFofoRetailers(true).get(fofoId);
+                } catch (ProfitMandiBusinessException e) {
+                    // TODO Auto-generated catch block
+                    return null;
+                }
+            }).filter(x -> x != null).collect(Collectors.toMap(x -> x.getPartnerId(), x -> x));
+            model.addAttribute("customRetailersMap", customRetailersMap);
+        }
+        model.addAttribute("isRBM", isRBM);
         model.addAttribute("isAdmin", isAdmin);
         return "partner-loan-statement";
+    }

Subversion Repositories SmartDukaan

(root)/trunk/profitmandi-fofo/src/main/java/com/spice/profitmandi/web/controller/ReportsController.java – Rev 34011 → 34066