Subversion Repositories SmartDukaan

import java.util.concurrent.atomic.AtomicInteger;

import org.springframework.http.HttpStatus;

import com.eclipsesource.json.JsonArray;

import com.eclipsesource.json.JsonValue;

import com.mongodb.BasicDBObject;

import com.mongodb.DBObject;

import com.spice.profitmandi.dao.entity.catalog.Category;

	@ApiImplicitParams({

			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })

	@RequestMapping(value = "/partnerStock", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	public ResponseEntity<?> partnerStock(HttpServletRequest request,

			@RequestParam(value = "categoryId", required = false, defaultValue = "(3 OR 6)") String categoryId,

			@RequestParam(value = "offset") String offset, @RequestParam(value = "limit") String limit,

			@RequestParam(value = "sort", required = false) String sort,

			@RequestParam(value = "brand", required = false) String brand,

			@RequestParam(value = "subCategoryId", required = false) int subCategoryId,

			@RequestParam(value = "q", required = false) String queryTerm,

			@RequestParam(value = " ", required = false, defaultValue = "true") boolean partnerStockOnly)

			throws Throwable {

		List<FofoCatalogResponse> dealResponse = new ArrayList<>();

		UserInfo userInfo = (UserInfo) request.getAttribute("userInfo");

		UserCart uc = userAccountRepository.getUserCart(userInfo.getUserId());

		logger.info("Retiler Id ==> {}", uc.getUserId());

		if (roleManagerService.isPartner(userInfo.getRoleIds())) {

			if (partnerStockOnly) {

			}

			List<Integer> tagIds = pricingService.getTagsIdsByRetailerId(userInfo.getRetailerId());

			RestClient rc = new RestClient();

			Map<String, String> params = new HashMap<>();

			List<String> mandatoryQ = new ArrayList<>();

			if (queryTerm != null && !queryTerm.equals("null")) {

				mandatoryQ.add(String.format("+(%s)", queryTerm));

			} else {

				queryTerm = null;

			}

			if (subCategoryId != 0) {

				mandatoryQ

						.add(String.format("+(subCategoryId_i:%s) +{!parent which=\"subCategoryId_i:%s\"} tagId_i:(%s)",

								subCategoryId, subCategoryId, StringUtils.join(tagIds, " ")));

			} else if (StringUtils.isNotBlank(brand)) {

				mandatoryQ.add(

						String.format("+(categoryId_i:%s) +(brand_ss:%s) +{!parent which=\"brand_ss:%s\"} tagId_i:(%s)",

								categoryId, brand, brand, StringUtils.join(tagIds, " ")));

			} else {

				mandatoryQ.add(

						String.format("+{!parent which=\"id:catalog*\"} tagId_i:(%s)", StringUtils.join(tagIds, " ")));

			}

			params.put("q", StringUtils.join(mandatoryQ, " "));

			params.put("fl", "*, [child parentFilter=id:catalog*]");

			if (queryTerm == null) {

				params.put("sort", "create_s desc");

			}

			params.put("start", String.valueOf(offset));

			params.put("rows", String.valueOf(limit));

			params.put("wt", "json");

			String response = null;

			try {

				response = rc.get(SchemeType.HTTP, "50.116.10.120", 8984, "solr/demo/select", params);

			} catch (HttpHostConnectException e) {

				throw new ProfitMandiBusinessException("", "", "Could not connect to host");

			}

			JSONObject solrResponseJSONObj = new JSONObject(response).getJSONObject("response");

			JSONArray docs = solrResponseJSONObj.getJSONArray("docs");

			dealResponse = getCatalogResponse(docs, false);

			/*

			 * if (Mongo.PARTNER_BLoCKED_BRANDS.containsKey(userInfo.getEmail())) {

			 * dealResponse.stream() .filter(x ->

			 * Mongo.PARTNER_BLoCKED_BRANDS.get(userInfo.getEmail()).contains(x.getBrand()))

			 * ; }

			 */

		} else {

			return responseSender.badRequest(

					new ProfitMandiBusinessException("Retailer id", userInfo.getUserId(), "NOT_FOFO_RETAILER"));

		}

		return responseSender.ok(dealResponse);

	}

	@RequestMapping(value = ProfitMandiConstants.URL_UNIT_DEAL, method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	@ApiImplicitParams({

			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })

	@ApiOperation(value = "Get unit deal object")

	public ResponseEntity<?> getUnitDeal(HttpServletRequest request, @PathVariable(value = "id") long id)

			throws ProfitMandiBusinessException {

		String response = null;

		// TODO: move to properties

		String uri = "getDealById/" + id;

		System.out.println("Unit deal " + uri);

		RestClient rc = new RestClient();

		Map<String, String> params = new HashMap<>();

		DealsResponse dealsResponse = null;

		try {

			response = rc.get(SchemeType.HTTP, host, port, uri, params);

		} catch (HttpHostConnectException e) {

			throw new ProfitMandiBusinessException("", "", "Could not connect to host");

		}

		JsonObject result_json = Json.parse(response).asObject();

		if (!result_json.isEmpty()) {

			dealsResponse = new Gson().fromJson(response, DealsResponse.class);

			Iterator<AvailabilityInfo> iter = dealsResponse.getAvailabilityInfo().iterator();

			while (iter.hasNext()) {

				AvailabilityInfo ai = iter.next();

				if (ai.getAvailability() <= 0)

					iter.remove();

			}

		}

		return responseSender.ok(dealsResponse);

	}

	@RequestMapping(value = "/partnerdeals/{id}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	@ApiImplicitParams({

			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })

	@ApiOperation(value = "Get unit deal object")

	public ResponseEntity<?> getUnitFocoDeal(HttpServletRequest request, @PathVariable(value = "id") long id)

			throws ProfitMandiBusinessException {

		List<FofoCatalogResponse> dealResponse = new ArrayList<>();

		List<Integer> tagIds = Arrays.asList(4);

		UserInfo userInfo = (UserInfo) request.getAttribute("userInfo");

		if (roleManagerService.isPartner(userInfo.getRoleIds())) {

			String categoryId = "(3 OR 6)";

			UserCart uc = userAccountRepository.getUserCart(userInfo.getUserId());

			RestClient rc = new RestClient();

			Map<String, String> params = new HashMap<>();

			List<String> mandatoryQ = new ArrayList<>();

			String catalogString = "catalog" + id;

			mandatoryQ.add(String.format("+(categoryId_i:%s) +(id:%s) +{!parent which=\"id:%s\"} tagId_i:(%s)",

					categoryId, catalogString, catalogString, StringUtils.join(tagIds, " ")));

			params.put("q", StringUtils.join(mandatoryQ, " "));

			params.put("fl", "*, [child parentFilter=id:catalog*]");

			params.put("sort", "rank_i asc, create_s desc");

			params.put("wt", "json");

			String response = null;

			try {

				response = rc.get(SchemeType.HTTP, "50.116.10.120", 8984, "solr/demo/select", params);

			} catch (HttpHostConnectException e) {

				throw new ProfitMandiBusinessException("", "", "Could not connect to host");

			}

			JSONObject solrResponseJSONObj = new JSONObject(response).getJSONObject("response");

			JSONArray docs = solrResponseJSONObj.getJSONArray("docs");

			dealResponse = getCatalogResponse(docs, false);

		} else {

			return responseSender.badRequest(

					new ProfitMandiBusinessException("Retailer id", userInfo.getUserId(), "NOT_FOFO_RETAILER"));

		}

		return responseSender.ok(dealResponse.get(0));

	}

	@RequestMapping(value = "/fofo/brands", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	public ResponseEntity<?> getBrandsToDisplay(HttpServletRequest request,

			@RequestParam(required = false, defaultValue = "0") int categoryId) throws Exception {

		UserInfo userInfo = (UserInfo) request.getAttribute("userInfo");

		logger.info("userInfo [{}]", userInfo);

		List<DBObject> brandsDisplay = mongoClient.getMongoBrands(userInfo.getRetailerId(), userInfo.getEmail(),

				categoryId);

		return new ResponseEntity<>(brandsDisplay, HttpStatus.OK);

	}

	@RequestMapping(value = "/fofo/accessory/all-categories", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	public ResponseEntity<?> getSubCategoriesToDisplay(HttpServletRequest request) throws Exception {

		UserInfo userInfo = (UserInfo) request.getAttribute("userInfo");

		logger.info("userInfo [{}]", userInfo);

		List<DBObject> subCateogriesDisplay = this.getSubCategoriesToDisplay();

		return new ResponseEntity<>(subCateogriesDisplay, HttpStatus.OK);

	}

	private List<DBObject> getSubCategoriesToDisplay() throws Exception {

		List<DBObject> subCategories = new ArrayList<>();

		RestClient rc = new RestClient();

		Map<String, String> params = new HashMap<>();

		params.put("q", "categoryId_i:6");

		params.put("group", "true");

		params.put("group.field", "subCategoryId_i");

		params.put("wt", "json");

		params.put("rows", "50");

		params.put("fl", "subCategoryId_i");

		String response = null;

		try {

			response = rc.get(SchemeType.HTTP, "50.116.10.120", 8984, "solr/demo/select", params);

		} catch (HttpHostConnectException e) {

			throw new ProfitMandiBusinessException("", "", "Could not connect to host");

		}

		JSONObject solrResponseJSONObj = new JSONObject(response).getJSONObject("grouped");

		JSONArray groups = solrResponseJSONObj.getJSONObject("subCategoryId_i").getJSONArray("groups");

		List<Integer> categoryIds = new ArrayList<>();

		for (int i = 0; i < groups.length(); i++) {

			JSONObject groupObject = groups.getJSONObject(i);

			int subCategoryId = groupObject.getInt("groupValue");

			int quantity = groupObject.getJSONObject("doclist").getInt("numFound");

			categoryIds.add(subCategoryId);

		}

		List<Category> categories = categoryRepository.selectByIds(categoryIds);

		AtomicInteger i = new AtomicInteger(0);

		categories.forEach(x -> {

			DBObject dbObject = new BasicDBObject();

			dbObject.put("name", x.getLabel());

			dbObject.put("subCategoryId", x.getId());

			dbObject.put("rank", i.incrementAndGet());

			dbObject.put("categoryId", 6);

			dbObject.put("url", "https://images.smartdukaan.com/uploads/campaigns/" + x.getId() + ".png");

			subCategories.add(dbObject);

		});

		return subCategories;

	}

	@RequestMapping(value = "/banners/{bannerType}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	public ResponseEntity<?> getBanners(@PathVariable String bannerType) {

		return new ResponseEntity<>(mongoClient.getBannersByType(bannerType), HttpStatus.OK);

	}

	@RequestMapping(value = "/deals/subCategories", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	public ResponseEntity<?> getSubcategoriesToDisplay() {

		return new ResponseEntity<>(mongoClient.getSubcategoriesToDisplay(), HttpStatus.OK);

	}

	@ApiImplicitParams({

			@ApiImplicitParam(name = "Auth-Token", value = "Auth-Token", required = true, dataType = "string", paramType = "header") })

	@RequestMapping(value = "/deals/skus/{skus}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

	public ResponseEntity<?> getDealsBySkus(@PathVariable String skus) throws ProfitMandiBusinessException {

		StringBuffer sb = new StringBuffer("/getDealsForNotification/");

		String uri = sb.append(skus).toString();

		RestClient rc = new RestClient();

		String response;

		try {

			response = rc.get(SchemeType.HTTP, host, port, uri, new HashMap<>());

		} catch (HttpHostConnectException e) {

			throw new ProfitMandiBusinessException("", "", "Could not connect to host");

		}

		JsonArray result_json = Json.parse(response).asArray();

		List<Object> responseObject = new ArrayList<>();

		for (JsonValue j : result_json) {

			// logger.info("res " + j.asArray());

			List<Object> innerObject = new ArrayList<>();

			for (JsonValue jsonObject : j.asArray()) {

				innerObject.add(toDealObject(jsonObject.asObject()));

			}

			if (innerObject.size() > 0) {

				responseObject.add(innerObject);

			}

		}

		return responseSender.ok(responseObject);

	}

	private List<FofoCatalogResponse> getCatalogSingleSkuResponse(JSONArray docs, Map<Integer, Integer> itemFilter,

			boolean hotDeal) throws ProfitMandiBusinessException {

		Map<Integer, TagListing> itemTagListingMap = null;

		List<FofoCatalogResponse> dealResponse = new ArrayList<>();

		List<Integer> tagIds = Arrays.asList(4);

		itemTagListingMap = tagListingRepository.selectByItemIdsAndTagIds(itemFilter.keySet(), new HashSet<>(tagIds))

				.stream().collect(Collectors.toMap(x -> x.getItemId(), x -> x));

		for (int i = 0; i < docs.length(); i++) {

			Map<Integer, FofoAvailabilityInfo> fofoAvailabilityInfoMap = new HashMap<>();

			JSONObject doc = docs.getJSONObject(i);

			for (int j = 0; j < doc.getJSONArray("_childDocuments_").length(); j++) {

				JSONObject childItem = doc.getJSONArray("_childDocuments_").getJSONObject(j);

				int itemId = childItem.getInt("itemId_i");

				TagListing tl = itemTagListingMap.get(itemId);

				if (tl == null) {

					continue;

				}

				if (hotDeal) {

					if (!tl.isHotDeals()) {

						continue;

					}

				}

				float sellingPrice = (float) childItem.getDouble("sellingPrice_f");

				if (fofoAvailabilityInfoMap.containsKey(itemId)) {

					if (fofoAvailabilityInfoMap.get(itemId).getSellingPrice() > sellingPrice) {

						fofoAvailabilityInfoMap.get(itemId).setSellingPrice(sellingPrice);

						fofoAvailabilityInfoMap.get(itemId).setMop((float) childItem.getDouble("mop_f"));

					}

				} else {

					FofoAvailabilityInfo fdi = new FofoAvailabilityInfo();

					fdi.setSellingPrice((float) childItem.getDouble("sellingPrice_f"));

					fdi.setMop((float) childItem.getDouble("mop_f"));

					fdi.setColor(childItem.has("color_s") ? childItem.getString("color_s") : "");

					fdi.setTagId(childItem.getInt("tagId_i"));

					fdi.setItem_id(itemId);

					Item item = itemRepository.selectById(itemId);

					// In case its tampered glass moq should be 5

					if (item.getCategoryId() == 10020) {

						fdi.setMinBuyQuantity(10);

					} else {

						fdi.setMinBuyQuantity(1);

					}

					fdi.setAvailability(itemFilter.get(itemId));

					fdi.setQuantityStep(1);

					fdi.setMaxQuantity(Math.min(fdi.getAvailability(), 100));

					fofoAvailabilityInfoMap.put(itemId, fdi);

				}

			}

			if (fofoAvailabilityInfoMap.values().size() > 0) {

				for (FofoAvailabilityInfo fofoAvailabilityInfo : fofoAvailabilityInfoMap.values()) {

					FofoCatalogResponse ffdr = new FofoCatalogResponse();

					ffdr.setCatalogId(doc.getInt("catalogId_i"));

					ffdr.setImageUrl(doc.getString("imageUrl_s"));

					ffdr.setTitle(doc.getString("title_s"));

					try {

						ffdr.setFeature(doc.getString("feature_s"));

					} catch (Exception e) {

						ffdr.setFeature(null);

						logger.info("Could not find Feature_s for {}", ffdr.getCatalogId());

					}

					ffdr.setBrand(doc.getJSONArray("brand_ss").getString(0));

					ffdr.setItems(Arrays.asList(fofoAvailabilityInfo));

					dealResponse.add(ffdr);

				}

			}

		}

		return dealResponse;

	}