From 7ba2b557e34fa422c68bc5fc32d69dc557c4e040 Mon Sep 17 00:00:00 2001
From: Paul Campbell <pcampbell@kemitix.net>
Date: Sat, 10 Mar 2018 23:41:35 +0000
Subject: [PATCH] builder: scan class path in java 9

---
 .../builder/DefaultPackageScanner.java        | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/builder/src/main/java/net/kemitix/checkstyle/ruleset/builder/DefaultPackageScanner.java b/builder/src/main/java/net/kemitix/checkstyle/ruleset/builder/DefaultPackageScanner.java
index 1e7c246..795271a 100644
--- a/builder/src/main/java/net/kemitix/checkstyle/ruleset/builder/DefaultPackageScanner.java
+++ b/builder/src/main/java/net/kemitix/checkstyle/ruleset/builder/DefaultPackageScanner.java
@@ -22,6 +22,7 @@
 package net.kemitix.checkstyle.ruleset.builder;
 
 import com.google.common.reflect.ClassPath;
+import io.github.lukehutch.fastclasspathscanner.FastClasspathScanner;
 import lombok.RequiredArgsConstructor;
 import net.kemitix.conditional.Value;
 import org.springframework.stereotype.Service;
@@ -44,23 +45,25 @@ public class DefaultPackageScanner implements PackageScanner {
     @Override
     public final Stream<String> apply(final RuleSource ruleSource) {
         return Value.<Stream<String>>where(isJava8())
-                .then(scanPackageByClassPath(ruleSource))
-                .otherwise(scanPackageByModulePath());
+                .then(scanPackageByClassPath(ruleSource.getBasePackage()))
+                .otherwise(scanPackageByModulePath(ruleSource.getBasePackage()));
     }
 
     private boolean isJava8() {
         return getClass().getClassLoader() instanceof URLClassLoader;
     }
 
-    private Supplier<Stream<String>> scanPackageByClassPath(final RuleSource ruleSource) {
-        return () -> classPath.getTopLevelClassesRecursive(ruleSource.getBasePackage())
+    private Supplier<Stream<String>> scanPackageByClassPath(final String basePackage) {
+        return () -> classPath.getTopLevelClassesRecursive(basePackage)
                 .stream()
                 .map(ClassPath.ClassInfo::getName);
     }
 
-    private Supplier<Stream<String>> scanPackageByModulePath() {
-        return () -> {
-            throw new UnsupportedOperationException("Java 9 Module Path in unsupported");
-        };
+    private Supplier<Stream<String>> scanPackageByModulePath(final String basePackage) {
+        return () -> new FastClasspathScanner(basePackage)
+                .scan()
+                .getNamesOfAllStandardClasses()
+                .stream()
+                .filter(packageName -> packageName.startsWith(basePackage));
     }
 }