org.allurefw.report.entity.WithLabels Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of plugin-api Show documentation
The newest version!
package org.allurefw.report.entity;

import org.allurefw.report.ModelUtils;

import java.util.List;
import java.util.Optional;
import java.util.stream.Collector;
import java.util.stream.Collectors;

import static org.allurefw.report.ModelUtils.createLabel;

/**
 * @author Dmitry Baev [email protected]
 *         Date: 31.01.16
 */
public interface WithLabels {

    List getLabels();

    void setLabels(List labels);

    default  T findAll(LabelName name, Collector collector) {
        return findAll(name.value(), collector);
    }

    default  T findAll(String name, Collector collector) {
        return getLabels().stream()
                .filter(label -> name.equals(label.getName()))
                .map(Label::getValue)
                .collect(collector);
    }

    default List findAll(LabelName name) {
        return findAll(name, Collectors.toList());
    }

    default List findAll(String name) {
        return findAll(name, Collectors.toList());
    }

    default Optional findOne(LabelName name) {
        return findOne(name.value());
    }

    default Optional findOne(String name) {
        return getLabels().stream()
                .filter(label -> name.equals(label.getName()))
                .map(Label::getValue)
                .findAny();
    }

    default void addLabelIfNotExists(LabelName name, String value) {
        addLabelIfNotExists(name.value(), value);
    }

    default void addLabelIfNotExists(String name, String value) {
        if (value == null) {
            return;
        }
        Optional any = getLabels().stream()
                .map(Label::getName)
                .filter(name::equals)
                .findAny();
        if (!any.isPresent()) {
            addLabel(name, value);
        }
    }

    default void addLabel(String name, String value) {
        getLabels().add(createLabel(name, value));
    }
}