com.tinkerpop.gremlin.groovy.SecurityCustomizerProvider 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.kohsuke.groovy.sandbox.GroovyInterceptor;
import org.kohsuke.groovy.sandbox.SandboxTransformer;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author Stephen Mallette (http://stephen.genoprime.com)
 */
public class SecurityCustomizerProvider implements CompilerCustomizerProvider {

    private final List interceptors;

    public SecurityCustomizerProvider() {
        this.interceptors = new ArrayList<>();
    }

    public SecurityCustomizerProvider(final GroovyInterceptor... interceptors) {
        this.interceptors = Arrays.asList(interceptors);
    }

    public void addInterceptor(final GroovyInterceptor interceptor) {
        this.interceptors.add(interceptor);
    }

    public void removeInterceptor(final GroovyInterceptor interceptor) {
        this.interceptors.remove(interceptor);
    }

    @Override
    public CompilationCustomizer getCompilationCustomizer() {
        return new SandboxTransformer();
    }

    public void registerInterceptors() {
        interceptors.forEach(GroovyInterceptor::register);
    }

    public void unregisterInterceptors() {
        interceptors.forEach(GroovyInterceptor::unregister);
    }


}