plugin bug

Author: 80002462

android/app/build.gradle

@@ -1,5 +1,5 @@
 apply plugin: 'com.android.application'
-//apply plugin: 'io.lecon.debug_tools.plugin'
+apply plugin: 'io.lecon.debug_tools.plugin'
 
 android {
     compileSdkVersion 26
@@ -17,6 +17,18 @@ android {
             proguardFiles getDefaultProguardFile('proguard-android.txt'), 'proguard-rules.pro'
         }
     }
+
+    flavorDimensions "default"
+
+    productFlavors {
+        sit {
+
+        }
+
+        prod {
+
+        }
+    }
 }
 
 //debug_tools {
@@ -39,14 +51,14 @@ dependencies {
 //    implementation 'io.lecon.debugtools:debugtools:1.0.5'
 }
 
-//debug_tools {
-//    database {
-//        services {
-//            path '/sdcard/services.db'
-//        }
-//        extra {
-//            path '/sdcard/.fxg_waybill_rule/waybill_rule.db'
-////            password '"fengxiaoge123456"'
-//        }
-//    }
-//}
+debug_tools {
+    database {
+        services {
+            path '/sdcard/services.db'
+        }
+        extra {
+            path '/sdcard/.fxg_waybill_rule/waybill_rule.db'
+//            password '"fengxiaoge123456"'
+        }
+    }
+}

android/build.gradle

@@ -11,7 +11,7 @@ buildscript {
     dependencies {
         classpath 'com.android.tools.build:gradle:3.1.4'
         classpath 'com.novoda:bintray-release:0.8.0'
-//        classpath 'io.lecon.debug_tools.plugin:plugin:1.0.0'
+        classpath 'io.lecon.debugtools.plugin:debugtoolsplugin:1.0.9.1'
         // NOTE: Do not place your application dependencies here; they belong
         // in the individual module build.gradle files
     }

android/plugin/plugin/build.gradle

@@ -75,7 +75,7 @@ publish {
     userOrg = 'leconio'//bintray.com你的用户名
     groupId = 'io.lecon.debugtools.plugin'//jcenter上的路径
     artifactId = 'debugtoolsplugin'//项目名称
-    publishVersion = '1.0.8'//版本号
+    publishVersion = '1.0.9'//版本号
     desc = 'Tools help you ez code'//描述，不重要
     website = 'https://github.com/leconio/AndroidDebugTools.git'
 }

android/plugin/plugin/src/main/java/io/lecon/debugtools.plugin/PluginMain.kt

@@ -1,7 +1,8 @@
 package io.lecon.debugtools.plugin
 
-import com.android.build.gradle.AppExtension
+import com.android.build.gradle.*
 import com.android.build.gradle.api.BaseVariant
+import com.android.build.gradle.api.LibraryVariant
 import com.squareup.javapoet.ClassName
 import com.squareup.javapoet.JavaFile
 import com.squareup.javapoet.MethodSpec
@@ -10,6 +11,7 @@ import groovy.util.XmlSlurper
 import io.lecon.debugtools.plugin.domain.DebugTools
 import org.gradle.api.Plugin
 import org.gradle.api.Project
+import org.gradle.api.UnknownDomainObjectException
 import org.gradle.api.plugins.ExtensionContainer
 import org.gradle.internal.reflect.Instantiator
 import org.gradle.invocation.DefaultGradle
@@ -23,8 +25,12 @@ import kotlin.reflect.KClass
  */
 class PluginMain : Plugin<Project> {
 
-    private operator fun <T : Any> ExtensionContainer.get(type: KClass<T>): T {
-        return getByType(type.java)
+    private operator fun <T : Any> ExtensionContainer.get(type: KClass<T>): T? {
+        try {
+            return getByType(type.java)
+        } catch (_ : UnknownDomainObjectException) {
+            return null
+        }
     }
 
     private lateinit var project: Project
@@ -66,20 +72,32 @@ class PluginMain : Plugin<Project> {
         val javaFile = JavaFile.builder("io.lecon.debug_tools", helloWorld)
                 .build()
 
-        project.extensions[AppExtension::class].apply {
-            this.applicationVariants.all { variant ->
-                val outputDir = project.buildDir.resolve(
-                        "generated/source/debug_tools/${variant.dirName}")
-                javaFile.writeTo(outputDir)
-                variant.outputs.all {
-                    project.tasks.create("generate${variant.name.capitalize()}ExtraDatabaseBuilder") {
-                        variant.registerJavaGeneratingTask(it, outputDir)
-                    }
-                }
+        project.extensions[AppExtension::class].let {
+            it?.let { extension ->
+                extension.applicationVariants.all { variant->
+                  getConfig(variant,javaFile)
+              }
+            }
+        }
 
+        project.extensions[LibraryExtension::class].let {
+            it?.let { extension ->
+                extension.libraryVariants.all { variant ->
+                    getConfig(variant,javaFile)
+                }
             }
         }
+    }
 
+    private fun getConfig(variant: BaseVariant,javaFile: JavaFile) {
+        val outputDir = project.buildDir.resolve(
+                "generated/source/debug_tools/${variant.dirName}")
+        javaFile.writeTo(outputDir)
+        variant.outputs.all {
+            project.tasks.create("generate${variant.name.capitalize()}ExtraDatabaseBuilder") {
+                variant.registerJavaGeneratingTask(it, outputDir)
+            }
+        }
     }
 
     private fun getPackageName(variant: BaseVariant): String {

android/plugin/publish.sh

@@ -0,0 +1 @@
+./gradlew clean build bintrayUpload -PbintrayUser=$BINTRAY_USER -PbintrayKey=$BINTRAY_API_KEY -PdryRun=false