com.github.triceo.robozonky.remote.Ratings Maven / Gradle / Ivy

Go to download
/*
 *
 *  * Copyright 2016 Lukáš Petrovický
 *  *
 *  * Licensed under the Apache License, Version 2.0 (the "License");
 *  * you may not use this file except in compliance with the License.
 *  *
 *  *      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 com.github.triceo.robozonky.remote;

import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.EnumSet;
import java.util.LinkedHashSet;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;

public class Ratings {

    public static Ratings valueOf(final String ratings) {
        // trim the surrounding []
        final String trimmed = ratings.trim();
        if (!trimmed.startsWith("[") || !trimmed.endsWith("]")) {
            throw new IllegalArgumentException("Expecting string in the format of [\"A\", \"B\"], got " + ratings);
        }
        if (trimmed.length() == 2) { // only contains []
            return Ratings.of();
        }
        final String[] parts = trimmed.substring(1, trimmed.length() - 1).split("\\Q,\\E");
        if (parts.length == 1 && parts[0].trim().length() == 0) { // only contains whitespace
            return Ratings.of();
        }
        // trim the parts and remove surrounding quotes
        final Collection strings = new LinkedHashSet<>(parts.length);
        for (final String part: parts) {
            final String trimmedPart = part.trim();
            if (!trimmedPart.startsWith("\"") && !trimmedPart.endsWith("\"") && trimmedPart.length() < 3) {
                throw new IllegalArgumentException("Expecting part of string to be quoted, got " + part);
            }
            strings.add(trimmedPart.substring(1, trimmedPart.length() - 1));
        }
        // convert string representations to actual instances
        final Collection converted = strings.stream().map(Rating::valueOf).collect(Collectors.toList());
        return Ratings.of(converted);
    }

    public static Ratings of(final Rating... ratings) {
        return Ratings.of(Arrays.asList(ratings));
    }

    public static Ratings of(final Collection ratings) {
        return new Ratings(ratings);
    }

    public static Ratings all() {
        return Ratings.of(Rating.values());
    }

    private final Set ratings;

    private Ratings(final Collection ratings) {
        this.ratings = ratings.isEmpty() ? Collections.emptySet() : EnumSet.copyOf(ratings);
    }

    public Set getRatings() {
        return Collections.unmodifiableSet(ratings);
    }

    @Override
    public String toString() {
        return ratings.stream().collect(Collectors.mapping(Rating::name, Collectors.joining("\", \"", "[\"", "\"]")));
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Ratings ratings1 = (Ratings) o;
        return Objects.equals(ratings, ratings1.ratings);
    }

    @Override
    public int hashCode() {
        return Objects.hash(ratings);
    }
}