From ffc93e8b43b094e0166cc6bc999a95c485521fc0 Mon Sep 17 00:00:00 2001
From: Domagoj Kriskovic <dom.kriskovic@overleaf.com>
Date: Fri, 10 Nov 2023 11:51:45 +0100
Subject: [PATCH] Add onboarding-flow in "register" and "user-registered"
 events (#15651)

* onboarding-flow in register event

* onboarding-flow in user-registered event

* fix typos

GitOrigin-RevId: aff8a7157062db0ec1c99499353a7c08eaa7b8b6
---
 .../web/app/src/Features/StaticPages/HomeController.js     | 3 ++-
 services/web/app/src/Features/User/UserCreator.js          | 7 +++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/services/web/app/src/Features/StaticPages/HomeController.js b/services/web/app/src/Features/StaticPages/HomeController.js
index cb1d42cca3..f2b9585cd8 100644
--- a/services/web/app/src/Features/StaticPages/HomeController.js
+++ b/services/web/app/src/Features/StaticPages/HomeController.js
@@ -67,8 +67,9 @@ module.exports = HomeController = {
 
       return res.render('external/home/v2', {
         designSystemUpdatesVariant: designSystemUpdatesAssignment.variant,
+        onboardingFlowVariant: onboardingFlowAssignment.variant,
         hideNewsletterCheckbox:
-          onboardingFlowAssignment?.variant === 'token-confirmation-odc',
+          onboardingFlowAssignment.variant === 'token-confirmation-odc',
       })
     } else {
       return res.redirect('/login')
diff --git a/services/web/app/src/Features/User/UserCreator.js b/services/web/app/src/Features/User/UserCreator.js
index 34a3ea7cf3..98f94b6b2a 100644
--- a/services/web/app/src/Features/User/UserCreator.js
+++ b/services/web/app/src/Features/User/UserCreator.js
@@ -51,10 +51,17 @@ async function recordRegistrationEvent(user) {
     )
   }
 
+  const onboardingFlowAssignment =
+    await SplitTestHandler.promises.getAssignmentForUser(
+      user._id,
+      'onboarding-flow'
+    )
+
   try {
     const segmentation = {
       'home-registration': 'default',
       'split-test-design-system-updates': designSystemUpdatesAssignment.variant,
+      'onboarding-flow': onboardingFlowAssignment.variant,
     }
     if (user.thirdPartyIdentifiers && user.thirdPartyIdentifiers.length > 0) {
       segmentation.provider = user.thirdPartyIdentifiers[0].providerId