Subversion Repositories SmartDukaan

package com.spice.profitmandi.dao.service.solr;

import com.fasterxml.jackson.core.JsonProcessingException;

import com.fasterxml.jackson.databind.ObjectMapper;

import com.google.gson.Gson;

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

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

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

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

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

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

import com.spice.profitmandi.dao.entity.inventory.SaholicCIS;

import com.spice.profitmandi.dao.entity.inventory.SaholicPOItem;

import com.spice.profitmandi.dao.repository.catalog.*;

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

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

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

import com.spice.profitmandi.dao.repository.similarModel.SimilarModelRepository;

import com.spice.profitmandi.service.catalog.SuperCatalogModel;

import com.spice.profitmandi.service.inventory.SaholicInventoryService;

import com.spice.profitmandi.service.tag.ItemTagModel;

import in.shop2020.model.v1.catalog.status;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.apache.solr.client.solrj.SolrClient;

import org.apache.solr.client.solrj.impl.HttpSolrClient;

import org.apache.solr.common.SolrInputDocument;

import org.json.JSONArray;

import org.json.JSONException;

import org.json.JSONObject;

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

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

import org.springframework.stereotype.Component;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.net.HttpURLConnection;

import java.net.URL;

import java.net.URLConnection;

import java.time.ZoneId;

import java.util.*;

import java.util.Map.Entry;

import java.util.stream.Collectors;

import java.util.zip.GZIPInputStream;

@Component

public class FofoSolr {

    @Autowired

    private Gson gson;

    @Autowired

    private TagListingRepository tagListingRepository;

    @Autowired

    private CatalogRepository catalogRepository;

    @Autowired

    TagRankingRepository tagRankingRepository;

    @Autowired

    private Mongo contentMongoClient;

    @Autowired

    private WebListingRepository webListingRepository;

    @Autowired

    private WebProductListingRepository webProductListingRepository;

    @Autowired

    private SuperCatalogMappingRepository superCatalogMappingRepository;

    @Value("${new.solr.url}")

    private String solrUrl;

    @Value("${reportico.url}")

    private String reporticoUrl;

    private static final Logger logger = LogManager.getLogger(FofoSolr.class);

    private static final List<Integer> CATEGORY_MASTER = Arrays.asList(10006, 10007, 10010, 14202, 14203);

    @Autowired

    ModelFlagshipRepository modelFlagshipRepository;

    @Autowired

    ModelHotDealRepository modelHotDealRepository;

    private static String getUrlContent(String urlString) {

        BufferedReader reader = null;

        try {

            URL url = new URL(urlString);

            URLConnection conn = url.openConnection();

            conn.setConnectTimeout(5000);

            conn.setReadTimeout(5000);

            reader = new BufferedReader(new InputStreamReader(conn.getInputStream()));

            return reader.lines().collect(Collectors.joining(System.lineSeparator()));

        } catch (IOException e) {

            e.printStackTrace();

        } finally {

            if (reader != null) {

                try {

                    reader.close();

                } catch (IOException e) {

                    e.printStackTrace();

                }

            }

        }

        return null;

    }

    public JSONArray getAvailability() throws IOException, JSONException {

        String url = reporticoUrl + "?execute_mode=EXECUTE&xmlin=warehousecisnew.xml&project=FOCOR&project_password=focor&target_format=JSON";

        HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection();

        connection.setRequestProperty("Accept-Encoding", "gzip");

        BufferedReader reader = null;

        try {

            InputStream inputStream = connection.getInputStream();

            if ("gzip".equals(connection.getContentEncoding())) {

                inputStream = new GZIPInputStream(inputStream);

            }

            reader = new BufferedReader(new InputStreamReader(inputStream, "UTF-8"));

            StringBuilder responseBuilder = new StringBuilder();

            String line;

            while ((line = reader.readLine()) != null) {

                responseBuilder.append(line);

            }

            JSONObject responseJson = new JSONObject(responseBuilder.toString());

            return responseJson.getJSONArray("data");

        } finally {

            if (reader != null) {

                try {

                    reader.close();

                } catch (IOException e) {

                    // Ignore

                }

            }

        }

    }

    public JSONArray getPendingPO() throws IOException, JSONException {

        String url = reporticoUrl + "?execute_mode=EXECUTE&xmlin=UnfulfilledPOItemsNew.xml&project=FOCOR&project_password=focor&target_format=JSON";

        HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection();

        connection.setRequestProperty("Accept-Encoding", "gzip");

        BufferedReader reader = null;

        try {

            InputStream inputStream = connection.getInputStream();

            if ("gzip".equals(connection.getContentEncoding())) {

                inputStream = new GZIPInputStream(inputStream);

            }

            reader = new BufferedReader(new InputStreamReader(inputStream, "UTF-8"));

            StringBuilder responseBuilder = new StringBuilder();

            String line;

            while ((line = reader.readLine()) != null) {

                responseBuilder.append(line);

            }

            JSONObject responseJson = new JSONObject(responseBuilder.toString());

            return responseJson.getJSONArray("data");

        } finally {

            if (reader != null) {

                try {

                    reader.close();

                } catch (IOException e) {

                    // Ignore

                }

            }

        }

    }

    @Autowired

    SaholicInventoryService saholicInventoryService;

    public Map<String, Map<String, Map<String, Integer>>> getItemMap() throws Exception {

//        JSONArray availabilityJSONArray = this.getAvailability();

        List<SaholicCIS> saholicCISList = saholicInventoryService.getSaholicStockListWithoutCatalogMovingStatus();

//        JSONArray pendingPOJSONArray = this.getPendingPO();

        List<SaholicPOItem> saholicPOItemList = saholicInventoryService.getSaholicPOItemList();

        Map<String, Map<String, Map<String, Integer>>> availabilityItemMap = new HashMap<>();

        for (SaholicCIS rawAvailability : saholicCISList) {

            String itemId = String.valueOf(rawAvailability.getItemId());

            String warehouseId = String.valueOf(rawAvailability.getWarehouseId());

            if (!availabilityItemMap.containsKey(itemId)) {

                availabilityItemMap.put(itemId, new HashMap<>());

            }

            if (!availabilityItemMap.get(itemId).containsKey(warehouseId)) {

                availabilityItemMap.get(itemId).put(warehouseId, new HashMap<String, Integer>());

                availabilityItemMap.get(itemId).get(warehouseId).put("netAvailability", 0);

                availabilityItemMap.get(itemId).get(warehouseId).put("netPo", 0);

            }

            int netAvailability = availabilityItemMap.get(itemId).get(warehouseId).get("netAvailability") + rawAvailability.getNetavailability();

            availabilityItemMap.get(itemId).get(warehouseId).put("netAvailability", netAvailability);

        }

        for (SaholicPOItem rawPendingPo : saholicPOItemList) {

            String itemId = String.valueOf(rawPendingPo.getItemId());

            String warehouseId = String.valueOf(rawPendingPo.getWarehouseId());

            if (!availabilityItemMap.containsKey(itemId)) {

                availabilityItemMap.put(itemId, new HashMap<String, Map<String, Integer>>());

            }

            if (!availabilityItemMap.get(itemId).containsKey(warehouseId)) {

                availabilityItemMap.get(itemId).put(warehouseId, new HashMap<String, Integer>() {

                    {

                        put("netAvailability", 0);

                        put("netPo", 0);

                    }

                });

            }

            availabilityItemMap.get(itemId).get(warehouseId).put("netPo",

                    availabilityItemMap.get(itemId).get(warehouseId).get("netPo") + rawPendingPo.getUnfulfilledQty());

        }

        return availabilityItemMap;

    }

    public Map<Integer, List<String>> getLabels() throws ProfitMandiBusinessException {

        List<String> labels = Arrays.asList("partner-best-sellers", "partner-price-drop", "new-launches",

                "upgrade-offer", "special-support");

        Map<Integer, String> webListing = webListingRepository.selectByUrls(labels).stream()

                .collect(Collectors.toMap(x -> x.getId(), x -> x.getUrl()));

        Map<Integer, List<WebProductListing>> webProductListingMap = webProductListingRepository

                .selectAllByWebListingIds(new ArrayList<>(webListing.keySet())).stream()

                .collect(Collectors.groupingBy(x -> x.getWebListingId()));

        Map<Integer, List<String>> catalogLabelMap = new HashMap<>();

        for (Entry<Integer, List<WebProductListing>> webProductListingEntry : webProductListingMap.entrySet()) {

            for (WebProductListing webProduct : webProductListingEntry.getValue()) {

                if (!catalogLabelMap.containsKey(webProduct.getEntityId())) {

                    catalogLabelMap.put(webProduct.getEntityId(), new ArrayList<>());

                }

                List<String> itemLabels = catalogLabelMap.get(webProduct.getEntityId());

                itemLabels.add(webListing.get(webProduct.getWebListingId()));

                catalogLabelMap.put(webProduct.getEntityId(), itemLabels);

            }

        }

        return catalogLabelMap;

    }

    @Autowired

    SimilarModelRepository similarModelRepository;

    public void populateTagItems() throws Exception {

        Map<String, Map<String, Map<String, Integer>>> availabilityItemMap = getItemMap();

        List<TagRanking> tagRankingList = tagRankingRepository.getAllTagRanking();

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

        Map<Integer, String> featureMap = new HashMap<>();

        for (TagRanking tagRanking : tagRankingList) {

            rankingList.add(tagRanking.getCatalogItemId());

            featureMap.put(tagRanking.getCatalogItemId(), tagRanking.getFeature());

        }

        //Get Similar Models

        /*List<SimilarModel> similarModels = similarModelRepository.selectAll();

        Map<Integer, List<Integer>> similarModelsMap = similarModels.stream().collect(Collectors.groupingBy(SimilarModel::getCatalogId, Collectors.mapping(x -> x.getSimilarCatalogId(), Collectors.toList())));*/

        Map<Integer, Map<String, Object>> catalogMap = new HashMap<>();

        List<status> statuses = Arrays.asList(status.ACTIVE, status.PAUSED_BY_RISK, status.PARTIALLY_ACTIVE);

        Map<Integer, List<String>> catalogLabelMap = this.getLabels();

        //logger.info("catalogLabelMap {}", catalogLabelMap);

        List<ItemTagModel> itemTagModels = tagListingRepository.getAllItemTagByStatus(statuses);

        logger.info("itemTagModels {}" , itemTagModels.size());

        List<Integer> eolWithOutStock = catalogRepository.findAllWithEOLWithOutStock();

        List<Integer> eolWithBadQty = catalogRepository.findAllWithNoGoodStock();

        Set<Integer> eolSet = new HashSet<>(eolWithOutStock);

        eolSet.addAll(eolWithBadQty);

        // Flagship models active as of today (catalog.model_flagship start_date/end_date)

        Set<Integer> flagshipSet = modelFlagshipRepository.selectActiveFlagshipCatalogIds();

        // Hot deal membership IS brand membership (catalog.brand = 'Hot Deal'). There is no

        // date window any more; model_hot_deal supplies only the pill attributes and the

        // pre-move brand, which is pushed separately as oem_brand_s.

        Set<Integer> hotDealSet = catalogRepository

                .selectCatalogIdsByBrand(ProfitMandiConstants.HOT_DEAL_BRAND);

        // oem_brand_s drives the brand chips INSIDE the hot-deal listing. It is kept out of

        // brand_ss on purpose: brand_ss feeds SolrService.brandExclusionFq, so an OEM value

        // there would hide the SKU from a partner blocked on that OEM.

        Map<Integer, String> oemBrandByCatalog = new HashMap<>();

        for (com.spice.profitmandi.dao.entity.catalog.ModelHotDeal hotDealRow : modelHotDealRepository

                .selectByCatalogItemIds(new ArrayList<>(hotDealSet))) {

            if (hotDealRow.getOemBrand() != null) {

                oemBrandByCatalog.put(hotDealRow.getCatalogItemId(), hotDealRow.getOemBrand());

            }

        }

        // Flagship is surfaced as a label (prepended so it shows first, ahead of all other flags)

        for (Integer flagshipCatalogId : flagshipSet) {

            catalogLabelMap.computeIfAbsent(flagshipCatalogId, k -> new ArrayList<>()).add(0, "flagship");

        }

        logger.info("eolSetIds.size {}" , eolSet.size());

        Map<String, Object> projection = new HashMap<>();

        projection.put("defaultImageUrl", 1);

        List<String> excludeBrands = Arrays.asList("Dummy", "FOC HANDSET");

        Map<Float, List<Integer>> priceModelMap = new TreeMap<>();

        for (ItemTagModel itemTagModel : itemTagModels) {

            TagListing tagListing = itemTagModel.getTagListing();

            com.spice.profitmandi.dao.entity.catalog.Item item = itemTagModel.getItem();

            if (excludeBrands.contains(item.getBrand())) {

                continue;

            }

            if (!catalogMap.containsKey(item.getCatalogItemId())) {

                if (!priceModelMap.containsKey(tagListing.getMop()) && item.getCategoryId() == 10006) {

                    priceModelMap.put(tagListing.getMop(), new ArrayList<>());

                }

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

                    List<Integer> priceModels = priceModelMap.get(tagListing.getMop());

                    priceModels.add(item.getCatalogItemId());

                }

                Map<String, Object> catalogObj = new HashMap<>();

                catalogObj.put("stockColor", 0);

                catalogObj.put("feature", "");

                catalogObj.put("hsnCode", item.getHsnCode());

                catalogObj.put("title",

                        String.join(" ", Arrays.asList(item.getBrand(), item.getModelName(), item.getModelNumber())

                                .stream().filter(s -> s != null && !s.isEmpty()).collect(Collectors.toList())));

                catalogObj.put("brand", new String[]{item.getBrand()});

                if (item.getBrand().equals("Huawei") || item.getBrand().equals("Honor")) {

                    catalogObj.put("brand", new String[]{"Huawei", "Honor"});

                }

                if (item.getBrand().equals("Mi") || item.getBrand().equals("Xiaomi")

                        || item.getBrand().equals("Redmi")) {

                    catalogObj.put("brand", new String[]{"Mi", "Xiaomi", "Redmi"});

                }

                catalogObj.put("identifier", item.getCatalogItemId());

                catalogObj.put("items", new HashMap<Integer, Map<String, Object>>());

                Map<String, Object> filterMap = new HashMap<>();

                filterMap.put("_id", item.getCatalogItemId());

                // Don't include if catalog not available

                try {

                    String imageUrl = contentMongoClient.getEntityById(item.getCatalogItemId()).getDefaultImageUrl();

                    if (imageUrl != null) {

                        catalogObj.put("imageUrl", imageUrl);

                    } else {

                        catalogObj.put("imageUrl",

                                "https://images.smartdukaan.com/uploads/campaigns/" + item.getCatalogItemId() + ".jpg");

                    }

                } catch (Exception e) {

                    try {

                        catalogObj.put("imageUrl",

                                "https://images.smartdukaan.com/uploads/campaigns/" + item.getCatalogItemId() + ".jpg");

                    } catch (Exception e2) {

                        e2.printStackTrace();

                    }

                }

                try {

                    catalogObj.put("rank", rankingList.indexOf(item.getCatalogItemId()));

                } catch (Exception e) {

                    // A very big number

                    e.printStackTrace();

                    catalogObj.put("rank", 50000000);

                }

                if (featureMap.containsKey(item.getCatalogItemId())

                        && featureMap.get(item.getCatalogItemId()) != null) {

                    catalogObj.put("feature", featureMap.get(item.getCatalogItemId()));

                }

                catalogObj.put("categoryId", CATEGORY_MASTER.contains(item.getCategoryId()) ? item.getCategoryId() : 6);

                if (item.getCategoryId() == 10006 || item.getCategoryId() == 10007) {

                    catalogObj.put("categoryId", 3);

                }

                catalogObj.put("subCategoryId", item.getCategoryId());

                catalogObj.put("create_timestamp",

                        tagListing.getCreatedTimestamp().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());

                if (catalogLabelMap.containsKey(item.getCatalogItemId())

                        && catalogLabelMap.get(item.getCatalogItemId()) != null) {

                    catalogObj.put("labels", catalogLabelMap.get(item.getCatalogItemId()));

                }

                catalogMap.put(item.getCatalogItemId(), catalogObj);

                catalogObj.put("avColor", new HashMap<>());

            }

            Map<String, Object> catalogObj = catalogMap.get(item.getCatalogItemId());

            Map<Integer, Integer> avColorMap = (Map<Integer, Integer>) catalogObj.get("avColor");

            if (availabilityItemMap.containsKey(String.valueOf(item.getId()))) {

                int itemStockColor = 0;

                for (Map.Entry<String, Map<String, Integer>> entry : availabilityItemMap

                        .get(String.valueOf(item.getId())).entrySet()) {

                    String warehouseId = entry.getKey();

                    Map<String, Integer> avMap = entry.getValue();

                    if (!avColorMap.containsKey(Integer.parseInt(warehouseId))) {

                        avColorMap.put(Integer.parseInt(warehouseId), 0);

                    }

                    int color = avColorMap.get(Integer.parseInt(warehouseId));

                    if (avMap.get("netAvailability") > 0) {

                        color = 2;

                    } else if (avMap.get("netPo") > 0) {

                        color = Math.max(color, 1);

                    } else {

                        color = Math.max(color, 0);

                    }

                    avColorMap.put(Integer.parseInt(warehouseId), color);

                    catalogObj.put("avColor", avColorMap);

                    if (color > itemStockColor) {

                        itemStockColor = color;

                    }

                }

                if (itemStockColor > 0 && itemStockColor > (Integer) catalogObj.get("stockColor")) {

                    catalogObj.put("stockColor", itemStockColor);

                }

            }

            Map<Integer, Object> itemColorMap = (Map<Integer, Object>) catalogObj.get("items");

            if (!itemColorMap.containsKey(item.getId())) {

                itemColorMap.put(item.getId(), new HashMap<String, Object>() {

                    {

                        put("color", item.getColor().replace("f_", ""));

                        put("tagPricing", new ArrayList<TagListing>());

                    }

                });

            }

            Map<String, Object> itemMap = (Map<String, Object>) itemColorMap.get(item.getId());

            List<TagListing> tagPricing = (List<TagListing>) itemMap.get("tagPricing");

            tagPricing.add(tagListing);

        }

        //logger.info("catalogObj {}", catalogMap);

        List<SuperCatalogModel> superCatalogMappingList = superCatalogMappingRepository.selectJoinedData();

        List<SuperCatalogMapping> superCatalogs = superCatalogMappingRepository.selectAll();

        List<SolrInputDocument> catalogSolrObjs = new ArrayList<>();

        for (Entry<Integer, Map<String, Object>> entry : catalogMap.entrySet()) {

            int catalogId = entry.getKey();

            Optional<SuperCatalogModel> superCatalogMapping = findMappingByCatalogId(superCatalogMappingList, catalogId, itemTagModels);

            Map<String, Object> catalogValMap = entry.getValue();

            Map<Integer, Object> itemsMap = (Map<Integer, Object>) catalogValMap.get("items");

            boolean active = false;

            // Map<Integer, Object> itemsMap = (Map<String, Object>)

            // catalogValMap.get("items");

            List<SolrInputDocument> itemObjs = new ArrayList<>();

            float mop = 0;

            float dp = 0;

            for (Entry<Integer, Object> itemEntry : itemsMap.entrySet()) {

                int itemId = itemEntry.getKey();

                Map<String, Object> itemMap = (Map<String, Object>) itemEntry.getValue();

                List<TagListing> tags = (List<TagListing>) itemMap.get("tagPricing");

                SolrInputDocument itemObj = new SolrInputDocument();

                for (TagListing tagListing : tags) {

                    active = active || tagListing.isActive();

                    itemObj.setField("id", "itemtag-" + itemId + "-" + tagListing.getTagId());

                    itemObj.setField("color_s", itemMap.get("color"));

                    itemObj.setField("itemId_i", itemId);

                    itemObj.setField("tagId_i", tagListing.getTagId());

                    itemObj.setField("mrp_f", tagListing.getMrp());

                    itemObj.setField("mop_f", tagListing.getMop());

                    itemObj.setField("sellingPrice_f", tagListing.getSellingPrice());

                    itemObj.setField("active_b", tagListing.isActive());

                    itemObj.setField("hot_deal_b", hotDealSet.contains(catalogId));

                    if (oemBrandByCatalog.containsKey(catalogId)) {

                        itemObj.setField("oem_brand_s", oemBrandByCatalog.get(catalogId));

                    }

                    mop = tagListing.getMop();

                    dp = tagListing.getSellingPrice();

                }

                itemObjs.add(itemObj);

            }

            SolrInputDocument catalogSolrObj = new SolrInputDocument();

            catalogSolrObj.setField("id", "catalog" + catalogId);

            catalogSolrObj.setField("rank_i", catalogValMap.get("rank"));

            catalogSolrObj.setField("title_s", catalogValMap.get("title"));

            catalogSolrObj.addChildDocuments(itemObjs);

            catalogSolrObj.setField("child_b", itemObjs.size() > 0);

            catalogSolrObj.setField("catalogId_i", catalogId);

            logger.info("superCatalogMapping1- {}", superCatalogMapping);

            if (superCatalogMapping.isPresent()) {

                int superCatalogId = superCatalogMapping.get().getSuperCatalogId();

                String superCatalogMappingVariants = findMappingVariantsByCatalogId(superCatalogs, superCatalogId, itemTagModels);

                catalogSolrObj.setField("superCatalog_i", superCatalogId);

                catalogSolrObj.setField("superCatalog_s", superCatalogMapping.get().getSuperCatalogName());

                catalogSolrObj.setField("superCatalogVariants_s", superCatalogMappingVariants);

            } else {

                catalogSolrObj.setField("superCatalog_i", 0);

                catalogSolrObj.setField("superCatalog_s", catalogValMap.get("title"));

                catalogSolrObj.setField("superCatalogVariants_s", "[]");

            }

            catalogSolrObj.setField("imageUrl_s",

                    catalogValMap.get("imageUrl").toString().replace("saholic", "smartdukaan"));

            catalogSolrObj.setField("feature_s", catalogValMap.get("feature"));

            catalogSolrObj.setField("brand_ss", catalogValMap.get("brand"));

            catalogSolrObj.setField("create_s", catalogValMap.get("create_timestamp"));

            catalogSolrObj.setField("categoryId_i", catalogValMap.get("categoryId"));

            catalogSolrObj.setField("subCategoryId_i", catalogValMap.get("subCategoryId"));

            catalogSolrObj.setField("label_ss", catalogValMap.get("labels"));

            catalogSolrObj.setField("mop_f", mop);

            catalogSolrObj.setField("dp_f", dp);

            catalogSolrObj.setField("hsncode_s", catalogValMap.get("hsnCode"));

            catalogSolrObj.setField("show_default_b", true);

            String[] brands = (String[]) catalogValMap.get("brand");

            for (String brand : brands) {

                if (brand.equals("FOC")) {

                    catalogSolrObj.setField("show_default_b", false);

                }

                if (brand.equals("Live Demo")) {

                    catalogSolrObj.setField("show_default_b", false);

                }

            }

            Map<Integer, Integer> avColorMap = (Map<Integer, Integer>) catalogValMap.get("avColor");

            for (Integer warehouseId : ProfitMandiConstants.WAREHOUSE_MAP.keySet()) {

                catalogSolrObj.setField("w" + warehouseId + "_i", 0);

            }

            for (Entry<Integer, Integer> avColorEntry : avColorMap.entrySet()) {

                int whId = avColorEntry.getKey();

                int color = avColorEntry.getValue();

                catalogSolrObj.setField("w" + whId + "_i", color);

            }

            List<Integer> similalarModels = null;

            if ((Integer) catalogValMap.get("categoryId") == 3) {

                float starPrice = mop - 500f;

                float endPrice = mop + 1000f;

                similalarModels = new ArrayList<>();

                for (Entry<Float, List<Integer>> floatListEntry : priceModelMap.entrySet()) {

                    float modelPrice = floatListEntry.getKey();

                    if (modelPrice >= starPrice && modelPrice <= endPrice) {

                        List<Integer> instock = floatListEntry.getValue().stream().filter(x->catalogMap.get(x).get("stockColor") != null && (Integer) (catalogMap.get(x).get("stockColor"))>0).collect(Collectors.toList());

                        //logger.info("Entry before - {}, after - {}", floatListEntry.getValue().size(), instock.size());

                        similalarModels.addAll(instock);

                    }

                    if (modelPrice > endPrice) break;

                }

                similalarModels.remove(Integer.valueOf(catalogId));

            }

            catalogSolrObj.setField("similarModels_ii", similalarModels == null ? Collections.EMPTY_LIST : similalarModels);

            catalogSolrObj.setField("active_b", active);

            if(eolSet.contains(catalogId)){

                catalogSolrObj.setField("eol_no_stock_b", true);

            }else {

                catalogSolrObj.setField("eol_no_stock_b", false);

            }

            catalogSolrObj.setField("flagship_b", flagshipSet.contains(catalogId));

            catalogSolrObj.setField("hot_deal_b", hotDealSet.contains(catalogId));

            if (oemBrandByCatalog.containsKey(catalogId)) {

                catalogSolrObj.setField("oem_brand_s", oemBrandByCatalog.get(catalogId));

            }

            catalogSolrObjs.add(catalogSolrObj);

        }

        String solrPath = "http://" + solrUrl + ":8984/solr/demo";

        SolrClient solr = new HttpSolrClient.Builder(solrPath).build();

        solr.deleteByQuery("*:*");

        solr.add(catalogSolrObjs);

        org.apache.solr.client.solrj.response.UpdateResponse updateResponse = solr.commit();

        logger.info("solr {}", updateResponse.getStatus());

    }

    public void pushData() throws Exception {

        this.populateTagItems();

    }

    /**

     * Updates a single catalog in Solr based on catalogId.

     * Used for real-time updates when TagListing changes.

     *

     * @param catalogId The catalog ID to update

     */

    public void updateSingleCatalog(int catalogId) throws Exception {

        logger.info("Updating single catalog in Solr: catalogId={}", catalogId);

        List<status> statuses = Arrays.asList(status.ACTIVE, status.PAUSED_BY_RISK, status.PARTIALLY_ACTIVE);

        boolean flagship = !modelFlagshipRepository

                .selectActiveFlagshipCatalogIds(Collections.singletonList(catalogId)).isEmpty();

        // Membership from the brand; oem_brand_s from the attributes row, which may be

        // absent on a freshly migrated SKU whose pills ops has not filled in yet.

        // Fully qualified: this package has its own Catalog class, which shadows the entity.

        com.spice.profitmandi.dao.entity.catalog.Catalog hotDealCatalog =

                catalogRepository.selectCatalogById(catalogId);

        boolean hotDeal = hotDealCatalog != null

                && ProfitMandiConstants.HOT_DEAL_BRAND.equals(hotDealCatalog.getBrand());

        com.spice.profitmandi.dao.entity.catalog.ModelHotDeal hotDealRow =

                hotDeal ? modelHotDealRepository.selectByCatalogItemId(catalogId) : null;

        String oemBrand = hotDealRow == null ? null : hotDealRow.getOemBrand();

        // Get all TagListings for this catalog

        List<ItemTagModel> itemTagModels = tagListingRepository.getItemTagByCatalogIdAndStatus(catalogId, statuses);

        if (itemTagModels.isEmpty()) {

            logger.warn("No active TagListings found for catalogId={}, removing from Solr", catalogId);

            String solrPath = "http://" + solrUrl + ":8984/solr/demo";

            SolrClient solr = new HttpSolrClient.Builder(solrPath).build();

            solr.deleteById("catalog" + catalogId);

            solr.commit();

            return;

        }

        // Get availability for items in this catalog

        Map<String, Map<String, Map<String, Integer>>> availabilityItemMap = getItemMap();

        // Get ranking info

        List<TagRanking> tagRankingList = tagRankingRepository.getAllTagRanking();

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

        Map<Integer, String> featureMap = new HashMap<>();

        for (TagRanking tagRanking : tagRankingList) {

            rankingList.add(tagRanking.getCatalogItemId());

            featureMap.put(tagRanking.getCatalogItemId(), tagRanking.getFeature());

        }

        // Get labels

        Map<Integer, List<String>> catalogLabelMap = this.getLabels();

        // Flagship is surfaced as a label (prepended so it shows first, ahead of all other flags)

        if (flagship) {

            catalogLabelMap.computeIfAbsent(catalogId, k -> new ArrayList<>()).add(0, "flagship");

        }

        // Get EOL info

        List<Integer> eolWithOutStock = catalogRepository.findAllWithEOLWithOutStock();

        List<Integer> eolWithBadQty = catalogRepository.findAllWithNoGoodStock();

        Set<Integer> eolSet = new HashSet<>(eolWithOutStock);

        eolSet.addAll(eolWithBadQty);

        // Build catalog document

        com.spice.profitmandi.dao.entity.catalog.Item firstItem = itemTagModels.get(0).getItem();

        TagListing firstTagListing = itemTagModels.get(0).getTagListing();

        Map<Integer, Integer> avColorMap = new HashMap<>();

        List<SolrInputDocument> itemObjs = new ArrayList<>();

        boolean active = false;

        float mop = 0;

        float dp = 0;

        int stockColor = 0;

        for (ItemTagModel itemTagModel : itemTagModels) {

            TagListing tagListing = itemTagModel.getTagListing();

            com.spice.profitmandi.dao.entity.catalog.Item item = itemTagModel.getItem();

            active = active || tagListing.isActive();

            // Build item document

            SolrInputDocument itemObj = new SolrInputDocument();

            itemObj.setField("id", "itemtag-" + item.getId() + "-" + tagListing.getTagId());

            itemObj.setField("color_s", item.getColor().replace("f_", ""));

            itemObj.setField("itemId_i", item.getId());

            itemObj.setField("tagId_i", tagListing.getTagId());

            itemObj.setField("mrp_f", tagListing.getMrp());

            itemObj.setField("mop_f", tagListing.getMop());

            itemObj.setField("sellingPrice_f", tagListing.getSellingPrice());

            itemObj.setField("active_b", tagListing.isActive());

            itemObj.setField("hot_deal_b", hotDeal);

            if (oemBrand != null) {

                itemObj.setField("oem_brand_s", oemBrand);

            }

            mop = tagListing.getMop();

            dp = tagListing.getSellingPrice();

            itemObjs.add(itemObj);

            // Calculate availability color

            if (availabilityItemMap.containsKey(String.valueOf(item.getId()))) {

                int itemStockColor = 0;

                for (Map.Entry<String, Map<String, Integer>> entry : availabilityItemMap

                        .get(String.valueOf(item.getId())).entrySet()) {

                    String warehouseId = entry.getKey();

                    Map<String, Integer> avMap = entry.getValue();

                    if (!avColorMap.containsKey(Integer.parseInt(warehouseId))) {

                        avColorMap.put(Integer.parseInt(warehouseId), 0);

                    }

                    int color = avColorMap.get(Integer.parseInt(warehouseId));

                    if (avMap.get("netAvailability") > 0) {

                        color = 2;

                    } else if (avMap.get("netPo") > 0) {

                        color = Math.max(color, 1);

                    } else {

                        color = Math.max(color, 0);

                    }

                    avColorMap.put(Integer.parseInt(warehouseId), color);

                    if (color > itemStockColor) {

                        itemStockColor = color;

                    }

                }

                if (itemStockColor > stockColor) {

                    stockColor = itemStockColor;

                }

            }

        }

        // Build catalog Solr document

        SolrInputDocument catalogSolrObj = new SolrInputDocument();

        catalogSolrObj.setField("id", "catalog" + catalogId);

        catalogSolrObj.setField("rank_i", rankingList.indexOf(catalogId));

        catalogSolrObj.setField("title_s", String.join(" ", Arrays.asList(

                firstItem.getBrand(), firstItem.getModelName(), firstItem.getModelNumber())

                .stream().filter(s -> s != null && !s.isEmpty()).collect(Collectors.toList())));

        catalogSolrObj.addChildDocuments(itemObjs);

        catalogSolrObj.setField("child_b", itemObjs.size() > 0);

        catalogSolrObj.setField("catalogId_i", catalogId);

        // Super catalog mapping

        List<SuperCatalogModel> superCatalogMappingList = superCatalogMappingRepository.selectJoinedData();

        List<SuperCatalogMapping> superCatalogs = superCatalogMappingRepository.selectAll();

        Optional<SuperCatalogModel> superCatalogMapping = findMappingByCatalogId(superCatalogMappingList, catalogId, itemTagModels);

        if (superCatalogMapping.isPresent()) {

            int superCatalogId = superCatalogMapping.get().getSuperCatalogId();

            String superCatalogMappingVariants = findMappingVariantsByCatalogId(superCatalogs, superCatalogId, itemTagModels);

            catalogSolrObj.setField("superCatalog_i", superCatalogId);

            catalogSolrObj.setField("superCatalog_s", superCatalogMapping.get().getSuperCatalogName());

            catalogSolrObj.setField("superCatalogVariants_s", superCatalogMappingVariants);

        } else {

            catalogSolrObj.setField("superCatalog_i", 0);

            catalogSolrObj.setField("superCatalog_s", catalogSolrObj.getFieldValue("title_s"));

            catalogSolrObj.setField("superCatalogVariants_s", "[]");

        }

        // Image URL

        try {

            String imageUrl = contentMongoClient.getEntityById(catalogId).getDefaultImageUrl();

            if (imageUrl != null) {

                catalogSolrObj.setField("imageUrl_s", imageUrl.replace("saholic", "smartdukaan"));

            } else {

                catalogSolrObj.setField("imageUrl_s",

                        "https://images.smartdukaan.com/uploads/campaigns/" + catalogId + ".jpg");

            }

        } catch (Exception e) {

            catalogSolrObj.setField("imageUrl_s",

                    "https://images.smartdukaan.com/uploads/campaigns/" + catalogId + ".jpg");

        }

        // Feature

        if (featureMap.containsKey(catalogId) && featureMap.get(catalogId) != null) {

            catalogSolrObj.setField("feature_s", featureMap.get(catalogId));

        } else {

            catalogSolrObj.setField("feature_s", "");

        }

        // Brand

        String[] brands = new String[]{firstItem.getBrand()};

        if (firstItem.getBrand().equals("Huawei") || firstItem.getBrand().equals("Honor")) {

            brands = new String[]{"Huawei", "Honor"};

        }

        if (firstItem.getBrand().equals("Mi") || firstItem.getBrand().equals("Xiaomi")

                || firstItem.getBrand().equals("Redmi")) {

            brands = new String[]{"Mi", "Xiaomi", "Redmi"};

        }

        catalogSolrObj.setField("brand_ss", brands);

        // Category

        catalogSolrObj.setField("categoryId_i", CATEGORY_MASTER.contains(firstItem.getCategoryId())

                ? ((firstItem.getCategoryId() == 10006 || firstItem.getCategoryId() == 10007) ? 3 : firstItem.getCategoryId()) : 6);

        catalogSolrObj.setField("subCategoryId_i", firstItem.getCategoryId());

        // Timestamps and labels

        catalogSolrObj.setField("create_s", firstTagListing.getCreatedTimestamp()

                .atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());

        if (catalogLabelMap.containsKey(catalogId) && catalogLabelMap.get(catalogId) != null) {

            catalogSolrObj.setField("label_ss", catalogLabelMap.get(catalogId));

        }

        catalogSolrObj.setField("mop_f", mop);

        catalogSolrObj.setField("dp_f", dp);

        catalogSolrObj.setField("hsncode_s", firstItem.getHsnCode());

        catalogSolrObj.setField("show_default_b", true);

        for (String brand : brands) {

            if (brand.equals("FOC") || brand.equals("Live Demo")) {

                catalogSolrObj.setField("show_default_b", false);

            }

        }

        // Warehouse availability

        for (Integer warehouseId : ProfitMandiConstants.WAREHOUSE_MAP.keySet()) {

            catalogSolrObj.setField("w" + warehouseId + "_i", 0);

        }

        for (Entry<Integer, Integer> avColorEntry : avColorMap.entrySet()) {

            int whId = avColorEntry.getKey();

            int color = avColorEntry.getValue();

            catalogSolrObj.setField("w" + whId + "_i", color);

        }

        // Similar models - simplified for single update

        catalogSolrObj.setField("similarModels_ii", Collections.EMPTY_LIST);

        catalogSolrObj.setField("active_b", active);

        catalogSolrObj.setField("eol_no_stock_b", eolSet.contains(catalogId));

        catalogSolrObj.setField("flagship_b", flagship);

        catalogSolrObj.setField("hot_deal_b", hotDeal);

        if (oemBrand != null) {

            catalogSolrObj.setField("oem_brand_s", oemBrand);

        }

        // Push to Solr

        String solrPath = "http://" + solrUrl + ":8984/solr/demo";

        SolrClient solr = new HttpSolrClient.Builder(solrPath).build();

        // Delete existing and add updated document

        solr.deleteById("catalog" + catalogId);

        solr.add(catalogSolrObj);

        solr.commit();

        logger.info("Successfully updated catalogId={} in Solr", catalogId);

    }

    public Optional<SuperCatalogModel> findMappingByCatalogId(List<SuperCatalogModel> superCatalogMappingList, int id, List<ItemTagModel> itemTagModels) {

        if (itemTagModels.stream().anyMatch(tag -> tag.getItem().getCatalogItemId() == id && tag.getTagListing().isActive())) {

            return superCatalogMappingList.stream().filter(mapping -> mapping.getCatalogId() == id).findFirst();

        }

        return Optional.empty();

    }

    public String findMappingVariantsByCatalogId(List<SuperCatalogMapping> superCatalogMappingList, int id, List<ItemTagModel> itemTagModels) throws JsonProcessingException {

        List<Map<String, Object>> variantList = new ArrayList<>();

        List<SuperCatalogMapping> matchingVariantNames = superCatalogMappingList.parallelStream()

                .filter(mapping -> mapping.getSuperCatalogId() == id)

                .collect(Collectors.toList());

        if (!matchingVariantNames.isEmpty()) {

            for (SuperCatalogMapping matchingVariant : matchingVariantNames) {

                logger.info("matchingVariantName: "+matchingVariant);

                ItemTagModel itemTagModel = itemTagModels.stream().filter(tag -> tag.getItem().getCatalogItemId() == matchingVariant.getCatalogId()).findFirst().orElse(null);

                   if (matchingVariant != null && matchingVariant.getVariantName() != null && itemTagModels.stream().anyMatch(tag -> tag.getItem().getCatalogItemId() == matchingVariant.getCatalogId() && tag.getTagListing().isActive())) {

                    Map<String, Object> variantObj = new HashMap<>();

                    variantObj.put("catalogId", matchingVariant.getCatalogId());

                    variantObj.put("variantName", matchingVariant.getVariantName());

                    variantObj.put("mrp", itemTagModel != null ? itemTagModel.getTagListing().getMrp() : 0.0);

                    variantObj.put("mop", itemTagModel != null ? itemTagModel.getTagListing().getMop() : 0.0);

                    variantObj.put("sellingPrice", itemTagModel != null ? itemTagModel.getTagListing().getSellingPrice() : 0.0);

                    variantList.add(variantObj);

                }

            }

        }

        if (variantList.isEmpty()) {

            return "[]";

        }

        variantList.sort((v1, v2) -> {

            int[] n1 = extractNumbers((String) v1.get("variantName"));

            int[] n2 = extractNumbers((String) v2.get("variantName"));

            int cmp = Integer.compare(n1[0], n2[0]);

            return cmp != 0 ? cmp : Integer.compare(n1[1], n2[1]);

        });

        ObjectMapper mapper = new ObjectMapper();

        return mapper.writeValueAsString(variantList);

    }

//(4GB 64GB)

    private int[] extractNumbers(String input) {

        String[] parts = input.split("\\s+");

        int[] result = new int[2];

        for (int i = 0; i < Math.min(parts.length, 2); i++) {

            String part = (parts[i].toUpperCase()).replace("(", "").replace(")", "");

            if (part.endsWith("GB")) {

                result[i] = Integer.parseInt(part.replace("GB", "").trim());

            } else if (part.endsWith("TB")) {

                result[i] = Integer.parseInt(part.replace("TB", "").trim()) * 1024;

            }

        }

        return result;

    }

}