com.peterphi.std.guice.web.rest.ServletEndpointDiscoveryImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of stdlib-guice-webapp Show documentation

stdlib guice webapp

There is a newer version: 10.1.5

package com.peterphi.std.guice.web.rest;

import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.peterphi.std.guice.apploader.GuiceProperties;
import com.peterphi.std.guice.common.serviceprops.composite.GuiceConfig;
import com.peterphi.std.guice.serviceregistry.LocalEndpointDiscovery;
import org.apache.log4j.Logger;

import java.net.URI;

@Singleton
public class ServletEndpointDiscoveryImpl implements LocalEndpointDiscovery
{
	private static final Logger log = Logger.getLogger(ServletEndpointDiscoveryImpl.class);

	private final GuiceConfig config;


	@Inject
	public ServletEndpointDiscoveryImpl(GuiceConfig configuration)
	{
		this.config = configuration;
	}


	@Override
	public URI getLocalEndpoint()
	{
		if (config.containsKey(GuiceProperties.STATIC_ENDPOINT_CONFIG_NAME))
		{
			final String uri = config.get(GuiceProperties.STATIC_ENDPOINT_CONFIG_NAME);

			return URI.create(uri);
		}
		else
		{
			// Need to build up endpoint
			String baseUri = config.get(GuiceProperties.STATIC_CONTAINER_PREFIX_CONFIG_NAME);
			String contextPath;

			if (config.containsKey(GuiceProperties.STATIC_CONTEXTPATH_CONFIG_NAME))
			{
				contextPath = config.get(GuiceProperties.STATIC_CONTEXTPATH_CONFIG_NAME);
			}
			else
			{
				// need to figure out webapp name
				contextPath = config.get(GuiceProperties.SERVLET_CONTEXT_NAME);
			}

			// Try to avoid double-slashing if the baseUri ends with /
			if (baseUri.endsWith("/") && contextPath.startsWith("/"))
			{
				contextPath = contextPath.substring(1); // both have slashes, remove it from the context path
			}
			else if (!baseUri.endsWith("/") && !contextPath.startsWith("/"))
			{
				contextPath = "/" + contextPath; // neither have slashes, add one to the context path
			}

			return URI.create(baseUri + contextPath);
		}
	}
}