001// --------------------------------------------------------------------------------
002// Copyright 2002-2024 Echo Three, LLC
003//
004// Licensed under the Apache License, Version 2.0 (the "License");
005// you may not use this file except in compliance with the License.
006// You may obtain a copy of the License at
007//
008//     http://www.apache.org/licenses/LICENSE-2.0
009//
010// Unless required by applicable law or agreed to in writing, software
011// distributed under the License is distributed on an "AS IS" BASIS,
012// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013// See the License for the specific language governing permissions and
014// limitations under the License.
015// --------------------------------------------------------------------------------
016
017package com.echothree.model.control.warehouse.server.indexer;
018
019import com.echothree.model.control.index.common.IndexFields;
020import com.echothree.model.control.index.server.indexer.FieldTypes;
021import com.echothree.model.control.party.common.PartyTypes;
022import com.echothree.model.control.party.server.indexer.PartyIndexer;
023import com.echothree.model.control.warehouse.server.analysis.WarehouseAnalyzer;
024import com.echothree.model.control.warehouse.server.control.WarehouseControl;
025import com.echothree.model.data.index.server.entity.Index;
026import com.echothree.model.data.party.server.entity.Party;
027import com.echothree.util.server.message.ExecutionErrorAccumulator;
028import com.echothree.util.server.persistence.Session;
029import org.apache.lucene.analysis.Analyzer;
030import org.apache.lucene.document.Document;
031import org.apache.lucene.document.Field;
032
033public class WarehouseIndexer
034        extends PartyIndexer {
035
036    WarehouseControl warehouseControl = Session.getModelController(WarehouseControl.class);
037
038    /** Creates a new instance of WarehouseIndexer */
039    public WarehouseIndexer(final ExecutionErrorAccumulator eea, final Index index) {
040        super(eea, index, PartyTypes.WAREHOUSE.name(), IndexFields.warehouseName.name());
041    }
042
043    @Override
044    protected Analyzer getAnalyzer() {
045        return new WarehouseAnalyzer(eea, language, entityType, entityAliasTypes, entityAttributes, tagScopes, partyType, entityNameIndexField);
046    }
047
048    @Override
049    protected String getEntityNameFromParty(final Party party) {
050        var warehouse = warehouseControl.getWarehouse(party);
051
052        return warehouse == null ? null : warehouse.getWarehouseName();
053    }
054
055    @Override
056    protected void addPartyFieldsToDocument(final Party party, final String entityName, final Document document) {
057        super.addPartyFieldsToDocument(party, entityName, document);
058
059        var warehouse = warehouseControl.getWarehouse(party);
060        if(warehouse != null) {
061            document.add(new Field(IndexFields.warehouseTypeName.name(), warehouse.getWarehouseType().getLastDetail().getWarehouseTypeName(), FieldTypes.NOT_STORED_TOKENIZED));
062        }
063    }
064
065}