Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
import com.spice.profitmandi.common.util.FormattingUtils;
import com.spice.profitmandi.common.util.Utils;
import com.spice.profitmandi.dao.entity.transaction.Order;
import com.spice.profitmandi.dao.repository.transaction.OrderRepository;
import com.spice.profitmandi.service.authentication.RoleManager;
import com.spice.profitmandi.web.model.LoginDetails;
import com.spice.profitmandi.web.util.CookiesProcessor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;

import javax.servlet.http.HttpServletRequest;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.YearMonth;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Controller
@Transactional(rollbackFor = Throwable.class)
public class InvoiceController {
    @Autowired
    OrderRepository orderRepository;

    @Autowired
    CookiesProcessor cookiesProcessor;

    @Autowired
    RoleManager roleManager;

    @RequestMapping(value = {"invoice-summary", "/invoice-summary/{yearMonth}" })
    public String addAddress(HttpServletRequest request, @PathVariable(required = false) YearMonth yearMonth, Model model, @RequestParam(defaultValue = "0") int fofoId) throws ProfitMandiBusinessException {
        if (yearMonth == null) {
            yearMonth = YearMonth.now();
        }
        LocalDateTime startDate = yearMonth.atDay(1).atStartOfDay();
        LocalDateTime endDate = yearMonth.atEndOfMonth().atTime(LocalTime.MAX);
        LoginDetails loginDetails = cookiesProcessor.getCookiesObject(request);
        boolean isAdmin = false;
        if (roleManager.isAdmin(loginDetails.getRoleIds())) {
            isAdmin = true;
        } else {
            fofoId = loginDetails.getFofoId();
        }
        List<Order> orders;
        if (fofoId == 0) {
            orders = new ArrayList<>();
        } else {
            orders = orderRepository.selectAllByBillingDatesBetween(fofoId, startDate,
                    endDate);
        }
        Map<String, LocalDateTime> invoiceBillingDateMap = orders.stream().filter(Utils.distinctByKey(x -> x.getInvoiceNumber())).collect(Collectors.toMap(x -> x.getInvoiceNumber(), x -> x.getBillingTimestamp()));
        String yearMonthString = FormattingUtils.formatYearMonth(startDate);
        model.addAttribute("yearMonth", yearMonth);
        model.addAttribute("invoiceMap", invoiceBillingDateMap);
        model.addAttribute("isAdmin", isAdmin);

        return "invoice-summary";
    }

}