CB-1447 Automatic SSO redirect

server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/DBWConstants.java

@@ -28,6 +28,13 @@ public class DBWConstants {
     public static final String PERMISSION_EDIT_STRUCTURE = "edit-meta";
     public static final String PERMISSION_EDIT_DATA = "edit-data";
 
+    public static final String STATE_ATTR_SIGN_IN_STATE = "state.signin";
+
+    public enum SignInState {
+        GLOBAL,
+        EMBEDDED
+    }
+
     //public static final String PERMISSION_USER = "user";
 
 }

server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/server/jetty/CBStaticServlet.java

@@ -1,12 +1,23 @@
 package io.cloudbeaver.server.jetty;
 
+import io.cloudbeaver.DBWConstants;
+import io.cloudbeaver.auth.DBWAuthProvider;
+import io.cloudbeaver.auth.DBWAuthProviderFederated;
+import io.cloudbeaver.auth.provider.AuthProviderConfig;
+import io.cloudbeaver.model.session.WebSession;
+import io.cloudbeaver.registry.WebAuthProviderDescriptor;
+import io.cloudbeaver.registry.WebServiceRegistry;
+import io.cloudbeaver.server.CBAppConfig;
 import io.cloudbeaver.server.CBApplication;
+import io.cloudbeaver.server.CBPlatform;
 import org.eclipse.jetty.http.HttpContent;
 import org.eclipse.jetty.http.HttpField;
 import org.eclipse.jetty.http.HttpHeader;
 import org.eclipse.jetty.server.ResourceService;
 import org.eclipse.jetty.servlet.DefaultServlet;
 import org.eclipse.jetty.util.resource.Resource;
+import org.jkiss.dbeaver.Log;
+import org.jkiss.utils.CommonUtils;
 import org.jkiss.utils.IOUtils;
 
 import javax.servlet.ServletException;
@@ -15,22 +26,84 @@ import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.*;
 import java.nio.charset.StandardCharsets;
+import java.util.Collections;
 import java.util.Enumeration;
+import java.util.Map;
 
 @WebServlet(urlPatterns = "/")
 public class CBStaticServlet extends DefaultServlet {
 
     public static final int STATIC_CACHE_SECONDS = 60 * 60 * 24 * 3;
 
+    private static final Log log = Log.getLog(CBStaticServlet.class);
+
     public CBStaticServlet() {
         super(makeResourceService());
     }
 
     @Override
     protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
+        String uri = request.getPathInfo();
+        if ((CommonUtils.isEmpty(uri) || uri.equals("/") || uri.equals("/index.html")) && request.getParameterMap().isEmpty()) {
+            if (processSessionStart(request, response)) {
+                return;
+            }
+        }
+
         super.doGet(request, response);
     }
 
+    private boolean processSessionStart(HttpServletRequest request, HttpServletResponse response) {
+        CBApplication application = CBApplication.getInstance();
+        if (application.isConfigurationMode()) {
+            return false;
+        }
+        CBAppConfig appConfig = application.getAppConfiguration();
+        String[] authProviders = appConfig.getEnabledAuthProviders();
+        if (authProviders.length == 1) {
+            String authProviderId = authProviders[0];
+            WebAuthProviderDescriptor authProvider = WebServiceRegistry.getInstance().getAuthProvider(authProviderId);
+            if (authProvider.isConfigurable()) {
+                String configId = null;
+                AuthProviderConfig activeAuthConfig = null;
+                for (Map.Entry<String, AuthProviderConfig> cfg : appConfig.getAuthProviderConfigurations().entrySet()) {
+                    if (!cfg.getValue().isDisabled() && cfg.getValue().getProvider().equals(authProviderId)) {
+                        if (activeAuthConfig != null) {
+                            return false;
+                        }
+                        configId = cfg.getKey();
+                        activeAuthConfig = cfg.getValue();
+                    }
+                }
+                if (activeAuthConfig == null) {
+                    return false;
+                }
+
+                try {
+                    // We have the only provider
+                    // Forward to signon URL
+                    DBWAuthProvider<?> authProviderInstance = authProvider.getInstance();
+                    if (authProviderInstance instanceof DBWAuthProviderFederated) {
+                        WebSession webSession = CBPlatform.getInstance().getSessionManager().getWebSession(request, response, false);
+                        if (webSession.getUser() == null) {
+                            String signInLink = ((DBWAuthProviderFederated) authProviderInstance).getSignInLink(configId, Collections.emptyMap());
+                            if (!CommonUtils.isEmpty(signInLink)) {
+                                // Redirect to it
+                                request.getSession().setAttribute(DBWConstants.STATE_ATTR_SIGN_IN_STATE, DBWConstants.SignInState.GLOBAL);
+                                response.sendRedirect(signInLink);
+                                return true;
+                            }
+                        }
+                    }
+                } catch (Exception e) {
+                    log.debug("Error reading auth provider configuration", e);
+                }
+            }
+        }
+
+        return false;
+    }
+
     private static ResourceService makeResourceService() {
         ResourceService resourceService = new ProxyResourceService();
         resourceService.setCacheControl(new HttpField(HttpHeader.CACHE_CONTROL, "public, max-age=" + STATIC_CACHE_SECONDS));