Integrate role asset studio into custom world agent flow

server-node/src/services/customWorldSessionStore.ts

@@ -4,12 +4,13 @@ import type { JsonObject } from '../../../packages/shared/src/contracts/common.j
 import type {
   CustomWorldGenerationMode,
   CustomWorldQuestion,
+  CustomWorldSessionRecord,
   CustomWorldSessionStatus,
 } from '../../../packages/shared/src/contracts/runtime.js';
+import type { RuntimeRepositoryPort } from '../repositories/runtimeRepository.js';
 
 export type CustomWorldSession = {
   sessionId: string;
-  userId: string;
   status: CustomWorldSessionStatus;
   settingText: string;
   creatorIntent: JsonObject | null;
@@ -25,6 +26,36 @@ function cloneSession(session: CustomWorldSession) {
   return JSON.parse(JSON.stringify(session)) as CustomWorldSession;
 }
 
+function toSessionRecord(session: CustomWorldSession): CustomWorldSessionRecord {
+  return {
+    sessionId: session.sessionId,
+    status: session.status,
+    settingText: session.settingText,
+    creatorIntent: session.creatorIntent,
+    generationMode: session.generationMode,
+    questions: session.questions,
+    result: session.result,
+    lastError: session.lastError,
+    createdAt: session.createdAt,
+    updatedAt: session.updatedAt,
+  };
+}
+
+function toSession(record: CustomWorldSessionRecord) {
+  return cloneSession({
+    sessionId: record.sessionId,
+    status: record.status,
+    settingText: record.settingText,
+    creatorIntent: record.creatorIntent ?? null,
+    generationMode: record.generationMode,
+    questions: record.questions,
+    result: record.result,
+    lastError: record.lastError,
+    createdAt: record.createdAt,
+    updatedAt: record.updatedAt,
+  });
+}
+
 function hasPendingQuestion(questions: CustomWorldQuestion[]) {
   return questions.some((question) => !question.answer?.trim());
 }
@@ -79,9 +110,11 @@ function buildClarificationQuestions(
 }
 
 export class CustomWorldSessionStore {
-  private readonly sessions = new Map<string, Map<string, CustomWorldSession>>();
+  constructor(
+    private readonly runtimeRepository: RuntimeRepositoryPort,
+  ) {}
 
-  create(
+  async create(
     userId: string,
     settingText: string,
     creatorIntent: JsonObject | null,
@@ -91,7 +124,6 @@ export class CustomWorldSessionStore {
     const now = new Date().toISOString();
     const session: CustomWorldSession = {
       sessionId,
-      userId,
       status: 'ready_to_generate',
       settingText,
       creatorIntent,
@@ -105,19 +137,34 @@ export class CustomWorldSessionStore {
       session.status = 'clarifying';
     }
 
-    const userSessions = this.sessions.get(userId) ?? new Map<string, CustomWorldSession>();
-    userSessions.set(sessionId, session);
-    this.sessions.set(userId, userSessions);
+    await this.runtimeRepository.upsertCustomWorldSession(
+      userId,
+      sessionId,
+      toSessionRecord(session),
+    );
     return cloneSession(session);
   }
 
-  get(userId: string, sessionId: string) {
-    const session = this.sessions.get(userId)?.get(sessionId);
-    return session ? cloneSession(session) : null;
+  async list(userId: string) {
+    const sessions = await this.runtimeRepository.listCustomWorldSessions(userId);
+    return sessions.map((session) => toSession(session));
   }
 
-  answer(userId: string, sessionId: string, questionId: string, answer: string) {
-    const session = this.sessions.get(userId)?.get(sessionId);
+  async get(userId: string, sessionId: string) {
+    const session = await this.runtimeRepository.getCustomWorldSession(
+      userId,
+      sessionId,
+    );
+    return session ? toSession(session) : null;
+  }
+
+  async answer(
+    userId: string,
+    sessionId: string,
+    questionId: string,
+    answer: string,
+  ) {
+    const session = await this.get(userId, sessionId);
     if (!session) {
       return null;
     }
@@ -132,16 +179,21 @@ export class CustomWorldSessionStore {
       ? 'clarifying'
       : 'ready_to_generate';
     session.updatedAt = new Date().toISOString();
+    await this.runtimeRepository.upsertCustomWorldSession(
+      userId,
+      sessionId,
+      toSessionRecord(session),
+    );
     return cloneSession(session);
   }
 
-  updateStatus(
+  async updateStatus(
     userId: string,
     sessionId: string,
     status: CustomWorldSessionStatus,
     lastError = '',
   ) {
-    const session = this.sessions.get(userId)?.get(sessionId);
+    const session = await this.get(userId, sessionId);
     if (!session) {
       return null;
     }
@@ -149,11 +201,16 @@ export class CustomWorldSessionStore {
     session.status = status;
     session.lastError = lastError || undefined;
     session.updatedAt = new Date().toISOString();
+    await this.runtimeRepository.upsertCustomWorldSession(
+      userId,
+      sessionId,
+      toSessionRecord(session),
+    );
     return cloneSession(session);
   }
 
-  setResult(userId: string, sessionId: string, result: JsonObject) {
-    const session = this.sessions.get(userId)?.get(sessionId);
+  async setResult(userId: string, sessionId: string, result: JsonObject) {
+    const session = await this.get(userId, sessionId);
     if (!session) {
       return null;
     }
@@ -162,6 +219,11 @@ export class CustomWorldSessionStore {
     session.lastError = undefined;
     session.result = JSON.parse(JSON.stringify(result)) as JsonObject;
     session.updatedAt = new Date().toISOString();
+    await this.runtimeRepository.upsertCustomWorldSession(
+      userId,
+      sessionId,
+      toSessionRecord(session),
+    );
     return cloneSession(session);
   }
 }