com.enonic.xp.index.PatternIndexConfigDocument 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.index;

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.SortedSet;
import java.util.TreeSet;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSortedSet;

import com.enonic.xp.annotation.PublicApi;
import com.enonic.xp.data.PropertyPath;
import com.enonic.xp.util.GlobPatternMatcher;

import static com.google.common.base.Strings.nullToEmpty;

@PublicApi
public final class PatternIndexConfigDocument
    extends AbstractIndexConfigDocument
{
    private final ImmutableSortedSet pathIndexConfigs;

    private final Map pathIndexConfigMap;

    private final IndexConfig defaultConfig;

    private final AllTextIndexConfig allTextConfig;

    public IndexConfig getDefaultConfig()
    {
        return defaultConfig;
    }

    private PatternIndexConfigDocument( final Builder builder )
    {
        super( builder );
        this.pathIndexConfigs = ImmutableSortedSet.copyOf( builder.pathIndexConfigs );
        this.pathIndexConfigMap = ImmutableMap.copyOf( builder.stringPathIndexConfigMap );
        this.defaultConfig = builder.defaultConfig;
        this.allTextConfig = builder.allTextIndexConfig.build();
    }

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

    public static Builder create( final PatternIndexConfigDocument source )
    {
        return new Builder( source );
    }

    public ImmutableSortedSet getPathIndexConfigs()
    {
        return pathIndexConfigs;
    }

    @Override
    public IndexConfig getConfigForPath( final PropertyPath dataPath )
    {
        return doGetConfigForPath( dataPath.resetAllIndexesTo( 0 ).toString() );
    }

    @Override
    public IndexConfig getConfigForPath( final IndexPath indexPath )
    {
        return doGetConfigForPath( indexPath.toString() );
    }

    private IndexConfig doGetConfigForPath( final String path )
    {
        final PathIndexConfig exactMatch = pathIndexConfigMap.get( path.toLowerCase() );

        if ( exactMatch != null )
        {
            return exactMatch.getIndexConfig();
        }

        for ( final PathIndexConfig pathIndexConfig : pathIndexConfigs )
        {
            if ( GlobPatternMatcher.match( pathIndexConfig.getPath().toString(), path, "." ) )
            {
                return pathIndexConfig.getIndexConfig();
            }

            if ( pathIndexConfig.matches( path ) )
            {
                return pathIndexConfig.getIndexConfig();
            }
        }

        return defaultConfig;
    }

    @Override
    public AllTextIndexConfig getAllTextConfig()
    {
        return allTextConfig;
    }

    @Override
    public boolean equals( final Object o )
    {
        if ( this == o )
        {
            return true;
        }
        if ( o == null || getClass() != o.getClass() )
        {
            return false;
        }
        if ( !super.equals( o ) )
        {
            return false;
        }
        final PatternIndexConfigDocument that = (PatternIndexConfigDocument) o;
        return Objects.equals( pathIndexConfigs, that.pathIndexConfigs ) &&
            Objects.equals( defaultConfig, that.defaultConfig ) && Objects.equals( allTextConfig, that.allTextConfig );
    }

    @Override
    public int hashCode()
    {
        return Objects.hash( super.hashCode(), pathIndexConfigs, defaultConfig, allTextConfig );
    }

    public static final class Builder
        extends AbstractIndexConfigDocument.Builder
    {
        private SortedSet pathIndexConfigs = new TreeSet<>();

        private Map stringPathIndexConfigMap = new HashMap<>();

        private IndexConfig defaultConfig = IndexConfig.BY_TYPE;

        private AllTextIndexConfig.Builder allTextIndexConfig = AllTextIndexConfig.create();

        private Builder()
        {
        }

        private Builder( final PatternIndexConfigDocument source )
        {
            this.pathIndexConfigs = new TreeSet<>( source.pathIndexConfigs );
            this.stringPathIndexConfigMap = new HashMap<>( source.pathIndexConfigMap );
            this.defaultConfig = IndexConfig.create( source.defaultConfig ).build();
            this.allTextIndexConfig = AllTextIndexConfig.create( source.allTextConfig );
        }

        public Builder add( final String path, final IndexConfig indexConfig )
        {
            add( PathIndexConfig.create().
                path( PropertyPath.from( path ) ).
                indexConfig( indexConfig ).
                build() );

            return this;
        }

        public Builder add( final PropertyPath path, final IndexConfig indexConfig )
        {
            add( PathIndexConfig.create().
                path( path ).
                indexConfig( indexConfig ).
                build() );

            return this;
        }

        public Builder add( final PathIndexConfig pathIndexConfig )
        {
            this.pathIndexConfigs.add( pathIndexConfig );
            this.stringPathIndexConfigMap.put( pathIndexConfig.getPath().toString().toLowerCase(), pathIndexConfig );

            return this;
        }

        public Builder remove( final PathIndexConfig pathIndexConfig )
        {
            this.pathIndexConfigs.remove( pathIndexConfig );
            this.stringPathIndexConfigMap.remove( pathIndexConfig.getPath().toString().toLowerCase() );

            return this;
        }

        public Builder addPattern( final PathIndexConfig pathIndexConfig )
        {
            this.pathIndexConfigs.add( pathIndexConfig );
            this.stringPathIndexConfigMap.put( pathIndexConfig.getPath().resetAllIndexesTo( 0 ).toString().toLowerCase(), pathIndexConfig );
            return this;
        }

        public Builder defaultConfig( IndexConfig defaultConfig )
        {
            this.defaultConfig = defaultConfig;
            return this;
        }

        public Builder addAllTextConfigLanguage( final String language )
        {
            if ( !nullToEmpty( language ).isBlank() )
            {
                this.allTextIndexConfig.addLanguage( language );
            }
            return this;
        }

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