Subversion Repositories SmartDukaan

	@RequestMapping(value = "/downloadNlcByBrand")

	public ResponseEntity<ByteArrayResource> downloadNlcByBrand(HttpServletRequest request,

																@RequestParam String brand, Model model, @RequestParam(defaultValue = "0", required = false) int fofoId) throws Exception {

		LoginDetails loginDetails = cookiesProcessor.getCookiesObject(request);

		if (fofoId > 0 && roleManager.isAdmin(loginDetails.getRoleIds())) {

		} else {

			fofoId = loginDetails.getFofoId();

		}

		PartnerType partnerType = partnerTypeChangeService.getTypeOnDate(fofoId, LocalDate.now());

		List<PartnerType> partnerTypes = new ArrayList<>();

		partnerTypes.add(partnerType);

		partnerTypes.add(PartnerType.ALL);

		PriceCircularModel priceCircular = priceCircularService.getPriceCircularByOffer(fofoId, brand);

		ByteArrayOutputStream baos = getNlcBaos(brand, priceCircular);

		final HttpHeaders headers = new HttpHeaders();

		headers.set("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");

		headers.set("Content-disposition", "inline; filename=\"" + brand + ".csv\"");

		byte[] byteArray = baos.toByteArray();

		headers.setContentLength(byteArray.length);

		return new ResponseEntity<>(new ByteArrayResource(byteArray), headers, HttpStatus.OK);

	}

	private ByteArrayOutputStream getNlcBaos(String brand, PriceCircularModel priceCircular) throws Exception {

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

		for (PriceCircularItemModel priceCircularItemModel : priceCircular.getPriceCircularItemModels()) {

			List<Serializable> row = new ArrayList<>();

			row.add(priceCircularItemModel.getCatalogId());

			row.add(brand);

			row.add(priceCircularItemModel.getModelName());

			row.add(priceCircularItemModel.getModelNumber());

			row.add(priceCircularItemModel.getNetPrice());

			rows.add(row);

		}

		return FileUtil.getCSVByteStream(Arrays.asList("Model Id", "Brand", "Model Name", "Model Number", "Partner Landing"), rows);

	}