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

 package com.spice.profitmandi.web.controller;
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Map.Entry;
 import java.util.Set;
 import java.util.stream.Collectors;
 import javax.servlet.http.HttpServletRequest;
-import javax.servlet.http.HttpServletResponse;
 import org.apache.commons.csv.CSVRecord;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.core.io.InputStreamResource;
 import org.springframework.format.annotation.DateTimeFormat;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Controller;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.ui.Model;
 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.mongodb.DBObject;
 import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
+import com.spice.profitmandi.common.model.BrandStockPrice;
 import com.spice.profitmandi.common.model.CustomRetailer;
 import com.spice.profitmandi.common.model.ProfitMandiConstants;
 import com.spice.profitmandi.common.util.FileUtil;
 import com.spice.profitmandi.common.web.util.ResponseSender;
 import com.spice.profitmandi.dao.entity.auth.AuthUser;
 import com.spice.profitmandi.dao.repository.dtr.FofoStoreRepository;
 import com.spice.profitmandi.dao.repository.dtr.FranchiseeActivityRepository;
 import com.spice.profitmandi.dao.repository.dtr.FranchiseeVisitRepository;
 import com.spice.profitmandi.dao.repository.dtr.LeadActivityRepository;
 import com.spice.profitmandi.dao.repository.dtr.LeadRepository;
+import com.spice.profitmandi.dao.repository.dtr.Mongo;
 import com.spice.profitmandi.dao.repository.dtr.PartnerOnBoardingPanelRepository;
 import com.spice.profitmandi.dao.repository.dtr.UserAccountRepository;
 import com.spice.profitmandi.dao.repository.dtr.UserRepository;
 import com.spice.profitmandi.dao.repository.fofo.PartnerDailyInvestmentRepository;
 import com.spice.profitmandi.dao.repository.transaction.OrderRepository;
 import com.spice.profitmandi.dao.repository.transaction.UserWalletRepository;
 import com.spice.profitmandi.service.PartnerCollectionService;
 import com.spice.profitmandi.service.user.RetailerService;
 import com.spice.profitmandi.service.user.StoreTimelineTatService;
 import com.spice.profitmandi.web.req.CreateFranchiseeRequest;
 import com.spice.profitmandi.web.res.Partner;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 	private TicketSubCategoryRepository ticketSubCategoryRepository;
 	@Autowired
 	private PartnerCollectionRemarkRepository partnerCollectionRemarkRepository;
+	@Autowired
+	private Mongo mongoClient;
+	@Autowired
+	private OrderRepository orderRepository;
+	@Autowired
+	private PositionRepository positionRepository;
+	@Autowired
+	private UserWalletRepository userWalletRepository;
+	@Autowired
+	private PartnerCollectionPlanRepository partnerCollectionPlanRepository;
+	@Autowired
+	private PartnerSecondaryPlanRepository partnerSecondaryPlanRepository;
+	@Autowired
+	private PartnerDailyInvestmentRepository partnerDailyInvestmentRepository;
+	@Autowired
+	private AuthUserPartnerMappingRepository authUserPartnerMappingRepository;
 	@RequestMapping(value = "/lead", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_VALUE)
 	@ApiImplicitParams({
 			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })
 	public ResponseEntity<?> LeadUser(HttpServletRequest request,
 			@RequestBody CreateRefferalRequest createRefferalRequest) throws Exception {
 		franchiseeVisit.setInvestment(createFranchiseeRequest.getInvestment());
 		franchiseeVisit.setMtd(createFranchiseeRequest.getMtd());
 		franchiseeVisit.setHygiene(createFranchiseeRequest.getHygiene());
 		franchiseeVisit.setCreatedTimestamp(LocalDateTime.now());
 		franchiseeVisit.setUpdatedTimestamp(LocalDateTime.now());
+		franchiseeVisit.setInformedAboutOnline(createFranchiseeRequest.getOnline());
+		franchiseeVisit.setPendingBilling(createFranchiseeRequest.getPendingBilling());
 		if (createFranchiseeRequest.getAction().equals(FranchiseeActivityStatus.FOLLOWUP)) {
 			franchiseeVisit.setStatus(FranchiseeVisitStatus.OPEN);
 		} else {
 			franchiseeVisit.setStatus(FranchiseeVisitStatus.CLOSE);
+		}
 			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })
 	public ResponseEntity<?> getFranchiseActivity(HttpServletRequest request, @RequestParam(name = "id") int id)
 			throws ProfitMandiBusinessException {
 		List<FranchiseeActivity> franchiseeActivities = franchiseeActivityRepository.selectByFranchiseeVisitId(id);
 		return responseSender.ok(franchiseeActivities);
+	}
 	@RequestMapping(value = "/getFranchiseeInfo", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)
 			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })
 	public ResponseEntity<?> getFranchiseeInfo(HttpServletRequest request, @RequestParam(name = "id") int id)
 			throws ProfitMandiBusinessException {
 		FranchiseeVisit franchiseeVisit = franchiseeVisitRepository.selectById(id);
+		List<DBObject> mobileBrands = mongoClient.getAllBrandsToDisplay(3);
+		List<String> brands = mobileBrands.stream().filter(x -> (boolean) x.get("active"))
+				.map(x -> (String) x.get("name")).collect(Collectors.toList());
+		franchiseeVisit.setBrands(brands);
 		return responseSender.ok(franchiseeVisit);
+	}
 	@RequestMapping(value = "/franchise-visit-update", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_VALUE)
 		FranchiseeVisit franchiseeVisit = franchiseeVisitRepository.selectById(id);
 		FranchiseeActivity franchiseeActivity = new FranchiseeActivity();
 		if (action == FranchiseeActivityStatus.FOLLOWUP) {
-			franchiseeActivity.setResolution(resolution);
-			franchiseeActivity.setFranchiseeVisitd(franchiseeVisit.getId());
-			franchiseeActivity.setAction(action);
 			franchiseeActivity.setSchelduleTimestamp(schelduleTimestamp);
-			franchiseeActivity.setCreatedTimestamp(LocalDateTime.now());
-			franchiseeActivityRepository.persist(franchiseeActivity);
-			franchiseeVisit.setFranchiseActivityId(franchiseeActivity.getId());
 			franchiseeVisit.setStatus(FranchiseeVisitStatus.OPEN);
 			franchiseeVisit.setUpdatedTimestamp(LocalDateTime.now());
 			franchiseeVisitRepository.persist(franchiseeVisit);
 		} else {
-			franchiseeActivity.setResolution(resolution);
-			franchiseeActivity.setFranchiseeVisitd(franchiseeVisit.getId());
-			franchiseeActivity.setAction(action);
 			franchiseeActivity.setSchelduleTimestamp(null);
-			franchiseeActivity.setCreatedTimestamp(LocalDateTime.now());
-			franchiseeActivityRepository.persist(franchiseeActivity);
 			franchiseeVisit.setFranchiseActivityId(franchiseeActivity.getId());
 			franchiseeVisit.setStatus(FranchiseeVisitStatus.CLOSE);
 			franchiseeVisit.setUpdatedTimestamp(LocalDateTime.now());
 			franchiseeVisitRepository.persist(franchiseeVisit);
+		}
+		franchiseeActivity.setResolution(resolution);
+		franchiseeActivity.setFranchiseeVisitd(franchiseeVisit.getId());
+		franchiseeActivity.setAction(action);
+		franchiseeActivity.setCreatedTimestamp(LocalDateTime.now());
+		franchiseeActivityRepository.persist(franchiseeActivity);
+		franchiseeVisit.setFranchiseActivityId(franchiseeActivity.getId());
+		franchiseeVisit.setStatus(FranchiseeVisitStatus.OPEN);
+		franchiseeVisit.setUpdatedTimestamp(LocalDateTime.now());
+		franchiseeVisitRepository.persist(franchiseeVisit);
 		return responseSender.ok(true);
+	}
 	@RequestMapping(value = "/onBoardingTimelineStatus", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)
 		return responseSender.ok(status);
+	}
-	@Autowired
-	private OrderRepository orderRepository;
-	@Autowired
-	private PositionRepository positionRepository;
-	@Autowired
-	private UserWalletRepository userWalletRepository;
-	@Autowired
-	private PartnerCollectionPlanRepository partnerCollectionPlanRepository;
-	@Autowired
-	private PartnerSecondaryPlanRepository partnerSecondaryPlanRepository;
-	@Autowired
-	private PartnerDailyInvestmentRepository partnerDailyInvestmentRepository;
-	@Autowired
-	private AuthUserPartnerMappingRepository authUserPartnerMappingRepository;
 	@RequestMapping(value = "/getPartnerTarget", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)
 	@ApiImplicitParams({
 			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })
 	public ResponseEntity<?> getPartnerTarget(HttpServletRequest request,
 			@RequestParam(name = "gmailId") String gmailId, @RequestParam String dayValue)
+		}
 		return responseSender.ok(true);
+	}
+	@RequestMapping(value = "/downloadFranchiseeVisitTemplate", method = RequestMethod.GET)
+	public ResponseEntity<?> downloadFranchiseeVisitTemplate(HttpServletRequest request) throws Exception {
+		// ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
+		List<List<?>> rows = new ArrayList<>();
+		org.apache.commons.io.output.ByteArrayOutputStream byteArrayOutputStream = FileUtil
+				.getCSVByteStream(Arrays.asList("Partner Id", "Agenda", "Schedule Timestamp"), rows);
+		try {
+			byteArrayOutputStream.close();
+		} catch (IOException e) {
+			// TODO Auto-generated catch block
+			e.printStackTrace();
+		}
+		final HttpHeaders headers = new HttpHeaders();
+		// headers.setContentType(TEXT_CSV_TYPE);
+		// headers.setCacheControl("must-revalidate, post-check=0, pre-check=0");
+		headers.setContentType(MediaType.parseMediaType("application/csv"));
+		headers.set("Content-disposition", "inline; filename=visit-tempplate.csv");
+		headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
+		headers.add("Pragma", "no-cache");
+		headers.add("Expires", "0");
+		headers.setContentLength(byteArrayOutputStream.toByteArray().length);
+		final InputStream inputStream = new ByteArrayInputStream(byteArrayOutputStream.toByteArray());
+		final InputStreamResource inputStreamResource = new InputStreamResource(inputStream);
+		return new ResponseEntity<>(inputStreamResource, headers, HttpStatus.OK);
+	}
+}

Subversion Repositories SmartDukaan

(root)/trunk/profitmandi-web/src/main/java/com/spice/profitmandi/web/controller/LeadController.java – Rev 30461 → 30487