Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;
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.transaction.annotation.Transactional;
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 com.spice.profitmandi.common.enumuration.ContentType;
import com.spice.profitmandi.common.enumuration.CounterSize;
import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
import com.spice.profitmandi.common.model.AddLocationModel;
import com.spice.profitmandi.common.model.ProfitMandiConstants;
import com.spice.profitmandi.common.model.SendNotificationModel;
import com.spice.profitmandi.common.model.UpdateRetailerRequest;
import com.spice.profitmandi.common.util.Utils;
import com.spice.profitmandi.common.web.util.ResponseSender;
import com.spice.profitmandi.dao.entity.dtr.Document;
import com.spice.profitmandi.dao.entity.dtr.Retailer;
import com.spice.profitmandi.dao.entity.dtr.Shop;
import com.spice.profitmandi.dao.entity.user.User;
import com.spice.profitmandi.dao.entity.user.Location;
import com.spice.profitmandi.dao.repository.dtr.DocumentRepository;
import com.spice.profitmandi.dao.repository.dtr.RetailerRepository;
import com.spice.profitmandi.dao.repository.dtr.ShopRepository;
import com.spice.profitmandi.dao.repository.user.LocationRepository;
import com.spice.profitmandi.dao.repository.user.UserRepository;
import com.spice.profitmandi.service.user.RetailerService;
import com.spice.profitmandi.web.util.MVCResponseSender;

@Controller
@Transactional(rollbackFor=Throwable.class)
public class RetailerController {

        private static final Logger LOGGER = LogManager.getLogger(RetailerController.class);
        
        @Autowired
        private RetailerService retailerService;
        
        @Autowired
        private RetailerRepository retailerRepository;
        
        @Autowired
        private ShopRepository shopRepository;
        
        @Autowired
        private DocumentRepository documentRepository;
        
        @Autowired
        private UserRepository userRepository;
        
        @Autowired
        private LocationRepository locationRepository;
        
        @Autowired
        private ResponseSender<?> responseSender;
        
        @Autowired
        private MVCResponseSender mvcResponseSender;
        
        @RequestMapping(value = "/retailerDetails", method = RequestMethod.GET)
        public String retailerInfoByEmailIdOrMobileNumber(HttpServletRequest request, @RequestParam(name = ProfitMandiConstants.EMAIL_ID_OR_MOBILE_NUMBER) String emailIdOrMobileNumber, Model model)  throws ProfitMandiBusinessException{
                LOGGER.info("Request Received at url {} with emailIdOrMobileNumber {}", request.getRequestURI(), emailIdOrMobileNumber);
                Map<String, Object> map = retailerService.getByEmailIdOrMobileNumber(emailIdOrMobileNumber);
                model.addAllAttributes(map);
                model.addAttribute("counterSizes", CounterSize.values());
                User user = userRepository.selectByEmailId(emailIdOrMobileNumber);
                if(user.getLocation() != null){
                Location location = locationRepository.selectById(user.getLocation());
                model.addAttribute("locationdetail",location);
                 LOGGER.info("location"+location );
                }
                LOGGER.info("map detail"+ map) ;
      
                return "retailer-details";
        }
        
        @RequestMapping(value = "/retailerDetails", method = RequestMethod.PUT)
        public String updateRetailerDetails(HttpServletRequest request, @RequestBody UpdateRetailerRequest updateRetailerRequest, Model model)  throws ProfitMandiBusinessException{
                LOGGER.info("Request Received at url {} with body {}", request.getRequestURI(), updateRetailerRequest);
                Map<String, Object> map = retailerService.updateRetailerDetails(updateRetailerRequest);
                model.addAllAttributes(map);
                model.addAttribute("counterSizes", CounterSize.values());
                return "retailer-details";
        }
        

        @RequestMapping(value = "/retailerInfo", method = RequestMethod.GET)
        public String retailerInfo(HttpServletRequest request)  throws Exception{
                return "retailer-info";
        }
        
        @RequestMapping(value = "/district/all/stateName", method = RequestMethod.GET)
        public ResponseEntity<?> getAllDistrict(@RequestParam(name = "stateName") String stateName){
                return responseSender.ok(retailerService.getAllDistrictMaster(stateName));
        }
        
        @RequestMapping(value = "/retailerDocument/documentId", method = RequestMethod.GET)
        public ResponseEntity<?> retailerDocumentById(HttpServletRequest request, @RequestParam(name = ProfitMandiConstants.DOCUMENT_ID) int documentId, @RequestParam(name = ProfitMandiConstants.RETAILER_ID) int retailerId) throws ProfitMandiBusinessException{
                Document document = documentRepository.selectById(documentId);
                Retailer retailer = retailerRepository.selectById(retailerId);
                
                if(retailer.getDocumentId() == null) {
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.RETAILER_ID, retailer.getId(), "RTLR_1012");
                }
                if(retailer.getDocumentId() != documentId) {
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.DOCUMENT_ID, documentId, "RTLR_1014");
                }
                return responseSender.ok(document);
        }
        
        @RequestMapping(value = "/retailerDocument/download", method = RequestMethod.GET)
        public ResponseEntity<?> downloadRetailerDocument(HttpServletRequest request, @RequestParam(name = ProfitMandiConstants.RETAILER_ID) int retailerId, Model model) throws ProfitMandiBusinessException{
                
                Retailer retailer = retailerRepository.selectById(retailerId);
                
                if(retailer.getDocumentId() == null) {
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.RETAILER_ID, retailer.getId(), "RTLR_1012");
                }
                
                Document document = documentRepository.selectById(retailer.getDocumentId());
                
                FileInputStream file = null;
                try {
                        file = new FileInputStream(document.getPath() + File.separator + document.getName());
                } catch (FileNotFoundException e) {
                        LOGGER.error("Retailer Document file not found : ", e);
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.DOCUMENT_ID, document.getId(), "RTLR_1013");
                }
                //ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
                //ExcelUtils.writeSchemeModels(schemeModels, byteArrayOutputStream);
                
                final HttpHeaders headers=new HttpHeaders();
                String contentType = "";
                if(document.getContentType() == ContentType.JPEG) {
                        contentType = "image/jpeg";
                }else if(document.getContentType() == ContentType.PNG) {
                        contentType = "image/png";
                }else if(document.getContentType() == ContentType.PDF) {
                        contentType = "application/pdf";
                }
        headers.set("Content-Type", contentType);
                headers.set("Content-disposition", "inline; filename="+document.getName());
        headers.setContentLength(document.getSize());
        final InputStreamResource inputStreamResource=new InputStreamResource(file);
        return new ResponseEntity<InputStreamResource>(inputStreamResource, headers, HttpStatus.OK);
                
                //return responseSender.ok(ResponseCodeHolder.getMessage("ITM_AGNG_OK_1000"));
        }
        
        @RequestMapping(value = "/retailerShopDocument/shopId", method = RequestMethod.GET)
        public ResponseEntity<?> retailerShopDocumentById(HttpServletRequest request, @RequestParam(name = ProfitMandiConstants.SHOP_ID) int shopId, @RequestParam(name = ProfitMandiConstants.RETAILER_ID) int retailerId) throws ProfitMandiBusinessException{
                Shop shop = shopRepository.selectById(shopId);
                
                if(shop.getRetailerId() != retailerId) {
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.SHOP_ID, shop.getId(), "SHP_1004");
                }
                
                if(shop.getDocumentId() == null) {
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.RETAILER_ID, shop.getId(), "SHP_1005");
                }
                
                Document document = documentRepository.selectById(shop.getDocumentId());
                return responseSender.ok(document);
        }
        
        @RequestMapping(value = "/retailerShopDocument/download", method = RequestMethod.GET)
        public ResponseEntity<?> downloadRetailerShopDocument(HttpServletRequest request, @RequestParam(name = ProfitMandiConstants.SHOP_ID) int shopId, @RequestParam(name = ProfitMandiConstants.RETAILER_ID) int retailerId, Model model) throws ProfitMandiBusinessException{
                
                Shop shop = shopRepository.selectById(shopId);
                
                if(shop.getRetailerId() != retailerId) {
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.SHOP_ID, shop.getId(), "SHP_1004");
                }
                
                if(shop.getDocumentId() == null) {
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.RETAILER_ID, shop.getId(), "SHP_1005");
                }
                
                Document document = documentRepository.selectById(shop.getDocumentId());
                
                FileInputStream file = null;
                try {
                        file = new FileInputStream(document.getPath() + File.separator + document.getName());
                } catch (FileNotFoundException e) {
                        LOGGER.error("Retailer Document file not found : ", e);
                        throw new ProfitMandiBusinessException(ProfitMandiConstants.DOCUMENT_ID, document.getId(), "RTLR_1013");
                }
                //ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
                //ExcelUtils.writeSchemeModels(schemeModels, byteArrayOutputStream);
                
                final HttpHeaders headers=new HttpHeaders();
                String contentType = "";
                if(document.getContentType() == ContentType.JPEG) {
                        contentType = "image/jpeg";
                }else if(document.getContentType() == ContentType.PNG) {
                        contentType = "image/png";
                }else if(document.getContentType() == ContentType.PDF) {
                        contentType = "application/pdf";
                }
        headers.set("Content-Type", contentType);
                headers.set("Content-disposition", "inline; filename="+document.getName());
        headers.setContentLength(document.getSize());
        final InputStreamResource inputStreamResource=new InputStreamResource(file);
        return new ResponseEntity<InputStreamResource>(inputStreamResource, headers, HttpStatus.OK);
                
                //return responseSender.ok(ResponseCodeHolder.getMessage("ITM_AGNG_OK_1000"));
        }
        
        @RequestMapping(value = "/getAddLocation", method = RequestMethod.GET)
        public String getAddLocationModal(HttpServletRequest request , Model model)  throws ProfitMandiBusinessException{
                model.addAttribute("stateNames", Utils.getAllStateNames());
      
                return "add-location-modal";
        }
        
        @RequestMapping(value = "/addLocation", method = RequestMethod.POST)
        public String addLocation(HttpServletRequest request , @RequestBody AddLocationModel addLocationModel,Model model)  throws Exception{
                
        Location location = new Location();
        location.setName(addLocationModel.getName());
        location.setLine1(addLocationModel.getLine1());
        location.setLine2(addLocationModel.getLine2());
        location.setCity(addLocationModel.getCity());
        location.setState(addLocationModel.getState());
        location.setPin(addLocationModel.getPin());
        
        LOGGER.info("PostLocation"+location);
       locationRepository.persist(location);
        
        model.addAttribute("response", mvcResponseSender.createResponseString(true));
                return "response";
      
        }
}