Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
import com.spice.profitmandi.common.util.FileUtil;
import com.spice.profitmandi.common.util.FormattingUtils;
import com.spice.profitmandi.dao.entity.fofo.FofoStore;
import com.spice.profitmandi.dao.entity.fofo.SamsungPCM;
import com.spice.profitmandi.dao.repository.dtr.FofoStoreRepository;
import com.spice.profitmandi.dao.repository.fofo.SamsungPCMRepository;
import com.spice.profitmandi.dao.repository.warehouse.BilledImeiModel;
import com.spice.profitmandi.dao.repository.warehouse.WarehouseInventoryItemRepository;
import com.spice.profitmandi.service.user.RetailerService;
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.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 java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.time.LocalDate;
import java.util.Arrays;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.stream.Collectors;

@Controller
@Transactional(rollbackFor = Throwable.class)
public class BrandPCMController {
    private static final Logger LOGGER = LogManager.getLogger(AuthUserController.class);

    @Autowired
    SamsungPCMRepository samsungPCMRepository;
    @Autowired
    RetailerService retailerService;

    @Autowired
    FofoStoreRepository fofoStoreRepository;

    @Autowired
    WarehouseInventoryItemRepository warehouseInventoryItemRepository;

    @RequestMapping(value = "/brand-pcm", method = RequestMethod.GET)
    public String brandPcm(Model model) throws ProfitMandiBusinessException {
        Map<Integer, LocalDate> samsungPCMMap = samsungPCMRepository.selectAll().stream().collect(Collectors.toMap(x -> x.getFofoId(), x -> x.getPcmDate()));
        model.addAttribute("samsungPCMMap", samsungPCMMap);
        model.addAttribute("retailerMap", retailerService.getAllFofoRetailers());
        return "manage-pcm";
    }

    @RequestMapping(value = "/brand-pcm", method = RequestMethod.POST)
    public String addBrandPCM(Model model, @RequestBody BrandPcmDateModel brandPcmDateModel) throws Exception {

        SamsungPCM samsungPCM = new SamsungPCM();
        samsungPCM.setFofoId(brandPcmDateModel.getFofoId());
        samsungPCM.setPcmDate(brandPcmDateModel.getPcmDate());
        samsungPCMRepository.persist(samsungPCM);

        model.addAttribute("response1", true);
        return "response";
    }

    @RequestMapping(value = "/brand-pcm/download", method = RequestMethod.GET)
    public ResponseEntity<InputStreamResource> downloadBrandPCM(@RequestParam int fofoId) throws Exception {

        FofoStore fs = fofoStoreRepository.selectByRetailerId(fofoId);

        List<BilledImeiModel> billedImeiModels = warehouseInventoryItemRepository.findStockByFofoIdBrandToRebill(fofoId, "Samsung");
        List<List<?>> rows = billedImeiModels.stream().map(x -> Arrays.asList(x.getFofoId(), x.getStoreCode(), x.getStoreName(), x.getStoreCity(), x.getBrand(), x.getModelName(), x.getModelNumber(), x.getColor(), x.getSerialNumber(),
                FormattingUtils.formatDate(x.getPcmDate()), FormattingUtils.formatDate(x.getInvoiceDate())
        )).collect(Collectors.toList());

        org.apache.commons.io.output.ByteArrayOutputStream baos = FileUtil.getCSVByteStream(Arrays.asList("Store Id",
                "Store Code", "Business Name", "City", "Brand", "Model Name", "Model Number", "Color", "Imei", "Invoice Date", "PCM Date"), rows);

        final HttpHeaders headers = new HttpHeaders();
        headers.set("Content-Type", "text/csv");
        headers.set("Content-disposition", "inline; filename=pcm-" + fs.getCode().toLowerCase(Locale.ROOT) + ".csv");
        headers.setContentLength(baos.toByteArray().length);

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

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


}