/*
 * Copyright 2015 Open Networking Laboratory
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.onosproject.ovsdb.rfc.schema.type;

import static com.google.common.base.MoreObjects.toStringHelper;

import java.util.Objects;
import java.util.Set;

import com.google.common.collect.Sets;

/**
 * One of the strings "integer", "real", "boolean", "string", or "uuid",
 * representing the specified scalar type. Refer to RFC 7047 Section 3.2.
 */
public final class IntegerBaseType implements BaseType {
    private final int min;
    private final int max;
    private final Set<Integer> enums;

    /**
     * Constructs a IntegerBaseType object.
     */
    public IntegerBaseType() {
        this.min = Integer.MIN_VALUE;
        this.max = Integer.MAX_VALUE;
        this.enums = Sets.newHashSet();
    }

    /**
     * Constructs a IntegerBaseType object.
     * @param min min constraint
     * @param max max constraint
     * @param enums enums constraint
     */
    public IntegerBaseType(int min, int max, Set<Integer> enums) {
        this.min = min;
        this.max = max;
        this.enums = enums;
    }

    /**
     * Get min.
     * @return min
     */
    public int min() {
        return min;
    }

    /**
     * Get max.
     * @return max
     */
    public int max() {
        return max;
    }

    /**
     * Get enums.
     * @return enums
     */
    public Set<Integer> enums() {
        return enums;
    }

    @Override
    public int hashCode() {
        return Objects.hash(min, max, enums);
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj instanceof IntegerBaseType) {
            final IntegerBaseType other = (IntegerBaseType) obj;
            return Objects.equals(this.enums, other.enums)
                    && Objects.equals(this.min, other.min)
                    && Objects.equals(this.max, other.max);
        }
        return false;
    }

    @Override
    public String toString() {
        return toStringHelper(this).add("min", min).add("max", max)
                .add("enums", enums).toString();
    }
}
