refactor: handle EULAAccepted with db override

SpecterOps · Dec 24, 2024 · e2660a5 · e2660a5
1 parent e568316
commit e2660a5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/cmd/api/src/api/registration/registration.go b/cmd/api/src/api/registration/registration.go
@@ -52,7 +52,7 @@ func RegisterFossGlobalMiddleware(routerInst *router.Router, cfg config.Configur
 func RegisterFossRoutes(
 	routerInst *router.Router,
 	cfg config.Configuration,
-	rdms *database.BloodhoundDB,
+	rdms database.Database,
 	graphDB *graph.DatabaseSwitch,
 	graphQuery queries.Graph,
 	apiCache cache.Cache,

diff --git a/cmd/api/src/api/v2/auth/oidc.go b/cmd/api/src/api/v2/auth/oidc.go
@@ -246,7 +246,7 @@ func (s ManagementResource) OIDCCallbackHandler(response http.ResponseWriter, re
 									PrincipalName: claims.Email,
 									Roles:         model.Roles{role},
 									SSOProviderID: null.Int32From(ssoProvider.ID),
-									EULAAccepted:  true, // Need to find a work around since BHE cannot auto accept EULA as true
+									EULAAccepted:  true, // EULA Acceptance does not pertain to Bloodhound Community Edition; this flag is used for Bloodhound Enterprise users.
 									FirstName:     null.StringFrom(claims.Email),
 									LastName:      null.StringFrom("Last name not found"),
 								}

diff --git a/cmd/api/src/api/v2/auth/saml.go b/cmd/api/src/api/v2/auth/saml.go
@@ -469,7 +469,7 @@ func (s ManagementResource) SAMLCallbackHandler(response http.ResponseWriter, re
 									PrincipalName: principalName,
 									Roles:         model.Roles{role},
 									SSOProviderID: null.Int32From(ssoProvider.ID),
-									EULAAccepted:  true, // Need to find a work around since BHE cannot auto accept EULA as true
+									EULAAccepted:  true, // EULA Acceptance does not pertain to Bloodhound Community Edition; this flag is used for Bloodhound Enterprise users.
 									FirstName:     null.StringFrom(principalName),
 									LastName:      null.StringFrom("Last name not found"),
 								}