Initial TrueGrowth source import

packages/drawnix/src/utils/image-generation-anchor-controller.ts

@@ -0,0 +1,75 @@
+import type { WorkflowMessageData } from '../types/chat.types';
+import type {
+  ImageGenerationAnchorPhase,
+  ImageGenerationAnchorViewModel,
+  PlaitImageGenerationAnchor,
+} from '../types/image-generation-anchor.types';
+import type { Task } from '../types/task.types';
+import {
+  buildImageGenerationAnchorViewModel,
+  deriveImageGenerationAnchorPhase,
+} from './image-generation-anchor-view-model';
+
+export interface ImageGenerationAnchorControllerOptions {
+  anchor: PlaitImageGenerationAnchor;
+  task?: Task | null;
+  tasks?: Task[];
+  workflow?: WorkflowMessageData | null;
+  postProcessingStatus?: WorkflowMessageData['postProcessingStatus'];
+  isInserting?: boolean;
+  hasInserted?: boolean;
+  taskDisplayProgress?: number | null;
+}
+
+export interface ImageGenerationAnchorControllerResult {
+  phase: ImageGenerationAnchorPhase;
+  nextPatch: Partial<PlaitImageGenerationAnchor>;
+  viewModel: ImageGenerationAnchorViewModel;
+}
+
+export function getImageGenerationAnchorControllerResult(
+  options: ImageGenerationAnchorControllerOptions
+): ImageGenerationAnchorControllerResult {
+  const {
+    anchor,
+    task,
+    tasks,
+    taskDisplayProgress,
+    workflow,
+    postProcessingStatus,
+    isInserting,
+    hasInserted,
+  } = options;
+
+  const phase = deriveImageGenerationAnchorPhase({
+    anchor,
+    task,
+    tasks,
+    workflow,
+    postProcessingStatus,
+    isInserting,
+    hasInserted,
+  });
+
+  const viewModel = buildImageGenerationAnchorViewModel({
+    anchor,
+    task,
+    tasks,
+    taskDisplayProgress,
+    workflow,
+    postProcessingStatus,
+    isInserting,
+    hasInserted,
+  });
+
+  return {
+    phase,
+    nextPatch: {
+      phase,
+      progress: viewModel.progress,
+      subtitle: viewModel.subtitle,
+      error: viewModel.error,
+    },
+    viewModel,
+  };
+}