From 35bb15808901d7fc89032cf76d21ee0763713de6 Mon Sep 17 00:00:00 2001
From: Harshith Mullapudi <harshithmullapudi@gmail.com>
Date: Mon, 8 Sep 2025 19:12:48 +0530
Subject: [PATCH] fix: save onboarding answers

---
 .../onboarding/onboarding-question.tsx        |  4 +--
 .../components/onboarding/onboarding-utils.ts | 13 +++++---
 .../components/spaces/space-pattern-card.tsx  | 33 ++++++++++++++++---
 apps/webapp/app/models/user.server.ts         |  4 ++-
 apps/webapp/app/routes/onboarding.tsx         |  3 ++
 apps/webapp/prisma/schema.prisma              |  2 ++
 packages/database/prisma/schema.prisma        | 15 ++-------
 7 files changed, 49 insertions(+), 25 deletions(-)

diff --git a/apps/webapp/app/components/onboarding/onboarding-question.tsx b/apps/webapp/app/components/onboarding/onboarding-question.tsx
index c8c2f13..9bf4b50 100644
--- a/apps/webapp/app/components/onboarding/onboarding-question.tsx
+++ b/apps/webapp/app/components/onboarding/onboarding-question.tsx
@@ -130,7 +130,7 @@ export default function OnboardingQuestionComponent({
               </div>
             )}
 
-            <div className="flex justify-between pt-4">
+            <div className="flex justify-end gap-2 pt-4">
               {!isFirst && (
                 <Button
                   type="button"
@@ -143,8 +143,6 @@ export default function OnboardingQuestionComponent({
                 </Button>
               )}
 
-              <div className="flex-1" />
-
               <Button
                 type="button"
                 variant="secondary"
diff --git a/apps/webapp/app/components/onboarding/onboarding-utils.ts b/apps/webapp/app/components/onboarding/onboarding-utils.ts
index f20a969..01c4859 100644
--- a/apps/webapp/app/components/onboarding/onboarding-utils.ts
+++ b/apps/webapp/app/components/onboarding/onboarding-utils.ts
@@ -83,19 +83,24 @@ export const ONBOARDING_QUESTIONS: OnboardingQuestion[] = [
         value: "Power an AI assistant / agent with context",
       },
       {
-        id: "explore-graphs",
-        label: "Explore / learn about reified graphs",
-        value: "Explore / learn about reified graphs",
+        id: "explore-core",
+        label: "Explore core",
+        value: "Explore core",
       },
     ],
     required: true,
   },
   {
     id: "tools",
-    title: "Which tools or data sources do you care about most?",
+    title: "Which tools do you care about most?",
     description: "Context → lets you connect integration nodes live",
     type: "multi-select",
     options: [
+      { id: "claude", label: "Claude", value: "Claude" },
+      { id: "claude-code", label: "Claude Code", value: "Claude Code" },
+      { id: "cursor", label: "Cursor", value: "Cursor" },
+      { id: "windsurf", label: "Windsurf", value: "Windsurf" },
+      { id: "zed", label: "Zed", value: "Zed" },
       { id: "github", label: "GitHub", value: "GitHub" },
       { id: "slack", label: "Slack", value: "Slack" },
       { id: "notion", label: "Notion", value: "Notion" },
diff --git a/apps/webapp/app/components/spaces/space-pattern-card.tsx b/apps/webapp/app/components/spaces/space-pattern-card.tsx
index a278d87..92138ef 100644
--- a/apps/webapp/app/components/spaces/space-pattern-card.tsx
+++ b/apps/webapp/app/components/spaces/space-pattern-card.tsx
@@ -6,6 +6,7 @@ import { useState } from "react";
 import { Dialog, DialogContent, DialogHeader, DialogTitle } from "../ui/dialog";
 import { Button } from "../ui";
 import { useFetcher } from "@remix-run/react";
+import { getTailwindColor, getTeamColor } from "../ui/color-utils";
 
 interface SpacePatternCardProps {
   pattern: SpacePattern;
@@ -46,10 +47,22 @@ export function SpacePatternCard({ pattern }: SpacePatternCardProps) {
                 <div className={cn("truncate text-left")}>{displayText}</div>
               </div>
               <div className="text-muted-foreground flex shrink-0 items-center justify-end gap-2 text-xs">
-                <Badge variant="secondary" className="rounded text-xs">
+                <Badge
+                  variant="secondary"
+                  className="rounded text-xs"
+                  style={{
+                    color: getTailwindColor(pattern.type),
+                  }}
+                >
                   {pattern.type}
                 </Badge>
-                <Badge variant="secondary" className="rounded text-xs">
+                <Badge
+                  variant="secondary"
+                  className="rounded text-xs"
+                  style={{
+                    color: getTailwindColor(pattern.name),
+                  }}
+                >
                   {pattern.name}
                 </Badge>
               </div>
@@ -66,10 +79,22 @@ export function SpacePatternCard({ pattern }: SpacePatternCardProps) {
 
           <div className="flex flex-col gap-2">
             <div className="flex gap-2">
-              <Badge variant="secondary" className="rounded text-xs">
+              <Badge
+                variant="secondary"
+                className="rounded text-xs"
+                style={{
+                  color: getTailwindColor(pattern.type),
+                }}
+              >
                 {pattern.type}
               </Badge>
-              <Badge variant="secondary" className="rounded text-xs">
+              <Badge
+                variant="secondary"
+                className="rounded text-xs"
+                style={{
+                  color: getTailwindColor(pattern.name),
+                }}
+              >
                 {pattern.name}
               </Badge>
             </div>
diff --git a/apps/webapp/app/models/user.server.ts b/apps/webapp/app/models/user.server.ts
index 8fe6093..f5aa837 100644
--- a/apps/webapp/app/models/user.server.ts
+++ b/apps/webapp/app/models/user.server.ts
@@ -203,7 +203,8 @@ export function updateUser({
   marketingEmails,
   referralSource,
   onboardingComplete,
-}: Pick<User, "id" | "onboardingComplete"> & {
+  metadata,
+}: Pick<User, "id" | "onboardingComplete" | "metadata"> & {
   marketingEmails?: boolean;
   referralSource?: string;
 }) {
@@ -214,6 +215,7 @@ export function updateUser({
       referralSource,
       confirmedBasicDetails: true,
       onboardingComplete,
+      metadata: metadata ? metadata : {},
     },
   });
 }
diff --git a/apps/webapp/app/routes/onboarding.tsx b/apps/webapp/app/routes/onboarding.tsx
index 203f2ca..6be8882 100644
--- a/apps/webapp/app/routes/onboarding.tsx
+++ b/apps/webapp/app/routes/onboarding.tsx
@@ -61,6 +61,9 @@ export async function action({ request }: ActionFunctionArgs) {
     await updateUser({
       id: userId,
       onboardingComplete: true,
+      metadata: {
+        answers,
+      },
     });
 
     await addToQueue(
diff --git a/apps/webapp/prisma/schema.prisma b/apps/webapp/prisma/schema.prisma
index fff83cc..b68eeb0 100644
--- a/apps/webapp/prisma/schema.prisma
+++ b/apps/webapp/prisma/schema.prisma
@@ -531,6 +531,8 @@ model User {
 
   referralSource String?
 
+  metadata Json @default("{}")
+
   personalAccessTokens PersonalAccessToken[]
   InvitationCode       InvitationCode?        @relation(fields: [invitationCodeId], references: [id])
   invitationCodeId     String?
diff --git a/packages/database/prisma/schema.prisma b/packages/database/prisma/schema.prisma
index fde5c0e..b68eeb0 100644
--- a/packages/database/prisma/schema.prisma
+++ b/packages/database/prisma/schema.prisma
@@ -210,18 +210,6 @@ model InvitationCode {
   createdAt DateTime @default(now())
 }
 
-model MCPSession {
-  id           String   @id @default(cuid())
-  source       String
-  integrations String[]
-
-  workspace   Workspace? @relation(references: [id], fields: [workspaceId])
-  workspaceId String?
-
-  createdAt DateTime  @default(now())
-  deleted   DateTime?
-}
-
 model OAuthAuthorizationCode {
   id String @id @default(cuid())
 
@@ -543,6 +531,8 @@ model User {
 
   referralSource String?
 
+  metadata Json @default("{}")
+
   personalAccessTokens PersonalAccessToken[]
   InvitationCode       InvitationCode?        @relation(fields: [invitationCodeId], references: [id])
   invitationCodeId     String?
@@ -638,7 +628,6 @@ model Workspace {
   OAuthRefreshToken       OAuthRefreshToken[]
   RecallLog               RecallLog[]
   Space                   Space[]
-  MCPSession              MCPSession[]
 }
 
 enum AuthenticationMethod {