org.joinfaces.tomcat.JsfTomcatApplicationListener.java Source code

Introduction

Here is the source code for org.joinfaces.tomcat.JsfTomcatApplicationListener.java

Source

/*
 * Copyright 2016-2016 the original author or authors.
 *
 * 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.joinfaces.tomcat;

import java.net.URL;

import lombok.Builder;
import org.apache.catalina.Context;
import org.apache.catalina.WebResourceRoot;
import org.apache.catalina.WebResourceSet;
import org.apache.catalina.webresources.JarResourceSet;
import org.apache.catalina.webresources.JarWarResourceSet;
import org.reflections.util.ClasspathHelper;

import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.ApplicationListener;

/**
 * Jsf Tomcat application listener to add resources to jsf access resources at
 * integration tests or embedded jar.
 *
 * @author Marcelo Fernandes
 */
@Builder
public class JsfTomcatApplicationListener implements ApplicationListener<ApplicationReadyEvent> {

    private Context context;

    private boolean isRunningTesting(WebResourceRoot resources) {
        return resources.getJarResources().length <= 1;
    }

    private boolean isRunningEmbeddedJar(WebResourceRoot resources) {
        boolean result = false;
        for (WebResourceSet resourceSet : resources.getJarResources()) {
            if (resourceSet instanceof JarWarResourceSet && resourceSet.getBaseUrl().getFile().endsWith(".jar")) {
                result = true;
                break;
            }
        }
        return result;
    }

    private boolean containsAppResources(WebResourceRoot resources) {
        boolean result = false;
        for (WebResourceSet resourceSet : resources.getJarResources()) {
            if (resourceSet instanceof JarResourceSet) {
                result = true;
                break;
            }
        }
        return result;
    }

    private URL mainFile(WebResourceRoot resources) {
        URL result = null;
        for (WebResourceSet resourceSet : resources.getJarResources()) {
            if (resourceSet instanceof JarWarResourceSet) {
                result = resourceSet.getBaseUrl();
                break;
            }
        }
        return result;
    }

    private String base(URL url) {
        String result;
        if (url.getProtocol().equals("jar")) {
            result = url.getFile();
            result = result.substring("file:".length());
            result = result.substring(0, result.indexOf("!/"));
        } else {
            result = url.getFile();
        }
        return result;
    }

    @Override
    public void onApplicationEvent(ApplicationReadyEvent event) {
        if (this.context != null) {
            WebResourceRoot resources = this.context.getResources();

            if (resources != null && resources.getJarResources() != null) {
                if (isRunningEmbeddedJar(resources) && !containsAppResources(resources)) {
                    String webAppMount = "/";
                    String archivePath = null;
                    String internalPath = "/META-INF/resources";

                    resources.createWebResourceSet(WebResourceRoot.ResourceSetType.POST, webAppMount,
                            base(mainFile(resources)), archivePath, internalPath);
                } else if (isRunningTesting(resources)) {
                    String webAppMount = "/";
                    String archivePath = null;
                    String internalPath = "/META-INF/resources";

                    for (URL url : ClasspathHelper.forResource("META-INF/resources/",
                            this.getClass().getClassLoader())) {
                        resources.createWebResourceSet(WebResourceRoot.ResourceSetType.POST, webAppMount, base(url),
                                archivePath, internalPath);
                    }
                }
            }
        }
    }
}