Subversion Repositories SmartDukaan

    @RequestMapping(value = "/account/closing-statements", method = RequestMethod.GET)

    public ResponseEntity<?> getAccountStatement(HttpServletRequest request, @RequestParam LocalDate closingDate, Model model) throws Exception {

        boolean isAdmin = roleManager.isAdmin(cookiesProcessor.getCookiesObject(request).getRoleIds());

        if (!isAdmin) {

            throw new ProfitMandiBusinessException("Unauthorised access", "PartnerId", "Permission Denied");

        }

        LocalDateTime closingDateTime = closingDate.atStartOfDay().plusDays(1);

        Map<Integer, CustomRetailer> customRetailerMap = retailerService.getAllFofoRetailers();

        Set<Integer> retailersSet = customRetailerMap.keySet();

        Map<Integer, Float> closingPurchaseMap = orderRepository.selectOpeningAmount(closingDateTime, retailersSet);

        Map<Integer, UserWallet> retailerWalletMap = walletService.getRetailerIdUserWalletMap(retailersSet);

        Map<Integer, Integer> walletRetailerMap = retailerWalletMap.entrySet().stream().collect(Collectors.toMap(x -> x.getValue().getId(), x -> x.getKey()));

        Set<Integer> walletSet = walletRetailerMap.keySet();

        Map<Integer, Float> closingBalanceMap = userWalletHistoryRepository.getSumTillDateExcludingPurchase(closingDateTime, walletSet);

        List<List<?>> rows = new ArrayList<>();

        for (Map.Entry<Integer, Float> closingWalletBalance : closingBalanceMap.entrySet()) {

            int walletId = closingWalletBalance.getKey();

            int retailerId = walletRetailerMap.get(walletId);

            if (!closingPurchaseMap.containsKey(retailerId)) {

                closingPurchaseMap.put(retailerId, closingWalletBalance.getValue());

            } else {

                closingPurchaseMap.put(retailerId, closingWalletBalance.getValue() - closingPurchaseMap.get(retailerId));

            }

            float closingValue = closingPurchaseMap.get(retailerId);

            CustomRetailer cr = customRetailerMap.get(retailerId);

            rows.add(Arrays.asList(retailerId, cr.getBusinessName(), cr.getAddress().getCity(), cr.getAddress().getState(), closingValue));

        }

        org.apache.commons.io.output.ByteArrayOutputStream byteArrayOutputStream = FileUtil.getCSVByteStream(Arrays.asList("Id", "Partner Name", "City", "State", "Closing Balance"), rows);

        final HttpHeaders headers = new HttpHeaders();

        headers.set("Content-Type", "text/csv");

        headers.set("Content-disposition", "inline; filename=account-statement-closing-." + StringUtils.toHyphenatedString(closingDate) + ".csv");

        headers.setContentLength(byteArrayOutputStream.toByteArray().length);

        final InputStream inputStream = new ByteArrayInputStream(byteArrayOutputStream.toByteArray());

        final InputStreamResource inputStreamResource = new InputStreamResource(inputStream);

        return new ResponseEntity<InputStreamResource>(inputStreamResource, headers, HttpStatus.OK);

    }