org.mayocat.multitenancy.DefaultTenantResolver.java Source code

Introduction

Here is the source code for org.mayocat.multitenancy.DefaultTenantResolver.java

Source

/*
 * Copyright (c) 2012, Mayocat <hello@mayocat.org>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.mayocat.multitenancy;

import java.util.HashMap;
import java.util.Map;

import javax.inject.Inject;
import javax.inject.Singleton;
import javax.servlet.ServletRequestEvent;
import javax.servlet.ServletRequestListener;

import org.apache.commons.lang3.StringUtils;
import org.mayocat.configuration.MultitenancySettings;
import org.mayocat.accounts.model.Tenant;
import org.mayocat.accounts.AccountsService;
import org.mayocat.configuration.SiteSettings;
import org.mayocat.store.EntityAlreadyExistsException;
import org.slf4j.Logger;
import org.xwiki.component.annotation.Component;

import com.google.common.base.Strings;
import com.google.common.net.InternetDomainName;

@Component
@Singleton
public class DefaultTenantResolver implements TenantResolver, ServletRequestListener {
    /**
     * Request-scoped cache so that we don't call the store multiple times per request for the same host. FIXME: maybe
     * setup a global cache instead.
     */
    private ThreadLocal<Map<String, Tenant>> resolved = new ThreadLocal<Map<String, Tenant>>();

    @Inject
    private AccountsService accountsService;

    @Inject
    private Logger logger;

    @Inject
    private MultitenancySettings configuration;

    @Inject
    private SiteSettings siteSettings;

    public void requestDestroyed() {
        this.resolved.set(null);
        this.resolved.remove();
    }

    @Override
    public Tenant resolve(String host) {
        return this.resolve(host, null);
    }

    @Override
    public Tenant resolve(String host, String path) {
        if (this.resolved.get() == null) {
            this.resolved.set(new HashMap<String, Tenant>());
        }
        if (!this.resolved.get().containsKey(host)) {
            Tenant tenant = null;
            try {
                if (!this.configuration.isActivated()) {
                    // Mono-tenant

                    tenant = this.accountsService.findTenant(this.configuration.getDefaultTenantSlug());
                    if (tenant == null) {
                        tenant = this.accountsService.createDefaultTenant();
                    }
                    this.resolved.get().put(host, tenant);
                } else {
                    // Multi-tenant

                    if (path != null && path.startsWith("/tenant/")) {
                        String tmp = StringUtils.substringAfter(path, "/tenant/");
                        String slug = StringUtils.substringBefore(tmp, "/");
                        tenant = this.accountsService.findTenant(slug);
                    }

                    if (tenant == null) {
                        tenant = this.accountsService.findTenantByDefaultHost(host);
                        if (tenant == null) {
                            tenant = this.accountsService.findTenant(this.extractSlugFromHost(host));
                            if (tenant == null) {
                                return null;
                            }
                        }
                    }
                    this.resolved.get().put(host, tenant);
                }
            } catch (EntityAlreadyExistsException e) {
                // Has been created in between ?
                this.logger.warn("Failed attempt at creating a tenant that already exists for host {}", host);
            }
        }
        logger.debug("Resolved tenant [{}] ", this.resolved.get().get(host));
        return this.resolved.get().get(host);
    }

    private String extractSlugFromHost(String host) {
        String rootDomain;
        String siteName = siteSettings.getWebDomainName().or(siteSettings.getDomainName());
        if (Strings.emptyToNull(siteName) == null) {
            InternetDomainName domainName = InternetDomainName.from(host);
            if (domainName.hasPublicSuffix()) {
                // Domain is under a valid TLD, extract the TLD + first child
                rootDomain = domainName.topPrivateDomain().name();
            } else if (host.indexOf(".") > 0 && host.indexOf(".") < host.length()) {
                // Otherwise, best guess : strip everything before the first dot.
                rootDomain = host.substring(host.indexOf(".") + 1);
            } else {
                rootDomain = host;
            }
        } else {
            rootDomain = StringUtils.substringBefore(siteSettings.getDomainName(), ":");
        }
        if (host.indexOf("." + rootDomain) > 0) {
            return host.substring(0, host.indexOf("." + rootDomain));
        } else {
            return host;
        }
    }

    @Override
    public void requestDestroyed(ServletRequestEvent sre) {
        this.resolved.remove();
    }

    @Override
    public void requestInitialized(ServletRequestEvent sre) {
    }
}