PortalWebResourcesUtil.java

gistfile1.java

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package com.liferay.portal.kernel.servlet;

import com.liferay.registry.Registry;
import com.liferay.registry.RegistryUtil;
import com.liferay.registry.ServiceReference;
import com.liferay.registry.ServiceTracker;
import com.liferay.registry.ServiceTrackerCustomizer;

import java.net.MalformedURLException;
import java.net.URL;

import java.util.Collection;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import javax.servlet.ServletContext;

/**
 * @author Peter Fellwock
 */
public class PortalWebResourcesUtil {

	public static String getContextPath(String resourceType) {
		return getPortalWebResources(resourceType).getContextPath();
	}

	public static long getLastModified(String resourceType) {
		return getPortalWebResources(resourceType).getLastModified();
	}

	public static PortalWebResources getPortalWebResources(
		String resourceType) {

		return _instance._portalWebResourcesMap.get(resourceType);
	}

	public static ServletContext getServletContext(String resourceType) {
		return getPortalWebResources(resourceType).getServletContext();
	}

	public static URL getServletContextResource(String resourceName) {
		for (PortalWebResources portalWebResources :
				_instance._getPortalWebResourcesList() ) {

			ServletContext servletContext =
				portalWebResources.getServletContext();

			String contextPath = servletContext.getContextPath();

			if (resourceName.startsWith(contextPath)) {
				resourceName = resourceName.substring(contextPath.length());
			}

			try {
				URL url = servletContext.getResource(resourceName);

				if (url != null) {
					return url;
				}
			}
			catch (MalformedURLException murle) {
			}
		}

		return null;
	}

	public static boolean isResourceAvailable(
		String resourceName, String resourceType) {

		String contextPath = getContextPath(resourceType);

		if (resourceName.startsWith(contextPath)) {
			resourceName = resourceName.substring(contextPath.length());
		}

		try {
			ServletContext servletContext = getServletContext(resourceType);

			URL url = servletContext.getResource(resourceName);

			if (url != null) {
				return true;
			}
		}
		catch (MalformedURLException murle) {
		}

		return false;
	}

	public static boolean isResourceContextPath(String requestURI) {
		for (PortalWebResources portalWebResources :
				_instance._getPortalWebResourcesList() ) {

			if (requestURI.startsWith(portalWebResources.getContextPath())) {
				return true;
			}
		}

		return false;
	}

	private PortalWebResourcesUtil() {
		Registry registry = RegistryUtil.getRegistry();

		_serviceTracker = registry.trackServices(
			PortalWebResources.class,
			new PortalWebResourcesServiceTrackerCustomizer());

		_serviceTracker.open();
	}

	private Collection<PortalWebResources> _getPortalWebResourcesList() {
		return _portalWebResourcesMap.values();
	}

	private static final PortalWebResourcesUtil _instance =
		new PortalWebResourcesUtil();

	private final Map<String, PortalWebResources>
		_portalWebResourcesMap = new ConcurrentHashMap<>();
	private final ServiceTracker<PortalWebResources, PortalWebResources>
		_serviceTracker;

	private class PortalWebResourcesServiceTrackerCustomizer
		implements ServiceTrackerCustomizer
			<PortalWebResources, PortalWebResources> {

		@Override
		public PortalWebResources addingService(
			ServiceReference<PortalWebResources> serviceReference) {

			Registry registry = RegistryUtil.getRegistry();

			PortalWebResources portalWebResources = registry.getService(
				serviceReference);

			String resourceType = portalWebResources.getResourceType();

			_portalWebResourcesMap.put(resourceType, portalWebResources);

			return portalWebResources;
		}

		@Override
		public void modifiedService(
			ServiceReference<PortalWebResources> serviceReference,
			PortalWebResources portalWebResources) {
		}

		@Override
		public void removedService(
			ServiceReference<PortalWebResources> serviceReference,
			PortalWebResources portalWebResources) {

			Registry registry = RegistryUtil.getRegistry();

			registry.ungetService(serviceReference);

			_portalWebResourcesMap.remove(portalWebResources.getResourceType());
		}

	}

}