User based onboarding

Author: csansoon

apps/console/src/models.ts

@@ -42,5 +42,4 @@ export {
   webhooks,
 } from '@latitude-data/core/schema/models/webhooks'
 export { workspaceFeatures } from '@latitude-data/core/schema/models/workspaceFeatures'
-export { workspaceOnboarding } from '@latitude-data/core/schema/models/workspaceOnboarding'
 export { workspaces } from '@latitude-data/core/schema/models/workspaces'

…/actions/workspaceOnboarding/complete.ts …s/web/src/actions/onboarding/complete.tsapps/web/src/actions/workspaceOnboarding/complete.ts renamed to apps/web/src/actions/onboarding/complete.ts apps/web/src/actions/workspaceOnboarding/complete.ts renamed to apps/web/src/actions/onboarding/complete.ts

@@ -1,16 +1,12 @@
 'use server'
 
-import { markWorkspaceOnboardingComplete } from '@latitude-data/core/services/workspaceOnboarding/update'
-import { getWorkspaceOnboarding } from '@latitude-data/core/services/workspaceOnboarding/get'
+import { markUserOnboardingComplete } from '@latitude-data/core/services/users/completeOnboarding'
 import { authProcedure } from '../procedures'
 import { frontendRedirect } from '$/lib/frontendRedirect'
 import { ROUTES } from '$/services/routes'
 import { z } from 'zod'
 import { RunSourceGroup } from '@latitude-data/constants'
 
-/**
- * Mark onboarding as complete
- */
 export const completeOnboardingAction = authProcedure
   .inputSchema(
     z.object({
@@ -20,15 +16,11 @@ export const completeOnboardingAction = authProcedure
   )
   .action(async ({ parsedInput, ctx }) => {
     const { projectId, commitUuid } = parsedInput
-    const onboarding = await getWorkspaceOnboarding({
-      workspace: ctx.workspace,
-    }).then((r) => r.unwrap())
 
-    await markWorkspaceOnboardingComplete({
-      onboarding,
+    await markUserOnboardingComplete({
+      user: ctx.user,
     }).then((r) => r.unwrap())
 
-    // Default redirect to the project's runs page
     return frontendRedirect(
       ROUTES.projects
         .detail({ id: projectId })
@@ -39,3 +31,4 @@ export const completeOnboardingAction = authProcedure
         }),
     )
   })
+

apps/web/src/actions/onboarding/skip.ts

@@ -0,0 +1,15 @@
+'use server'
+
+import { markUserOnboardingComplete } from '@latitude-data/core/services/users/completeOnboarding'
+import { authProcedure } from '../procedures'
+import { frontendRedirect } from '$/lib/frontendRedirect'
+import { ROUTES } from '$/services/routes'
+
+export const skipOnboardingAction = authProcedure.action(async ({ ctx }) => {
+  await markUserOnboardingComplete({
+    user: ctx.user,
+  }).then((r) => r.unwrap())
+
+  return frontendRedirect(ROUTES.dashboard.root)
+})
+

apps/web/src/actions/user/setupAction.ts

@@ -11,8 +11,7 @@ import { errorHandlingProcedure } from '../procedures'
 import { frontendRedirect } from '$/lib/frontendRedirect'
 import { UserTitle } from '@latitude-data/constants/users'
 import { env } from '@latitude-data/env'
-import { markWorkspaceOnboardingComplete } from '@latitude-data/core/services/workspaceOnboarding/update'
-import { getWorkspaceOnboarding } from '@latitude-data/core/services/workspaceOnboarding/get'
+import { markUserOnboardingComplete } from '@latitude-data/core/services/users/completeOnboarding'
 
 export const setupAction = errorHandlingProcedure
   .inputSchema(
@@ -62,14 +61,9 @@ export const setupAction = errorHandlingProcedure
       if (isCloud) {
         return frontendRedirect(ROUTES.onboarding.choice)
       }
-      // If user is self-hosted and they're in the new prompt engineering onboarding, we complete the onboarding and redirect to the dashboard as they cannot generate the dataset with copilot
-      const onboarding = await getWorkspaceOnboarding({
-        workspace,
-      }).then((r) => r.unwrap())
-
-      await markWorkspaceOnboardingComplete({
-        onboarding,
-      }).then((r) => r.unwrap())
+      // If user is self-hosted, complete the onboarding and redirect to the dashboard
+      // as they cannot generate the dataset with copilot
+      await markUserOnboardingComplete({ user }).then((r) => r.unwrap())
 
       return frontendRedirect(ROUTES.dashboard.root)
     }

apps/web/src/actions/workspaceOnboarding/skip.ts

@@ -1,6 +1,6 @@
 'use server'
 
-import { getOnboardingResources } from '$/data-access/workspaceOnboarding'
+import { getOnboardingResources } from '$/data-access/onboarding'
 import { redirect } from 'next/navigation'
 import { ROUTES } from '$/services/routes'
 import { ProjectProvider } from '$/app/providers/ProjectProvider'

apps/web/src/app/(onboarding)/onboarding-prompt-engineering/layout.tsx

@@ -15,9 +15,9 @@ import { useExperiments } from '$/stores/experiments'
 import useDatasets from '$/stores/datasets'
 import { envClient } from '$/envClient'
 import { usePromptEngineeringOnboarding } from '$/app/(onboarding)/onboarding-prompt-engineering/usePromptEngineeringOnboarding'
-import useWorkspaceOnboarding from '$/stores/workspaceOnboarding'
 import useLatitudeAction from '$/hooks/useLatitudeAction'
 import { publishEventAction } from '$/actions/events/publishEventAction'
+import { completeOnboardingAction } from '$/actions/onboarding/complete'
 import { User } from '@latitude-data/core/schema/models/types/User'
 import { Workspace } from '@latitude-data/core/schema/models/types/Workspace'
 import {
@@ -41,7 +41,7 @@ export default function RunExperimentBody({
   user: User
   workspace: Workspace
 }) {
-  const { executeCompleteOnboarding } = useWorkspaceOnboarding()
+  const { execute: completeOnboarding } = useLatitudeAction(completeOnboardingAction)
   const { execute: publishEvent } = useLatitudeAction(publishEventAction)
   const { project } = useCurrentProject()
   const { commit } = useCurrentCommit()
@@ -77,7 +77,7 @@ export default function RunExperimentBody({
     {
       onCreate: async () => {
         setReplayOnboarding(false)
-        executeCompleteOnboarding({
+        completeOnboarding({
           projectId: project.id,
           commitUuid: commit.uuid,
         })

apps/web/src/app/(onboarding)/onboarding-prompt-engineering/run-experiment/_components/RunExperiment/index.tsx

@@ -12,7 +12,7 @@ import { ROUTES } from '$/services/routes'
 import { useHover } from '@latitude-data/web-ui/browser'
 import { cn } from '@latitude-data/web-ui/utils'
 
-type OnboardingChoice = 'scratch' | 'existing' | null
+type OnboardingChoice = 'product_engineer' | 'developer' | null
 
 function ChoiceCard({
   icon,
@@ -66,12 +66,12 @@ export default function OnboardingChoicePage() {
   })
 
   const handleStartFromScratch = useCallback(() => {
-    setOnboardingChoice('scratch')
+    setOnboardingChoice('product_engineer')
     router.push(ROUTES.onboarding.promptEngineering.pasteYourPrompt)
   }, [setOnboardingChoice, router])
 
   const handleExistingProject = useCallback(() => {
-    setOnboardingChoice('existing')
+    setOnboardingChoice('developer')
     router.push(ROUTES.onboarding.install)
   }, [setOnboardingChoice, router])
 

apps/web/src/app/(onboarding)/onboarding/choice/page.tsx

@@ -15,6 +15,8 @@ import { Text } from '@latitude-data/web-ui/atoms/Text'
 import { TableSkeleton } from '@latitude-data/web-ui/molecules/TableSkeleton'
 import useUsers from '$/stores/users'
 import useCurrentWorkspace from '$/stores/currentWorkspace'
+import useLatitudeAction from '$/hooks/useLatitudeAction'
+import { publishEventAction } from '$/actions/events/publishEventAction'
 import { useCallback, useState } from 'react'
 
 function NewUserRow({
@@ -75,10 +77,17 @@ export default function InviteDeveloperModal({
 }) {
   const { data: workspace } = useCurrentWorkspace()
   const { data: users, isLoading, invite } = useUsers()
+  const { execute: publishEvent } = useLatitudeAction(publishEventAction)
 
   const onInviteUser = useCallback(
-    (name: string, email: string) => invite({ name, email }),
-    [invite],
+    (name: string, email: string) => {
+      invite({ name, email })
+      publishEvent({
+        eventType: 'onboardingUserInvited',
+        payload: { invitedEmail: email },
+      })
+    },
+    [invite, publishEvent],
   )
 
   return (