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

import com.yahoo.schema.Schema;
import com.yahoo.vespa.documentmodel.SummaryField;
import com.yahoo.vespa.documentmodel.SummaryTransform;
import com.yahoo.vespa.config.search.summary.JuniperrcConfig;

import java.util.Set;

/**
 * Generated juniperrc-config for controlling juniper.
 *
 * @author Simon Thoresen Hult
 */
public class Juniperrc extends Derived implements JuniperrcConfig.Producer {

    private static final int Mb = 1024 * 1024;

    /** List of all fields that should be bolded. */
    private final Set<String> boldingFields = new java.util.LinkedHashSet<>();

    /**
     * Constructs a new juniper rc instance for a given search object.
     * This will derive the configuration automatically,
     * so there is no need to call {@link #derive(Schema)}.
     *
     * @param schema the search model to use for deriving
     */
    public Juniperrc(Schema schema) {
        derive(schema);
    }

    @Override
    protected void derive(Schema schema) {
        super.derive(schema);
        for (SummaryField summaryField : schema.getUniqueNamedSummaryFields().values()) {
            if (summaryField.getTransform() == SummaryTransform.BOLDED) {
                boldingFields.add(summaryField.getName());
            }
        }
    }

    @Override
    protected String getDerivedName() { return "juniperrc"; }

    @Override
    public void getConfig(JuniperrcConfig.Builder builder) {
        if (boldingFields.size() != 0) {
            builder.prefix(true);
            for (String name : boldingFields) {
                builder.override(new JuniperrcConfig.Override.Builder()
                    .fieldname(name)
                    .length(64*Mb)
                    .max_matches(1)
                    .min_length(8192)
                    .surround_max(64*Mb));
            }
        }
    }

}