datahub/datahub-frontend/app/controllers/CentralLogoutController.java

package controllers;

import com.typesafe.config.Config;
import java.net.URLEncoder;
import lombok.extern.slf4j.Slf4j;
import org.pac4j.play.LogoutController;
import play.mvc.Http;
import play.mvc.Result;
import play.mvc.Results;

import javax.inject.Inject;
import java.nio.charset.StandardCharsets;

/**
 * Responsible for handling logout logic with oidc providers
 */
@Slf4j
public class CentralLogoutController extends LogoutController {
  private static final String DEFAULT_BASE_URL_PATH = "/login";
  private static Boolean _isOidcEnabled = false;

  @Inject
  public CentralLogoutController(Config config) {
    _isOidcEnabled = config.hasPath("auth.oidc.enabled") && config.getBoolean("auth.oidc.enabled");

    setDefaultUrl(DEFAULT_BASE_URL_PATH);
    setLogoutUrlPattern(DEFAULT_BASE_URL_PATH + ".*");
    setLocalLogout(true);
    setCentralLogout(true);

  }

  /**
   * logout() method should not be called if oidc is not enabled
   */
  public Result executeLogout(Http.Request request) {
    if (_isOidcEnabled) {
      try {
        return Results.redirect(DEFAULT_BASE_URL_PATH)
                .removingFromSession(request);
      } catch (Exception e) {
        log.error("Caught exception while attempting to perform SSO logout! It's likely that SSO integration is mis-configured.", e);
        return redirect(
            String.format("/login?error_msg=%s",
                URLEncoder.encode("Failed to sign out using Single Sign-On provider. Please contact your DataHub Administrator, "
                    + "or refer to server logs for more information.", StandardCharsets.UTF_8)));
      }
    }
    return Results.redirect(DEFAULT_BASE_URL_PATH)
            .withNewSession();
  }
}