com.tinkerpop.gremlin.groovy.EmptyImportCustomizerProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gremlin-groovy Show documentation
The newest version!
package com.tinkerpop.gremlin.groovy;

import org.codehaus.groovy.control.customizers.CompilationCustomizer;
import org.codehaus.groovy.control.customizers.ImportCustomizer;

import java.util.Collections;
import java.util.HashSet;
import java.util.Set;

/**
 * This {@link com.tinkerpop.gremlin.groovy.ImportCustomizerProvider} is empty and comes with no pre-defined imports
 * at all.
 *
 * @author Stephen Mallette (http://stephen.genoprime.com)
 */
public class EmptyImportCustomizerProvider extends DefaultImportCustomizerProvider {

    /**
     * Utilizes imports defined by the supplied arguments.
     */
    public EmptyImportCustomizerProvider(final Set extraImports, final Set extraStaticImports) {
        this.extraStaticImports.addAll(extraStaticImports);
        this.extraImports.addAll(extraImports);
    }

    /**
     * Utilizes imports defined by the supplied arguments.
     *
     * @param baseCustomizer Imports from this customizer get added to the new one.
     */
    public EmptyImportCustomizerProvider(final ImportCustomizerProvider baseCustomizer,
                                         final Set extraImports, final Set extraStaticImports) {
        this(extraImports, extraStaticImports);
        this.extraImports.addAll(baseCustomizer.getImports());
        this.extraImports.addAll(baseCustomizer.getExtraImports());
        this.extraStaticImports.addAll(baseCustomizer.getStaticImports());
        this.extraStaticImports.addAll(baseCustomizer.getExtraStaticImports());
    }

    @Override
    public CompilationCustomizer getCompilationCustomizer() {
        final ImportCustomizer ic = new ImportCustomizer();

        processImports(ic, extraImports);
        processStaticImports(ic, extraStaticImports);

        return ic;
    }

    @Override
    public Set getImports() {
        return Collections.emptySet();
    }

    @Override
    public Set getStaticImports() {
        return Collections.emptySet();
    }

    @Override
    public Set getAllImports() {
        final Set allImports = new HashSet<>();
        allImports.addAll(extraImports);
        allImports.addAll(extraStaticImports);

        return allImports;
    }
}