1

server-node/src/services/customWorldSessionStore.ts

@@ -1,28 +1,21 @@
 import crypto from 'node:crypto';
 
-export type CustomWorldSessionStatus =
-  | 'clarifying'
-  | 'ready_to_generate'
-  | 'generating'
-  | 'completed'
-  | 'generation_error';
-
-export type CustomWorldQuestion = {
-  id: string;
-  label: string;
-  question: string;
-  answer?: string;
-};
+import type { JsonObject } from '../../../packages/shared/src/contracts/common.js';
+import type {
+  CustomWorldGenerationMode,
+  CustomWorldQuestion,
+  CustomWorldSessionStatus,
+} from '../../../packages/shared/src/contracts/runtime.js';
 
 export type CustomWorldSession = {
   sessionId: string;
   userId: string;
   status: CustomWorldSessionStatus;
   settingText: string;
-  creatorIntent: Record<string, unknown> | null;
-  generationMode: 'fast' | 'full';
+  creatorIntent: JsonObject | null;
+  generationMode: CustomWorldGenerationMode;
   questions: CustomWorldQuestion[];
-  result?: Record<string, unknown>;
+  result?: JsonObject;
   lastError?: string;
   createdAt: string;
   updatedAt: string;
@@ -38,7 +31,7 @@ function hasPendingQuestion(questions: CustomWorldQuestion[]) {
 
 function buildClarificationQuestions(
   settingText: string,
-  creatorIntent: Record<string, unknown> | null,
+  creatorIntent: JsonObject | null,
 ) {
   const questions: CustomWorldQuestion[] = [];
   const worldHook =
@@ -91,8 +84,8 @@ export class CustomWorldSessionStore {
   create(
     userId: string,
     settingText: string,
-    creatorIntent: Record<string, unknown> | null,
-    generationMode: 'fast' | 'full',
+    creatorIntent: JsonObject | null,
+    generationMode: CustomWorldGenerationMode,
   ) {
     const sessionId = `custom-world-session-${crypto.randomBytes(16).toString('hex')}`;
     const now = new Date().toISOString();
@@ -159,7 +152,7 @@ export class CustomWorldSessionStore {
     return cloneSession(session);
   }
 
-  setResult(userId: string, sessionId: string, result: Record<string, unknown>) {
+  setResult(userId: string, sessionId: string, result: JsonObject) {
     const session = this.sessions.get(userId)?.get(sessionId);
     if (!session) {
       return null;
@@ -167,7 +160,7 @@ export class CustomWorldSessionStore {
 
     session.status = 'completed';
     session.lastError = undefined;
-    session.result = JSON.parse(JSON.stringify(result)) as Record<string, unknown>;
+    session.result = JSON.parse(JSON.stringify(result)) as JsonObject;
     session.updatedAt = new Date().toISOString();
     return cloneSession(session);
   }