Subversion Repositories SmartDukaan

package com.smartdukaan.cron.migrations;

import java.io.File;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeMessage;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;

import com.spice.profitmandi.common.model.CustomRetailer;
import com.spice.profitmandi.common.util.FormattingUtils;
import com.spice.profitmandi.dao.entity.fofo.FofoStore;
import com.spice.profitmandi.dao.entity.fofo.Purchase;
import com.spice.profitmandi.dao.entity.transaction.LineItem;
import com.spice.profitmandi.dao.entity.transaction.LineItemImei;
import com.spice.profitmandi.dao.entity.transaction.Order;
import com.spice.profitmandi.dao.repository.dtr.FofoStoreRepository;
import com.spice.profitmandi.dao.repository.fofo.PurchaseRepository;
import com.spice.profitmandi.dao.repository.transaction.LineItemImeisRepository;
import com.spice.profitmandi.dao.repository.transaction.LineItemRepository;
import com.spice.profitmandi.dao.repository.transaction.OrderRepository;
import com.spice.profitmandi.service.inventory.InventoryService;
import com.spice.profitmandi.service.transaction.TransactionService;
import com.spice.profitmandi.service.user.RetailerService;
import com.spice.profitmandi.service.wallet.WalletService;

@Component
@Transactional(rollbackFor = Throwable.class)
public class RunOnceTasks {

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

        @Autowired
        private LineItemRepository lineItemRepository;

        @Autowired
        private JavaMailSender mailSender;
        
        @Autowired
        private WalletService walletService;

        @Autowired
        private InventoryService inventoryService;

        @Autowired
        private TransactionService transactionService;

        @Autowired
        private OrderRepository orderRepository;

        @Autowired
        private FofoStoreRepository fofoStoreRepository;

        @Autowired
        private LineItemImeisRepository lineItemImeisRepository;

        @Autowired
        private RetailerService retailerService;

        @Autowired
        private PurchaseRepository purchaseRepository;

        public void populateGrnTimestamp() {
                List<Purchase> allPurchases = purchaseRepository.selectAll();
                for (Purchase p : allPurchases) {
                        String invoiceNumber = p.getPurchaseReference();
                        if (p.getCompleteTimestamp() == null) {
                                LOGGER.info("GRN for invoice {} is delivered but partially Completed.", p.getPurchaseReference());
                        } else {
                                List<Order> orders = orderRepository.selectByAirwayBillOrInvoiceNumber(invoiceNumber, p.getFofoId());
                                for (Order order : orders) {
                                        if (order.getPartnerGrnTimestamp() == null) {
                                                order.setPartnerGrnTimestamp(p.getCompleteTimestamp());
                                                orderRepository.persist(order);
                                        }
                                }
                        }
                }

        }

        public void migarateLineItemsToNewTable() {
                LOGGER.info("Before Migrated LineItems Successfully");
                int lineItemImeiId = 0;
                LocalDateTime startDate = null;
                try {
                        lineItemImeiId = lineItemImeisRepository.selectMaxId();
                        LineItem lineItem = lineItemRepository
                                        .selectById(lineItemImeisRepository.selectById(lineItemImeiId).getLineItemId());
                        Order order = orderRepository.selectById(lineItem.getOrderId());
                        startDate = order.getBillingTimestamp();
                } catch (Exception e) {
                        LOGGER.info("Running before first time");
                        startDate = LocalDateTime.of(LocalDate.of(2017, 7, 1), LocalTime.MIDNIGHT);
                }
                List<Order> orders = orderRepository.selectAllByBillingDatesBetween(startDate, LocalDateTime.now());
                Collections.reverse(orders);

                for (Order order : orders) {
                        try {
                                String serialNumbers = order.getLineItem().getSerialNumber();
                                if (!StringUtils.isEmpty(serialNumbers)) {
                                        List<String> serialNumberList = Arrays.asList(serialNumbers.split(","));
                                        for (String serialNumber : serialNumberList) {
                                                int lineItemId = order.getLineItem().getId();
                                                LineItemImei lineItemImei = new LineItemImei();
                                                lineItemImei.setSerialNumber(serialNumber);
                                                lineItemImei.setLineItemId(lineItemId);
                                                lineItemImeisRepository.persist(lineItemImei);
                                        }
                                } else {
                                        LOGGER.info("Serial Numbers dont exist for Order {}", order.getId());
                                }
                        } catch (Exception e) {
                                LOGGER.info("Error occurred while creating lineitem imei {}, because of {}", order.getId(),
                                                e.getMessage());
                        }
                }
                LOGGER.info("Migrated LineItems Successfully");
        }

        public void getInvestmentDetails() throws Exception {
                File f = new File("/tmp/InvestmentSummary-" + FormattingUtils.formatDate(LocalDateTime.now()) + ".csv");
                List<String> lines = new ArrayList<>();
                lines.add(
                                "Code\tStoreName\tEmail\tMobile\tminimumInvestment\twalletAmount\tinStockAmount\tunbilledStockAmount\tgrnPendingStockAmount\tTotalInvested");
                List<FofoStore> fofoStores = fofoStoreRepository.selectAll();
                Map<Integer, CustomRetailer> customRetailerMap = retailerService
                                .getFofoRetailers(fofoStores.stream().map(x -> x.getId()).collect(Collectors.toList()));
                for (FofoStore fofoStore : fofoStores) {
                        try {
                                CustomRetailer retailer = customRetailerMap.get(fofoStore.getId());
                                float walletAmount = walletService.getUserWallet(fofoStore.getId()).getAmount();
                                float inStockAmount = inventoryService.getTotalAmountInStock(fofoStore.getId());

                                float unbilledStockAmount = 0;
                                List<Order> unbilledOrders = transactionService.getInTransitOrders(fofoStore.getId());
                                for (Order unBilledOrder : unbilledOrders) {
                                        unbilledStockAmount += unBilledOrder.getTotalAmount();
                                }

                                float grnPendingStockAmount = 0;
                                List<Order> grnPendingOrders = orderRepository.selectPendingGrnOrders(fofoStore.getId());
                                for (Order grnPendingOrder : grnPendingOrders) {
                                        grnPendingStockAmount += grnPendingOrder.getTotalAmount();
                                }
                                float totalInvestedAmount = walletAmount + inStockAmount + unbilledStockAmount + grnPendingStockAmount;
                                lines.add(String.join("\t", fofoStore.getCode(), retailer.getBusinessName(), retailer.getMobileNumber(),
                                                String.valueOf(fofoStore.getMinimumInvestment()), String.valueOf(walletAmount),
                                                String.valueOf(inStockAmount), String.valueOf(unbilledStockAmount),
                                                String.valueOf(grnPendingStockAmount), String.valueOf(totalInvestedAmount)));
                        } catch (Exception e) {
                                e.printStackTrace();
                                continue;
                        }
                        FileUtils.writeLines(f, lines);
                        this.sendMailWithAttachments("amit.gupta@shop2020.in", "Partner Investment Summary", "PFA", Arrays.asList(f));
                }

        }
        
        private void sendMailWithAttachments(String email, String subject, String body, List<File> attachments) throws Exception {

        MimeMessage message = mailSender.createMimeMessage();
        MimeMessageHelper helper = new MimeMessageHelper(message,true);
        helper.setSubject(subject);
        helper.setText(body);
        //String[] cc = {"backup@shop2020.in", "gadgetcops.01@gmail.com"};
        //String[] cc = {"amit.gupta@shop2020.in"};
        //helper.setCc(cc);
        helper.setTo(email);
        InternetAddress senderAddress = new InternetAddress("noreply@smartdukaan.com", "ProfitMandi Admin");
        //helper.setTo("help@smartdukaan.com");
        helper.setFrom(senderAddress);
        for (File file : attachments) {
                helper.addAttachment(file.getName(), file);
        }
        mailSender.send(message);
                
        }
}