io.protostuff.generator.ExtensibleStCompiler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of protostuff-generator Show documentation
There is a newer version: 3.1.40
package io.protostuff.generator;

import com.google.inject.Inject;
import com.google.inject.assistedinject.Assisted;
import io.protostuff.compiler.model.Module;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import org.stringtemplate.v4.AttributeRenderer;
import org.stringtemplate.v4.Interpreter;
import org.stringtemplate.v4.ST;
import org.stringtemplate.v4.STGroup;
import org.stringtemplate.v4.misc.ObjectModelAdaptor;

public class ExtensibleStCompiler implements ProtoCompiler {

    private final List compilers;


    @Inject
    protected ExtensibleStCompiler(StCompilerFactory compilerFactory,
                                   @Assisted Collection templates,
                                   @Assisted ExtensionProvider extensionProvider) {
        this.compilers = new ArrayList<>();
        for (String template : templates) {
            StCompiler compiler = (StCompiler) compilerFactory.create(template);
            STGroup group = compiler.getStGroup();
            addRenderExtensions(group, extensionProvider);
            addPropertyExtensions(group, extensionProvider);
            compilers.add(compiler);
        }
    }

    private void addPropertyExtensions(STGroup group, ExtensionProvider extensionProvider) {
        Map, PropertyProvider> extenderMap = extensionProvider.propertyProviders();
        for (Map.Entry, PropertyProvider> entry : extenderMap.entrySet()) {
            Class objectClass = entry.getKey();
            PropertyProvider extender = entry.getValue();
            group.registerModelAdaptor(objectClass, new ObjectModelAdaptor() {
                @Override
                public synchronized Object getProperty(Interpreter interp, ST self, Object o, Object property, String propertyName) {
                    if (extender.hasProperty(propertyName)) {
                        return extender.getProperty(o, propertyName);
                    }
                    return super.getProperty(interp, self, o, property, propertyName);
                }
            });
        }
    }

    private void addRenderExtensions(STGroup group, ExtensionProvider extensionProvider) {
        Map, AttributeRenderer> attributeRendererMap = extensionProvider.attributeRenderers();
        for (Map.Entry, AttributeRenderer> entry : attributeRendererMap.entrySet()) {
            group.registerRenderer(entry.getKey(), entry.getValue());
        }
    }

    @Override
    public void compile(Module module) {
        for (StCompiler compiler : compilers) {
            compiler.compile(module);
        }
    }

}