@spotpatch/dev-server 0.7.0 → 0.8.0

package/dist/index.d.cts

@@ -155,6 +155,7 @@ declare const DEFAULT_OPTIONS: Readonly<{
             readonly graphMaxCallsites: 100;
             readonly analysisTimeoutMs: 2000;
             readonly reportMaxFields: 100;
+            readonly sourceMaxComponents: 256;
         }>;
     }>;
     externalAgent: Readonly<{
@@ -216,7 +217,7 @@ type SourceIdFactory = () => string;
 
 interface SourceRegistry {
     register(absolutePath: string): string;
-    registerDataFlowComponents(absolutePath: string, sourceVersion: string, components: readonly DataFlowComponentAnchorInput[]): void;
+    registerDataFlowComponents(absolutePath: string, sourceVersion: string, components: readonly DataFlowComponentAnchorInput[]): string;
     resolve(fileId: string): string | undefined;
     resolveDataFlowComponent(componentSourceId: string): DataFlowComponentAnchor | undefined;
     clear(): void;

package/dist/index.d.ts

@@ -155,6 +155,7 @@ declare const DEFAULT_OPTIONS: Readonly<{
             readonly graphMaxCallsites: 100;
             readonly analysisTimeoutMs: 2000;
             readonly reportMaxFields: 100;
+            readonly sourceMaxComponents: 256;
         }>;
     }>;
     externalAgent: Readonly<{
@@ -216,7 +217,7 @@ type SourceIdFactory = () => string;
 
 interface SourceRegistry {
     register(absolutePath: string): string;
-    registerDataFlowComponents(absolutePath: string, sourceVersion: string, components: readonly DataFlowComponentAnchorInput[]): void;
+    registerDataFlowComponents(absolutePath: string, sourceVersion: string, components: readonly DataFlowComponentAnchorInput[]): string;
     resolve(fileId: string): string | undefined;
     resolveDataFlowComponent(componentSourceId: string): DataFlowComponentAnchor | undefined;
     clear(): void;

package/dist/index.js

@@ -2478,6 +2478,7 @@ function createSourceRegistry(options = {}) {
         );
       }
       componentIdsByPath.set(normalizedPath, currentIds);
+      return fileId;
     },
     resolve(fileId) {
       return idToPath.get(fileId);
@@ -4161,17 +4162,33 @@ function createSpotPatchMiddleware(options) {
 
 // src/server/source-registration.ts
 import { timingSafeEqual as timingSafeEqual3 } from "crypto";
-import { lstat as lstat4, realpath as realpath6 } from "fs/promises";
+import { lstat as lstat4, readFile as readFile4, realpath as realpath6 } from "fs/promises";
 import path8 from "path";
-import { createSourceFilter } from "@spotpatch/compiler";
+import {
+  createDataFlowSourceFilter,
+  createDataFlowSourceVersion,
+  createSourceFilter
+} from "@spotpatch/compiler";
+import { DEFAULT_DATA_FLOW_LIMITS as DEFAULT_DATA_FLOW_LIMITS2 } from "@spotpatch/shared";
 import { z as z2 } from "zod";
-var REGISTRATION_BODY_LIMIT_BYTES = 4096;
+var REGISTRATION_BODY_LIMIT_BYTES = DEFAULT_DATA_FLOW_LIMITS2.protocolRequestMaxBytes;
 var REGISTRATION_IDENTITY_PATTERN = /^[A-Za-z0-9_-]{16,128}$/u;
+var DATA_FLOW_IDENTITY_PATTERN = /^[A-Za-z0-9_-]{1,128}$/u;
 var INTERNAL_SECRET_HEADER = "x-spotpatch-internal";
 var FORBIDDEN_SOURCE_SEGMENTS = /* @__PURE__ */ new Set([".next", "node_modules"]);
 var registrationRequestSchema = z2.strictObject({
   epoch: z2.string().regex(REGISTRATION_IDENTITY_PATTERN),
-  resourcePath: z2.string().min(1).max(3072)
+  resourcePath: z2.string().min(1).max(3072),
+  dataFlow: z2.strictObject({
+    sourceVersion: z2.string().regex(DATA_FLOW_IDENTITY_PATTERN),
+    components: z2.array(
+      z2.strictObject({
+        componentSourceId: z2.string().regex(DATA_FLOW_IDENTITY_PATTERN),
+        line: z2.number().int().positive().max(1e7),
+        column: z2.number().int().positive().max(1e7)
+      })
+    ).max(DEFAULT_DATA_FLOW_LIMITS2.sourceMaxComponents)
+  }).optional()
 });
 function getSingleHeader3(request, name) {
   const value = request.headers[name.toLowerCase()];
@@ -4235,6 +4252,33 @@ async function createSourceRegistrationService(input) {
   }
   const root = await realpath6(input.root);
   const sourceFilter = createSourceFilter(root, input.options);
+  const dataFlowFilter = createDataFlowSourceFilter(root, input.options);
+  const registrationQueues = /* @__PURE__ */ new Map();
+  async function registerCurrentSource(sourcePath, dataFlow) {
+    if (dataFlow === void 0) {
+      return input.registry.register(sourcePath);
+    }
+    let fileId;
+    const previous = registrationQueues.get(sourcePath) ?? Promise.resolve();
+    const current = previous.catch(() => void 0).then(async () => {
+      const currentSource = await readFile4(sourcePath, "utf8");
+      if (createDataFlowSourceVersion(currentSource) !== dataFlow.sourceVersion) {
+        return;
+      }
+      fileId = input.registry.registerDataFlowComponents(
+        sourcePath,
+        dataFlow.sourceVersion,
+        dataFlow.components
+      );
+    });
+    registrationQueues.set(sourcePath, current);
+    await current.finally(() => {
+      if (registrationQueues.get(sourcePath) === current) {
+        registrationQueues.delete(sourcePath);
+      }
+    });
+    return fileId;
+  }
   const handler = (request, response) => {
     const handle = async () => {
       const contentType = getSingleHeader3(request, "content-type")?.split(";", 1)[0]?.trim().toLowerCase();
@@ -4255,16 +4299,18 @@ async function createSourceRegistrationService(input) {
       const sourcePath = await resolveAuthorizedSource(
         root,
         parsed.data.resourcePath,
-        (absolutePath) => sourceFilter.shouldTransform(absolutePath, "<")
+        (absolutePath) => parsed.data.dataFlow === void 0 ? sourceFilter.shouldTransform(absolutePath, "<") : dataFlowFilter.shouldTransform(absolutePath, "")
       );
       if (sourcePath === void 0) {
         writeJson3(response, 403, { ok: false });
         return;
       }
-      writeJson3(response, 200, {
-        epoch: input.registryEpoch,
-        fileId: input.registry.register(sourcePath)
-      });
+      const fileId = await registerCurrentSource(sourcePath, parsed.data.dataFlow);
+      if (fileId === void 0) {
+        writeJson3(response, 409, { ok: false });
+        return;
+      }
+      writeJson3(response, 200, { epoch: input.registryEpoch, fileId });
     };
     void handle().catch(() => {
       if (!response.headersSent) {