Merge pull request #13 from BryanSharp/support_library_ext

support for library

Author: BryanSharp

README.md

@@ -12,7 +12,8 @@ Beaver，即河狸，是一种日日忙碌于在自己栖息河流上修建和
 
 该插件已经上传到Jcenter,可直接引用最新版本如下：
 
-    classpath 'com.bryansharp:hibeaver:1.2.5'
+    classpath 'com.bryansharp:hibeaver:1.2.6'
+
 
 [Link to Jcenter](https://bintray.com/bsp0911932/maven/HiBeaver)
 
@@ -97,7 +98,7 @@ Basically, HiBeaver is an Android plugin for modifying your java byte code durin
 
 This plugin has been uploaded to jcenter. You can use this by adding the following code to your buildScripts:
 
-    classpath 'com.bryansharp:HiBeaver:1.2.5'
+    classpath 'com.bryansharp:HiBeaver:1.2.6'
 
 [Link to Jcenter](https://bintray.com/bsp0911932/maven/HiBeaver)
 

build.gradle

@@ -24,7 +24,7 @@ repositories {
     mavenCentral()
 }
 group = 'com.bryansharp'
-version = '1.2.5'
+version = '1.2.6'
 uploadArchives {
 //    version = version + '-SNAPSHOT'//if you are testing the demo I provide locally, you can uncomment this.
     repositories {

src/main/groovy/com/bryansharp/gradle/hibeaver/HiBeaverPluginImpl.groovy

@@ -1,6 +1,8 @@
 package com.bryansharp.gradle.hibeaver
 
+import com.android.build.gradle.AppExtension
 import com.android.build.gradle.BaseExtension
+import com.android.build.gradle.LibraryExtension
 import com.bryansharp.gradle.hibeaver.utils.DataHelper
 import com.bryansharp.gradle.hibeaver.utils.Log
 import com.bryansharp.gradle.hibeaver.utils.ModifyFiles
@@ -36,6 +38,13 @@ class HiBeaverPluginImpl implements Plugin<Project> {
     def static registerTransform(Project project) {
 //        def isApp = project.plugins.hasPlugin("com.android.application")
         BaseExtension android = project.extensions.getByType(BaseExtension)
+        if (android instanceof LibraryExtension){
+            DataHelper.ext.projectType = DataHelper.TYPE_LIB;
+        } else if(android instanceof AppExtension){
+            DataHelper.ext.projectType = DataHelper.TYPE_APP;
+        } else {
+            DataHelper.ext.projectType = -1
+        }
         InjectTransform transform = new InjectTransform()
         android.registerTransform(transform)
     }

src/main/groovy/com/bryansharp/gradle/hibeaver/InjectTransform.groovy

@@ -34,6 +34,11 @@ public class InjectTransform extends Transform {
 
     @Override
     Set<QualifiedContent.Scope> getScopes() {
+        if(DataHelper.ext.projectType == DataHelper.TYPE_APP){
+            return TransformManager.SCOPE_FULL_PROJECT
+        }else if (DataHelper.ext.projectType == DataHelper.TYPE_LIB){
+            return TransformManager.SCOPE_FULL_LIBRARY
+        }
         return TransformManager.SCOPE_FULL_PROJECT
     }
 

src/main/groovy/com/bryansharp/gradle/hibeaver/utils/DataHelper.groovy

@@ -1,6 +1,8 @@
 package com.bryansharp.gradle.hibeaver.utils
 
 public class DataHelper {
+    public static final int TYPE_APP = 1;
+    public static final int TYPE_LIB = 2;
     static Map getExt() {
         return objMap
     }