源码快照 @ 2026-09-01 (source-only, 剔除非源码素材)

packages/agent/gui/agent-gui/agentGuiNode/queuedPromptImageLoadOwner.ts

@@ -0,0 +1,113 @@
+import type { AgentGUIRuntime } from "../../agentActivityRuntime";
+
+export interface QueuedPromptImageLoadRequest {
+  agentSessionId: string;
+  attachmentId: string;
+  imageKey: string;
+  mimeType: string;
+  name: string;
+  path: string;
+  remoteUrl: string;
+  runtime: AgentGUIRuntime;
+  workspaceId: string;
+}
+
+export function queuedPromptImageLoadRequestIdentity(
+  request: Omit<QueuedPromptImageLoadRequest, "runtime">
+): string {
+  return [
+    request.imageKey,
+    request.workspaceId,
+    request.agentSessionId,
+    request.attachmentId,
+    request.path,
+    request.mimeType,
+    request.name,
+    request.remoteUrl
+  ].join("\u0000");
+}
+
+export function queuedPromptImageHasSafeRemoteUrl(value: string): boolean {
+  try {
+    const url = new URL(value);
+    return (
+      url.protocol === "https:" &&
+      Boolean(url.hostname) &&
+      !url.username &&
+      !url.password
+    );
+  } catch {
+    return false;
+  }
+}
+
+export class QueuedPromptImageLoadOwner {
+  private canceled = false;
+
+  constructor(
+    private readonly request: QueuedPromptImageLoadRequest,
+    private readonly onSource: (source: string | null) => void
+  ) {}
+
+  start(): void {
+    const {
+      agentSessionId,
+      attachmentId,
+      mimeType,
+      name,
+      path,
+      runtime,
+      workspaceId
+    } = this.request;
+    const readAsset = async (): Promise<
+      { data: string; mimeType: string } | undefined
+    > => {
+      if (attachmentId) {
+        return await runtime.readSessionAttachment?.({
+          workspaceId,
+          agentSessionId,
+          attachmentId
+        });
+      }
+      return await runtime.readPromptAsset?.({
+        workspaceId,
+        agentSessionId,
+        mimeType,
+        name,
+        path
+      });
+    };
+    void readAsset()
+      .then((asset) => {
+        if (this.canceled || !asset) {
+          return;
+        }
+        this.onSource(`data:${asset.mimeType};base64,${asset.data}`);
+      })
+      .catch((error: unknown) => {
+        if (this.canceled) {
+          return;
+        }
+        this.onSource(null);
+        console.warn(
+          "[agent-gui]",
+          JSON.stringify({
+            event: "agent.gui.queued_prompt_image_load_failed",
+            level: "warn",
+            source: "agent-gui",
+            workspaceId,
+            details: {
+              agentSessionId: agentSessionId || null,
+              attachmentId: attachmentId || null,
+              path: path || null,
+              error: error instanceof Error ? error.message : String(error)
+            }
+          })
+        );
+      });
+  }
+
+  dispose(): void {
+    this.canceled = true;
+  }
+}