Subversion Repositories SmartDukaan

import com.spice.profitmandi.service.transaction.TransactionService;

	@Autowired

	TransactionService transactionService;

	@RequestMapping(value = "/account/reco", method = RequestMethod.GET)

	public ResponseEntity<?> accountReco(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 = LocalDate.now().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);

		Map<Integer, Float> closingWalletMap = userWalletHistoryRepository.getSumTillDate(closingDateTime, walletSet);

		Map<Integer, Float> inTransitStock = transactionService.getIntransitStockValue();

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

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

			int walletId = closingBalance.getKey();

			int retailerId = walletRetailerMap.get(walletId);

			float accountClosing = 0f;

			if (!closingPurchaseMap.containsKey(retailerId)) {

				accountClosing = closingBalance.getValue();

			} else {

				accountClosing = closingBalance.getValue() - closingPurchaseMap.get(retailerId);

			}

			CustomRetailer cr = customRetailerMap.get(retailerId);

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

					cr.getAddress().getState(), accountClosing, closingWalletMap.get(retailerId), inTransitStock.get(retailerId)));

		}

		org.apache.commons.io.output.ByteArrayOutputStream byteArrayOutputStream = FileUtil

				.getCSVByteStream(Arrays.asList("Id", "Partner Name", "City", "State", "Closing Balance", "Closing Wallet", "In Transit"), 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);

	}