Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import com.mongodb.DBObject;

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

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.web.util.ResponseSender;

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

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

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

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

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

import com.spice.profitmandi.dao.entity.dtr.User;

import com.spice.profitmandi.dao.entity.dtr.UserAccount;

import com.spice.profitmandi.dao.entity.fofo.*;

import com.spice.profitmandi.dao.entity.user.*;

import com.spice.profitmandi.dao.enumuration.auth.CollectionRemark;

import com.spice.profitmandi.dao.enumuration.cs.EscalationType;

import com.spice.profitmandi.dao.enumuration.dtr.*;

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

import com.spice.profitmandi.dao.repository.auth.*;

import com.spice.profitmandi.dao.repository.cs.*;

import com.spice.profitmandi.dao.repository.dtr.*;

import com.spice.profitmandi.dao.repository.fofo.PartnerDailyInvestmentRepository;

import com.spice.profitmandi.dao.repository.fofo.PurchaseRepository;

import com.spice.profitmandi.dao.repository.fofo.RbmRatingRepository;

import com.spice.profitmandi.dao.repository.fofo.SalesRatingRepository;

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

import com.spice.profitmandi.dao.repository.user.AddressRepository;

import com.spice.profitmandi.service.AuthService;

import com.spice.profitmandi.service.NotificationService;

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.req.RbmSalesRatingRequest;

import com.spice.profitmandi.web.res.Partner;

import io.swagger.annotations.ApiImplicitParam;

import io.swagger.annotations.ApiImplicitParams;

import org.apache.commons.csv.CSVRecord;

import org.apache.commons.io.output.ByteArrayOutputStream;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

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

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.mail.javamail.JavaMailSender;

import org.springframework.stereotype.Controller;

import org.springframework.transaction.annotation.Transactional;

import org.springframework.ui.Model;

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

import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;

import java.io.IOException;

import java.time.LocalDate;

import java.time.LocalDateTime;

import java.time.YearMonth;

import java.time.format.DateTimeFormatter;

import java.util.*;

import java.util.HashSet;

import java.util.Map.Entry;

import java.util.stream.Collectors;

@Controller

@Transactional(rollbackFor = Throwable.class)

public class LeadController {

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

    @Autowired

    private ResponseSender<?> responseSender;

    @Autowired

    private AuthRepository authRepository;

    @Autowired

    private LeadRepository leadRepository;

    @Autowired

    DocumentRepository documentRepository;

    @Autowired

    PositionRepository positionRepository;

    @Autowired

    private ActivityAttachmentRepository activityAttachmentRepository;

    @Autowired

    private CsService csService;

    @Autowired

    private UserRepository userRepository;

    @Autowired

    private AddressRepository addressRepository;

    @Autowired

    private UserRoleRepository userRoleRepository;

    @Autowired

    private UserAccountRepository userAccountRepository;

    @Autowired

    private com.spice.profitmandi.dao.repository.user.UserRepository userUserRepository;

    @Autowired

    private RetailerService retailerService;

    @Autowired

    private LeadActivityRepository leadActivityRepository;

    @Autowired

    private FranchiseeVisitRepository franchiseeVisitRepository;

    @Autowired

    private FranchiseeActivityRepository franchiseeActivityRepository;

    @Autowired

    private PartnerOnBoardingPanelRepository partnerOnBoardingPanelRepository;

    @Autowired

    private FofoStoreRepository fofoStoreRepository;

    @Autowired

    private StoreTimelineTatService storeTimelineTatService;

    @Autowired

    private PartnerCollectionService partnerCollectionService;

    @Autowired

    private TicketRepository ticketRepository;

    @Autowired

    private TicketCategoryRepository ticketCategoryRepository;

    @Autowired

    private TicketSubCategoryRepository ticketSubCategoryRepository;

    @Autowired

    private PartnerCollectionRemarkRepository partnerCollectionRemarkRepository;

    @Autowired

    private Mongo mongoClient;

    @Autowired

    private OrderRepository orderRepository;

    @Autowired

    private PartnerCollectionPlanRepository partnerCollectionPlanRepository;

    @Autowired

    private PartnerSecondaryPlanRepository partnerSecondaryPlanRepository;

    @Autowired

    private PartnerDailyInvestmentRepository partnerDailyInvestmentRepository;

    @Autowired

    private AuthUserPartnerMappingRepository authUserPartnerMappingRepository;

    @Autowired

    private VisitRequestRepository visitRequestRepository;

    @Autowired

    private AuthService authService;

    @Autowired

    private NotificationService notificationService;

    @Autowired

    JavaMailSender mailSender;

    @Autowired

    private TicketAssignedRepository ticketAssignedRepository;

    @Autowired

    LeadDetailRepository leadDetailRepository;

    @Autowired

    PurchaseRepository purchaseRepository;

    @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(@RequestBody CreateRefferalRequest createRefferalRequest) throws Exception {

        LOGGER.info("CreateReferralRequest - " + createRefferalRequest);

        if (createRefferalRequest.getOutletName() == null || createRefferalRequest.getPotential() == 0) {

            throw new Exception("Outlet should not be empty and potential should be greater > 0");

        }

        Lead lead;

        AuthUser authUser = authRepository.selectByGmailId(createRefferalRequest.getReffereeEmail());

        if (createRefferalRequest.getId() > 0) {

            lead = leadRepository.selectById(createRefferalRequest.getId());

            lead.setAssignTo(authUser.getId());

        } else {

            lead = new Lead();

            lead.setFirstName(createRefferalRequest.getFirstName());

            lead.setLastName(createRefferalRequest.getLastName());

            lead.setLeadMobile(createRefferalRequest.getMobile());

            lead.setState(createRefferalRequest.getState());

            lead.setCity(createRefferalRequest.getCity());

            lead.setAddress(createRefferalRequest.getAddress());

            lead.setCreatedTimestamp(LocalDateTime.now());

            lead.setOutLetName(createRefferalRequest.getOutletName());

            lead.setUpdatedTimestamp(LocalDateTime.now());

            lead.setStatus(createRefferalRequest.getStatus());

            lead.setSource(createRefferalRequest.getSource());

            lead.setNotinterestedReason(createRefferalRequest.getReason());

            lead.setPotential(createRefferalRequest.getPotential());

            if (createRefferalRequest.getColorCheck() == true) {

                lead.setColor("Green");

            } else {

                lead.setColor("Yellow");

            }

            String authUserName = authUser.getFullName();

            lead.setAuthId(authUser.getId());

            lead.setAssignTo(authUser.getId());

            lead.setCreatedBy(authUserName);

            leadRepository.persist(lead);

        }

        LeadActivity leadActivity = new LeadActivity();

        leadActivity.setLeadId(lead.getId());

        leadActivity.setRemark((createRefferalRequest.getId() > 0 ? "Self Assigned." : "") + createRefferalRequest.getRemark());

        leadActivity.setAuthId(authUser.getId());

        if (createRefferalRequest.getStatus().equals(LeadStatus.followUp)) {

            leadActivity.setSchelduleTimestamp(createRefferalRequest.getSchelduleTimestamp());

            leadActivity.setCommunicationType(createRefferalRequest.getCommunicationType());

            if (leadActivity.getCommunicationType().equals(CommunicationType.VISIT)) {

                visitRequestRepository.createVisitRequest(lead.getId(), "lead", lead.getAssignTo(),

                        createRefferalRequest.getSchelduleTimestamp());

            }

            //

            leadActivity.setClosureTimestamp(createRefferalRequest.getClosureTimestamp());

            lead.setClosureTimestamp(createRefferalRequest.getClosureTimestamp());

        } else {

            leadActivity.setSchelduleTimestamp(null);

            leadActivity.setClosureTimestamp(null);

            lead.setClosureTimestamp(null);

        }

        leadActivity.setCreatedTimestamp(LocalDateTime.now());

        leadActivityRepository.persist(leadActivity);

        return responseSender.ok(true);

    }

    @RequestMapping(value = "/lead-description", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({

            @ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> leadDescription(HttpServletRequest request, @RequestParam(name = "gmailId") String

                                                     gmailId,

                                             @RequestParam(name = "status") LeadStatus status) throws ProfitMandiBusinessException {

        AuthUser authUser = authRepository.selectByGmailId(gmailId);

        if (authUser == null) {

            throw new ProfitMandiBusinessException("gmailId", gmailId, "Invalid gmailId");

        }

        List<Lead> leads = null;

        LOGGER.info("emails" + status);

        if (status.equals(LeadStatus.followUp)) {

            leads = leadRepository.selectLeadsScheduledBetweenDate(Arrays.asList(authUser.getId()), null, null);

            // Only fetch visitRequests when needed for followUp status

            List<VisitRequest> visitRequests = visitRequestRepository.selectByAuthIdAndDate(authUser.getId(),

                    LocalDate.now());

            if (!visitRequests.isEmpty()) {

                Map<Integer, List<VisitRequest>> visitRequestMap = visitRequests.stream()

                        .collect(Collectors.groupingBy(VisitRequest::getVisitId));

                Comparator<VisitRequest> visitComparator = Comparator.comparing(VisitRequest::getCreatedTimestamp);

                for (Lead lead : leads) {

                    List<VisitRequest> visitRequest = visitRequestMap.get(lead.getId());

                    if (visitRequest != null) {

                        if (visitRequest.size() > 1) {

                            VisitRequest youngestVisit = visitRequest.stream().max(visitComparator).get();

                            lead.setVisitStatus(youngestVisit.getStatus());

                        } else {

                            lead.setVisitStatus(visitRequest.get(0).getStatus());

                        }

                    }

                }

            }

            leads = leads.stream()

                    .sorted(Comparator.comparing(Lead::getScheduledTimestamp, Comparator.nullsFirst(Comparator.reverseOrder())))

                    .collect(Collectors.toList());

        } else {

            leads = leadRepository.selectByAssignAuthIdAndStatus(authUser.getId(), status);

        }

        return responseSender.ok(leads);

    }

    @RequestMapping(value = "/getlead", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> getLead(HttpServletRequest request, @RequestParam(name = "id") int id) throws

            ProfitMandiBusinessException {

        List<LeadActivity> leadActivities = leadActivityRepository.selectBYLeadId(id);

        Lead lead = leadRepository.selectById(id);

        lead.setLeadActivities(leadActivities);

        return responseSender.ok(lead);

    }

    @RequestMapping(value = "/check-existing-lead", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> getLeadByMobile(HttpServletRequest request, @RequestParam(name = "mobile") String

            mobile) throws ProfitMandiBusinessException {

        int userId = (int) request.getAttribute("userId");

        LOGGER.info("userId: " + userId);

        Lead lead = leadRepository.selectByMobileNumber(mobile);

        return responseSender.ok(lead);

    }

    @RequestMapping(value = "/leadUpdate", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> leadUpdate(HttpServletRequest request, @RequestParam(name = "id") int id,

                                        @RequestParam(name = "status") LeadStatus status, @RequestParam(name = "colorCheck") Boolean colorCheck,

                                        @RequestParam(name = "remark") String remark, @RequestParam(name = "reason") String reason,

                                        @RequestParam(name = "schelduleTimestamp") @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime schelduleTimestamp,

                                        @RequestParam(name = "closureTimestamp") @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime closureTimestamp,

                                        @RequestParam(name = "communicationType") CommunicationType communicationType) throws

            ProfitMandiBusinessException {

        Lead lead = leadRepository.selectById(id);

        LeadActivity leadActivity = new LeadActivity();

        lead.setStatus(status);

        lead.setNotinterestedReason(reason);

        leadActivity.setRemark(remark);

        leadActivity.setLeadId(id);

        leadActivity.setOutletName(lead.getOutLetName());

        leadActivity.setPotential(lead.getPotential());

        leadActivity.setCreatedTimestamp(LocalDateTime.now());

        leadActivity.setSchelduleTimestamp(null);

        leadActivity.setClosureTimestamp(null);

        leadActivity.setAuthId(lead.getAssignTo());

        lead.setUpdatedTimestamp(LocalDateTime.now());

        if (colorCheck == true) {

            lead.setColor("Green");

        } else {

            lead.setColor("Yellow");

        }

        if (status == LeadStatus.followUp) {

            if (schelduleTimestamp != null) {

                leadActivity.setCommunicationType(communicationType);

                if (leadActivity.getCommunicationType().equals(CommunicationType.VISIT)) {

                    visitRequestRepository.createVisitRequest(lead.getId(), "lead", lead.getAssignTo(), schelduleTimestamp);

                }

            }

            leadActivity.setSchelduleTimestamp(schelduleTimestamp);

            leadActivity.setClosureTimestamp(closureTimestamp);

            lead.setClosureTimestamp(closureTimestamp);

        } else {

            leadActivity.setSchelduleTimestamp(null);

            leadActivity.setClosureTimestamp(null);

            lead.setClosureTimestamp(null);

        }

        leadActivityRepository.persist(leadActivity);

        return responseSender.ok(true);

    }

    @RequestMapping(value = ProfitMandiConstants.URL_NEW_LEAD, method = RequestMethod.POST)

    public ResponseEntity<?> newLead(HttpServletRequest request, @RequestBody CreateRefferalRequest

            createRefferalRequest) throws ProfitMandiBusinessException {

        LOGGER.info("requested url : " + request.getRequestURL().toString());

        Lead lead = new Lead();

        lead.setAddress(createRefferalRequest.getCity());

        Long.parseLong(createRefferalRequest.getMobile());

        if (createRefferalRequest.getMobile().length() != 10) {

            throw new ProfitMandiBusinessException("Mobile Number", createRefferalRequest.getMobile(), "Number should be of 10 digits");

        }

        lead.setLeadMobile(createRefferalRequest.getMobile());

        lead.setCity(createRefferalRequest.getCity());

        lead.setState(createRefferalRequest.getState());

        lead.setLastName(createRefferalRequest.getLastName());

        if (lead.getState().equals("Uttar Pradesh")) {

            lead.setAssignTo(53);

        } else if (lead.getState().equals("Haryana")) {

            lead.setAssignTo(53);

        } else if (lead.getState().equals("Delhi")) {

            lead.setAssignTo(53);

        } else {

            // Assign to sm

            lead.setAssignTo(53);

            // Assign to neha

            // lead.setAssignTo(1);

        }

        lead.setAuthId(lead.getAssignTo());

        lead.setCreatedBy("daily-sync");

        lead.setSource("SD-WEB");

        lead.setFirstName(createRefferalRequest.getFirstName());

        lead.setStatus(LeadStatus.followUp);

        lead.setColor("yellow");

        lead.setCreatedTimestamp(LocalDateTime.now());

        lead.setUpdatedTimestamp(LocalDateTime.now());

        leadRepository.persist(lead);

        return responseSender.ok(true);

    }

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

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> getPartners(HttpServletRequest request, @RequestParam(name = "gmailId") String gmailId)

            throws ProfitMandiBusinessException {

        AuthUser authUser = authRepository.selectByGmailId(gmailId);

        Map<String, Set<String>> storeGuyMap = csService.getAuthUserPartnerEmailMapping();

        Set<String> emails = storeGuyMap.get(authUser.getEmailId());

        LOGGER.info("emails" + emails);

        List<User> users = userRepository.selectAllByEmailIds(new ArrayList<>(emails));

        List<Partner> partners = new ArrayList<>();

        for (User user : users) {

            UserAccount uc = userAccountRepository.selectSaholicByUserId(user.getId());

            com.spice.profitmandi.dao.entity.user.User userInfo = userUserRepository.selectById(uc.getAccountKey());

            CustomRetailer customRetailer = retailerService.getFofoRetailer(userInfo.getId());

            Partner partner = new Partner();

            partner.setBusinessName(customRetailer.getBusinessName());

            partner.setPartnerId(customRetailer.getPartnerId());

            partner.setCartId(customRetailer.getCartId());

            partner.setEmail(customRetailer.getEmail());

            partner.setGstNumber(customRetailer.getGstNumber());

            partner.setDisplayName(customRetailer.getDisplayName());

            partner.setCity(customRetailer.getAddress().getCity());

            partner.setUserId(user.getId());

            partners.add(partner);

        }

        LOGGER.info("partners" + partners);

        return responseSender.ok(partners);

    }

    @RequestMapping(value = "/franchise-first-visit", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> FranchiseFirstVisit(HttpServletRequest request, @RequestBody CreateFranchiseeRequest

            createFranchiseeRequest) throws Exception {

        FranchiseeVisit franchiseeVisit = new FranchiseeVisit();

        franchiseeVisit.setFofoId(createFranchiseeRequest.getFofoId());

        CustomRetailer customRetailer = retailerService.getFofoRetailer(createFranchiseeRequest.getFofoId());

        franchiseeVisit.setPartnerName(customRetailer.getBusinessName());

        franchiseeVisit.setAgenda(createFranchiseeRequest.getAgenda());

        franchiseeVisit.setCreatedTimestamp(LocalDateTime.now());

        franchiseeVisit.setUpdatedTimestamp(LocalDateTime.now());

        franchiseeVisit.setStatus(FranchiseeVisitStatus.OPEN);

        franchiseeVisit.setScheduleTimestamp(createFranchiseeRequest.getFirstSchelduleTimestamp());

        // change

        AuthUser authUser = authRepository.selectByGmailId(createFranchiseeRequest.getCreatedBy());

        String authUserName = authUser.getFirstName() + " " + authUser.getLastName();

        franchiseeVisit.setCreatedBy(authUserName);

        franchiseeVisit.setAuthId(authUser.getId());

        franchiseeVisitRepository.persist(franchiseeVisit);

        FranchiseeActivity franchiseeActivity = new FranchiseeActivity();

        franchiseeActivity.setAction(FranchiseeActivityStatus.FOLLOWUP);

        franchiseeActivity.setFranchiseeVisitd(franchiseeVisit.getId());

        franchiseeActivity.setResolution(createFranchiseeRequest.getResolution());

        franchiseeActivity.setSchelduleTimestamp(createFranchiseeRequest.getFirstSchelduleTimestamp());

        franchiseeActivity.setCreatedTimestamp(LocalDateTime.now());

        franchiseeActivityRepository.persist(franchiseeActivity);

        franchiseeVisit.setFranchiseActivityId(franchiseeActivity.getId());

        visitRequestRepository.createVisitRequest(franchiseeVisit.getId(), "franchiseeVisit", authUser.getId(), createFranchiseeRequest.getFirstSchelduleTimestamp());

        return responseSender.ok(true);

    }

    @RequestMapping(value = "/franchise-visit", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> FranchiseVisit(HttpServletRequest request,

                                            @RequestBody CreateFranchiseeRequest createFranchiseeRequest) throws Exception {

        int rbmL1 = csService.getAuthUserId(ProfitMandiConstants.TICKET_CATEGORY_RBM, EscalationType.L1, createFranchiseeRequest.getFofoId());

        FranchiseeVisit franchiseeVisit = franchiseeVisitRepository.selectById(createFranchiseeRequest.getId());

        franchiseeVisit.setFofoId(createFranchiseeRequest.getFofoId());

        CustomRetailer customRetailer = retailerService.getFofoRetailer(createFranchiseeRequest.getFofoId());

        LOGGER.info("rbmL1 {}", rbmL1);

        franchiseeVisit.setPartnerName(customRetailer.getBusinessName());

        franchiseeVisit.setAgenda(createFranchiseeRequest.getAgenda());

        franchiseeVisit.setPartnerRemark(createFranchiseeRequest.getPartnerRemark());

        franchiseeVisit.setOutsideVisibity(createFranchiseeRequest.getOutsideVisibity());

        franchiseeVisit.setInstoreVisibility(createFranchiseeRequest.getInstoreVisibility());

        franchiseeVisit.setOutsideStock(createFranchiseeRequest.getOutsideStock());

        franchiseeVisit.setSystemKnowledge(createFranchiseeRequest.getSystemKnowledge());

        franchiseeVisit.setWorkingDevice(createFranchiseeRequest.getWorkingDevice());

        franchiseeVisit.setWorkingPrinter(createFranchiseeRequest.getWorkingPrinter());

        franchiseeVisit.setCarryBags(createFranchiseeRequest.getCarryBags());

        franchiseeVisit.setSmartdukaanTshirt(createFranchiseeRequest.getSmartdukaanTshirt());

        franchiseeVisit.setLatestDummies(createFranchiseeRequest.getLatestDummies());

        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());

        franchiseeVisit.setRbmId(rbmL1);

        franchiseeVisit.setRbmRating(createFranchiseeRequest.getRbmRating());

        if (createFranchiseeRequest.getAction().equals(FranchiseeActivityStatus.FOLLOWUP)) {

            franchiseeVisit.setStatus(FranchiseeVisitStatus.OPEN);

        } else {

            franchiseeVisit.setStatus(FranchiseeVisitStatus.CLOSE);

        }

        // AuthUser authUser =

        // authRepository.selectByGmailId(createFranchiseeRequest.getCreatedBy());

        // change

        AuthUser authUser = authRepository.selectByGmailId(createFranchiseeRequest.getCreatedBy());

        String authUserName = authUser.getFirstName() + " " + authUser.getLastName();

        franchiseeVisit.setCreatedBy(authUserName);

        franchiseeVisit.setAuthId(authUser.getId());

        franchiseeVisitRepository.persist(franchiseeVisit);

        FranchiseeActivity franchiseeActivity = new FranchiseeActivity();

        franchiseeActivity.setAction(createFranchiseeRequest.getAction());

        franchiseeActivity.setFranchiseeVisitd(franchiseeVisit.getId());

        franchiseeActivity.setResolution(createFranchiseeRequest.getResolution());

        if (createFranchiseeRequest.getAction().equals(FranchiseeActivityStatus.FOLLOWUP)) {

            franchiseeActivity.setSchelduleTimestamp(createFranchiseeRequest.getSchelduleTimestamp());

            visitRequestRepository.createVisitRequest(franchiseeVisit.getId(), "franchiseeVisit",

                    franchiseeVisit.getAuthId(), createFranchiseeRequest.getSchelduleTimestamp());

        } else {

            franchiseeActivity.setSchelduleTimestamp(null);

        }

        franchiseeActivity.setCreatedTimestamp(LocalDateTime.now());

        franchiseeActivityRepository.persist(franchiseeActivity);

        franchiseeVisit.setFranchiseActivityId(franchiseeActivity.getId());

        if (!createFranchiseeRequest.getOutsideVisibityReason().isEmpty()) {

            csService.createTicket(createFranchiseeRequest.getFofoId(), ProfitMandiConstants.TICKET_CATEGORY_DESIGN, ProfitMandiConstants.TICKET_CATEGORY_DESIGNING_BRANDING_EXTERNAL_VISIBILITY, createFranchiseeRequest.getOutsideVisibityReason(), authUser.getId());

        }

        if (!createFranchiseeRequest.getInstoreVisibilityReason().isEmpty()) {

            csService.createTicket(createFranchiseeRequest.getFofoId(), ProfitMandiConstants.TICKET_CATEGORY_DESIGN, ProfitMandiConstants.TICKET_CATEGORY_DESIGNING_BRANDING_INTERNAL_VISIBILITY, createFranchiseeRequest.getInstoreVisibilityReason(), authUser.getId());

        }

        if (!createFranchiseeRequest.getOutsideStockReason().isEmpty()) {

            csService.createTicket(createFranchiseeRequest.getFofoId(), ProfitMandiConstants.TICKET_CATEGORY_SALES, ProfitMandiConstants.TICKET_CATEGORY_SALES_OUTSIDE_STOCK, createFranchiseeRequest.getOutsideStockReason(), authUser.getId());

        }

        if (!createFranchiseeRequest.getSystemKnowledgeReason().isEmpty()) {

            csService.createTicket(createFranchiseeRequest.getFofoId(), ProfitMandiConstants.TICKET_CATEGORY_TRAINING, ProfitMandiConstants.TICKET_CATEGORY_TRAINING_NEED, createFranchiseeRequest.getSystemKnowledgeReason(), authUser.getId());

        }

        if (!createFranchiseeRequest.getLatestDummiesReason().isEmpty()) {

            csService.createTicket(createFranchiseeRequest.getFofoId(), ProfitMandiConstants.TICKET_CATEGORY_MARKETING, ProfitMandiConstants.TICKET_CATEGORY_MARKETING_DUMMIES, createFranchiseeRequest.getLatestDummiesReason(), authUser.getId());

        }

        if (!createFranchiseeRequest.getInvestmentReason().isEmpty()) {

            csService.createTicket(createFranchiseeRequest.getFofoId(), ProfitMandiConstants.TICKET_CATEGORY_SALES, ProfitMandiConstants.TICKET_CATEGORY_SALES_LOW_INVESTMENT, createFranchiseeRequest.getInvestmentReason(), authUser.getId());

        }

        if (!createFranchiseeRequest.getHygieneReason().isEmpty()) {

            csService.createTicket(createFranchiseeRequest.getFofoId(), ProfitMandiConstants.TICKET_CATEGORY_SALES, ProfitMandiConstants.TICKET_CATEGORY_SALES_LOW_DATA_ACCURACY, createFranchiseeRequest.getHygieneReason(), authUser.getId());

        }

        return responseSender.ok(true);

    }

    @RequestMapping(value = "/getFranchiseVisit", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> getFranchiseVisit(HttpServletRequest request,

                                               @RequestParam(name = "gmailId") String gmailId, @RequestParam(name = "status") FranchiseeVisitStatus

                                                       status,

                                               @RequestParam(name = "offset", defaultValue = "0") int offset,

                                               @RequestParam(name = "limit", defaultValue = "10") int limit) throws ProfitMandiBusinessException {

        AuthUser authUser = authRepository.selectByGmailId(gmailId);

        List<VisitRequest> visitRequests = visitRequestRepository.selectByAuthIdAndDate(authUser.getId(), LocalDate.now());

        Map<Integer, List<VisitRequest>> visitRequestMap = null;

        if (!visitRequests.isEmpty()) {

            visitRequestMap = visitRequests.stream().collect(Collectors.groupingBy(x -> x.getVisitId()));

        }

        List<FranchiseeVisit> franchiseeVisits = franchiseeVisitRepository.selectByAuthIdAndStatus(authUser.getId(), status, offset, limit);

        // Batch fetch all FranchiseeActivity in one query instead of N+1

        List<Integer> activityIds = franchiseeVisits.stream()

                .filter(fv -> fv.getFranchiseActivityId() != 0)

                .map(FranchiseeVisit::getFranchiseActivityId)

                .collect(Collectors.toList());

        Map<Integer, FranchiseeActivity> activityMap = new HashMap<>();

        if (!activityIds.isEmpty()) {

            activityMap = franchiseeActivityRepository.selectByIds(activityIds).stream()

                    .collect(Collectors.toMap(FranchiseeActivity::getId, x -> x));

        }

        for (FranchiseeVisit fv : franchiseeVisits) {

            if (visitRequestMap != null) {

                List<VisitRequest> visitRequest = visitRequestMap.get(fv.getId());

                if (visitRequest != null) {

                    if (visitRequest.size() > 1) {

                        Comparator<VisitRequest> visitComparato = Comparator.comparing(VisitRequest::getCreatedTimestamp);

                        VisitRequest youngestVisit = visitRequest.stream().max(visitComparato).get();

                        fv.setVisitStatus(youngestVisit.getStatus());

                    } else {

                        fv.setVisitStatus(visitRequest.get(0).getStatus());

                    }

                }

            }

            if (fv.getFranchiseActivityId() != 0) {

                FranchiseeActivity fA = activityMap.get(fv.getFranchiseActivityId());

                fv.setFranchiseeActivity(fA);

            }

        }

        return responseSender.ok(franchiseeVisits);

    }

    @RequestMapping(value = "/getFranchiseActivity", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@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)

    @ApiImplicitParams({@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)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> franchiseVisitUpdate(HttpServletRequest request, @RequestParam(name = "id") int id,

                                                  @RequestParam(name = "action") FranchiseeActivityStatus action,

                                                  @RequestParam(name = "resolution") String resolution,

                                                  @RequestParam(name = "schelduleTimestamp") @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime schelduleTimestamp)

            throws ProfitMandiBusinessException {

        FranchiseeVisit franchiseeVisit = franchiseeVisitRepository.selectById(id);

        FranchiseeActivity franchiseeActivity = new FranchiseeActivity();

        if (action == FranchiseeActivityStatus.FOLLOWUP) {

            if (schelduleTimestamp != null) {

                visitRequestRepository.createVisitRequest(franchiseeVisit.getId(), "franchiseeVisit",

                        franchiseeVisit.getAuthId(), schelduleTimestamp);

            }

            franchiseeActivity.setSchelduleTimestamp(schelduleTimestamp);

            franchiseeVisit.setStatus(FranchiseeVisitStatus.OPEN);

            franchiseeVisit.setUpdatedTimestamp(LocalDateTime.now());

            franchiseeVisitRepository.persist(franchiseeVisit);

        } else {

            franchiseeActivity.setSchelduleTimestamp(null);

            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)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> onBoardingTimelineStatus(HttpServletRequest request, Model model)

            throws ProfitMandiBusinessException {

        //This userId is of dtr.users

        int userId = (int) request.getAttribute("userId");

        //UserCart contains userId/retailerId in context of user.user as

        //uc contains retailerId and cartId of the retailerId that belongs to user schema

        UserCart uc = userAccountRepository.getUserCart(userId);

        FofoStore fs = fofoStoreRepository.selectByRetailerId(uc.getUserId());

        PartnerOnBoardingPanel partnerOnBoardingPanel = partnerOnBoardingPanelRepository.selectByCode(fs.getCode());

        Map<StoreTimeline, OnBoardingTimelineModel> timelineStatus = null;

        LOGGER.info("partnerOnBoardingPanel" + partnerOnBoardingPanel);

        if (partnerOnBoardingPanel != null) {

            timelineStatus = storeTimelineTatService.getTimeline(partnerOnBoardingPanel.getId());

        }

        List<OnBoardingTimelineModel> onBoardingModel = new ArrayList<>();

        if (timelineStatus != null) {

            onBoardingModel = timelineStatus.values().stream().skip(2).map(x -> x).collect(Collectors.toList());

        }

        return responseSender.ok(onBoardingModel);

    }

    @RequestMapping(value = "/purchaseExist", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({

            @ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> purchaseExist(HttpServletRequest request) throws ProfitMandiBusinessException {

        int userId = (int) request.getAttribute("userId");

        UserCart uc = userAccountRepository.getUserCart(userId);

        return responseSender.ok(purchaseRepository.hasPurchaseForFofoId(uc.getUserId()));

    }

    @RequestMapping(value = "/onBoardingTimelineVisibility", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ApiImplicitParams({@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header")})

    public ResponseEntity<?> onBoardingTimelineVisibility(HttpServletRequest request, Model model)

            throws ProfitMandiBusinessException {

        int userId = (int) request.getAttribute("userId");

        UserCart uc = userAccountRepository.getUserCart(userId);

        FofoStore fs = fofoStoreRepository.selectByRetailerId(uc.getUserId());

        PartnerOnBoardingPanel partnerOnBoardingPanel = partnerOnBoardingPanelRepository.selectByCode(fs.getCode());

        boolean status = true;

        if (partnerOnBoardingPanel != null) {

            status = storeTimelineTatService.getTimelineCompleted(partnerOnBoardingPanel.getId());

        }

        LOGGER.info("status" + status);

        return responseSender.ok(status);

    }

    @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 String

            gmailId, @RequestParam String dayValue, @RequestParam ActivationType activationType) throws

            ProfitMandiBusinessException {

        AuthUser authUser = authRepository.selectByGmailId(gmailId);

        Map<String, Set<Integer>> storeGuyMap = csService.getAuthUserPartnerIdMapping();

        Set<Integer> fofoIds = storeGuyMap.get(authUser.getEmailId());

        List<String> brands = Arrays.asList("Vivo", "Samsung", "Oppo", "Itel", "Almost New", "Others");

        float totalPartnerTargetSecondary = 0;

        float totalPartnerTargetCollection = 0;

        float totalPartnerAchievementSecondary = 0;

        float totalPartnerAchievementCollection = 0;

        TargetModel tm = new TargetModel();

        Map<Integer, PartnerDailyInvestment> partnerDailyInvestmentMap = new HashMap<>();

        Map<Integer, Long> partnerTicketCount = ticketRepository.selectAllOpenTicketsGroupByRetailer();

        LOGGER.info("partnerTicketCount {}", partnerTicketCount);

        List<PartnerTargetAchievementModel> ptams = new ArrayList<>();

        if (fofoIds != null && fofoIds.size() > 0) {

            LOGGER.info("fofoIds {}", fofoIds);

            List<Integer> fofoIdList = fofoStoreRepository.selectByRetailerIds(new ArrayList<>(fofoIds)).stream()

                    .filter(x -> (!x.isInternal() && (activationType == null || x.getActivationType().equals(activationType))))

                    .map(x -> x.getId()).collect(Collectors.toList());

            LOGGER.info("fofoIdList {}", fofoIdList);

            LocalDateTime startDate = LocalDate.now().atStartOfDay();

            if (dayValue.equals("previous")) {

                startDate = LocalDate.now().minusDays(1).atStartOfDay();

            }

            if (fofoIdList.size() > 0) {

                List<PartnerDailyInvestment> partnerDailyInvestments = partnerDailyInvestmentRepository.selectAll(fofoIdList, startDate.toLocalDate().minusDays(1));

                if (!partnerDailyInvestments.isEmpty()) {

                    partnerDailyInvestmentMap = partnerDailyInvestments.stream().collect(Collectors.toMap(x -> x.getFofoId(), x -> x));

                }

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

                Map<Integer, CustomRetailer> customRetailers = fofoIdList.stream().map(x -> customRetailerMap.get(x))

                        .filter(x -> x != null).collect(Collectors.toList()).stream()

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

                List<Integer> remarkIds = partnerCollectionRemarkRepository.selectMaxRemarkId(fofoIdList);

                LOGGER.info("remarkIds {}", remarkIds);

                long todayCollectionCount = 0;

                if (!remarkIds.isEmpty()) {

                    todayCollectionCount = partnerCollectionRemarkRepository

                            .selectByAuthIdAndIds(authUser.getId(), remarkIds).stream()

                            .filter(x -> x.getCreateTimestamp().toLocalDate().equals(LocalDate.now()))

                            .collect(Collectors.counting());

                }

                Map<Integer, PartnerCollectionPlanModel> collectionMap = partnerCollectionService.getCollectionMap(fofoIdList, startDate);

                Map<Integer, List<PartnerSecondaryPlanModel>> partnerSecondaryPlans = orderRepository

                        .selectPartnerSecondaryGroupByBrand(fofoIdList, startDate.toLocalDate()).stream()

                        .collect(Collectors.groupingBy(x -> x.getFofoId()));

                // Pre-fetch all AuthUsers that might be needed (instead of N+1 queries)

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

                collectionMap.values().stream()

                        .map(PartnerCollectionPlanModel::getAuthId)

                        .filter(id -> id != null && id != authUser.getId())

                        .forEach(allAuthIds::add);

                partnerSecondaryPlans.values().stream()

                        .flatMap(List::stream)

                        .map(PartnerSecondaryPlanModel::getAuthId)

                        .filter(id -> id != null && id != authUser.getId())

                        .forEach(allAuthIds::add);

                Map<Integer, AuthUser> authUserMap = new HashMap<>();

                if (!allAuthIds.isEmpty()) {

                    authUserMap = authRepository.selectByIds(new ArrayList<>(allAuthIds)).stream()

                            .collect(Collectors.toMap(AuthUser::getId, x -> x));

                }

                LOGGER.info("partnerSecondayPlans {}", partnerSecondaryPlans);

                for (Entry<Integer, CustomRetailer> customRetailerEntry : customRetailers.entrySet()) {

                    int fofoId = customRetailerEntry.getKey();

                    CustomRetailer customRetailer = customRetailerEntry.getValue();

                    float totalSecondaryPlan = 0;

                    float totalSecondaryAchivement = 0;

                    PartnerTargetAchievementModel ptam = new PartnerTargetAchievementModel();

                    ptam.setFofoId(fofoId);

                    ptam.setBusinessName(customRetailer.getBusinessName());

                    ptam.setMobile(customRetailer.getMobileNumber());

                    if (partnerDailyInvestmentMap.get(fofoId) != null) {

                        ptam.setWalletAmount(partnerDailyInvestmentMap.get(fofoId).getWalletAmount());

                        ptam.setShortInvestment(partnerDailyInvestmentMap.get(fofoId).getShortInvestment());

                    }

                    if (collectionMap.get(fofoId) != null) {

                        PartnerCollectionPlanModel collectionPlan = collectionMap.get(fofoId);

                        ptam.setRemark(collectionPlan.getRemark());

                        ptam.setMessage(collectionPlan.getMessage());

                        ptam.setRemarkTimestamp(collectionPlan.getRemarkTimestamp());

                        ptam.setRank(collectionPlan.getRank());

                        Integer authId = collectionPlan.getAuthId();

                        Map<Integer, String> rankColorMap = ProfitMandiConstants.Rank_Color_Map;

                        ptam.setCollectionColor(rankColorMap.get(collectionPlan.getRank()));

                        LOGGER.info("authId" + authId);

                        if (collectionPlan.getTargetPlan() != null && collectionPlan.getCommittedDate() != null) {

                            float targetCollection = 0;

                            if (collectionPlan.getRank() == 2 && collectionPlan.getAchievementPlan() != null) {

                                targetCollection = collectionPlan.getTargetPlan() - collectionPlan.getAchievementPlan();

                            } else {

                                targetCollection = collectionPlan.getTargetPlan();

                            }

                            if (authId != null && authId == authUser.getId()) {

                                if (!collectionPlan.getCommittedDate().isAfter(startDate)) {

                                    totalPartnerTargetCollection += targetCollection;

                                }

                            }

                            ptam.setCollectionTarget(targetCollection);

                            if (collectionPlan.getRank() == 2 && collectionPlan.getAchievementPlan() != null

                                    && collectionPlan.getWalletTimestamp() != null

                                    && collectionPlan.getWalletTimestamp().toLocalDate().equals(startDate.toLocalDate())) {

                                if (authId != null && authId == authUser.getId()) {

                                    totalPartnerAchievementCollection += collectionPlan.getAchievementPlan();

                                }

                                ptam.setCollectionAchievement(collectionPlan.getAchievementPlan());

                            } else if (collectionPlan.getCommittedDate().toLocalDate().isEqual(startDate.toLocalDate())

                                    && collectionPlan.getAchievementPlan() != null) {

                                if (authId != null && authId == authUser.getId()) {

                                    totalPartnerAchievementCollection += collectionPlan.getAchievementPlan();

                                }

                                ptam.setCollectionAchievement(collectionPlan.getAchievementPlan());

                            }

                        }

                        if (authId != null && authId != authUser.getId()) {

                            ptam.setAuthUser(authUserMap.get(authId));

                        }

                        if (collectionPlan.getCommittedDate() != null) {

                            ptam.setCollectionCommitmentDate(collectionPlan.getCommittedDate().toLocalDate());

                        }

                    }

                    // Secondary

                    PartnerSecondaryPlanModel otherPartnerSecondaryPlanModel = null;

                    Map<String, PartnerSecondaryPlanModel> secondaryModelMap = new HashMap<>();