GriffonArtifactASTInjector.java

/*
 * Copyright 2008-2017 the original author or authors.
 *
 * 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.
 */
package org.codehaus.griffon.compile.core.ast.transform;

import griffon.core.GriffonApplication;
import griffon.core.artifact.GriffonClass;
import org.codehaus.griffon.compile.core.BaseConstants;
import org.codehaus.groovy.ast.ClassNode;
import org.codehaus.groovy.ast.FieldNode;
import org.codehaus.groovy.ast.MethodNode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.annotation.Nonnull;
import javax.inject.Inject;

import static java.lang.reflect.Modifier.FINAL;
import static java.lang.reflect.Modifier.PRIVATE;
import static java.lang.reflect.Modifier.PUBLIC;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.NO_ARGS;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.NO_EXCEPTIONS;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.NO_PARAMS;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.THIS;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.args;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.call;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.constx;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.field;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.injectField;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.injectMethod;
import static org.codehaus.griffon.compile.core.ast.GriffonASTUtils.returns;
import static org.codehaus.groovy.ast.ClassHelper.CLASS_Type;

/**
 * @author Andres Almiray
 * @since 2.0.0
 */
public class GriffonArtifactASTInjector extends AbstractASTInjector implements BaseConstants {
    private static final ClassNode GRIFFON_APPLICATION_TYPE = makeClassSafe(GriffonApplication.class);
    private static final ClassNode GRIFFON_CLASS_TYPE = makeClassSafe(GriffonClass.class);
    private static final ClassNode LOGGER_TYPE = makeClassSafe(Logger.class);
    private static final ClassNode LOGGER_FACTORY_TYPE = makeClassSafe(LoggerFactory.class);
    private static final ClassNode INJECT_TYPE = makeClassSafe(Inject.class);

    public void inject(@Nonnull ClassNode classNode, @Nonnull String artifactType) {
        injectedField(classNode, GRIFFON_APPLICATION_TYPE, APPLICATION);

        // GriffonClass getGriffonClass()
        injectMethod(classNode, new MethodNode(
            "getGriffonClass",
            PUBLIC,
            GRIFFON_CLASS_TYPE,
            NO_PARAMS,
            NO_EXCEPTIONS,
            returns(call(
                applicationProperty(classNode, "artifactManager"),
                "findGriffonClass",
                args(call(THIS, "getClass", NO_ARGS))))
        ));
        String loggerCategory = "griffon.app." + artifactType + "." + classNode.getName();
        FieldNode loggerField = injectField(classNode,
            "this$logger",
            FINAL | PRIVATE,
            LOGGER_TYPE,
            call(
                LOGGER_FACTORY_TYPE,
                "getLogger",
                args(constx(loggerCategory)))
        );

        // Class getTypeClass()
        injectMethod(classNode, new MethodNode(
            "getTypeClass",
            PUBLIC,
            makeClassSafe(CLASS_Type),
            NO_PARAMS,
            NO_EXCEPTIONS,
            returns(call(ClassNode.THIS, "getClass", NO_ARGS))
        ));

        // Logger getLog()
        injectMethod(classNode, new MethodNode(
            "getLog",
            PUBLIC,
            LOGGER_TYPE,
            NO_PARAMS,
            NO_EXCEPTIONS,
            returns(field(loggerField))
        ));

        ThreadingAwareASTTransformation.apply(classNode);
        MVCAwareASTTransformation.apply(classNode);
        ResourcesAwareASTTransformation.apply(classNode);
    }
}