modify PuppetManager use reflections

Author: redmaple1

wechaty/pom.xml

@@ -63,14 +63,16 @@
 <!--            <groupId>org.apache.logging.log4j</groupId>-->
 <!--            <artifactId>log4j-api</artifactId>-->
 <!--        </dependency>-->
-<!--        <dependency>-->
-<!--            <groupId>org.apache.logging.log4j</groupId>-->
-<!--            <artifactId>log4j-core</artifactId>-->
-<!--        </dependency>-->
-<!--        <dependency>-->
-<!--            <groupId>org.apache.logging.log4j</groupId>-->
-<!--            <artifactId>log4j-slf4j-impl</artifactId>-->
-<!--        </dependency>-->
+        <dependency>
+            <groupId>org.apache.logging.log4j</groupId>
+            <artifactId>log4j-core</artifactId>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.logging.log4j</groupId>
+            <artifactId>log4j-slf4j-impl</artifactId>
+            <scope>test</scope>
+        </dependency>
 
         <!-- https://mvnrepository.com/artifact/com.lmax/disruptor -->
 <!--        <dependency>-->
@@ -110,6 +112,11 @@
             <artifactId>junit</artifactId>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.reflections</groupId>
+            <artifactId>reflections</artifactId>
+            <version>0.9.12</version>
+        </dependency>
 
     </dependencies>
 

wechaty/src/main/kotlin/io/github/wechaty/WechatyOptions.kt

@@ -9,7 +9,8 @@ class WechatyOptions {
 
     var name:String = "Wechaty"
 
-    var puppet:String = "wechaty-puppet-hostie"
+//    var puppet:String = "wechaty-puppet-hostie"
+    var puppet:String = "io.github.wechaty.grpc.GrpcPuppet"
 
     var puppetOptions:PuppetOptions? = null
 

wechaty/src/main/kotlin/io/github/wechaty/user/manager/PuppetManager.kt

@@ -4,11 +4,15 @@ import io.github.wechaty.MockPuppet
 import io.github.wechaty.Puppet
 import io.github.wechaty.WechatyOptions
 import io.github.wechaty.grpc.GrpcPuppet
+import io.github.wechaty.schemas.PuppetOptions
 import io.github.wechaty.utils.JsonUtils
+import org.reflections.Reflections
 import org.slf4j.LoggerFactory
 import java.util.concurrent.CompletableFuture
 import java.util.concurrent.Future
 
+const val REFLECTION_BASE_PACKAGE = "io.github.wechaty"
+
 class PuppetManager {
 
     companion object {
@@ -18,13 +22,48 @@ class PuppetManager {
         fun resolveInstance(wechatyOptions: WechatyOptions): Future<Puppet> {
             log.info("PuppetManager resolveInstance(${JsonUtils.write(wechatyOptions)})")
 
-            return if ("wechaty-puppet-hostie" == wechatyOptions.puppet) {
-                CompletableFuture.completedFuture(GrpcPuppet(wechatyOptions.puppetOptions!!))
-            } else {
-                CompletableFuture.completedFuture(MockPuppet(wechatyOptions.puppetOptions!!))
+//            return if ("io.github.wechaty.grpc.GrpcPuppet" == wechatyOptions.puppet) {
+//                CompletableFuture.completedFuture(GrpcPuppet(wechatyOptions.puppetOptions!!))
+//            } else {
+//                CompletableFuture.completedFuture(MockPuppet(wechatyOptions.puppetOptions!!))
+//            }
+            val reflections = Reflections(REFLECTION_BASE_PACKAGE)
+            val subTypes: Set<*> = reflections.getSubTypesOf(Puppet::class.java)
+
+            for (subType in subTypes) {
+                val subTypeClass = subType as Class<*>
+                if (wechatyOptions.puppet == subTypeClass.canonicalName) {
+                    val declaredConstructor = subTypeClass.getDeclaredConstructor(PuppetOptions::class.java)
+                    return CompletableFuture.completedFuture(declaredConstructor.newInstance(wechatyOptions.puppetOptions!!) as Puppet)
+                }
             }
+            throw RuntimeException("instant puppet implementation error. Please check your wechatyOptions.puppet")
         }
     }
 
 
 }
+
+fun main() {
+//    val serviceLoader:ServiceLoader<Puppet> = ServiceLoader.load(Puppet::class.java)
+//    for (puppet in serviceLoader) {
+//        println(puppet)
+//    }
+
+//    val kClass:KClass<Puppet> = Puppet::class
+//    print(kClass)
+    val reflections: Reflections = Reflections("io.github.wechaty")
+    val subTypes: Set<*> = reflections.getSubTypesOf(Puppet::class.java)
+    for (subType in subTypes) {
+        val subTypeClass = subType as Class<*>
+        if ("io.github.wechaty.grpc.GrpcPuppet" == subTypeClass.canonicalName) {
+
+                val declaredConstructor = subTypeClass.getDeclaredConstructor(PuppetOptions::class.java)
+
+                val newInstance = declaredConstructor.newInstance(PuppetOptions())
+
+        }
+    }
+
+}
+

wechaty/src/test/kotlin/io/github/wechaty/user/RoomTest.kt

@@ -36,7 +36,7 @@ class RoomTest {
         puppet = MockPuppet(PuppetOptions())
         val wechatyOptions = WechatyOptions()
         wechatyOptions.name = "MockWechaty"
-        wechatyOptions.puppet = "wechaty-puppet-mock"
+        wechatyOptions.puppet = "io.github.wechaty.MockPuppet"
         wechatyOptions.puppetOptions = PuppetOptions()
         wechaty = Wechaty.instance(wechatyOptions)
         wechaty.start()