Create demo KYC pricing contexts

src/schemas/m2m.ts

@@ -1,3 +1,4 @@
+import { createHash } from 'node:crypto'
 import { prisma } from '../db.js'
 
 export const m2mTypeDefs = `#graphql
@@ -22,9 +23,47 @@ export const m2mTypeDefs = `#graphql
   type Mutation {
     createKycProfile(kycApplicationId: String!): CreateKycProfileResult
     createKycMonitoring(kycApplicationId: String!): CreateKycProfileResult
+    ensureDemoPricingContextForUser(userId: String!, countryCode: String!): KycPricingContext!
   }
 `
 
+function demoProfileUuid(userId: string): string {
+  const hex = createHash('sha256')
+    .update(`kyc:demo-pricing-context:${userId}`)
+    .digest('hex')
+    .slice(0, 32)
+    .split('')
+  hex[12] = '4'
+  hex[16] = '8'
+  const value = hex.join('')
+  return `${value.slice(0, 8)}-${value.slice(8, 12)}-${value.slice(12, 16)}-${value.slice(16, 20)}-${value.slice(20)}`
+}
+
+function normalizedCountryCode(countryCode: string): string {
+  const normalized = countryCode.trim().toUpperCase()
+  if (!/^[A-Z]{2}$/.test(normalized)) {
+    throw new Error('Demo pricing country code must contain two letters')
+  }
+  return normalized
+}
+
+function mapPricingContext(profile: {
+  userId: string
+  uuid: string
+  countryCode: string
+  updatedAt: Date
+}) {
+  if (profile.countryCode.length !== 2) {
+    throw new Error(`KYC pricing profile ${profile.uuid} has no valid country code`)
+  }
+  return {
+    userId: profile.userId,
+    profileUuid: profile.uuid,
+    countryCode: profile.countryCode.toUpperCase(),
+    resolvedAt: profile.updatedAt.toISOString(),
+  }
+}
+
 async function createProfile(_: unknown, args: { kycApplicationId: string }) {
   try {
     const app = await prisma.kYCApplication.findUnique({
@@ -72,7 +111,7 @@ async function pricingContextForUser(_: unknown, args: { userId: string }) {
     where: {
       userId: args.userId,
       workflowStatus: {
-        in: ['active', 'approved'],
+        in: ['active', 'approved', 'demo'],
       },
     },
     orderBy: {
@@ -80,15 +119,45 @@ async function pricingContextForUser(_: unknown, args: { userId: string }) {
     },
   })
   if (profile === null) return null
-  if (profile.countryCode.length !== 2) {
-    throw new Error(`Active KYC profile ${profile.uuid} has no valid country code`)
-  }
-  return {
-    userId: profile.userId,
-    profileUuid: profile.uuid,
-    countryCode: profile.countryCode.toUpperCase(),
-    resolvedAt: profile.updatedAt.toISOString(),
-  }
+  return mapPricingContext(profile)
+}
+
+async function ensureDemoPricingContextForUser(
+  _: unknown,
+  args: { userId: string; countryCode: string },
+) {
+  const existing = await prisma.kYCProfile.findFirst({
+    where: {
+      userId: args.userId,
+      workflowStatus: {
+        in: ['active', 'approved', 'demo'],
+      },
+    },
+    orderBy: {
+      updatedAt: 'desc',
+    },
+  })
+  if (existing !== null) return mapPricingContext(existing)
+
+  const countryCode = normalizedCountryCode(args.countryCode)
+  const profileUuid = demoProfileUuid(args.userId)
+  const profile = await prisma.kYCProfile.upsert({
+    where: { uuid: profileUuid },
+    create: {
+      uuid: profileUuid,
+      userId: args.userId,
+      teamName: 'Demo workspace',
+      countryCode,
+      workflowStatus: 'demo',
+    },
+    update: {
+      userId: args.userId,
+      teamName: 'Demo workspace',
+      countryCode,
+      workflowStatus: 'demo',
+    },
+  })
+  return mapPricingContext(profile)
 }
 
 export const m2mResolvers = {
@@ -99,5 +168,6 @@ export const m2mResolvers = {
   Mutation: {
     createKycProfile: createProfile,
     createKycMonitoring: createProfile,
+    ensureDemoPricingContextForUser,
   },
 }