Auto-open draft result after foundation completes

src/services/customWorldAgentUiState.ts

@@ -2,6 +2,8 @@ import type { CustomWorldAgentUiState } from '../types';
 
 export const CUSTOM_WORLD_AGENT_SESSION_QUERY_KEY = 'customWorldSessionId';
 export const CUSTOM_WORLD_AGENT_OPERATION_QUERY_KEY = 'customWorldOperationId';
+export const CUSTOM_WORLD_GENERATION_SOURCE_QUERY_KEY =
+  'customWorldGenerationSource';
 export const CUSTOM_WORLD_AGENT_UI_STATE_STORAGE_KEY =
   'genarrative.custom-world-agent-ui.v1';
 
@@ -50,6 +52,10 @@ function normalizeValue(value: unknown) {
   return typeof value === 'string' && value.trim() ? value.trim() : null;
 }
 
+function normalizeGenerationSource(value: unknown) {
+  return value === 'agent-draft-foundation' ? value : null;
+}
+
 export function readCustomWorldAgentUiState(
   env?: CustomWorldAgentUiEnvironment,
 ): CustomWorldAgentUiState {
@@ -62,9 +68,16 @@ export function readCustomWorldAgentUiState(
     activeOperationId: normalizeValue(
       params.get(CUSTOM_WORLD_AGENT_OPERATION_QUERY_KEY),
     ),
+    customWorldGenerationSource: normalizeGenerationSource(
+      params.get(CUSTOM_WORLD_GENERATION_SOURCE_QUERY_KEY),
+    ),
   };
 
-  if (stateFromQuery.activeSessionId || stateFromQuery.activeOperationId) {
+  if (
+    stateFromQuery.activeSessionId ||
+    stateFromQuery.activeOperationId ||
+    stateFromQuery.customWorldGenerationSource
+  ) {
     return stateFromQuery;
   }
 
@@ -80,6 +93,9 @@ export function readCustomWorldAgentUiState(
     return {
       activeSessionId: normalizeValue(parsed.activeSessionId),
       activeOperationId: normalizeValue(parsed.activeOperationId),
+      customWorldGenerationSource: normalizeGenerationSource(
+        parsed.customWorldGenerationSource,
+      ),
       ownerUserId: normalizeValue(parsed.ownerUserId),
     };
   } catch {
@@ -95,10 +111,14 @@ export function writeCustomWorldAgentUiState(
   const resolved = resolveEnvironment(env);
   const activeSessionId = normalizeValue(state.activeSessionId);
   const activeOperationId = normalizeValue(state.activeOperationId);
+  const customWorldGenerationSource = normalizeGenerationSource(
+    state.customWorldGenerationSource,
+  );
   const ownerUserId = normalizeValue(state.ownerUserId);
   const nextState: CustomWorldAgentUiState = {
     activeSessionId,
     activeOperationId,
+    customWorldGenerationSource,
     ownerUserId,
   };
 
@@ -116,6 +136,15 @@ export function writeCustomWorldAgentUiState(
       params.delete(CUSTOM_WORLD_AGENT_OPERATION_QUERY_KEY);
     }
 
+    if (customWorldGenerationSource) {
+      params.set(
+        CUSTOM_WORLD_GENERATION_SOURCE_QUERY_KEY,
+        customWorldGenerationSource,
+      );
+    } else {
+      params.delete(CUSTOM_WORLD_GENERATION_SOURCE_QUERY_KEY);
+    }
+
     const search = params.toString();
     const nextUrl = search
       ? `${resolved.location.pathname}?${search}`
@@ -124,7 +153,7 @@ export function writeCustomWorldAgentUiState(
   }
 
   if (resolved.sessionStorage) {
-    if (activeSessionId || activeOperationId) {
+    if (activeSessionId || activeOperationId || customWorldGenerationSource) {
       resolved.sessionStorage.setItem(
         CUSTOM_WORLD_AGENT_UI_STATE_STORAGE_KEY,
         JSON.stringify(nextState),