Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.time.LocalDate;

import java.time.LocalDateTime;

import java.time.format.DateTimeFormatter;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import java.util.Set;

import java.util.stream.Collector;

import java.util.stream.Collectors;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.transaction.Transactional;

import org.apache.commons.csv.CSVFormat;

import org.apache.commons.csv.CSVParser;

import org.apache.commons.csv.CSVRecord;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.apache.poi.ss.usermodel.CellStyle;

import org.apache.poi.ss.usermodel.CreationHelper;

import org.apache.poi.xssf.usermodel.XSSFSheet;

import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import org.json.JSONArray;

import org.json.JSONObject;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.core.io.InputStreamResource;

import org.springframework.http.HttpHeaders;

import org.springframework.http.HttpStatus;

import org.springframework.http.ResponseEntity;

import org.springframework.stereotype.Controller;

import org.springframework.ui.Model;

import org.springframework.web.bind.annotation.RequestBody;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestMethod;

import org.springframework.web.bind.annotation.RequestParam;

import org.springframework.web.bind.annotation.RequestPart;

import org.springframework.web.multipart.MultipartFile;

import com.google.gson.Gson;

import com.spice.profitmandi.common.enumuration.MessageType;

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.SendNotificationModel;

import com.spice.profitmandi.common.util.FileUtil;

import com.spice.profitmandi.common.util.StringUtils;

import com.spice.profitmandi.dao.entity.auth.AuthUser;

import com.spice.profitmandi.dao.entity.cs.Position;

import com.spice.profitmandi.dao.entity.logistics.Provider;

import com.spice.profitmandi.dao.entity.logistics.ProviderDetails;

import com.spice.profitmandi.dao.entity.logistics.ProviderTat;

import com.spice.profitmandi.dao.entity.logistics.PublicHolidays;

import com.spice.profitmandi.dao.entity.transaction.Order;

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

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

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

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

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.UserAccountRepository;

import com.spice.profitmandi.dao.repository.logistics.ProviderDetailsRepository;

import com.spice.profitmandi.dao.repository.logistics.ProviderRepository;

import com.spice.profitmandi.dao.repository.logistics.ProviderTatRepository;

import com.spice.profitmandi.dao.repository.logistics.PublicHolidaysRepository;

import com.spice.profitmandi.dao.repository.transaction.OrderRepository;

import com.spice.profitmandi.service.LogisticsService;

import com.spice.profitmandi.service.NotificationService;

import com.spice.profitmandi.web.model.LoginDetails;

import com.spice.profitmandi.web.util.CookiesProcessor;

import com.spice.profitmandi.web.util.MVCResponseSender;

import in.shop2020.logistics.DeliveryType;

import in.shop2020.model.v1.order.OrderStatus;

@Controller

@Transactional(rollbackOn = Throwable.class)

public class LogisticsController {

	@Autowired

	private OrderRepository orderRepository;

	@Autowired

	private ProviderRepository providerRepository;

	@Autowired

	private ProviderDetailsRepository providerDetailsRepository;

	@Autowired

	private MVCResponseSender mvcResponseSender;

	@Autowired

	private ProviderTatRepository providerTatRepository;

	@Autowired

	private PublicHolidaysRepository publicHolidaysRepository;

	@Autowired

	private LogisticsService logisticsService;

	@Autowired

	private UserAccountRepository userAccountRepository;

	@Autowired

	private NotificationService notificationService;

	@Autowired

	private AuthRepository authRepository;

	@Autowired

	private PositionRepository positionRepository;

	@Autowired

	private CookiesProcessor cookiesProcessor;

	@Autowired

	CsService csService;

	@Autowired

	private Gson gson;

	private static final Logger LOGGER = LogManager.getLogger(LogisticsController.class);

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

	public String logistics(HttpServletRequest request, Model model) throws Exception {

		ProviderTat pt = providerTatRepository.selectByProviderId(1, 7678, "110018");

		LocalDate date = logisticsService.calculateDeliveryTimeline(pt);

		LOGGER.info("date" + date);

		return "logistics";

	}

	@RequestMapping(value = "/authorisedWarehouses")

	public String authorisedWarehouses(HttpServletRequest request, Model model) throws ProfitMandiBusinessException {

		LoginDetails loginDetails = cookiesProcessor.getCookiesObject(request);

		String email = loginDetails.getEmailId();

		AuthUser authUser = authRepository.selectByEmailOrMobile(email);

		List<Position> positions = positionRepository.selectAll(authUser.getId());

		List<Position> warehousePositions = positions.stream()

				.filter(x -> x.getCategoryId() == ProfitMandiConstants.TICKET_CATEGORY_WAREHOUSE)

				.collect(Collectors.toList());

		Set<CustomRetailer> positionRetailers = new HashSet<>();

		csService.getPositionCustomRetailerMap(warehousePositions).values().forEach(customRetailers -> {

			positionRetailers.addAll(customRetailers);

		});

		Map<Integer, String> warehouseMap = this.getWarehouses(positionRetailers);

		JSONArray response = new JSONArray();

		warehouseMap.keySet().stream().filter(x -> warehouseMap.get(x) != null).forEach(x -> {

			response.put(new JSONObject().put("id", x).put("name", warehouseMap.get(x)));

		});

		LOGGER.info("response", response.toString());

		model.addAttribute("response", response.toString());

		return "response";

	}

	@RequestMapping(value = "/billedOrderFileUploader", method = RequestMethod.POST)

	public String getBilledOrderFileUploader(HttpServletRequest request, Model model, HttpServletResponse response,

			@RequestPart MultipartFile file) throws Throwable {

		LOGGER.info("file" + file.toString());

		String fileName = file.getName();

		String fileNames = file.getOriginalFilename();

		LOGGER.info("fileName" + fileName);

		LOGGER.info("fileNames" + fileNames);

		List<BilledOrderListModel> billedOrderList = this.readFile(file);

		LoginDetails loginDetails = cookiesProcessor.getCookiesObject(request);

		String email = loginDetails.getEmailId();

		AuthUser authUser = authRepository.selectByEmailOrMobile(email);

		List<Position> positions = positionRepository.selectAll(authUser.getId());

		List<Position> warehousePositions = positions.stream()

				.filter(x -> x.getCategoryId() == ProfitMandiConstants.TICKET_CATEGORY_WAREHOUSE)

				.collect(Collectors.toList());

		Set<CustomRetailer> positionRetailers = new HashSet<>();

		csService.getPositionCustomRetailerMap(warehousePositions).values().forEach(customRetailers -> {

			positionRetailers.addAll(customRetailers);

		});

		LOGGER.info("positionRetailers" + positionRetailers);

		List<Integer> fofoIds = new ArrayList<>();

		List<Integer> warehouseIds = new ArrayList<>();

		if (!positionRetailers.isEmpty()) {

			fofoIds.addAll(positionRetailers.stream().map(x -> x.getPartnerId()).collect(Collectors.toList()));

			warehouseIds = this.getWarehouses(positionRetailers).entrySet().stream().map(x -> x.getKey())

					.collect(Collectors.toList());

		}

		LOGGER.info("fofoIds" + fofoIds);

		Map<String, DispatchNotificationModel> dispatchNotication = new HashMap<>();

		if (!billedOrderList.isEmpty()) {

			List<Order> ords = orderRepository.selectByInvoiceNumbers(

					billedOrderList.stream().map(x -> x.getInvoiceNumber()).collect(Collectors.toList()));

			List<Integer> shippingWarehouseIds = ords.stream().map(x -> x.getWarehouseId())

					.collect(Collectors.toList());

			if (!warehouseIds.containsAll(shippingWarehouseIds)) {

				throw new ProfitMandiBusinessException("Uploaded File", "",

						"Partners are not belong to the same warehouse");

			}

			Map<String, Set<Integer>> invoicefofoIdmap = ords.stream().collect(Collectors.groupingBy(

					x -> x.getInvoiceNumber(), Collectors.mapping(Order::getRetailerId, Collectors.toSet())));

			Map<String, Set<String>> airbillInvoiceNumberMap = billedOrderList.stream()

					.collect(Collectors.groupingBy(x -> x.getAirwayBillNumber(),

							Collectors.mapping(BilledOrderListModel::getInvoiceNumber, Collectors.toSet())));

			for (Entry<String, Set<String>> aim : airbillInvoiceNumberMap.entrySet()) {

				Set<String> InvoceNumbers = aim.getValue();

				Set<Integer> fofoId = new HashSet<>();

				for (String InvoceNumber : InvoceNumbers) {

					fofoId.addAll(invoicefofoIdmap.get(InvoceNumber));

				}

				LOGGER.info("fofoId" + fofoId);

				if (fofoId.size() > 1) {

					throw new ProfitMandiBusinessException("Uploaded File", "",

							"Waybill " + aim.getKey() + " number cannot have invoices with different Retailers");

				}

			}

			for (BilledOrderListModel bl : billedOrderList) {

				List<Order> orders = orderRepository.selectByInvoiceNumber(bl.getInvoiceNumber());

				Provider provider = providerRepository.selectByName(bl.getLogisticsProviderName());

				if (provider == null) {

					throw new ProfitMandiBusinessException("Uploaded File", "", "Provider is not Available");

				}

				if (!orders.isEmpty()) {

					dispatchNotication = logisticsService.markedOrderShippedDetail(orders, provider,

							bl.getAirwayBillNumber());

				} else {

					throw new ProfitMandiBusinessException("Uploaded File", "", "No Order Found");

				}

				if (!dispatchNotication.isEmpty()) {

					for (Entry<String, DispatchNotificationModel> dn : dispatchNotication.entrySet()) {

						Provider pr = providerRepository.selectById(dn.getValue().getProviderId());

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

						invoiceNumber.addAll(dn.getValue().getInvoiceNumber());

						SendNotificationModel sendNotificationModel = new SendNotificationModel();

						sendNotificationModel.setCampaignName("Order Shipped");

						sendNotificationModel.setTitle("Order Shipped");

						sendNotificationModel.setMessage(String.format("Dear customer, your items with Invoice Number "

								+ String.join(",", invoiceNumber) + " worth Rs." + dn.getValue().getTotalAmount()

								+ " and qty " + dn.getValue().getTotalQty() + " has been shipped through "

								+ pr.getName() + " via your track id " + dn.getKey() + " and will be delivered by "

								+ dn.getValue().getDate().format(DateTimeFormatter.ofPattern("dd-MM-yyyy"))));

						sendNotificationModel.setType("url");

						sendNotificationModel.setUrl("https://app.smartdukaan.com/pages/home/dashboard");

						sendNotificationModel.setExpiresat(LocalDateTime.now().plusDays(1));

						sendNotificationModel.setMessageType(MessageType.notification);

						int userId = userAccountRepository.selectUserIdByRetailerId(dn.getValue().getFofoId());

						sendNotificationModel.setUserIds(Arrays.asList(userId));

						notificationService.sendNotification(sendNotificationModel);

					}

				}

				LOGGER.info("dispatchNotication" + dispatchNotication);

			}

		}

		LOGGER.info("dispatchNotication" + dispatchNotication);

		LOGGER.info("billedOrderList" + billedOrderList);

		model.addAttribute("response", mvcResponseSender.createResponseString(true));

		return "response";

	}

	private List<BilledOrderListModel> readFile(MultipartFile file) throws Exception {

		CSVParser parser = new CSVParser(new InputStreamReader(file.getInputStream()), CSVFormat.DEFAULT);

		List<CSVRecord> records = parser.getRecords();

		LOGGER.info("records" + records);

		LOGGER.info("parser" + parser);

		if (records.size() < 2) {

			parser.close();

			throw new ProfitMandiBusinessException("Uploaded File", "", "No records Found");

		}

		// Remove header

		records.remove(0);

		List<BilledOrderListModel> returnList = new ArrayList<BilledOrderListModel>();

		for (CSVRecord record : records) {

			BilledOrderListModel cp = null;

			BilledOrderListModel sgr = createBilleOrder(record);

			returnList.add(sgr);

			LOGGER.info("records" + record.get(1));

		}

		parser.close();

		return returnList;

	}

	private BilledOrderListModel createBilleOrder(CSVRecord record) {

		BilledOrderListModel bol = new BilledOrderListModel();

		bol.setInvoiceNumber(record.get(0));

		bol.setLogisticsProviderName(record.get(1));

		bol.setAirwayBillNumber(record.get(2));

		return bol;

	}

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

	public String provider(HttpServletRequest request, Model model) throws Exception {

		List<Provider> providers = providerRepository.selectAll();

		List<Integer> providerIds = providers.stream().map(x -> x.getId()).collect(Collectors.toList());

		Map<Integer, ProviderDetails> providerDetails = providerDetailsRepository

				.selectByProviderIds(providerIds, DeliveryType.PREPAID, true).stream()

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

		model.addAttribute("providers", providers);

		model.addAttribute("providerDetails", providerDetails);

		model.addAttribute("deliveryTypes", DeliveryType.values());

		return "provider";

	}

	@RequestMapping(value = "/createProvider", method = RequestMethod.POST)

	public String createProvider(HttpServletRequest request, @RequestBody ProviderDetailModel providerDetailModel,

			Model model) throws Exception {

		Provider provider = new Provider();

		provider.setName(providerDetailModel.getName());

		provider.setActive(providerDetailModel.isProvideractive());

		provider.setBundleWeightLimit(0);

		provider.setGroupShipmentAllowed(true);

		provider.setPickup(0);

		provider.setMaxCODLimit(0);

		providerRepository.persist(provider);

		ProviderDetails pd = new ProviderDetails();

		pd.setAccountNo(providerDetailModel.getAccountNo());

		pd.setEmail(providerDetailModel.getEmail());

		pd.setPhone(providerDetailModel.getMobile());

		pd.setProviderId(provider.getId());

		pd.setActive(true);

		pd.setType(DeliveryType.PREPAID);

		providerDetailsRepository.persist(pd);

		model.addAttribute("response", mvcResponseSender.createResponseString(true));

		return "response";

	}

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

	public String providerTat(HttpServletRequest request, Model model) throws Exception {

		List<Provider> providers = providerRepository.selectAll();

		model.addAttribute("providers", providers);

		return "provider-tat";

	}

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

	public String getProviderTat(HttpServletRequest request, @RequestParam int providerId, Model model)

			throws Exception {

		List<ProviderTat> providerTat = providerTatRepository.selectByProviderId(providerId);

		Map<Integer, String> warehouseMap = ProfitMandiConstants.WAREHOUSE_MAP;

		providerTat = providerTat.stream().filter(x -> warehouseMap.containsKey(x.getWarehouseLocation()))

				.collect(Collectors.toList());

		Map<Integer, Provider> providerIdMap = new HashMap<>();

		Provider provider = providerRepository.selectById(providerId);

		providerIdMap.put(provider.getId(), provider);

		LOGGER.info("providerTat" + providerTat);

		model.addAttribute("providerTat", providerTat);

		model.addAttribute("warehouseMap", warehouseMap);

		model.addAttribute("providerIdMap", providerIdMap);

		return "provider-tat-detail";

	}

	@RequestMapping(value = "/providerTatFileUploader", method = RequestMethod.POST)

	public String providerTatFileUploader(HttpServletRequest request, Model model, HttpServletResponse response,

			@RequestPart MultipartFile file) throws Throwable {

		LOGGER.info("file" + file.toString());

		String fileName = file.getName();

		String fileNames = file.getOriginalFilename();

		LOGGER.info("fileName" + fileName);

		LOGGER.info("fileNames" + fileNames);

		List<ProviderTatModel> providerTatList = this.readTatFile(file);

		if (!providerTatList.isEmpty()) {

			for (ProviderTatModel pt : providerTatList) {

				LOGGER.info("ProviderTatModel" + pt);

				Provider provider = providerRepository.selectByName(pt.getProviderName());

				if (provider == null) {

					throw new ProfitMandiBusinessException("Uploaded File", "", "Provider is not Available");

				}

				Map<String, Integer> warehouseMap = ProfitMandiConstants.WAREHOUSE_NAME_MAP;

				ProviderTat providerTat = providerTatRepository.selectByProviderId(provider.getId(),

						warehouseMap.get(pt.getWarehouseName()), pt.getPincode());

				LOGGER.info("providerTat" + providerTat);

				if (providerTat == null) {

					LOGGER.info("providerTat" + providerTat);

					providerTat = new ProviderTat();

					providerTat.setDestinationPin(pt.getPincode());

					providerTat.setProviderId(provider.getId());

					providerTat.setWarehouseLocation(warehouseMap.get(pt.getWarehouseName()));

					providerTat.setDeliveryTime(pt.getTat());

					providerTatRepository.persist(providerTat);

				} else {

					LOGGER.info("providerTat else" + providerTat);

					providerTat.setDeliveryTime(pt.getTat());

				}

			}

		}

		model.addAttribute("response", mvcResponseSender.createResponseString(true));

		return "response";

	}

	private List<ProviderTatModel> readTatFile(MultipartFile file) throws Exception {

		CSVParser parser = new CSVParser(new InputStreamReader(file.getInputStream()), CSVFormat.DEFAULT);

		List<CSVRecord> records = parser.getRecords();

		LOGGER.info("records" + records);

		LOGGER.info("parser" + parser);

		if (records.size() < 2) {

			parser.close();

			throw new ProfitMandiBusinessException("Uploaded File", "", "No records Found");

		}

		// Remove header

		records.remove(0);

		List<ProviderTatModel> returnList = new ArrayList<ProviderTatModel>();

		for (CSVRecord record : records) {

			ProviderTatModel sgr = createProviderTat(record);

			returnList.add(sgr);

			LOGGER.info("records" + record.get(1));

		}

		parser.close();

		return returnList;

	}

	private ProviderTatModel createProviderTat(CSVRecord record) {

		ProviderTatModel ptm = new ProviderTatModel();

		ptm.setProviderName(record.get(0));

		ptm.setWarehouseName(record.get(1));

		ptm.setPincode(record.get(2));

		ptm.setTat(Integer.parseInt(record.get(3)));

		return ptm;

	}

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

	public String publicProviderHolidays(HttpServletRequest request, Model model) throws Exception {

		List<PublicHolidays> publicHolidays = publicHolidaysRepository

				.selectAllByDate(LocalDate.of(LocalDate.now().getYear(), 1, 1));

		model.addAttribute("publicHolidays", publicHolidays);

		return "public-provider-holiday";

	}

	@RequestMapping(value = "/createPublicHoliday", method = RequestMethod.POST)

	public String createPublicHoliday(HttpServletRequest request, @RequestParam LocalDate date,

			@RequestParam String occassion, Model model) throws Exception {

		PublicHolidays ph = new PublicHolidays();

		ph.setDate(date);

		ph.setHolidayDate(LocalDate.now().atStartOfDay());

		ph.setOccasion(occassion);

		publicHolidaysRepository.persist(ph);

		model.addAttribute("response", mvcResponseSender.createResponseString(true));

		return "response";

	}

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

	public ResponseEntity<?> downloadUploadShippingTemplate(HttpServletRequest request, @RequestParam int warehouseId)

			throws Exception {

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

		List<Order> orders = orderRepository.selectAllByWarehouseAndStatus(warehouseId, OrderStatus.BILLED);

		if (orders.isEmpty()) {

			throw new ProfitMandiBusinessException("Template File", "", "No order has been billed");

		}

		Set<String> invoiceNumbers = orders.stream().map(x -> x.getInvoiceNumber()).collect(Collectors.toSet());

		for (String invoicdNumber : invoiceNumbers) {

			rows.add(Arrays.asList(invoicdNumber));

		}

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

				.getCSVByteStream(Arrays.asList("Invoice Number", "Provider", "AWB"), rows);

		final HttpHeaders headers = new HttpHeaders();

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

		headers.set("Content-disposition", "inline; filename=upload-shipping-template.csv");

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

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

		final InputStreamResource inputStreamResource = new InputStreamResource(inputStream);

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

	}

	private Map<Integer, String> getWarehouses(Set<CustomRetailer> positionRetailers) {

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

		positionRetailers.stream().forEach(x -> {

			if (x.getWarehouseId() != 0) {

				warehouses.put(x.getWarehouseId(), ProfitMandiConstants.WAREHOUSE_MAP.get(x.getWarehouseId()));

			}

		});

		return warehouses;

	}

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

	public ResponseEntity<?> downloadProviderTatTemplate(HttpServletRequest request) throws Exception {

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

		Map<Integer, String> warehouseMap = ProfitMandiConstants.WAREHOUSE_MAP;

		for (Entry<Integer, String> warehouse : warehouseMap.entrySet()) {

			rows.add(Arrays.asList("Delhivery", warehouse.getValue(), "110018", "-"));

		}

		org.apache.commons.io.output.ByteArrayOutputStream baos = FileUtil.getCSVByteStream(

				Arrays.asList("Provider Name", "Source Warehouse", "Destination Pincode", "TAT in days"), rows);

		final HttpHeaders headers = new HttpHeaders();

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

		headers.set("Content-disposition", "inline; filename=provider-tat-template.csv");

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

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

		final InputStreamResource inputStreamResource = new InputStreamResource(inputStream);

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

	}

}