Gradle插件自动插入埋点代码

做无埋点的时候需要 hook 每个 View 的 click 事件,大体办法有两种:

  • view.setAccessibilityDelegate
  • Gradle 插件修改 class 文件

一开始做的时候选择的是第一种办法,遍历ViewTree来解决,本文着重记录第二种办法。

新建 Gradle 插件

第一步

在app同级目录下新建文件夹,文件夹名字必须为 buildSrc,新建 src->main->groovy->com.xxx.xxx 以及一个 build.Gradle 文件,
最终目录如图

WX20180301-105442@2x.png

图中所示的文件,除了src以及build.gradle之外的,都是系统后期自动编译生成,这步忽略不计。

第二步

在build.gradle文件中填入如下内容:

apply plugin: 'groovy'

dependencies {
    compile gradleApi()//gradle sdk
    compile localGroovy()//groovy sdk
}

repositories {
    jcenter()
}

第三步

在 src 文件夹的包中新建具体的插件,gradle 插件基于 groovy 语言,所以新建的类文件拓展名为 .groovy

package com.netease.plugin

import org.gradle.api.Plugin
import org.gradle.api.Project

public class TestPlugin implements Plugin<Project> {

    @Override
    void apply(Project project) {

    }
}

第四步

在app模块的build.gradle中添加如下内容:

...
apply plugin: com.netease.plugin.TestPlugin
...

点击 clean project,然后 make project,当插件生效时,会调用 TestPlugin 的 apply 方法。

在 onClick 中增加代码

Gradle提供了一个Transform api,可以用来处理编译之后的class文件,大概原理如下图:

transform

每个类文件都会通过一个又一个Transform

Transform API 介绍

新建自定义 Transform

Transform 为抽象类,我们在 plugin 包中新建一个 ClickTransform,同样是 groovy 文件,代码如下:

package com.netease.plugin

import com.android.build.api.transform.*
import com.android.build.gradle.internal.pipeline.TransformManager
import com.netease.plugin.util.ClassUtil
import javassist.ClassPool
import javassist.CtClass
import javassist.CtMethod
import org.gradle.api.Project
import org.apache.commons.io.FileUtils

public class ClickTransform extends Transform {

    private Project mProject

    public ClickTransform(Project p) {
        mProject = p
    }
    /**
     * Returns the unique name of the transform.
     *
     * <p/>
     * This is associated with the type of work that the transform does. It does not have to be
     * unique per variant.
     */
    @Override
    String getName() {
        return "ClickTransformImpl"
    }
     /**
     * Returns the type(s) of data that is consumed by the Transform. This may be more than
     * one type.
     * <strong>This must be of type {@link QualifiedContent.DefaultContentType}</strong>
     * Transform的输入类型
     */
    @Override
    Set<QualifiedContent.ContentType> getInputTypes() {
        return TransformManager.CONTENT_CLASS
    }
    /**
     * Returns the scope(s) of the Transform. This indicates which scopes the transform consumes.
     * Transform的作用范围
     */
    @Override
    Set<QualifiedContent.Scope> getScopes() {
        return TransformManager.SCOPE_FULL_PROJECT
    }

    @Override
    boolean isIncremental() {
        return false
    }

    @Override
    void transform(Context context, Collection<TransformInput> inputs,
                   Collection<TransformInput> referencedInputs,
                   TransformOutputProvider outputProvider, boolean isIncremental)
            throws IOException, TransformException, InterruptedException {
    }
}

如代码中所示,我们主要关注 transform 函数,对类的操作在这个函数进行。具体代码如下:

    @Override
    void transform(Context context, Collection<TransformInput> inputs,
                   Collection<TransformInput> referencedInputs,
                   TransformOutputProvider outputProvider, boolean isIncremental)
            throws IOException, TransformException, InterruptedException {
        inputs.each { TransformInput input->
            input.directoryInputs.each { DirectoryInput directoryInput->
                //往类中注入代码
                injectClick(directoryInput.file.getAbsolutePath(), "com", mProject)

                def dest = outputProvider.getContentLocation(directoryInput.name,
                        directoryInput.contentTypes, directoryInput.scopes, Format.DIRECTORY)

                //将 input 的目录复制到 output 指定目录
                FileUtils.copyDirectory(directoryInput.file, dest)
            }

            input.jarInputs.each { JarInput jarInput ->
                //往类中注入代码
                injectClick(jarInput.file.getAbsolutePath(), "com.netease", mProject)

                //重命名输出文件(同目录 copyFile 会冲突)
                def jarName = jarInput.name
                def md5Name = jarInput.file.hashCode()
                if(jarName.endsWith(".jar")){
                    jarName = jarName.substring(0, jarName.length() - 4)
                }
                def dest = outputProvider.getContentLocation(jarName + md5Name,
                        jarInput.contentTypes, jarInput.scopes, Format.JAR)
                FileUtils.copyFile(jarInput.file, dest)
            }
        }
    }
    private void injectClick(String path, String packageName,Project project) {
        mPool.appendClassPath(path)
        mPool.appendClassPath(project.android.bootClasspath[0].toString())
        mPool.importPackage(IMPORT_CLASS_PATH)
        File dir = new File(path)
        if (dir.isDirectory()) {
            dir.eachFileRecurse {
                File file ->
                    String filePath = file.absolutePath
                    if (filePath.endsWith(".class") && !filePath.contains('R$')
                            && !filePath.contains('R.class') && !filePath.contains("BuildConfig.class")) {
                        int index = filePath.indexOf(packageName);
                        boolean isMyPackage = index != -1;
                        if (!isMyPackage) {
                            return
                        }
                        String className = ClassUtil.getClassName(index, filePath)

                        CtClass ctClass = mPool.getCtClass(className)

                        if (ctClass.isFrozen())
                            ctClass.defrost()
                             //遍历类中的所有方法,找到onClick函数
                        for (CtMethod method : ctClass.getDeclaredMethods()) {
                            //找到 onClick(View) 方法
                            if (checkOnClickMethod(method)) {
                                injectMethod(method)
                                ctClass.writeFile(path)
                            }
                        }
                    }
            }
        }
    }

    private static void injectMethod(CtMethod method) {
        method.insertAfter("YXSConfigManager.getInstance().onInvokeClick(\$1);")
    }

    private static boolean checkOnClickMethod(CtMethod method) {
        return method.getName().endsWith("onClick") && method.getParameterTypes().length == 1 &&
                method.getParameterTypes()[0].getName().equals("android.view.View")
    }    

需要在插件的 build.gradle 中增加以下依赖:

dependencies {
    compile 'com.android.tools.build:gradle:2.1.2'
    compile gradleApi()//gradle sdk
    compile localGroovy()//groovy sdk
    compile 'org.javassist:javassist:3.20.0-GA'
    compile 'com.android.tools.build:transform-api:1.5.0'
}

JavaAssist GuideLine

注册自定义Transform

我们写好了自己的 Transform 之后,需要在插件中注册。在自定义 Plugin 的 apply 函数中增加以下代码:

import com.android.build.gradle.AppExtension
import org.gradle.api.Plugin
import org.gradle.api.Project

public class MyPlugin implements Plugin<Project> {

    void apply(Project project) {

        def android = project.extensions.getByType(AppExtension)
        //注册一个Transform
        def classTransform = new ClickTransform(project);
        android.registerTransform(classTransform);
    }
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270

推荐阅读更多精彩内容