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

import com.yahoo.schema.document.Attribute;
import com.yahoo.document.DataType;
import com.yahoo.schema.document.SDField;
import com.yahoo.document.WeightedSetDataType;

/**
 * @author Einar M R Rosenvinge
 */
public class WeightedSetOperation implements FieldOperation {

    private Boolean createIfNonExistent;
    private Boolean removeIfZero;

    public Boolean getCreateIfNonExistent() {
        return createIfNonExistent;
    }

    public void setCreateIfNonExistent(Boolean createIfNonExistent) {
        this.createIfNonExistent = createIfNonExistent;
    }

    public Boolean getRemoveIfZero() {
        return removeIfZero;
    }

    public void setRemoveIfZero(Boolean removeIfZero) {
        this.removeIfZero = removeIfZero;
    }

    public void apply(SDField field) {
        WeightedSetDataType ctype = (WeightedSetDataType) field.getDataType();

        if (createIfNonExistent != null) {
            field.setDataType(DataType.getWeightedSet(ctype.getNestedType(), createIfNonExistent,
                                                      ctype.removeIfZero()));
        }

        ctype = (WeightedSetDataType) field.getDataType();
        if (removeIfZero != null) {
            field.setDataType(DataType.getWeightedSet(ctype.getNestedType(),
                                                      ctype.createIfNonExistent(), removeIfZero));
        }

        ctype = (WeightedSetDataType) field.getDataType();
        for (Object o : field.getAttributes().values()) {
            Attribute attribute = (Attribute) o;
            attribute.setRemoveIfZero(ctype.removeIfZero());
            attribute.setCreateIfNonExistent(ctype.createIfNonExistent());
        }
    }

    @Override
    public int compareTo(FieldOperation other) {
        // this operation should be executed first because it modifies the type of weighted sets, and other
        // operation depends on the type of the weighted set
        return -1;
    }
    
    @Override
    public String toString() {
        return "WeightedSetOperation{" +
               "createIfNonExistent=" + createIfNonExistent +
               ", removeIfZero=" + removeIfZero +
               '}';
    }

}