@replit/river 0.200.0-rc.2 → 0.200.0-rc.4

package/dist/{chunk-7CKIN3JT.js → chunk-SIRRYRLQ.js}

@@ -1,15 +1,19 @@
 import {
   ControlMessageCloseSchema,
   ControlMessagePayloadSchema,
+  abortMessage,
+  closeStreamMessage,
   coerceErrorString,
   createHandlerSpan,
   createProcTelemetryInfo,
+  generateId,
   getPropagationContext,
   isStreamAbort,
   isStreamClose,
   isStreamCloseRequest,
-  isStreamOpen
-} from "./chunk-S5RL45KH.js";
+  isStreamOpen,
+  requestCloseStreamMessage
+} from "./chunk-V57VWV5S.js";
 
 // router/services.ts
 import { Type } from "@sinclair/typebox";
@@ -213,468 +217,6 @@ var ServiceScaffold = class {
   }
 };
 
-// router/diff.ts
-function diffServerSchema(oldServer, newServer, options) {
-  const allServices = /* @__PURE__ */ new Set([
-    ...Object.keys(oldServer.services),
-    ...Object.keys(newServer.services)
-  ]);
-  const breakages = {};
-  for (const serviceName of allServices) {
-    const oldService = oldServer.services[serviceName];
-    const newService = newServer.services[serviceName];
-    const breakage = diffService(oldService, newService, options);
-    if (breakage) {
-      breakages[serviceName] = breakage;
-    }
-  }
-  if (Object.keys(breakages).length) {
-    return { serviceBreakages: breakages };
-  }
-  return null;
-}
-function diffService(oldService, newService, options) {
-  if (!newService) {
-    return options?.allowServiceRemoval ? null : { reason: "removed" };
-  }
-  if (!oldService) {
-    return null;
-  }
-  const allProcedures = /* @__PURE__ */ new Set([
-    ...Object.keys(oldService.procedures),
-    ...Object.keys(newService.procedures)
-  ]);
-  const breakages = {};
-  for (const procedureName of allProcedures) {
-    const aProcedure = oldService.procedures[procedureName];
-    const bProcedure = newService.procedures[procedureName];
-    const breakage = diffProcedure(aProcedure, bProcedure, options);
-    if (breakage) {
-      breakages[procedureName] = breakage;
-    }
-  }
-  if (Object.keys(breakages).length) {
-    return { reason: "modified", procedureBreakages: breakages };
-  }
-  return null;
-}
-function diffProcedure(oldProcedure, newProcedure, options) {
-  if (!newProcedure) {
-    return options?.allowProcedureRemoval ? null : { reason: "removed" };
-  }
-  if (!oldProcedure) {
-    return null;
-  }
-  if (oldProcedure.type !== newProcedure.type) {
-    return {
-      reason: "type-changed",
-      oldType: oldProcedure.type,
-      newType: newProcedure.type
-    };
-  }
-  const inputBreakage = diffProcedureField(
-    oldProcedure.input,
-    newProcedure.input,
-    "client"
-  );
-  const initBreakage = diffProcedureField(
-    oldProcedure.init,
-    newProcedure.init,
-    "client"
-  );
-  const outputBreakage = diffProcedureField(
-    oldProcedure.output,
-    newProcedure.output,
-    "server"
-  );
-  if (inputBreakage ?? initBreakage ?? outputBreakage) {
-    const result = {
-      reason: "modified"
-    };
-    if (inputBreakage) {
-      result.input = inputBreakage;
-    }
-    if (initBreakage) {
-      result.init = initBreakage;
-    }
-    if (outputBreakage) {
-      result.output = outputBreakage;
-    }
-    return result;
-  }
-  return null;
-}
-function diffProcedureField(oldSchema, newSchema, origin) {
-  if (!oldSchema && !newSchema) {
-    return null;
-  }
-  const diffBreakage = diffRequired(oldSchema, newSchema, origin, false, false);
-  if (diffBreakage) {
-    return diffBreakage;
-  }
-  if (!oldSchema || !newSchema) {
-    throw new Error("Appease typescript, this should never happen");
-  }
-  return diffJSONSchema(oldSchema, newSchema, origin);
-}
-function diffRequired(oldSchema, newSchema, origin, oldRequired, newRequired) {
-  if (!newSchema && !oldSchema) {
-    throw new Error("Both old and new schema are undefined");
-  }
-  if (!newSchema) {
-    if (!oldRequired && origin == "server") {
-      return null;
-    }
-    return { reason: "removed-required" };
-  }
-  if (!oldSchema) {
-    if (newRequired && origin === "client") {
-      return { reason: "new-required" };
-    }
-    return null;
-  }
-  if (origin === "client" && !oldRequired && newRequired) {
-    return { reason: "new-required" };
-  }
-  if (origin === "server" && oldRequired && !newRequired) {
-    return { reason: "removed-required" };
-  }
-  return null;
-}
-function diffJSONSchema(oldSchema, newSchema, origin) {
-  if (oldSchema.type !== newSchema.type) {
-    return {
-      reason: "type-changed",
-      oldType: getReportingType(oldSchema),
-      newType: getReportingType(newSchema)
-    };
-  }
-  if (getReportingType(oldSchema) !== getReportingType(newSchema)) {
-    return {
-      reason: "type-changed",
-      oldType: getReportingType(oldSchema),
-      newType: getReportingType(newSchema)
-    };
-  }
-  if ("const" in oldSchema && "const" in newSchema && oldSchema.const !== newSchema.const) {
-    return {
-      reason: "type-changed",
-      oldType: `${getReportingType(oldSchema)}-const-${oldSchema.const}`,
-      newType: `${getReportingType(newSchema)}-const-${newSchema.const}`
-    };
-  }
-  if ("const" in oldSchema && !("const" in newSchema) && origin === "server") {
-    return {
-      reason: "type-changed",
-      oldType: `${getReportingType(oldSchema)}-const-${oldSchema.const}`,
-      newType: getReportingType(newSchema)
-    };
-  }
-  if ("const" in newSchema && !("const" in oldSchema) && origin === "client") {
-    return {
-      reason: "type-changed",
-      oldType: getReportingType(oldSchema),
-      newType: `${getReportingType(newSchema)}-const-${newSchema.const}`
-    };
-  }
-  const breakages = {};
-  if ("$ref" in newSchema) {
-    if (newSchema.$ref !== oldSchema.$ref) {
-      return {
-        reason: "type-changed",
-        oldType: getReportingType(oldSchema),
-        newType: getReportingType(newSchema)
-      };
-    }
-  } else if ("not" in newSchema) {
-    const notBreakage = diffJSONSchema(
-      oldSchema.not,
-      newSchema.not,
-      origin
-    );
-    if (notBreakage) {
-      breakages.not = notBreakage;
-    }
-  } else if ("anyOf" in newSchema) {
-    const oldAnyOfStringified = oldSchema.anyOf.map((el) => JSON.stringify(el)).sort();
-    const newAnyOfStringified = newSchema.anyOf.map((el) => JSON.stringify(el)).sort();
-    const anyOfBreakages = {};
-    for (let i = 0; i < oldAnyOfStringified.length; i++) {
-      if (newAnyOfStringified.includes(oldAnyOfStringified[i])) {
-        continue;
-      }
-      if (!newAnyOfStringified[i]) {
-        if (origin === "server") {
-          continue;
-        }
-        anyOfBreakages[`old-${i}`] = { reason: "removed-required" };
-      } else {
-        const breakage = diffJSONSchema(
-          JSON.parse(oldAnyOfStringified[i]),
-          JSON.parse(newAnyOfStringified[i]),
-          origin
-        );
-        if (breakage) {
-          anyOfBreakages[`old-${i}`] = breakage;
-        }
-      }
-    }
-    for (let i = 0; i < newAnyOfStringified.length; i++) {
-      if (oldAnyOfStringified.includes(newAnyOfStringified[i])) {
-        continue;
-      }
-      if (!oldAnyOfStringified[i]) {
-        if (origin === "client") {
-          continue;
-        }
-        anyOfBreakages[`new-${i}`] = { reason: "new-required" };
-      } else {
-        const breakage = diffJSONSchema(
-          JSON.parse(oldAnyOfStringified[i]),
-          JSON.parse(newAnyOfStringified[i]),
-          origin
-        );
-        if (breakage) {
-          anyOfBreakages[`new-${i}`] = breakage;
-        }
-      }
-    }
-    if (Object.keys(anyOfBreakages).length > 0) {
-      breakages.anyOf = {
-        reason: "field-breakage",
-        fieldBreakages: anyOfBreakages
-      };
-    }
-  } else if ("oneOf" in newSchema) {
-    throw new Error("oneOf is not supported, typebox does not emit it");
-  } else if ("allOf" in newSchema) {
-    if (newSchema.allOf.length !== oldSchema.allOf.length) {
-      breakages.allOf = {
-        reason: "type-changed",
-        oldType: `${oldSchema.allOf}`,
-        newType: `${newSchema.allOf}`
-      };
-    } else {
-      for (let i = 0; i < newSchema.allOf.length; i++) {
-        const breakage = diffJSONSchema(
-          oldSchema.allOf[i],
-          newSchema.allOf[i],
-          origin
-        );
-        if (breakage) {
-          breakages.allOf = breakage;
-          break;
-        }
-      }
-    }
-  } else if (newSchema.type === "array") {
-    const itemsBreakages = diffJSONSchema(
-      oldSchema.items,
-      newSchema.items,
-      origin
-    );
-    if (itemsBreakages) {
-      breakages.items = itemsBreakages;
-    }
-    if (oldSchema.minItems < newSchema.minItems) {
-      if (origin === "client") {
-        breakages.minItems = {
-          reason: "type-changed",
-          oldType: `${oldSchema.minItems}`,
-          newType: `${newSchema.minItems}`
-        };
-      }
-    } else if (oldSchema.minItems > newSchema.minItems) {
-      if (origin === "server") {
-        breakages.minItems = {
-          reason: "type-changed",
-          oldType: `${oldSchema.minItems}`,
-          newType: `${newSchema.minItems}`
-        };
-      }
-    }
-    if (oldSchema.maxItems < newSchema.maxItems) {
-      if (origin === "server") {
-        breakages.maxItems = {
-          reason: "type-changed",
-          oldType: `${oldSchema.maxItems}`,
-          newType: `${newSchema.maxItems}`
-        };
-      }
-    } else if (oldSchema.maxItems > newSchema.maxItems) {
-      if (origin === "client") {
-        breakages.maxItems = {
-          reason: "type-changed",
-          oldType: `${oldSchema.maxItems}`,
-          newType: `${newSchema.maxItems}`
-        };
-      }
-    }
-    if (!oldSchema.uniqueItems && newSchema.uniqueItems && origin === "client") {
-      breakages.uniqueItems = {
-        reason: "type-changed",
-        oldType: `${!!oldSchema.uniqueItems}`,
-        newType: `${!!newSchema.uniqueItems}`
-      };
-    }
-    if ("contains" in newSchema !== "contains" in oldSchema) {
-      if ("contains" in newSchema && !("contains" in oldSchema) && origin === "client") {
-        breakages.contains = {
-          reason: "type-changed",
-          oldType: "no-contains",
-          newType: "contains"
-        };
-      }
-    } else if ("contains" in newSchema) {
-      const containsBreakage = diffJSONSchema(
-        oldSchema.contains,
-        newSchema.contains,
-        origin
-      );
-      if (containsBreakage) {
-        breakages.contains = containsBreakage;
-      }
-    }
-    if (oldSchema.minContains < newSchema.minContains) {
-      if (origin === "client") {
-        breakages.minContains = {
-          reason: "type-changed",
-          oldType: `${oldSchema.minContains}`,
-          newType: `${newSchema.minContains}`
-        };
-      }
-    } else if (oldSchema.minContains > newSchema.minContains) {
-      if (origin === "server") {
-        breakages.minContains = {
-          reason: "type-changed",
-          oldType: `${oldSchema.minContains}`,
-          newType: `${newSchema.minContains}`
-        };
-      }
-    }
-    if (oldSchema.maxContains < newSchema.maxContains) {
-      if (origin === "server") {
-        breakages.maxContains = {
-          reason: "type-changed",
-          oldType: `${oldSchema.maxContains}`,
-          newType: `${newSchema.maxContains}`
-        };
-      }
-    } else if (oldSchema.maxContains > newSchema.maxContains) {
-      if (origin === "client") {
-        breakages.maxContains = {
-          reason: "type-changed",
-          oldType: `${oldSchema.maxContains}`,
-          newType: `${newSchema.maxContains}`
-        };
-      }
-    }
-  } else if (newSchema.type === "object") {
-    if ("properties" in newSchema !== "properties" in oldSchema) {
-      return {
-        reason: "type-changed",
-        oldType: "properties" in oldSchema ? "probably-object" : "probably-record",
-        newType: "properties" in newSchema ? "probably-object" : "probably-record"
-      };
-    }
-    if ("properties" in newSchema) {
-      const propertiesBreakages = diffObjectProperties(
-        oldSchema.properties,
-        newSchema.properties,
-        origin,
-        oldSchema.required,
-        newSchema.required
-      );
-      if (Object.keys(propertiesBreakages).length) {
-        breakages.properties = {
-          reason: "field-breakage",
-          fieldBreakages: propertiesBreakages
-        };
-      }
-    }
-    if ("patternProperties" in newSchema) {
-      const patternPropertiesBreakages = diffObjectProperties(
-        oldSchema.patternProperties,
-        newSchema.patternProperties,
-        origin,
-        oldSchema.required,
-        newSchema.required
-      );
-      if (Object.keys(patternPropertiesBreakages).length) {
-        breakages.patternProperties = {
-          reason: "field-breakage",
-          fieldBreakages: patternPropertiesBreakages
-        };
-      }
-    }
-    if ("additionalProperties" in newSchema || "additionalProperties" in oldSchema) {
-      throw new Error("additionalProperties is not supported");
-    }
-    if ("minProperties" in newSchema || "minProperties" in oldSchema) {
-      throw new Error("minProperties is not supported");
-    }
-    if ("maxProperties" in newSchema || "maxProperties" in oldSchema) {
-      throw new Error("maxProperties is not supported");
-    }
-  }
-  if (Object.keys(breakages).length) {
-    return {
-      reason: "field-breakage",
-      fieldBreakages: breakages
-    };
-  }
-  return null;
-}
-function diffObjectProperties(oldProperties, newProperties, origin, oldRequiredProperties = [], newRequiredProperties = []) {
-  const allProperties = /* @__PURE__ */ new Set([
-    ...Object.keys(oldProperties),
-    ...Object.keys(newProperties)
-  ]);
-  const breakages = {};
-  for (const propertyName of allProperties) {
-    const requiredBreakage = diffRequired(
-      oldProperties[propertyName],
-      newProperties[propertyName],
-      origin,
-      oldRequiredProperties.includes(propertyName),
-      newRequiredProperties.includes(propertyName)
-    );
-    if (requiredBreakage) {
-      breakages[propertyName] = requiredBreakage;
-    } else if (oldProperties[propertyName] && newProperties[propertyName]) {
-      const propertyBreakage = diffJSONSchema(
-        oldProperties[propertyName],
-        newProperties[propertyName],
-        origin
-      );
-      if (propertyBreakage) {
-        breakages[propertyName] = propertyBreakage;
-      }
-    }
-  }
-  return breakages;
-}
-function getReportingType(schema) {
-  if ("not" in schema) {
-    return "not";
-  }
-  if ("anyOf" in schema) {
-    return "anyOf";
-  }
-  if ("allOf" in schema) {
-    return "allOf";
-  }
-  if ("$ref" in schema) {
-    return "$ref";
-  }
-  if (schema.type && typeof schema.type === "string") {
-    return schema.type;
-  }
-  throw new Error(
-    "Subschema not supported, probably a conditional subschema. Check logs."
-  );
-}
-
 // router/result.ts
 import {
   Type as Type2
@@ -1082,7 +624,6 @@ var Procedure = {
 };
 
 // router/client.ts
-import { nanoid } from "nanoid";
 import { Value } from "@sinclair/typebox/value";
 var OutputErrResultSchema = ErrResultSchema(OutputReaderErrorSchema);
 var noop = () => {
@@ -1115,7 +656,7 @@ function createClient(transport, serverId, providedClientOptions = {}) {
   }
   const clientOptions = { ...defaultClientOptions, ...providedClientOptions };
   if (clientOptions.eagerlyConnect) {
-    void transport.connect(serverId);
+    transport.connect(serverId);
   }
   return _createRecursiveProxy((opts) => {
     const [serviceName, procName, procMethod] = [...opts.path];
@@ -1125,8 +666,8 @@ function createClient(transport, serverId, providedClientOptions = {}) {
       );
     }
     const [init, callOptions] = opts.args;
-    if (clientOptions.connectOnInvoke && !transport.connections.has(serverId)) {
-      void transport.connect(serverId);
+    if (clientOptions.connectOnInvoke && !transport.sessions.has(serverId)) {
+      transport.connect(serverId);
     }
     if (procMethod !== "rpc" && procMethod !== "subscribe" && procMethod !== "stream" && procMethod !== "upload") {
       throw new Error(
@@ -1146,7 +687,7 @@ function createClient(transport, serverId, providedClientOptions = {}) {
 }
 function handleProc(procType, transport, serverId, init, serviceName, procedureName, abortSignal) {
   const procClosesWithInit = procType === "rpc" || procType === "subscription";
-  const streamId = nanoid();
+  const streamId = generateId();
   const { span, ctx } = createProcTelemetryInfo(
     transport,
     procType,
@@ -1166,14 +707,14 @@ function handleProc(procType, transport, serverId, init, serviceName, procedureN
   inputWriter.onClose(() => {
     span.addEvent("inputWriter closed");
     if (!procClosesWithInit && cleanClose) {
-      transport.sendCloseControl(serverId, streamId);
+      transport.send(serverId, closeStreamMessage(streamId));
     }
     if (outputReader.isClosed()) {
       cleanup();
     }
   });
   const outputReader = new ReadStreamImpl(() => {
-    transport.sendRequestCloseControl(serverId, streamId);
+    transport.send(serverId, requestCloseStreamMessage(streamId));
   });
   outputReader.onClose(() => {
     span.addEvent("outputReader closed");
@@ -1203,13 +744,15 @@ function handleProc(procType, transport, serverId, init, serviceName, procedureN
       outputReader.triggerClose();
     }
     inputWriter.close();
-    transport.sendAbort(
+    transport.send(
       serverId,
-      streamId,
-      Err({
-        code: ABORT_CODE,
-        message: "Aborted by client"
-      })
+      abortMessage(
+        streamId,
+        Err({
+          code: ABORT_CODE,
+          message: "Aborted by client"
+        })
+      )
     );
   }
   function onMessage(msg) {
@@ -1438,7 +981,9 @@ var RiverServer = class {
     controlFlags,
     initPayload,
     from,
-    tracingCtx
+    sessionId,
+    tracingCtx,
+    protocolVersion
   }) {
     this.openStreams.add(streamId);
     let cleanClose = true;
@@ -1501,7 +1046,7 @@ var RiverServer = class {
       if (isStreamCloseRequest(msg.controlFlags)) {
         outputWriter.triggerCloseRequest();
       }
-      if (isStreamAbort(msg.controlFlags)) {
+      if (isStreamAbortBackwardsCompat(msg.controlFlags, protocolVersion)) {
         let abortResult;
         if (Value2.Check(InputErrResultSchema, msg.payload)) {
           abortResult = msg.payload;
@@ -1568,7 +1113,7 @@ var RiverServer = class {
           })
         );
       }
-      if (isStreamClose(msg.controlFlags)) {
+      if (isStreamCloseBackwardsCompat(msg.controlFlags, protocolVersion)) {
         inputReader.triggerClose();
       }
     };
@@ -1587,25 +1132,37 @@ var RiverServer = class {
       });
       onFinishedCallbacks.length = 0;
     };
+    const procClosesWithResponse = procedure.type === "rpc" || procedure.type === "upload";
     const inputReader = new ReadStreamImpl(() => {
-      this.transport.sendRequestCloseControl(from, streamId);
+      this.transport.send(from, requestCloseStreamMessage(streamId));
     });
     inputReader.onClose(() => {
+      if (protocolVersion === "v1.1") {
+        if (!procClosesWithResponse && !outputWriter.isClosed()) {
+          outputWriter.close();
+        }
+      }
       if (outputWriter.isClosed()) {
         cleanup();
       }
     });
-    const procClosesWithResponse = procedure.type === "rpc" || procedure.type === "upload";
     const outputWriter = new WriteStreamImpl((response) => {
       this.transport.send(from, {
         streamId,
-        controlFlags: procClosesWithResponse ? 8 /* StreamClosedBit */ : 0,
+        controlFlags: procClosesWithResponse ? getStreamCloseBackwardsCompat(protocolVersion) : 0,
         payload: response
       });
     });
     outputWriter.onClose(() => {
       if (!procClosesWithResponse && cleanClose) {
-        this.transport.sendCloseControl(from, streamId);
+        const message = closeStreamMessage(streamId);
+        message.controlFlags = getStreamCloseBackwardsCompat(protocolVersion);
+        this.transport.send(from, closeStreamMessage(streamId));
+      }
+      if (protocolVersion === "v1.1") {
+        if (!inputReader.isClosed()) {
+          inputReader.triggerClose();
+        }
       }
       if (inputReader.isClosed()) {
         cleanup();
@@ -1622,7 +1179,7 @@ var RiverServer = class {
         })
       );
     };
-    if (isStreamClose(controlFlags)) {
+    if (isStreamCloseBackwardsCompat(controlFlags, protocolVersion)) {
       inputReader.triggerClose();
     } else if (procedure.type === "rpc" || procedure.type === "subscription") {
       this.log?.warn(`${procedure.type} sent an init without a stream close`, {
@@ -1633,6 +1190,7 @@ var RiverServer = class {
     const serviceContextWithTransportInfo = {
       ...this.getContext(service, serviceName),
       from,
+      sessionId,
       metadata: sessionMetadata,
       abortController: handlerAbortController,
       clientAbortSignal: clientAbortController.signal,
@@ -1788,7 +1346,9 @@ var RiverServer = class {
       );
       return null;
     }
-    const sessionMetadata = this.transport.sessionHandshakeMetadata.get(session);
+    const sessionMetadata = this.transport.sessionHandshakeMetadata.get(
+      session.to
+    );
     if (!sessionMetadata) {
       const errMessage = `session doesn't have handshake metadata`;
       this.log?.error(errMessage, {
@@ -1929,7 +1489,9 @@ var RiverServer = class {
       controlFlags: initMessage.controlFlags,
       tracingCtx: initMessage.tracing,
       initPayload: initMessage.payload,
-      from: initMessage.from
+      from: initMessage.from,
+      sessionId: session.id,
+      protocolVersion: session.protocolVersion
     };
   }
   abortStream(to, streamId, payload) {
@@ -1939,7 +1501,11 @@ var RiverServer = class {
       this.serverAbortedStreams.set(to, abortedForSession);
     }
     abortedForSession.add(streamId);
-    this.transport.sendAbort(to, streamId, payload);
+    this.transport.send(
+      to,
+      // TODO remove once clients migrate to v2
+      this.transport.sessions.get(to)?.protocolVersion === "v1.1" ? closeStreamMessage(streamId) : abortMessage(streamId, payload)
+    );
   }
 };
 var LRUSet = class {
@@ -1964,6 +1530,24 @@ var LRUSet = class {
     return this.items.has(item);
   }
 };
+function isStreamAbortBackwardsCompat(controlFlags, protocolVersion) {
+  if (protocolVersion === "v1.1") {
+    return false;
+  }
+  return isStreamAbort(controlFlags);
+}
+function isStreamCloseBackwardsCompat(controlFlags, protocolVersion) {
+  if (protocolVersion === "v1.1") {
+    return isStreamAbort(controlFlags);
+  }
+  return isStreamClose(controlFlags);
+}
+function getStreamCloseBackwardsCompat(protocolVersion) {
+  if (protocolVersion === "v1.1") {
+    return 4 /* StreamAbortBit */;
+  }
+  return 8 /* StreamClosedBit */;
+}
 function createServer(transport, services, providedServerOptions) {
   return new RiverServer(
     transport,
@@ -1985,7 +1569,6 @@ function createServerHandshakeOptions(schema, validate) {
 export {
   serializeSchema,
   ServiceSchema,
-  diffServerSchema,
   Ok,
   Err,
   ReadStreamImpl,
@@ -2001,4 +1584,4 @@ export {
   createClientHandshakeOptions,
   createServerHandshakeOptions
 };
-//# sourceMappingURL=chunk-7CKIN3JT.js.map
+//# sourceMappingURL=chunk-SIRRYRLQ.js.map