BytecodeEnhancerManager.java

package act;

/*-
 * #%L
 * ACT Framework
 * %%
 * Copyright (C) 2014 - 2017 ActFramework
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import act.app.App;
import act.asm.ClassWriter;
import act.util.AppByteCodeEnhancer;
import act.util.AsmByteCodeEnhancer;
import act.util.ByteCodeVisitor;
import act.util.DestroyableBase;
import org.osgl.$;
import org.osgl.util.C;

import javax.enterprise.context.ApplicationScoped;
import java.util.List;

import static act.Destroyable.Util.tryDestroyAll;

public class BytecodeEnhancerManager extends DestroyableBase {
    private List<AppByteCodeEnhancer> appEnhancers = C.newList();
    private List<AsmByteCodeEnhancer> generalEnhancers = C.newList();

    public BytecodeEnhancerManager() {
    }

    public void register(AsmByteCodeEnhancer enhancer) {
        if (enhancer instanceof AppByteCodeEnhancer) {
            appEnhancers.add((AppByteCodeEnhancer) enhancer);
        } else {
            generalEnhancers.add(enhancer);
        }
    }

    public void register(AppByteCodeEnhancer enhancer) {
        appEnhancers.add(enhancer);
    }

    public ByteCodeVisitor appEnhancer(App app, String className, $.Var<ClassWriter> cw) {
        List<AppByteCodeEnhancer> l = appFilter(app, className);
        return l.isEmpty() ? null : ByteCodeVisitor.chain(cw, l);
    }

    public ByteCodeVisitor generalEnhancer(String className, $.Var<ClassWriter> cw) {
        List<AsmByteCodeEnhancer> l = generalFilter(className);
        return l.isEmpty() ? null : ByteCodeVisitor.chain(cw, l);
    }

    private List<AppByteCodeEnhancer> appFilter(App app, String className) {
        List<AppByteCodeEnhancer> l = C.newList();
        for (AppByteCodeEnhancer e : appEnhancers) {
            AppByteCodeEnhancer e0 = (AppByteCodeEnhancer) e.clone();
            e0.app(app);
            if (e0.isTargetClass(className)) {
                l.add(e0);
            }
        }
        return l;
    }

    private List<AsmByteCodeEnhancer> generalFilter(String className) {
        List<AsmByteCodeEnhancer> l = C.newList();
        for (AsmByteCodeEnhancer e : generalEnhancers) {
            AsmByteCodeEnhancer e0 = e.clone();
            if (e0.isTargetClass(className)) {
                l.add(e0.clone());
            }
        }
        return l;
    }

    @Override
    protected void releaseResources() {
        tryDestroyAll(appEnhancers, ApplicationScoped.class);
        appEnhancers.clear();
        tryDestroyAll(generalEnhancers, ApplicationScoped.class);
        generalEnhancers.clear();
    }
}