Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.controller;

import java.util.HashSet;

import java.util.List;

import javax.servlet.http.HttpServletRequest;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Qualifier;

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.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.model.CreateWarehouseRequest;

import com.spice.profitmandi.common.model.ProfitMandiConstants;

import com.spice.profitmandi.common.util.Utils;

import com.spice.profitmandi.dao.entity.dtr.User;

import com.spice.profitmandi.dao.entity.nonbillable.Warehouse;

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

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

import com.spice.profitmandi.dao.repository.nonbillable.UserWarehouseRepository;

import com.spice.profitmandi.dao.repository.nonbillable.WarehouseRepository;

import com.spice.profitmandi.service.nonbillable.WarehouseService;

@Controller

@Transactional

public class WarehouseController {

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

	@Autowired

	private WarehouseRepository warehouseRepository;

	@Autowired

	private WarehouseService warehouseService;

	@Autowired

	@Qualifier("userRepository")

	private UserRepository userRepository;

	@Autowired

	private UserAccountRepository userAccountRepository;

	@Autowired

	private UserWarehouseRepository userWarehouseRepository;

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

	public String createWarehouse(HttpServletRequest request, Model model) {

		model.addAttribute("stateNames", Utils.getAllStateNames());

		return "create-warehouse";

	}

	@RequestMapping(value = "/createWarehouse", method = RequestMethod.POST)

	public String createWarehouse(HttpServletRequest request, @RequestBody CreateWarehouseRequest createWarehouseRequest, @RequestParam(name = "offset", defaultValue = "0") int offset, @RequestParam(name = "limit", defaultValue = "10") int limit, Model model) throws ProfitMandiBusinessException{

		LOGGER.info("Request Received at url {} with body {}", request.getRequestURI(), createWarehouseRequest);

		warehouseService.createWarehouse(createWarehouseRequest);

		List<Warehouse> warehouses = warehouseRepository.selectAll(offset, limit);

		long size = warehouseRepository.selectAllCount();

		model.addAttribute("warehouses", warehouses);

		model.addAttribute("start", offset + 1);

		model.addAttribute("size", size);

		if (warehouses.size() < limit){

			model.addAttribute("end", offset + warehouses.size());

		}else{

			model.addAttribute("end", offset + limit);

		}

		return "warehouses";

	}

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

	public String getWarehouses(HttpServletRequest request, @RequestParam(name = "offset", defaultValue = "0") int offset, @RequestParam(name = "limit", defaultValue = "10") int limit, Model model) throws ProfitMandiBusinessException{

		List<Warehouse> warehouses = warehouseRepository.selectAll(offset, limit);

		long size = warehouseRepository.selectAllCount();

		model.addAttribute("warehouses", warehouses);

		model.addAttribute("start", offset + 1);

		model.addAttribute("size", size);

		if (warehouses.size() < limit){

			model.addAttribute("end", offset + warehouses.size());

		}else{

			model.addAttribute("end", offset + limit);

		}

		return "warehouses";

	}

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

	public String getPaginatedWarehouses(HttpServletRequest request, @RequestParam(name = "offset", defaultValue = "0") int offset, @RequestParam(name = "limit", defaultValue = "10") int limit, Model model) throws ProfitMandiBusinessException{

		List<Warehouse> warehouses = warehouseRepository.selectAll(offset, limit);

		model.addAttribute("warehouses", warehouses);

		return "warehouses-paginated";

	}

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

	public String getAddRemoveWarehouse(HttpServletRequest request) {

		return "add-remove-warehouse";

	}

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

	public String userWarehouses(HttpServletRequest request, @RequestParam(name = ProfitMandiConstants.EMAIL_ID_OR_MOBILE_NUMBER) String emailIdOrMobileNumber, Model model)  throws Exception{

		User user = userRepository.selectByEmailIdOrMobileNumber(emailIdOrMobileNumber);

		int retailerId = userAccountRepository.selectRetailerIdByUserId(user.getId());

		List<Integer> warehouseIds = userWarehouseRepository.selectWarehouseIdsByUserId(retailerId);

		List<Warehouse> warehouses = warehouseRepository.selectAll();

		model.addAttribute("userWarehouseIds", warehouseIds);

		model.addAttribute("warehouses", warehouses);

		return "user-warehouses";

	}

	@RequestMapping(value = "/addRemoveWarehouses", method = RequestMethod.POST)

	public String addRemoveRoles(HttpServletRequest request, @RequestParam(name = ProfitMandiConstants.EMAIL_ID_OR_MOBILE_NUMBER) String emailIdOrMobileNumber, @RequestBody List<Integer> warehouseIds, Model model)  throws Exception{

		warehouseService.addRemoveWarehouses(emailIdOrMobileNumber, new HashSet<>(warehouseIds));

		return "add-remove-warehouse";

	}

}