Protei/PromucFlow_constructor
2
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: Creating default workspace for SSO signed up users before creating first application (#16579)

Browse Source
This commit is contained in:
Trisha Anand 2022-09-06 18:25:07 +05:30 committed by GitHub
parent ee35760097
commit 70bb40d5c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/server/appsmith-server/src/main/java/com/appsmith/server/authentication/handlers/AuthenticationSuccessHandler.java
View File

@ -2,12 +2,12 @@ package com.appsmith.server.authentication.handlers;
import com.appsmith.server.authentication.handlers.ce.AuthenticationSuccessHandlerCE;
import com.appsmith.server.helpers.RedirectHelper;
import com.appsmith.server.repositories.WorkspaceRepository;
import com.appsmith.server.repositories.UserRepository;
import com.appsmith.server.services.AnalyticsService;
import com.appsmith.server.services.ApplicationPageService;
import com.appsmith.server.services.SessionUserService;
import com.appsmith.server.services.UserDataService;
import com.appsmith.server.services.WorkspaceService;
import com.appsmith.server.solutions.ExamplesWorkspaceCloner;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
@ -22,9 +22,10 @@ public class AuthenticationSuccessHandler extends AuthenticationSuccessHandlerCE
AnalyticsService analyticsService,
UserDataService userDataService,
UserRepository userRepository,
WorkspaceRepository workspaceRepository,
WorkspaceService workspaceService,
ApplicationPageService applicationPageService) {
super(examplesWorkspaceCloner, redirectHelper, sessionUserService, analyticsService, userDataService,
userRepository, workspaceRepository, applicationPageService);
userRepository, workspaceService, applicationPageService);
}
}

31
app/server/appsmith-server/src/main/java/com/appsmith/server/authentication/handlers/ce/AuthenticationSuccessHandlerCE.java
View File

@ -1,19 +1,20 @@
package com.appsmith.server.authentication.handlers.ce;
import com.appsmith.server.authentication.handlers.CustomServerOAuth2AuthorizationRequestResolver;
import com.appsmith.external.constants.AnalyticsEvents;
import com.appsmith.server.authentication.handlers.CustomServerOAuth2AuthorizationRequestResolver;
import com.appsmith.server.constants.FieldName;
import com.appsmith.server.constants.Security;
import com.appsmith.server.domains.Application;
import com.appsmith.server.domains.LoginSource;
import com.appsmith.server.domains.User;
import com.appsmith.server.domains.Workspace;
import com.appsmith.server.helpers.RedirectHelper;
import com.appsmith.server.repositories.WorkspaceRepository;
import com.appsmith.server.repositories.UserRepository;
import com.appsmith.server.services.AnalyticsService;
import com.appsmith.server.services.ApplicationPageService;
import com.appsmith.server.services.SessionUserService;
import com.appsmith.server.services.UserDataService;
import com.appsmith.server.services.WorkspaceService;
import com.appsmith.server.solutions.ExamplesWorkspaceCloner;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
@ -52,7 +53,7 @@ public class AuthenticationSuccessHandlerCE implements ServerAuthenticationSucce
private final AnalyticsService analyticsService;
private final UserDataService userDataService;
private final UserRepository userRepository;
private final WorkspaceRepository workspaceRepository;
private final WorkspaceService workspaceService;
private final ApplicationPageService applicationPageService;
/**
@ -103,7 +104,7 @@ public class AuthenticationSuccessHandlerCE implements ServerAuthenticationSucce
}
if(isFromSignup) {
boolean finalIsFromSignup = isFromSignup;
redirectionMono = createDefaultApplication(defaultWorkspaceId)
redirectionMono = createDefaultApplication(defaultWorkspaceId, authentication)
.flatMap(defaultApplication->handleOAuth2Redirect(webFilterExchange, defaultApplication, finalIsFromSignup));
} else {
redirectionMono = handleOAuth2Redirect(webFilterExchange, null, isFromSignup);
@ -111,7 +112,7 @@ public class AuthenticationSuccessHandlerCE implements ServerAuthenticationSucce
} else {
boolean finalIsFromSignup = isFromSignup;
if(createDefaultApplication && isFromSignup) {
redirectionMono = createDefaultApplication(defaultWorkspaceId).flatMap(
redirectionMono = createDefaultApplication(defaultWorkspaceId, authentication).flatMap(
defaultApplication->handleRedirect(webFilterExchange, defaultApplication, finalIsFromSignup)
);
} else {
@ -167,12 +168,28 @@ public class AuthenticationSuccessHandlerCE implements ServerAuthenticationSucce
.then(redirectionMono);
}
private Mono<Application> createDefaultApplication(String defaultWorkspaceId) {
protected Mono<Application> createDefaultApplication(String defaultWorkspaceId, Authentication authentication) {
// need to create default application
Application application = new Application();
application.setWorkspaceId(defaultWorkspaceId);
application.setName("My first application");
return applicationPageService.createApplication(application);
Mono<Application> applicationMono = Mono.just(application);
if (defaultWorkspaceId == null) {
String email = ((User) authentication.getPrincipal()).getEmail();
// This could happen if the user is signing up via SSO methods
log.debug("Creating blank default workspace for user '{}'.", email);
applicationMono = userRepository.findByEmail(email)
.flatMap(user -> workspaceService.createDefault(new Workspace(), user))
.map(workspace -> {
application.setWorkspaceId(workspace.getId());
return application;
});
}
return applicationMono
.flatMap(application1 -> applicationPageService.createApplication(application1));
}
/**