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

package controllers;

import auth.sso.SsoManager;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import javax.inject.Inject;
import lombok.extern.slf4j.Slf4j;
import org.pac4j.play.LogoutController;
import play.mvc.Http;
import play.mvc.Result;
import play.mvc.Results;

/** Responsible for handling logout logic with oidc providers */
@Slf4j
public class CentralLogoutController extends LogoutController {
  private static final String AUTH_URL_CONFIG_PATH = "/login";
  private static final String DEFAULT_BASE_URL_PATH = "/";
  @Inject private SsoManager ssoManager;

  public CentralLogoutController() {
    setDefaultUrl(AUTH_URL_CONFIG_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 (ssoManager.isSsoEnabled()) {
      try {
        return logout(request).toCompletableFuture().get().withNewSession();
      } 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)))
            .withNewSession();
      }
    }
    return Results.redirect(AUTH_URL_CONFIG_PATH).withNewSession();
  }
}