/atlassian-plugins-osgi-testrunner-bundle/src/main/java/com/atlassian/plugins/osgi/test/util/SoyTransformer.java

package com.atlassian.plugins.osgi.test.util;

import java.util.ArrayList;
import java.util.List;

import com.atlassian.plugin.elements.ResourceLocation;
import com.atlassian.plugin.servlet.DownloadableResource;
import com.atlassian.plugin.webresource.transformer.AbstractStringTransformedDownloadableResource;
import com.atlassian.plugin.webresource.transformer.WebResourceTransformer;
import com.atlassian.sal.api.ApplicationProperties;
import com.atlassian.sal.api.message.I18nResolver;

import com.google.inject.*;
import com.google.inject.multibindings.Multibinder;
import com.google.template.soy.SoyFileSet;
import com.google.template.soy.SoyModule;
import com.google.template.soy.jssrc.SoyJsSrcOptions;
import com.google.template.soy.shared.restricted.SoyFunction;
import com.google.template.soy.xliffmsgplugin.XliffMsgPluginModule;

import org.dom4j.Element;

/**
 * @since version
 */
public class SoyTransformer implements WebResourceTransformer
{
    private final I18nResolver i18n;
    private final ApplicationProperties applicationProperties;

    public SoyTransformer(I18nResolver i18n, ApplicationProperties applicationProperties)
    {
        this.i18n = i18n;
        this.applicationProperties = applicationProperties;
    }

    @Override
    public DownloadableResource transform(Element element, ResourceLocation location, String filePath, DownloadableResource nextResource)
    {
        return new SoyResource(nextResource, location.getLocation());
    }

    private class SoyResource extends AbstractStringTransformedDownloadableResource
    {
        private final String location;

        private SoyResource(DownloadableResource originalResource, String location)
        {
            super(originalResource);
            this.location = location;
        }

        @Override
        public String getContentType()
        {
            return "text/javascript";//NON-NLS
        }

        @Override
        protected String transform(String originalContent)
        {
            List<Module> guiceModules = new ArrayList<Module>();
            guiceModules.add(new SoyModule());
            guiceModules.add(new XliffMsgPluginModule());
            guiceModules.add(new SpringBridgeModule());
            guiceModules.add(new OurFunctionsModule());

            final Injector injector = Guice.createInjector(guiceModules);

            SoyFileSet.Builder sfsBuilder = injector.getInstance(SoyFileSet.Builder.class);

            SoyJsSrcOptions jsSrcOptions = new SoyJsSrcOptions();
            jsSrcOptions.setShouldGenerateJsdoc(false);

            sfsBuilder.add(originalContent, location);

            SoyFileSet sfs = sfsBuilder.build();

            final List<String> output = sfs.compileToJsSrc(jsSrcOptions, null);
            if (output.size() != 1)
            {
                throw new IllegalStateException(i18n.getText("it.test.console.exception.soy.did.not.manage.to.compile.soy.template", output.size()));
            }
            return output.get(0);
        }
    }

    private class SpringBridgeModule extends AbstractModule
    {
        @Override
        public void configure()
        {
            Binder binder = binder();
            binder.bind(ApplicationProperties.class).toInstance(applicationProperties);
            binder.bind(I18nResolver.class).toInstance(i18n);
        }
    }
    
    private static class OurFunctionsModule extends AbstractModule
    {
        @Override
        public void configure()
        {
            Multibinder<SoyFunction> binder = Multibinder.newSetBinder(binder(), SoyFunction.class);
            binder.addBinding().to(ContextFunction.class);
            binder.addBinding().to(GetTextFunction.class);
        }
    }
}