com.enonic.xp.node.NodeComparisons Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core-api Show documentation
core-api
The newest version!
package com.enonic.xp.node;

import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import com.google.common.collect.ImmutableSet;

import com.enonic.xp.annotation.PublicApi;
import com.enonic.xp.content.CompareStatus;

@PublicApi
public class NodeComparisons
    implements Iterable
{
    private final Map comparisonMap;

    private NodeComparisons( final Builder builder )
    {
        this.comparisonMap = builder.nodeIdNodeComparisonMap;
    }

    public static Builder create()
    {
        return new Builder();
    }

    public NodeComparison get( final NodeId nodeId )
    {
        return this.comparisonMap.get( nodeId );
    }

    public NodeComparison getBySourcePath( final NodePath nodePath )
    {
        return this.comparisonMap.values().stream().
            filter( comparison -> comparison.getSourcePath().equals( nodePath ) ).
            findFirst().
            get();
    }

    @Override
    public Iterator iterator()
    {
        return this.comparisonMap.values().iterator();
    }

    public int getSize()
    {
        return this.comparisonMap.size();
    }

    public NodeIds getNodeIds()
    {
        return NodeIds.from( this.comparisonMap.values().stream().
            map( NodeComparison::getNodeId ).
            collect( ImmutableSet.toImmutableSet() ) );
    }

    public NodePaths getSourcePaths()
    {
        return NodePaths.from(
            this.comparisonMap.values().stream().map( NodeComparison::getSourcePath ).collect( ImmutableSet.toImmutableSet() ) );
    }

    public Set getWithStatus( final CompareStatus status )
    {
        return this.comparisonMap.values()
            .stream()
            .filter( nodeComparison -> nodeComparison.getCompareStatus() == status )
            .collect( Collectors.toSet() );
    }

    public Collection getComparisons()
    {
        return this.comparisonMap.values();
    }

    public static final class Builder
    {
        private final Map nodeIdNodeComparisonMap = new HashMap<>();

        private Builder()
        {
        }

        public Builder add( final NodeComparison nodeComparison )
        {
            this.nodeIdNodeComparisonMap.put( nodeComparison.getNodeId(), nodeComparison );
            return this;
        }

        public Builder addAll( final Collection nodeComparisons )
        {
            nodeComparisons.
                forEach( comparison -> this.nodeIdNodeComparisonMap.put( comparison.getNodeId(), comparison ) );

            return this;
        }

        public NodeComparisons build()
        {
            return new NodeComparisons( this );
        }
    }
}