// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.prelude;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import com.yahoo.search.config.IndexInfoConfig;
import com.yahoo.container.QrSearchersConfig;

/**
 * Parameter class used for construction IndexFacts.
 *
 * @author Steinar Knutsen
 * @author bratseth
 */
public final class IndexModel {

    // Copied from MasterClustersInfoUpdater. It's a temporary workaround for IndexFacts
    private Map<String, List<String>> masterClusters;
    private Map<String, SearchDefinition> searchDefinitions;
    private SearchDefinition unionSearchDefinition;

    /**
     * Create an index model.
     */
    public IndexModel(Map<String, List<String>> masterClusters, Collection<SearchDefinition> searchDefinitions) {
        this.masterClusters = masterClusters;
        this.searchDefinitions = searchDefinitions.stream().collect(Collectors.toMap(sd -> sd.getName(), sd -> sd));
        this.unionSearchDefinition = unionOf(searchDefinitions);
    }

    /**
     * Use IndexModel as a pure wrapper for the parameters given.
     */
    // TODO: Deprecate on Vespa 7 and remove on Vespa 8
    public IndexModel(Map<String, List<String>> masterClusters,
                      Map<String, SearchDefinition> searchDefinitions,
                      SearchDefinition unionSearchDefinition) {
        this.masterClusters = masterClusters;
        this.searchDefinitions = searchDefinitions;
        this.unionSearchDefinition = unionSearchDefinition;
    }

    public IndexModel(IndexInfoConfig indexInfo, QrSearchersConfig clusters) {
        this(indexInfo, toClusters(clusters));
    }

    public IndexModel(IndexInfoConfig indexInfo, Map<String, List<String>> clusters) {
        if (indexInfo != null) {
            searchDefinitions = toSearchDefinitions(indexInfo);
            unionSearchDefinition = unionOf(searchDefinitions.values());
        } else {
            searchDefinitions = null;
            unionSearchDefinition = null;
        }
        this.masterClusters = clusters;
    }

    private static Map<String, List<String>> toClusters(QrSearchersConfig config) {
        if (config == null) return new HashMap<>();

        Map<String, List<String>> clusters = new HashMap<>();
        for (int i = 0; i < config.searchcluster().size(); ++i) {
            List<String> docTypes = new ArrayList<>();
            String clusterName = config.searchcluster(i).name();
            for (int j = 0; j < config.searchcluster(i).searchdef().size(); ++j) {
                docTypes.add(config.searchcluster(i).searchdef(j));
            }
            clusters.put(clusterName, docTypes);
        }
        return clusters;
    }

    @SuppressWarnings("deprecation")
    private static Map<String, SearchDefinition> toSearchDefinitions(IndexInfoConfig c) {
        Map<String, SearchDefinition> searchDefinitions = new HashMap<>();

        for (Iterator<IndexInfoConfig.Indexinfo> i = c.indexinfo().iterator(); i.hasNext();) {
            IndexInfoConfig.Indexinfo info = i.next();
            SearchDefinition sd = new SearchDefinition(info.name());
            for (Iterator<IndexInfoConfig.Indexinfo.Command> j = info.command().iterator(); j.hasNext();) {
                IndexInfoConfig.Indexinfo.Command command = j.next();
                sd.addCommand(command.indexname(),command.command());
            }
            sd.fillMatchGroups();
            searchDefinitions.put(info.name(), sd);
        }

        for (IndexInfoConfig.Indexinfo info : c.indexinfo()) {
            SearchDefinition sd = searchDefinitions.get(info.name());
            for (IndexInfoConfig.Indexinfo.Alias alias : info.alias()) {
                String aliasString = alias.alias();
                String indexString = alias.indexname();

                sd.addAlias(aliasString, indexString);
            }
        }
        return searchDefinitions;
    }

    @SuppressWarnings("deprecation")
    private SearchDefinition unionOf(Collection<SearchDefinition> searchDefinitions) {
        SearchDefinition union = new SearchDefinition(IndexFacts.unionName);

        for (SearchDefinition sd : searchDefinitions) {
            for (Index index : sd.indices().values()) {
                union.getOrCreateIndex(index.getName());
                for (String command : index.allCommands())
                    union.addCommand(index.getName(), command);
                for (String alias : index.aliases())
                    union.addAlias(alias, index.getName());
            }

        }
        union.fillMatchGroups();
        return union;
    }

    public Map<String, List<String>> getMasterClusters() { return masterClusters; }

    public Map<String, SearchDefinition> getSearchDefinitions() { return searchDefinitions; }

    // TODO: Deprecate on Vespa 7 and make package scope on Vespa 8
    public SearchDefinition getUnionSearchDefinition() { return unionSearchDefinition; }

}