Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import java.io.IOException;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.apache.http.HttpResponse;

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.PathVariable;

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

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

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

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

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.services.ReporticoService;

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

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

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

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

import com.spice.profitmandi.service.authentication.RoleManager;

import com.spice.profitmandi.web.model.LoginDetails;

import com.spice.profitmandi.web.util.CookiesProcessor;

@Controller

@Transactional(rollbackFor = Throwable.class)

public class ReportsController {

	@Autowired

	private RoleRepository roleRepository;

	@Autowired

	private CookiesProcessor cookiesProcessor;

	@Autowired

	private ReporticoService reporticoService;

	@Autowired

	private RoleManager roleManager;

	@Autowired

	private CsService csService;

	@Autowired

	private AuthRepository authRepository;

	private static final Logger log = LogManager.getLogger(OrderController.class);

	@RequestMapping(value = "/reports/{projectName}/{fileName}")

	public ResponseEntity<?> fetchReport(HttpServletRequest request, @PathVariable String fileName,

			@PathVariable ReporticoProject projectName, @RequestBody(required = false) Map<String, String> paramsMap)

			throws ProfitMandiBusinessException, UnsupportedOperationException, IOException {

		LoginDetails loginDetails = cookiesProcessor.getCookiesObject(request);

		HttpResponse response;

		if (roleManager.isAdmin(loginDetails.getRoleIds())) {

			if (fileName.equalsIgnoreCase("LeadsReport")) {

				if (paramsMap == null) {

					paramsMap = new HashMap<String, String>();

				}

				Map<Integer, List<Integer>> mapping = csService.getL2L1Mapping();

				AuthUser authUser = authRepository.selectByEmailOrMobile(loginDetails.getEmailId());

				List<Integer> authIds = mapping.get(authUser.getId());

				if (authIds == null) {

					authIds = new ArrayList<>();

				}

				authIds.add(authUser.getId());

				paramsMap.put("authId", authIds + "");

			}

		} else {

			if (paramsMap == null) {

				paramsMap = new HashMap<String, String>();

			}

			paramsMap.put("MANUAL_fofoId", loginDetails.getFofoId() + "");

		}

		response = getAdminReportFile(loginDetails.getEmailId(), projectName, fileName + ".xml", paramsMap);

		HttpHeaders headers = new HttpHeaders();

		InputStreamResource is = new InputStreamResource(response.getEntity().getContent());

		headers.set("Content-Type", "application/vnd.ms-excel");

		headers.set("Content-disposition", "inline; filename=report-" + fileName + ".csv");

		headers.setContentLength(response.getEntity().getContentLength());

		return new ResponseEntity<InputStreamResource>(is, headers, HttpStatus.OK);

	}

	private HttpResponse getAdminReportFile(String email, ReporticoProject projectName, String fileName,

			Map<String, String> reportParams) throws ProfitMandiBusinessException, IOException {

		return reporticoService.getReportFile(projectName, fileName, reportParams);

	}

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

	public String reports(Model model) throws ProfitMandiBusinessException, UnsupportedOperationException, IOException {

		model.addAttribute("reporticoProjectMap", ReporticoProject.reporticoProjectMap);

		return "reports";

	}

}