@spotpatch/dev-server 0.7.0 → 0.8.0

package/dist/index.cjs

@@ -2472,6 +2472,7 @@ function createSourceRegistry(options = {}) {
         );
       }
       componentIdsByPath.set(normalizedPath, currentIds);
+      return fileId;
     },
     resolve(fileId) {
       return idToPath.get(fileId);
@@ -4095,14 +4096,26 @@ var import_node_crypto11 = require("crypto");
 var import_promises7 = require("fs/promises");
 var import_node_path8 = __toESM(require("path"), 1);
 var import_compiler = require("@spotpatch/compiler");
+var import_shared21 = require("@spotpatch/shared");
 var import_zod2 = require("zod");
-var REGISTRATION_BODY_LIMIT_BYTES = 4096;
+var REGISTRATION_BODY_LIMIT_BYTES = import_shared21.DEFAULT_DATA_FLOW_LIMITS.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 = import_zod2.z.strictObject({
   epoch: import_zod2.z.string().regex(REGISTRATION_IDENTITY_PATTERN),
-  resourcePath: import_zod2.z.string().min(1).max(3072)
+  resourcePath: import_zod2.z.string().min(1).max(3072),
+  dataFlow: import_zod2.z.strictObject({
+    sourceVersion: import_zod2.z.string().regex(DATA_FLOW_IDENTITY_PATTERN),
+    components: import_zod2.z.array(
+      import_zod2.z.strictObject({
+        componentSourceId: import_zod2.z.string().regex(DATA_FLOW_IDENTITY_PATTERN),
+        line: import_zod2.z.number().int().positive().max(1e7),
+        column: import_zod2.z.number().int().positive().max(1e7)
+      })
+    ).max(import_shared21.DEFAULT_DATA_FLOW_LIMITS.sourceMaxComponents)
+  }).optional()
 });
 function getSingleHeader3(request, name) {
   const value = request.headers[name.toLowerCase()];
@@ -4166,6 +4179,33 @@ async function createSourceRegistrationService(input) {
   }
   const root = await (0, import_promises7.realpath)(input.root);
   const sourceFilter = (0, import_compiler.createSourceFilter)(root, input.options);
+  const dataFlowFilter = (0, import_compiler.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 (0, import_promises7.readFile)(sourcePath, "utf8");
+      if ((0, import_compiler.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();
@@ -4186,16 +4226,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) {