WKStatisticsPlugin/buildSrc/src/main/groovy/com/gradle/ares/MethodAgentVisitor.groovy

package com.gradle.ares

import org.objectweb.asm.MethodVisitor
import org.objectweb.asm.Opcodes

/**
 * Created by Glen on 2017/2/8.
 */

public class MethodAgentVisitor extends MethodBaseVisitor {

    public final static MethodModel[] Modify = [
            new MethodModel("onClick", "(Landroid/view/View;)V", 1),//view
            new MethodModel("onClick", "(Landroid/content/DialogInterface;I)V", 2),//dialog
            new MethodModel("onItemClick", "(Landroid/widget/AdapterView;Landroid/view/View;IJ)V", 4),//list view
            new MethodModel("onTabSelected", "(Landroid/support/design/widget/TabLayout\$Tab;)V", 1),//design lib tab layout
            new MethodModel("onCheckedChanged", "(Landroid/widget/CompoundButton;Z)V", 2)
    ] as MethodModel[]

    public
    static MethodModel getAgent(int access, String className, String superName, String name, String desc) {
        for (MethodModel item : Modify) {
            if (item.name.equals(name) && item.desc.equals(desc)) {
                item.access = access
                item.className = className
                item.superName = superName
                return item
            }
        }
        return null
    }

    MethodAgentVisitor(MethodVisitor mv, MethodModel agent) {
        super(mv, agent)
    }

    @Override
    void visitCode() {
        super.visitCode()
        for (int i = 0; i < this.agent.count; i++) loadArg(i)
        visitMethodInsn(Opcodes.INVOKESTATIC, AresValue.AgentClass, this.agent.name + "Pre", this.agent.desc, false)
    }

    @Override
    public void visitInsn(int opcode) {
        if (opcode == Opcodes.RETURN) {
            for (int i = 0; i < this.agent.count; i++) loadArg(i)
            visitMethodInsn(Opcodes.INVOKESTATIC, AresValue.AgentClass, this.agent.name, this.agent.desc, false)
        }
        super.visitInsn(opcode);
    }
}