io.github.simple4tests.webdriver.utils.Groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of webdriver-overload Show documentation

WebDriver overload

The newest version!

package io.github.simple4tests.webdriver.utils;

import groovy.lang.Binding;
import groovy.lang.GroovyShell;
import org.codehaus.groovy.control.CompilerConfiguration;
import org.codehaus.groovy.control.customizers.ImportCustomizer;

import java.util.List;
import java.util.Objects;

public class Groovy {

    private static final ThreadLocal localGroovyShell = new ThreadLocal<>();

    private static List currentImports;

    public static final List IMPORT_EMPTY = List.of();
    public static final List IMPORT_HAMCREST_MATCHERS = List.of("org.hamcrest.MatcherAssert", "org.hamcrest.Matchers");

    private static void resetShell() {
        localGroovyShell.set(null);
    }

    private static GroovyShell initShell(final List classNames) {
        currentImports = classNames;

        final ImportCustomizer importCustomizer = new ImportCustomizer();
        for (String className : classNames) {
            importCustomizer.addStaticStars(className);
        }
        final CompilerConfiguration configuration = new CompilerConfiguration();
        configuration.addCompilationCustomizers(importCustomizer);

        return new GroovyShell(Groovy.class.getClassLoader(), new Binding(), configuration);
    }

    public static GroovyShell getShell() {
        return getShell(IMPORT_EMPTY);
    }

    public static GroovyShell getShell(final List classNames) {
        if (Objects.isNull(localGroovyShell.get())) {
            localGroovyShell.set(initShell(classNames));
        } else if (!currentImports.stream().sorted().equals(classNames.stream().sorted())) {
            resetShell();
            localGroovyShell.set(initShell(classNames));
        }
        return localGroovyShell.get();
    }
}