org.ambraproject.wombat.config.site.SiteTemplateLoader.java Source code

Introduction

Here is the source code for org.ambraproject.wombat.config.site.SiteTemplateLoader.java

Source

/*
 * Copyright (c) 2017 Public Library of Science
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

package org.ambraproject.wombat.config.site;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists;
import freemarker.cache.MultiTemplateLoader;
import freemarker.cache.TemplateLoader;
import freemarker.cache.WebappTemplateLoader;
import org.ambraproject.wombat.config.DelegatingTemplateLoader;
import org.ambraproject.wombat.config.theme.Theme;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletContext;
import java.io.IOException;
import java.util.List;

public class SiteTemplateLoader extends DelegatingTemplateLoader {
    private static final Logger log = LoggerFactory.getLogger(SiteTemplateLoader.class);

    private final ImmutableMap<String, TemplateLoader> loaders; // mapped by site key

    public SiteTemplateLoader(ServletContext servletContext, SiteSet siteSet) throws IOException {
        this.loaders = buildLoaders(servletContext, siteSet);
    }

    private static ImmutableMap<String, TemplateLoader> buildLoaders(ServletContext servletContext, SiteSet siteSet)
            throws IOException {
        ImmutableMap.Builder<String, TemplateLoader> builder = ImmutableMap.builder();

        // Add the loader for the application root page
        builder.put("", new WebappTemplateLoader(servletContext, "/WEB-INF/themes/root/app/"));

        // Add loader for each site
        for (Site site : siteSet.getSites()) {
            Theme leaf = site.getTheme();

            List<TemplateLoader> loaders = Lists.newArrayList();
            for (Theme theme : leaf.getInheritanceChain()) {
                loaders.add(theme.getTemplateLoader());
            }

            MultiTemplateLoader multiLoader = new MultiTemplateLoader(
                    loaders.toArray(new TemplateLoader[loaders.size()]));
            builder.put(site.getKey(), multiLoader);
        }

        return builder.build();
    }

    @Override
    protected TemplateLoader delegate(String key) {
        TemplateLoader loader = loaders.get(key);
        if (loader == null) {
            throw new TemplateNotFoundException("No loader found for site key: " + key);
        }
        return loader;
    }

}