project.getPluginManager().withPlugin("java", p -> {
            // We want to get any compilation error before running the pre-commit checks.
            project.getExtensions().getByType(JavaPluginExtension.class).getSourceSets().all(sourceSet ->
                    task.configure(t -> t.shouldRunAfter(sourceSet.getClassesTaskName()))
            );
        });
    }

    @PathSensitive(PathSensitivity.RELATIVE)
    @SkipWhenEmpty
    public FileCollection getClassDirectories() {
        return getProject().getExtensions().getByType(JavaPluginExtension.class)
            .getSourceSets()
            .stream()
            // Don't pick up all source sets like the java9 ones as logger-check doesn't support the class format
            .filter(

            project.getPlugins().withType(JavaBasePlugin.class, javaBasePlugin -> {
                final SourceSetContainer sourceSets = project.getExtensions().getByType(JavaPluginExtension.class).getSourceSets();
                licenseHeadersTask.getSourceFolders()
                    .addAll(providerFactory.provider(() -> sourceSets.stream().map(s -> s.getAllJava()).collect(Collectors.toList())));
            });
        });

                );
        project.getPluginManager().withPlugin("java", p -> {
            // run compilation as part of precommit
            project.getExtensions().getByType(JavaPluginExtension.class).getSourceSets().all(sourceSet ->
                    precommit.configure(t -> t.shouldRunAfter(sourceSet.getClassesTaskName()))
            );
            // make sure tests run after all precommit tasks

            }
        };
    }

    public static SourceSetContainer getJavaSourceSets(Project project) {
        return project.getExtensions().getByType(JavaPluginExtension.class).getSourceSets();
    }