org.jtwig.functions.SpringFunctions.java Source code

Introduction

Here is the source code for org.jtwig.functions.SpringFunctions.java

Source

/**
 * 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.jtwig.functions;

import org.jtwig.exceptions.AssetResolveException;
import org.jtwig.functions.annotations.JtwigFunction;
import org.jtwig.functions.annotations.Parameter;
import org.jtwig.functions.exceptions.FunctionException;
import org.jtwig.services.api.assets.AssetResolver;
import org.jtwig.util.UrlPath;
import org.jtwig.util.render.RenderHttpServletRequest;
import org.jtwig.util.render.RenderHttpServletResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.MessageSource;
import org.springframework.core.env.Environment;
import org.springframework.web.servlet.LocaleResolver;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

import static org.springframework.http.HttpMethod.GET;

public class SpringFunctions {
    @Autowired(required = false)
    private AssetResolver assetResolver;

    @Autowired(required = false)
    private MessageSource messageSource;

    @Autowired(required = false)
    private LocaleResolver localeResolver;

    @Autowired(required = false)
    private Environment environment;

    @JtwigFunction(name = "asset")
    public String asset(HttpServletRequest request, @Parameter String input)
            throws AssetResolveException, FunctionException {
        if (assetResolver == null)
            throw new FunctionException("In order to use the asset function, a bean of type "
                    + AssetResolver.class.getName() + " must be configured");
        return new UrlPath().append(request.getContextPath()).append(assetResolver.resolve(input)).toString();
    }

    @JtwigFunction(name = "path")
    public String path(HttpServletRequest request, @Parameter String input) {
        return new UrlPath().append(request.getContextPath()).append(input).toString();
    }

    @JtwigFunction(name = "path")
    public String path(HttpServletRequest request) {
        return new UrlPath().append(request.getContextPath()).toString();
    }

    @JtwigFunction(name = "translate", aliases = { "message", "trans" })
    public String translate(HttpServletRequest request, @Parameter String input, @Parameter Object... rest)
            throws FunctionException {
        if (messageSource == null)
            throw new FunctionException("In order to use the translate function, a bean of type "
                    + MessageSource.class.getName() + " must be configured");
        if (localeResolver == null)
            throw new FunctionException("In order to use the translate function, a bean of type "
                    + LocaleResolver.class.getName() + " must be configured");

        return messageSource.getMessage(input, rest, localeResolver.resolveLocale(request));
    }

    @JtwigFunction(name = "property")
    public Object property(@Parameter String name) throws FunctionException {
        if (environment == null)
            throw new FunctionException("Unable to retrieve Environment bean");
        else
            return environment.getProperty(name);
    }

    @JtwigFunction(name = "render")
    public String render(HttpServletRequest request, @Parameter String url) throws FunctionException {
        return render(request, url, new HashMap<String, String>());
    }

    @JtwigFunction(name = "render")
    public String render(HttpServletRequest request, @Parameter String url,
            @Parameter Map<String, String> parameters) throws FunctionException {
        RenderHttpServletResponse responseWrapper = new RenderHttpServletResponse();
        RenderHttpServletRequest builder = new RenderHttpServletRequest(request).to(url).withMethod(GET);

        for (Map.Entry<String, String> entry : parameters.entrySet()) {
            builder.withGetParameter(entry.getKey(), entry.getValue());
        }

        try {
            RequestDispatcher requestDispatcher = request.getRequestDispatcher(request.getServletPath());
            requestDispatcher.include(builder, responseWrapper);

            return responseWrapper.toString();
        } catch (ServletException | IOException e) {
            throw new FunctionException(e);
        }
    }
}