Add role in signup form (#1876)

This is the first task of the new onboarding, adding a new role parameter in the signup form called "role".

Author: learningbizz

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

@@ -9,6 +9,7 @@ import { unsafelyFindUserByEmail } from '@latitude-data/core/data-access/users'
 
 import { errorHandlingProcedure } from '../procedures'
 import { frontendRedirect } from '$/lib/frontendRedirect'
+import { UserTitle } from '@latitude-data/constants/users'
 
 export const setupAction = errorHandlingProcedure
   .inputSchema(
@@ -35,6 +36,7 @@ export const setupAction = errorHandlingProcedure
       companyName: z
         .string()
         .min(1, { error: 'Workspace name is a required field' }),
+      title: z.enum(UserTitle).optional(),
     }),
   )
   .action(async ({ parsedInput }) => {

apps/web/src/app/(public)/setup/_components/SetupForm/index.tsx

@@ -12,6 +12,25 @@ import { Separator } from '@latitude-data/web-ui/atoms/Separator'
 import { Text } from '@latitude-data/web-ui/atoms/Text'
 import { useToast } from '@latitude-data/web-ui/atoms/Toast'
 import Link from 'next/link'
+import { Select } from '@latitude-data/web-ui/atoms/Select'
+import { USER_TITLES, UserTitle } from '@latitude-data/constants/users'
+
+export const humanizeUserTitle = (title: UserTitle): string => {
+  switch (title) {
+    case UserTitle.Engineer:
+      return 'Engineer'
+    case UserTitle.DataAIAndML:
+      return 'Data/AI/ML'
+    case UserTitle.ProductManager:
+      return 'Product Manager'
+    case UserTitle.Designer:
+      return 'Designer'
+    case UserTitle.Founder:
+      return 'Founder'
+    case UserTitle.Other:
+      return 'Other'
+  }
+}
 
 export default function SetupForm({
   email,
@@ -74,6 +93,17 @@ export default function SetupForm({
           errors={errors?.companyName}
           defaultValue={data?.companyName || companyName}
         />
+        <Select
+          required
+          name='title'
+          label='Your role'
+          errors={errors?.title}
+          placeholder='Select'
+          options={USER_TITLES.map((title: UserTitle) => ({
+            label: humanizeUserTitle(title),
+            value: title,
+          }))}
+        />
         <div className='flex flex-col gap-6'>
           <Button fullWidth isLoading={isPending} fancy>
             Create account

apps/web/src/services/user/setupService.ts

@@ -1,17 +1,20 @@
 import setupServiceFn from '@latitude-data/core/services/users/setupService'
 import { env } from '@latitude-data/env'
 import { captureException } from '$/helpers/captureException'
+import { UserTitle } from '@latitude-data/constants/users'
 
 export default function setupService({
   email,
   name,
   companyName,
   source,
+  title,
 }: {
   email: string
   name: string
   companyName: string
   source?: string
+  title?: UserTitle
 }) {
   return setupServiceFn({
     email,
@@ -21,5 +24,6 @@ export default function setupService({
     defaultProviderName: env.NEXT_PUBLIC_DEFAULT_PROVIDER_NAME,
     defaultProviderApiKey: env.DEFAULT_PROVIDER_API_KEY,
     captureException,
+    title,
   })
 }

packages/constants/package.json

@@ -28,7 +28,8 @@
     "./promptl": "./src/promptl.ts",
     "./trigger": "./src/trigger.ts",
     "./documentTriggers": "./src/documentTriggers/schema.ts",
-    "./toolSources": "./src/toolSources.ts"
+    "./toolSources": "./src/toolSources.ts",
+    "./users": "./src/users.ts"
   },
   "devDependencies": {
     "@latitude-data/eslint-config": "workspace:*",
@@ -41,4 +42,4 @@
     "promptl-ai": "catalog:",
     "zod": "catalog:"
   }
-}
+}

packages/constants/src/users.ts

@@ -0,0 +1,10 @@
+export enum UserTitle {
+  Engineer = 'engineer',
+  DataAIAndML = 'data_ai_ml',
+  ProductManager = 'product_manager',
+  Designer = 'designer',
+  Founder = 'founder',
+  Other = 'other',
+}
+
+export const USER_TITLES = Object.values(UserTitle)

packages/core/drizzle/0227_new_user_roles.sql

@@ -0,0 +1,2 @@
+ALTER TABLE "latitude"."users" ADD COLUMN "title" varchar(128);--> statement-breakpoint
+CREATE INDEX "users_title_idx" ON "latitude"."users" USING btree ("title");