All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.github.robozonky.common.remote.Select Maven / Gradle / Ivy

/*
 * Copyright 2018 The RoboZonky Project
 *
 * 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 com.github.robozonky.common.remote;

import java.time.LocalDate;
import java.time.OffsetDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.Consumer;
import java.util.stream.Collectors;
import java.util.stream.LongStream;
import java.util.stream.Stream;

/**
 * Allows to filter Zonky API requests based on the values of individual fields.
 * @see Zonky API docs.
 */
public class Select implements Consumer {

    private final Map> conditions = new HashMap<>(0);

    public static Select unrestricted() {
        return new Select();
    }

    private void addObjects(final String field, final String operation, final Object... value) {
        final String key = field + "__" + operation;
        conditions.compute(key, (k, v) -> {
            final String val = Stream.of(value).map(Object::toString)
                    .collect(Collectors.joining("\",\"", "[\"", "\"]"));
            final List result = (v == null) ? new ArrayList<>(1) : v;
            result.add(val);
            return result;
        });
    }

    private void addLongs(final String field, final String operation, final long... value) {
        final String key = operation == null ? field : field + "__" + operation;
        conditions.compute(key, (k, v) -> {
            final String val = LongStream.of(value).mapToObj(String::valueOf)
                    .collect(Collectors.joining("\",\"", "[\"", "\"]"));
            final List result = (v == null) ? new ArrayList<>(1) : v;
            result.add(val);
            return result;
        });
    }

    private void addObject(final String field, final String operation, final Object value) {
        final String key = operation == null ? field : field + "__" + operation;
        conditions.compute(key, (k, v) -> {
            final List result = (v == null) ? new ArrayList<>(1) : v;
            result.add(value);
            return result;
        });
    }

    private void addLong(final String field, final String operation, final long value) {
        addObject(field, operation, String.valueOf(value));
    }

    private void addLocalDate(final String field, final String operation, final LocalDate value) {
        addObject(field, operation, DateTimeFormatter.ISO_DATE.format(value));
    }

    private void addOffsetDateTime(final String field, final String operation, final OffsetDateTime value) {
        addObject(field, operation, DateTimeFormatter.ISO_OFFSET_DATE_TIME.format(value));
    }

    public Select contains(final String field, final String value) {
        addObject(field, "contains", value);
        return this;
    }

    public Select containsCaseInsensitive(final String field, final String value) {
        addObject(field, "icontains", value);
        return this;
    }

    public Select startsWith(final String field, final String value) {
        addObject(field, "startswith", value);
        return this;
    }

    public Select startsWithCaseInsensitive(final String field, final String value) {
        addObject(field, "istartswith", value);
        return this;
    }

    public Select endsWith(final String field, final String value) {
        addObject(field, "endswith", value);
        return this;
    }

    public Select endsWithCaseInsensitive(final String field, final String value) {
        addObject(field, "iendswith", value);
        return this;
    }

    public Select in(final String field, final String... values) {
        addObjects(field, "in", (Object[]) values);
        return this;
    }

    public Select containsAll(final String field, final String... values) {
        addObjects(field, "containsall", (Object[]) values);
        return this;
    }

    public Select containsAny(final String field, final String... values) {
        addObjects(field, "containsany", (Object[]) values);
        return this;
    }

    public Select in(final String field, final long... values) {
        addLongs(field, "in", values);
        return this;
    }

    public Select containsAll(final String field, final long... values) {
        addLongs(field, "containsall", values);
        return this;
    }

    public Select containsAny(final String field, final long... values) {
        addLongs(field, "containsany", values);
        return this;
    }

    public Select equals(final String field, final Object value) {
        addObject(field, "eq", value);
        return this;
    }

    public Select equalsPlain(final String field, final Object value) {
        addObject(field, null, value);
        return this;
    }

    public Select notEquals(final String field, final Object value) {
        addObject(field, "noteq", value);
        return this;
    }

    public Select greaterThan(final String field, final long value) {
        addLong(field, "gt", value);
        return this;
    }

    public Select greaterThan(final String field, final LocalDate value) {
        addLocalDate(field, "gt", value);
        return this;
    }

    public Select greaterThan(final String field, final OffsetDateTime value) {
        addOffsetDateTime(field, "gt", value);
        return this;
    }

    public Select greaterThanOrEquals(final String field, final long value) {
        addLong(field, "gte", value);
        return this;
    }

    public Select greaterThanOrEquals(final String field, final LocalDate value) {
        addLocalDate(field, "gte", value);
        return this;
    }

    public Select greaterThanOrEquals(final String field, final OffsetDateTime value) {
        addOffsetDateTime(field, "gte", value);
        return this;
    }

    public Select greaterThanOrNull(final String field, final long value) {
        addLong(field, "gteornull", value);
        return this;
    }

    public Select lessThan(final String field, final long value) {
        addLong(field, "lt", value);
        return this;
    }

    public Select lessThan(final String field, final LocalDate value) {
        addLocalDate(field, "lt", value);
        return this;
    }

    public Select lessThan(final String field, final OffsetDateTime value) {
        addOffsetDateTime(field, "lt", value);
        return this;
    }

    public Select lessThanOrEquals(final String field, final long value) {
        addLong(field, "lte", value);
        return this;
    }

    public Select lessThanOrEquals(final String field, final LocalDate value) {
        addLocalDate(field, "lte", value);
        return this;
    }

    public Select lessThanOrEquals(final String field, final OffsetDateTime value) {
        addOffsetDateTime(field, "lte", value);
        return this;
    }

    public Select lessThanOrNull(final String field, final long value) {
        addLong(field, "lteornull", value);
        return this;
    }

    @Override
    public void accept(final RoboZonkyFilter roboZonkyFilter) {
        conditions.forEach((k, v) -> v.forEach(r -> roboZonkyFilter.setQueryParam(k, r)));
    }

    @Override
    public boolean equals(final Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || !Objects.equals(getClass(), o.getClass())) {
            return false;
        }
        final Select select = (Select) o;
        return Objects.equals(conditions, select.conditions);
    }

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

    @Override
    public String toString() {
        return "Select{" +
                "conditions=" + conditions +
                '}';
    }
}