dynamodb-reactive 0.4.0 → 0.6.0

package/dist/server.js

@@ -1,8 +1,8 @@
 import { DynamoDBClient, ExecuteStatementCommand } from '@aws-sdk/client-dynamodb';
 import { GetCommand, PutCommand, DeleteCommand, DynamoDBDocumentClient, UpdateCommand, TransactWriteCommand, QueryCommand } from '@aws-sdk/lib-dynamodb';
 import { createHash, randomUUID } from 'crypto';
-import { z } from 'zod';
 import { unmarshall } from '@aws-sdk/util-dynamodb';
+import { z } from 'zod';
 import { isDeepStrictEqual } from 'util';
 import { ApiGatewayManagementApiClient, PostToConnectionCommand, GoneException } from '@aws-sdk/client-apigatewaymanagementapi';
 import jsonpatch from 'fast-json-patch';
@@ -152,11 +152,9 @@ function encodeStructuredValue(value, seen) {
       return `function:${String(value)}`;
   }
   if (ArrayBuffer.isView(value)) {
-    return `binary:${Buffer.from(
-      value.buffer,
-      value.byteOffset,
-      value.byteLength
-    ).toString("base64url")}`;
+    return `binary:${Buffer.from(value.buffer, value.byteOffset, value.byteLength).toString(
+      "base64url"
+    )}`;
   }
   if (value instanceof ArrayBuffer) {
     return `binary:${Buffer.from(value).toString("base64url")}`;
@@ -168,9 +166,7 @@ function encodeStructuredValue(value, seen) {
   seen.add(value);
   let encoded;
   if (Array.isArray(value)) {
-    encoded = `array:${JSON.stringify(
-      value.map((item) => encodeStructuredValue(item, seen))
-    )}`;
+    encoded = `array:${JSON.stringify(value.map((item) => encodeStructuredValue(item, seen)))}`;
   } else if (value instanceof Set) {
     encoded = `set:${JSON.stringify(
       [...value].map((item) => encodeStructuredValue(item, seen)).sort()
@@ -222,15 +218,13 @@ function extractFromCondition(tableName, condition) {
       return { dependencies, safe: dependencies.length === 1 };
     }
     case "logical": {
-      if (condition.operator === "not" || !condition.conditions || condition.conditions.length === 0) {
+      if (condition.operator === "not" || // oxlint-disable-next-line typescript/no-unnecessary-condition -- Conditions can come from plain JavaScript callers.
+      !condition.conditions || condition.conditions.length === 0) {
         return { dependencies, safe: false };
       }
       let safe = true;
       for (const subCondition of condition.conditions) {
-        const extracted = extractFromCondition(
-          tableName,
-          subCondition
-        );
+        const extracted = extractFromCondition(tableName, subCondition);
         dependencies.push(...extracted.dependencies);
         safe &&= extracted.safe;
       }
@@ -255,11 +249,7 @@ function extractDependencies(operation) {
   return [...unique.values()];
 }
 function createDependencyKey(dependency) {
-  const components = [
-    dependency.tableName,
-    dependency.fieldName,
-    dependency.fieldValue
-  ];
+  const components = [dependency.tableName, dependency.fieldName, dependency.fieldValue];
   const encoded = components.map(
     (component) => Buffer.from(component, "utf8").toString("base64url")
   );
@@ -311,9 +301,7 @@ function parseDependencyKey(key) {
   return null;
 }
 function extractAffectedKeys(tableName, item) {
-  const keys = /* @__PURE__ */ new Set([
-    createDependencyKey(wildcardDependency(tableName))
-  ]);
+  const keys = /* @__PURE__ */ new Set([createDependencyKey(wildcardDependency(tableName))]);
   const visit = (fieldName, fieldValue) => {
     if (fieldValue !== null && fieldValue !== void 0) {
       const encodedValue = encodeDependencyValue(fieldValue);
@@ -327,10 +315,7 @@ function extractAffectedKeys(tableName, item) {
         );
       }
       if (typeof fieldValue === "string") {
-        const maxPrefixLen = Math.min(
-          fieldValue.length,
-          MAX_TRACKED_PREFIX_LENGTH
-        );
+        const maxPrefixLen = Math.min(fieldValue.length, MAX_TRACKED_PREFIX_LENGTH);
         for (let i = 1; i <= maxPrefixLen; i++) {
           keys.add(
             createDependencyKey({
@@ -687,6 +672,8 @@ var SystemTableNames = {
   dependencies: "ReactiveDependenciesV2",
   queries: "ReactiveConnectionQueriesV2"
 };
+
+// ../server/src/query-metadata.ts
 var InvalidQueryMetadataError = class extends Error {
   constructor(message) {
     super(message);
@@ -739,9 +726,7 @@ function attributeValue(value) {
       )
     };
   }
-  throw new InvalidQueryMetadataError(
-    `Unsupported query value type: ${typeof value}`
-  );
+  throw new InvalidQueryMetadataError(`Unsupported query value type: ${typeof value}`);
 }
 function quoteIdentifier(identifier) {
   if (!identifier) {
@@ -767,9 +752,7 @@ function compileCondition(condition, parameters) {
     };
     const operator = operators[condition.operator];
     if (!operator) {
-      throw new InvalidQueryMetadataError(
-        `Unsupported comparison operator: ${condition.operator}`
-      );
+      throw new InvalidQueryMetadataError(`Unsupported comparison operator: ${condition.operator}`);
     }
     parameters.push(attributeValue(condition.value));
     return `${quotePath(condition.field)} ${operator} ?`;
@@ -786,7 +769,7 @@ function compileCondition(condition, parameters) {
       return `${name}(${quotePath(condition.field)}, ?)`;
     }
     throw new InvalidQueryMetadataError(
-      `Unsupported function operator: ${condition.operator}`
+      `Unsupported function operator: ${String(condition.operator)}`
     );
   }
   if (condition.type === "logical" && condition.conditions?.length) {
@@ -806,9 +789,7 @@ function compileCondition(condition, parameters) {
 function parseQueryMetadata(metadata) {
   const parsed = QueryMetadataSchema.safeParse(metadata);
   if (!parsed.success) {
-    throw new InvalidQueryMetadataError(
-      `Invalid query metadata: ${parsed.error.message}`
-    );
+    throw new InvalidQueryMetadataError(`Invalid query metadata: ${parsed.error.message}`);
   }
   return parsed.data;
 }
@@ -825,9 +806,7 @@ function compileQueryMetadata(metadataInput, nextToken) {
   }
   if (metadata.filterConditions.length) {
     predicates.push(
-      ...metadata.filterConditions.map(
-        (condition) => compileCondition(condition, parameters)
-      )
+      ...metadata.filterConditions.map((condition) => compileCondition(condition, parameters))
     );
   }
   if (predicates.length) statement += ` WHERE ${predicates.join(" AND ")}`;
@@ -867,9 +846,7 @@ async function executeQueryMetadata(client, metadataInput) {
       new ExecuteStatementCommand(compileQueryMetadata(metadata, nextToken))
     );
     items.push(
-      ...(response.Items ?? []).map(
-        (item) => unmarshall(item)
-      )
+      ...(response.Items ?? []).map((item) => unmarshall(item))
     );
     nextToken = response.NextToken;
     if (nextToken && seenTokens.has(nextToken)) {
@@ -915,10 +892,7 @@ function createClients(config) {
 function createDbContext(config, dependencyTracker) {
   const { client, docClient } = createClients(config);
   async function executeQuery(operation) {
-    return executeQueryMetadata(
-      client,
-      operationToQueryMetadata(operation)
-    );
+    return executeQueryMetadata(client, operationToQueryMetadata(operation));
   }
   return {
     query(table) {
@@ -969,10 +943,7 @@ function createDbContext(config, dependencyTracker) {
       });
       const expected = current;
       const fields = [
-        .../* @__PURE__ */ new Set([
-          ...table.getFieldNames().map(String),
-          ...Object.keys(expected)
-        ])
+        .../* @__PURE__ */ new Set([...table.getFieldNames().map(String), ...Object.keys(expected)])
       ].sort();
       const conditions = [];
       const expressionAttributeNames = {};
@@ -997,9 +968,7 @@ function createDbContext(config, dependencyTracker) {
         "utf8"
       );
       if (Buffer.byteLength(conditionExpression, "utf8") > MAX_CONDITION_EXPRESSION_BYTES || conditions.length > MAX_CONDITION_OPERATORS || variableBytes > MAX_EXPRESSION_VARIABLE_BYTES) {
-        throw new Error(
-          "Item is too complex to update with a safe compare-and-swap"
-        );
+        throw new Error("Item is too complex to update with a safe compare-and-swap");
       }
       const command = new PutCommand({
         TableName: table.tableName,
@@ -1064,26 +1033,11 @@ function validateKeyString(name, value, maxBytes) {
   }
 }
 function validateRequest(request) {
-  validateKeyString(
-    "connectionId",
-    request.connectionId,
-    DYNAMODB_PARTITION_KEY_MAX_BYTES2
-  );
+  validateKeyString("connectionId", request.connectionId, DYNAMODB_PARTITION_KEY_MAX_BYTES2);
   if (request.type === "subscribe" || request.type === "unsubscribe") {
-    validateKeyString(
-      "subscriptionId",
-      request.subscriptionId,
-      DYNAMODB_SORT_KEY_MAX_BYTES
-    );
-    validateKeyString(
-      "registrationId",
-      request.registrationId,
-      DYNAMODB_SORT_KEY_MAX_BYTES
-    );
-    if (Buffer.byteLength(
-      `${request.connectionId}#${request.subscriptionId}`,
-      "utf8"
-    ) > DYNAMODB_SORT_KEY_MAX_BYTES) {
+    validateKeyString("subscriptionId", request.subscriptionId, DYNAMODB_SORT_KEY_MAX_BYTES);
+    validateKeyString("registrationId", request.registrationId, DYNAMODB_SORT_KEY_MAX_BYTES);
+    if (Buffer.byteLength(`${request.connectionId}#${request.subscriptionId}`, "utf8") > DYNAMODB_SORT_KEY_MAX_BYTES) {
       throw requestError(
         "connectionId and subscriptionId exceed DynamoDB dependency key limits",
         400
@@ -1216,18 +1170,11 @@ function createReactiveHandler(config) {
     if (procedure.type !== "query") {
       throw requestError("Only query procedures can be subscribed to", 400);
     }
-    const resolverResult = await config.router.execute(
-      request.path,
-      context,
-      request.input
-    );
+    const resolverResult = await config.router.execute(request.path, context, request.input);
     const operations = tracker.getOperations();
     const capturedResult = tracker.getCapturedResult();
     if (operations.length !== 1 || !capturedResult) {
-      throw requestError(
-        "A subscription must execute exactly one ctx.db query",
-        400
-      );
+      throw requestError("A subscription must execute exactly one ctx.db query", 400);
     }
     if (!Array.isArray(resolverResult) || resolverResult !== capturedResult.reference || !isDeepStrictEqual(resolverResult, capturedResult.snapshot)) {
       throw requestError(
@@ -1394,10 +1341,7 @@ function createReactiveHandler(config) {
     }
     let baseline;
     try {
-      baseline = await executeQueryMetadata(
-        ddbClient,
-        registrationEntry.queryMetadata
-      );
+      baseline = await executeQueryMetadata(ddbClient, registrationEntry.queryMetadata);
       const active = {
         ...registrationEntry,
         status: "active",
@@ -1461,9 +1405,7 @@ function createReactiveHandler(config) {
       )
     ]);
     const query2 = queryResponse.Item;
-    const currentConnection = SystemSchemas.connections.safeParse(
-      connectionResponse.Item
-    );
+    const currentConnection = SystemSchemas.connections.safeParse(connectionResponse.Item);
     if (query2?.status !== "active" || query2.generation !== request.generation || query2.registrationId !== request.registrationId || !currentConnection.success || currentConnection.data.status !== "open" || currentConnection.data.connectedAt !== connection.connectedAt) {
       return null;
     }
@@ -1563,11 +1505,7 @@ function createReactiveHandler(config) {
     };
   }
   async function registerConnection(connectionId, context) {
-    validateKeyString(
-      "connectionId",
-      connectionId,
-      DYNAMODB_PARTITION_KEY_MAX_BYTES2
-    );
+    validateKeyString("connectionId", connectionId, DYNAMODB_PARTITION_KEY_MAX_BYTES2);
     const now = Date.now();
     await docClient.send(
       new PutCommand({
@@ -1583,11 +1521,7 @@ function createReactiveHandler(config) {
     );
   }
   async function unregisterConnection(connectionId) {
-    validateKeyString(
-      "connectionId",
-      connectionId,
-      DYNAMODB_PARTITION_KEY_MAX_BYTES2
-    );
+    validateKeyString("connectionId", connectionId, DYNAMODB_PARTITION_KEY_MAX_BYTES2);
     await docClient.send(
       new UpdateCommand({
         TableName: connectionsTable,
@@ -1658,9 +1592,8 @@ function createReactiveHandler(config) {
     createCaller
   };
 }
-var { compare } = jsonpatch;
 function generatePatches(oldValue, newValue) {
-  const operations = compare(
+  const operations = jsonpatch.compare(
     oldValue,
     newValue
   );
@@ -1691,13 +1624,7 @@ function isConditionalFailure(error2) {
   return error2 instanceof Error && error2.name === "ConditionalCheckFailedException";
 }
 function createStreamUtils(config) {
-  const {
-    ddbClient,
-    docClient,
-    connectionsTable,
-    dependenciesTable,
-    queriesTable
-  } = config;
+  const { ddbClient, docClient, connectionsTable, dependenciesTable, queriesTable } = config;
   let apiClient = config.apiClient;
   function getApiClient() {
     apiClient ??= new ApiGatewayManagementApiClient({
@@ -1873,10 +1800,7 @@ function createStreamUtils(config) {
           messageId: randomUUID(),
           patches
         };
-        const pendingDelivery = Buffer.byteLength(
-          JSON.stringify(deliveryMessage(query, patchDelivery)),
-          "utf8"
-        ) > MAX_PATCH_MESSAGE_BYTES ? {
+        const pendingDelivery = Buffer.byteLength(JSON.stringify(deliveryMessage(query, patchDelivery)), "utf8") > MAX_PATCH_MESSAGE_BYTES ? {
           type: "resync",
           generation: query.generation,
           fromVersion,
@@ -2054,17 +1978,14 @@ function assertConnectionId(connectionId) {
   }
 }
 async function mapWithConcurrency(values, concurrency, mapper) {
-  const results = new Array(values.length);
+  const results = Array.from({ length: values.length });
   let nextIndex = 0;
-  const workers = Array.from(
-    { length: Math.min(concurrency, values.length) },
-    async () => {
-      while (nextIndex < values.length) {
-        const index = nextIndex++;
-        results[index] = await mapper(values[index], index);
-      }
+  const workers = Array.from({ length: Math.min(concurrency, values.length) }, async () => {
+    while (nextIndex < values.length) {
+      const index = nextIndex++;
+      results[index] = await mapper(values[index], index);
     }
-  );
+  });
   await Promise.all(workers);
   return results;
 }
@@ -2082,10 +2003,7 @@ function extractRecordAffectedKeys(record) {
   return affectedKeys;
 }
 function subscriptionKey(subscription) {
-  return JSON.stringify([
-    subscription.connectionId,
-    subscription.subscriptionId
-  ]);
+  return JSON.stringify([subscription.connectionId, subscription.subscriptionId]);
 }
 function failureIdentifier(record, cause) {
   const itemIdentifier = record.dynamodb?.SequenceNumber;
@@ -2187,9 +2105,7 @@ function createConnectHandler(config) {
         connectedAt: now,
         ttl: Math.floor(now / 1e3) + DEFAULT_CONNECTION_LEASE_SECONDS
       };
-      await docClient.send(
-        new PutCommand({ TableName: connectionsTable, Item: connectionEntry })
-      );
+      await docClient.send(new PutCommand({ TableName: connectionsTable, Item: connectionEntry }));
       debug("[stream] Connection established");
       return { statusCode: 200 };
     } catch (error2) {
@@ -2254,18 +2170,9 @@ function assertRequestKey(name, value, maxBytes) {
   return value;
 }
 function createLambdaHandlers(config = {}) {
-  const connectionsTable = requiredTableName(
-    config.connectionsTableName,
-    "CONNECTIONS_TABLE"
-  );
-  const dependenciesTable = requiredTableName(
-    config.dependenciesTableName,
-    "DEPENDENCIES_TABLE"
-  );
-  const queriesTable = requiredTableName(
-    config.queriesTableName,
-    "QUERIES_TABLE"
-  );
+  const connectionsTable = requiredTableName(config.connectionsTableName, "CONNECTIONS_TABLE");
+  const dependenciesTable = requiredTableName(config.dependenciesTableName, "DEPENDENCIES_TABLE");
+  const queriesTable = requiredTableName(config.queriesTableName, "QUERIES_TABLE");
   const apiGatewayEndpoint = config.apiGatewayEndpoint ?? process.env.WEBSOCKET_ENDPOINT ?? "";
   const ddbClient = config.dbConfig?.client ?? new DynamoDBClient({
     region: config.dbConfig?.region ?? process.env.AWS_REGION,