fix: improve segment group properties

novuhq · Dec 25, 2024 · f2a6ba6 · f2a6ba6
1 parent 53dd3ee
commit f2a6ba6
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 1 deletion.
diff --git a/apps/dashboard/src/components/auth/questionnaire-form.tsx b/apps/dashboard/src/components/auth/questionnaire-form.tsx
@@ -16,6 +16,8 @@ import { ROUTES } from '../../utils/routes';
 import { useMutation } from '@tanstack/react-query';
 import { useOrganization, useUser } from '@clerk/clerk-react';
 import { useEnvironment, useFetchEnvironments } from '../../context/environment/hooks';
+import { useSegment } from '../../context/segment';
+import { useAuth } from '../../context/auth/hooks';
 
 interface QuestionnaireFormData {
   jobTitle: JobTitleEnum;
@@ -241,6 +243,8 @@ export function QuestionnaireForm() {
 }
 
 function useSubmitQuestionnaire() {
+  const { currentUser, currentOrganization } = useAuth();
+  const segment = useSegment();
   const track = useTelemetry();
   const navigate = useNavigate();
   const { currentEnvironment } = useEnvironment();
@@ -271,6 +275,26 @@ function useSubmitQuestionnaire() {
         companySize: data.companySize,
         organizationType: data.organizationType,
       });
+
+      if (currentUser && currentOrganization) {
+        segment.identify(currentUser, {
+          organizationType: data.organizationType,
+          jobTitle: data.jobTitle,
+          companySize: data.companySize,
+        });
+
+        segment.group(
+          {
+            id: currentOrganization?._id,
+            name: currentOrganization?.name,
+            createdAt: currentOrganization?.createdAt,
+          },
+          {
+            organizationType: data.organizationType,
+            companySize: data.companySize,
+          }
+        );
+      }
     },
     onSuccess: () => {
       navigate(ROUTES.USECASE_SELECT);

diff --git a/apps/dashboard/src/utils/segment.ts b/apps/dashboard/src/utils/segment.ts
@@ -59,7 +59,7 @@ export class SegmentService {
     }
   }
 
-  identify(user: IUserEntity) {
+  identify(user: IUserEntity, extraProperties?: Record<string, unknown>) {
     if (!this.isSegmentEnabled()) {
       return;
     }
@@ -70,6 +70,19 @@ export class SegmentService {
       firstName: user.firstName,
       lastName: user.lastName,
       avatar: user.profilePicture,
+      ...(extraProperties || {}),
+    });
+  }
+
+  group(organization: { id: string; name: string; createdAt: string }, extraProperties?: Record<string, unknown>) {
+    if (!this.isSegmentEnabled()) {
+      return;
+    }
+
+    this._segment?.group(organization.id, {
+      name: organization.name,
+      createdAt: organization.createdAt,
+      ...(extraProperties || {}),
     });
   }