LoginFormHTMLInclude.java

/* (c) 2016 Open Source Geospatial Foundation - all rights reserved
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.web;

import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;

import org.apache.wicket.markup.ComponentTag;
import org.apache.wicket.markup.MarkupStream;
import org.apache.wicket.markup.html.include.Include;
import org.apache.wicket.request.resource.PackageResourceReference;
import org.geoserver.platform.GeoServerExtensions;
import org.geotools.util.logging.Logging;

import freemarker.template.Configuration;
import freemarker.template.Template;

/**
 * @author Alessio Fabiani, GeoSolutions S.A.S.
 *
 */
public class LoginFormHTMLInclude extends Include {

    protected static final Logger LOGGER = Logging.getLogger(LoginFormHTMLInclude.class);

    /** serialVersionUID */
    private static final long serialVersionUID = 2413413223248385722L;

    private static final String DEFAULT_AUTOCOMPLETE_VALUE = "on";

    public static final String GEOSERVER_LOGIN_AUTOCOMPLETE = "geoserver.login.autocomplete";

    private static Configuration templateConfig;

    static {
        // initialize the template engine, this is static to maintain a cache
        templateConfig = new Configuration();
    }

    private PackageResourceReference resourceReference;

    public LoginFormHTMLInclude(String id, PackageResourceReference packageResourceReference) {
        super(id);
        this.resourceReference = packageResourceReference;
    }

    @Override
    public void onComponentTagBody(final MarkupStream markupStream, final ComponentTag openTag) {
        String content = importAsString();
        replaceComponentTagBody(markupStream, openTag, content);
    }

    /**
     * Imports the contents of the url of the model object.
     * 
     * @return the imported contents
     */
    @Override
    protected String importAsString() {
        try {

            templateConfig.setClassForTemplateLoading(this.resourceReference.getScope(), "");
            Template template = templateConfig.getTemplate(this.resourceReference.getName());
            Map<String, Object> params = new HashMap<>();

            String autocompleteValue = GeoServerExtensions.getProperty(GEOSERVER_LOGIN_AUTOCOMPLETE);
            if (autocompleteValue == null) {
                autocompleteValue = DEFAULT_AUTOCOMPLETE_VALUE;
            }
            params.put("autocomplete", autocompleteValue);

            StringWriter writer = new StringWriter();
            template.process(params, writer);

            return writer.toString();

        } catch (Exception ex) {
            LOGGER.log(Level.FINEST, "Problem reading resource contents.", ex);
        }

        return "";
    }

}