@restatedev/restate-sdk 1.9.1 → 1.10.0

package/dist/endpoint/handlers/generic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generic.js","names":["endpoint: Endpoint","protocolMode: ProtocolMode","additionalDiscoveryFields: Partial<EndpointManifest>","vm.WasmIdentityVerifier","vm.WasmHeader","journalValueCodec: JournalValueCodec","vm.WasmVM","headers","invocationRequest: Request","error","invocationLoggers: Map<number, Logger>"],"sources":["../../../src/endpoint/handlers/generic.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\nimport {\n  ensureError,\n  logError,\n  RestateError,\n  RetryableError,\n  TerminalError,\n} from \"../../types/errors.js\";\nimport type {\n  Endpoint as EndpointManifest,\n  ProtocolMode,\n} from \"../discovery.js\";\nimport type { Component, ComponentHandler } from \"../components.js\";\nimport { parseUrlComponents } from \"../components.js\";\nimport { X_RESTATE_SERVER } from \"../../user_agent.js\";\nimport { type ReadableStream, TransformStream } from \"node:stream/web\";\nimport { OnceStream } from \"../../utils/streams.js\";\nimport { ContextImpl } from \"../../context_impl.js\";\nimport type { Request } from \"../../context.js\";\nimport * as vm from \"./vm/sdk_shared_core_wasm_bindings.js\";\nimport { CompletablePromise } from \"../../utils/completable_promise.js\";\nimport { HandlerKind } from \"../../types/rpc.js\";\nimport { createLogger, type Logger } from \"../../logging/logger.js\";\nimport {\n  DEFAULT_CONSOLE_LOGGER_LOG_LEVEL,\n  defaultLoggerTransport,\n} from \"../../logging/console_logger_transport.js\";\nimport {\n  LoggerContext,\n  type LoggerTransport,\n  LogSource,\n  RestateLogLevel,\n} from \"../../logging/logger_transport.js\";\nimport {\n  type JournalValueCodec,\n  millisOrDurationToMillis,\n} from \"@restatedev/restate-sdk-core\";\nimport type { Endpoint } from \"../endpoint.js\";\n\nexport interface Headers {\n  [name: string]: string | string[] | undefined;\n}\n\nexport interface ResponseHeaders {\n  [name: string]: string;\n}\n\nexport interface AdditionalContext {\n  [name: string]: string;\n}\n\nexport interface RestateRequest {\n  readonly url: string;\n  readonly headers: Headers;\n  readonly body: ReadableStream<Uint8Array> | null;\n  readonly extraArgs: unknown[];\n  readonly abortSignal: AbortSignal;\n}\n\nexport interface RestateResponse {\n  readonly headers: ResponseHeaders;\n  readonly statusCode: number;\n  readonly body: ReadableStream<Uint8Array>;\n}\n\nexport interface RestateHandler {\n  handle(\n    request: RestateRequest,\n    context?: AdditionalContext\n  ): Promise<RestateResponse>;\n}\n\nconst ENDPOINT_MANIFEST_V2 = \"application/vnd.restate.endpointmanifest.v2+json\";\nconst ENDPOINT_MANIFEST_V3 = \"application/vnd.restate.endpointmanifest.v3+json\";\nconst ENDPOINT_MANIFEST_V4 = \"application/vnd.restate.endpointmanifest.v4+json\";\n\nexport function tryCreateContextualLogger(\n  loggerTransport: LoggerTransport,\n  url: string,\n  headers: Headers,\n  additionalContext?: { [name: string]: string }\n): Logger | undefined {\n  try {\n    const path = new URL(url, \"https://example.com\").pathname;\n    const parsed = parseUrlComponents(path);\n    if (parsed.type !== \"invoke\") {\n      return undefined;\n    }\n    const invocationId = invocationIdFromHeaders(headers);\n    return createLogger(\n      loggerTransport,\n      LogSource.SYSTEM,\n      new LoggerContext(\n        invocationId,\n        parsed.componentName,\n        parsed.handlerName,\n        undefined,\n        undefined,\n        additionalContext\n      )\n    );\n  } catch {\n    return undefined;\n  }\n}\n\nfunction invocationIdFromHeaders(headers: Headers) {\n  const invocationIdHeader = headers[\"x-restate-invocation-id\"];\n  const invocationId =\n    typeof invocationIdHeader === \"string\"\n      ? invocationIdHeader\n      : Array.isArray(invocationIdHeader)\n        ? (invocationIdHeader[0] ?? \"unknown id\")\n        : \"unknown id\";\n  return invocationId;\n}\n\n/**\n * This is an internal API to support 'fetch' like handlers.\n * It supports both request-reply mode and bidirectional streaming mode.\n *\n * An individual handler will have to convert the shape of the incoming request\n * to a RestateRequest, and then pass it to this handler, and eventually convert back\n * the response.\n * Different runtimes have slightly different shapes of the incoming request, and responses.\n */\nexport class GenericHandler implements RestateHandler {\n  private readonly identityVerifier?: vm.WasmIdentityVerifier;\n\n  constructor(\n    readonly endpoint: Endpoint,\n    private readonly protocolMode: ProtocolMode,\n    private readonly additionalDiscoveryFields: Partial<EndpointManifest>\n  ) {\n    // Setup identity verifier\n    if (\n      this.endpoint.keySet === undefined ||\n      this.endpoint.keySet.length === 0\n    ) {\n      this.endpoint.rlog.warn(\n        `Accepting requests without validating request signatures; handler access must be restricted`\n      );\n    } else {\n      this.endpoint.rlog.info(\n        `Validating requests using signing keys [${this.endpoint.keySet}]`\n      );\n      this.identityVerifier = new vm.WasmIdentityVerifier(this.endpoint.keySet);\n    }\n\n    // Set the logging level in the shared core too!\n    vm.set_log_level(\n      restateLogLevelToWasmLogLevel(DEFAULT_CONSOLE_LOGGER_LOG_LEVEL)\n    );\n  }\n\n  // handle does not throw.\n  public async handle(\n    request: RestateRequest,\n    context?: AdditionalContext\n  ): Promise<RestateResponse> {\n    try {\n      return await this._handle(request, context);\n    } catch (e) {\n      const error = ensureError(e);\n      (\n        tryCreateContextualLogger(\n          this.endpoint.loggerTransport,\n          request.url,\n          request.headers\n        ) ?? this.endpoint.rlog\n      ).error(\n        \"Error while handling request: \" + (error.stack ?? error.message)\n      );\n      return this.toErrorResponse(\n        error instanceof RestateError ? error.code : 500,\n        error.message\n      );\n    }\n  }\n\n  private async _handle(\n    request: RestateRequest,\n    context?: AdditionalContext\n  ): Promise<RestateResponse> {\n    // this is the recommended way to get the relative path from a url that may be relative or absolute\n    const path = new URL(request.url, \"https://example.com\").pathname;\n    const parsed = parseUrlComponents(path);\n\n    if (parsed.type === \"unknown\") {\n      const msg = `Invalid path. Allowed are /health, or /discover, or /invoke/SvcName/handlerName, but was: ${path}`;\n      this.endpoint.rlog.trace(msg);\n      return this.toErrorResponse(404, msg);\n    }\n\n    if (parsed.type === \"health\") {\n      return {\n        body: OnceStream(new TextEncoder().encode(\"OK\")),\n        headers: {\n          \"content-type\": \"application/text\",\n          \"x-restate-server\": X_RESTATE_SERVER,\n        },\n        statusCode: 200,\n      };\n    }\n\n    const error = this.validateConnectionSignature(path, request.headers);\n    if (error !== null) {\n      return error;\n    }\n    if (parsed.type === \"discover\") {\n      return this.handleDiscovery(request.headers[\"accept\"]);\n    }\n    const serviceProtocolVersionString = request.headers[\"content-type\"];\n    if (typeof serviceProtocolVersionString !== \"string\") {\n      const errorMessage = \"Missing content-type header\";\n      this.endpoint.rlog.warn(errorMessage);\n      return this.toErrorResponse(415, errorMessage);\n    }\n    const service = this.endpoint.components.get(parsed.componentName);\n    if (!service) {\n      const msg = `No service found for URL: ${JSON.stringify(parsed)}`;\n      this.endpoint.rlog.error(msg);\n      return this.toErrorResponse(404, msg);\n    }\n    const handler = service?.handlerMatching(parsed);\n    if (!handler) {\n      const msg = `No service found for URL: ${JSON.stringify(parsed)}`;\n      this.endpoint.rlog.error(msg);\n      return this.toErrorResponse(404, msg);\n    }\n    if (!request.body) {\n      const msg = \"The incoming message body was null\";\n      this.endpoint.rlog.error(msg);\n      return this.toErrorResponse(400, msg);\n    }\n\n    return this.handleInvoke(\n      service,\n      handler,\n      request.body,\n      request.headers,\n      request.extraArgs,\n      request.abortSignal,\n      context ?? {}\n    );\n  }\n\n  private validateConnectionSignature(\n    path: string,\n    headers: Headers\n  ): RestateResponse | null {\n    if (!this.identityVerifier) {\n      // not validating\n      return null;\n    }\n\n    const vmHeaders = Object.entries(headers)\n      .filter(([, v]) => v !== undefined)\n      .map(\n        ([k, v]) =>\n          new vm.WasmHeader(k, v instanceof Array ? v[0]! : (v as string))\n      );\n\n    try {\n      this.identityVerifier.verify_identity(path, vmHeaders);\n      return null;\n    } catch (e) {\n      this.endpoint.rlog.error(\n        // eslint-disable-next-line @typescript-eslint/restrict-template-expressions\n        `Rejecting request as its JWT did not validate: ${e}`\n      );\n      return this.toErrorResponse(401, \"Unauthorized\");\n    }\n  }\n\n  private async handleInvoke(\n    service: Component,\n    handler: ComponentHandler,\n    body: ReadableStream<Uint8Array>,\n    headers: Headers,\n    extraArgs: unknown[],\n    abortSignal: AbortSignal,\n    additionalContext: AdditionalContext\n  ): Promise<RestateResponse> {\n    const journalValueCodec: JournalValueCodec = this.endpoint.journalValueCodec\n      ? await this.endpoint.journalValueCodec\n      : {\n          encode: (entry) => entry,\n          decode: (entry) => Promise.resolve(entry),\n        };\n    const loggerId = Math.floor(Math.random() * 4_294_967_295 /* u32::MAX */);\n\n    try {\n      // Instantiate core vm and prepare response headers\n      const vmHeaders = Object.entries(headers)\n        .filter(([, v]) => v !== undefined)\n        .map(\n          ([k, v]) =>\n            new vm.WasmHeader(k, v instanceof Array ? v[0]! : (v as string))\n        );\n      const coreVm = new vm.WasmVM(\n        vmHeaders,\n        restateLogLevelToWasmLogLevel(DEFAULT_CONSOLE_LOGGER_LOG_LEVEL),\n        loggerId,\n        this.endpoint.journalValueCodec !== undefined\n      );\n      const responseHead = coreVm.get_response_head();\n      const responseHeaders = responseHead.headers.reduce(\n        (headers, { key, value }) => ({\n          [key]: value,\n          ...headers,\n        }),\n        {\n          \"x-restate-server\": X_RESTATE_SERVER,\n        }\n      );\n\n      // Use a default logger that still respects the endpoint custom logger\n      // We will override this later with a logger that has a LoggerContext\n      // See vm_log below for more details\n      invocationLoggers.set(\n        loggerId,\n        createLogger(\n          this.endpoint.loggerTransport,\n          LogSource.JOURNAL,\n          new LoggerContext(\n            invocationIdFromHeaders(headers),\n            service.name(),\n            handler.name(),\n            undefined,\n            undefined,\n            additionalContext\n          )\n        )\n      );\n\n      const inputReader = body.getReader();\n      abortSignal.addEventListener(\n        \"abort\",\n        () => {\n          invocationLoggers.delete(loggerId);\n          void inputReader.cancel();\n        },\n        { once: true }\n      );\n\n      // Now buffer input entries\n      while (!coreVm.is_ready_to_execute()) {\n        const nextValue = await inputReader.read();\n        if (nextValue.value !== undefined) {\n          coreVm.notify_input(nextValue.value);\n        }\n        if (nextValue.done) {\n          coreVm.notify_input_closed();\n          break;\n        }\n      }\n\n      // Get input\n      const input = coreVm.sys_input();\n\n      const invocationRequest: Request = {\n        id: input.invocation_id,\n        headers: input.headers.reduce((headers, { key, value }) => {\n          headers.set(key, value);\n          return headers;\n        }, new Map()),\n        attemptHeaders: Object.entries(headers).reduce(\n          (headers, [key, value]) => {\n            if (value !== undefined) {\n              headers.set(key, value instanceof Array ? value[0] : value);\n            }\n            return headers;\n          },\n          new Map()\n        ),\n        body: input.input,\n        extraArgs,\n        attemptCompletedSignal: abortSignal,\n      };\n\n      // Prepare logger\n      const loggerContext = new LoggerContext(\n        input.invocation_id,\n        handler.component().name(),\n        handler.name(),\n        handler.kind() === HandlerKind.SERVICE ? undefined : input.key,\n        invocationRequest,\n        additionalContext\n      );\n      const ctxLogger = createLogger(\n        this.endpoint.loggerTransport,\n        LogSource.USER,\n        loggerContext,\n        () => !coreVm.is_processing()\n      );\n      const vmLogger = createLogger(\n        this.endpoint.loggerTransport,\n        LogSource.JOURNAL,\n        loggerContext\n        // Filtering is done within the shared core\n      );\n      // See vm_log below for more details\n      invocationLoggers.set(loggerId, vmLogger);\n      if (!coreVm.is_processing()) {\n        vmLogger.info(\"Replaying invocation.\");\n      } else {\n        vmLogger.info(\"Starting invocation.\");\n      }\n\n      // This promise is used to signal the end of the computation,\n      // which can be either the user returns a value,\n      // or an exception gets catched, or the state machine fails/suspends.\n      //\n      // The last case is handled internally within the ContextImpl.\n      const invocationEndPromise = new CompletablePromise<void>();\n\n      // Prepare response stream\n      const responseTransformStream = new TransformStream<Uint8Array>();\n      const outputWriter = responseTransformStream.writable.getWriter();\n\n      // Prepare context\n      const ctx = new ContextImpl(\n        coreVm,\n        input,\n        ctxLogger,\n        handler.kind(),\n        vmLogger,\n        invocationRequest,\n        invocationEndPromise,\n        inputReader,\n        outputWriter,\n        journalValueCodec,\n        service.options?.serde,\n        service.options?.asTerminalError\n      );\n\n      journalValueCodec\n        .decode(input.input)\n        .catch((e) =>\n          Promise.reject(\n            new TerminalError(\n              `Failed to decode input using journal value codec: ${\n                ensureError(e).message\n              }`,\n              {\n                errorCode: 400,\n              }\n            )\n          )\n        )\n        .then((decodedInput) =>\n          // Invoke user handler code\n          handler.invoke(ctx, decodedInput)\n        )\n        .then((output) => {\n          // Write output result\n          coreVm.sys_write_output_success(journalValueCodec.encode(output));\n          coreVm.sys_end();\n          vmLogger.info(\"Invocation completed successfully.\");\n        })\n        .catch((e) => {\n          // Convert to Error\n          const error = ensureError(e, service.options?.asTerminalError);\n          logError(vmLogger, error);\n\n          // If TerminalError, handle it here.\n          // NOTE: this can still fail!\n          if (error instanceof TerminalError) {\n            coreVm.sys_write_output_failure({\n              code: error.code,\n              message: error.message,\n            });\n            coreVm.sys_end();\n            return;\n          }\n\n          // Not a terminal error, have the below catch handle it\n          throw error;\n        })\n        .catch((e) => {\n          // Handle any other error now (retryable errors)\n          const error = ensureError(e);\n          if (error instanceof RetryableError) {\n            coreVm.notify_error_with_delay_override(\n              error.message,\n              error.stack,\n              error.retryAfter !== undefined\n                ? BigInt(millisOrDurationToMillis(error.retryAfter))\n                : undefined\n            );\n          } else {\n            coreVm.notify_error(error.message, error.stack);\n          }\n        })\n        .finally(() => {\n          invocationEndPromise.resolve();\n        });\n\n      // Let's wire up invocationEndPromise with consuming all the output and closing the streams.\n      invocationEndPromise.promise\n        .then(async () => {\n          // Consume output till the end, write it out, then close the stream\n          let nextOutput = coreVm.take_output() as\n            | Uint8Array\n            | null\n            | undefined;\n          while (nextOutput !== null && nextOutput !== undefined) {\n            await outputWriter.write(nextOutput);\n            nextOutput = coreVm.take_output() as Uint8Array | null | undefined;\n          }\n          await outputWriter.close();\n          // Let's cancel the input reader, if it's still here\n          inputReader.cancel().catch(() => {});\n        })\n        .finally(() => {\n          invocationLoggers.delete(loggerId);\n        })\n        .catch(() => {});\n\n      return {\n        headers: responseHeaders,\n        statusCode: responseHead.status_code,\n        body: responseTransformStream.readable as ReadableStream<Uint8Array>,\n      };\n    } catch (error) {\n      invocationLoggers.delete(loggerId);\n      throw error;\n    }\n  }\n\n  private handleDiscovery(\n    acceptVersionsString: string | string[] | undefined\n  ): RestateResponse {\n    if (typeof acceptVersionsString !== \"string\") {\n      const errorMessage = \"Missing accept header\";\n      this.endpoint.rlog.warn(errorMessage);\n      return this.toErrorResponse(415, errorMessage);\n    }\n\n    // Negotiate version to use\n    let manifestVersion;\n    if (acceptVersionsString.includes(ENDPOINT_MANIFEST_V4)) {\n      manifestVersion = 4;\n    } else if (acceptVersionsString.includes(ENDPOINT_MANIFEST_V3)) {\n      manifestVersion = 3;\n    } else if (acceptVersionsString.includes(ENDPOINT_MANIFEST_V2)) {\n      manifestVersion = 2;\n    } else {\n      const errorMessage = `Unsupported service discovery protocol version '${acceptVersionsString}'`;\n      this.endpoint.rlog.warn(errorMessage);\n      return this.toErrorResponse(415, errorMessage);\n    }\n\n    const discovery = {\n      ...this.endpoint.discoveryMetadata,\n      ...this.additionalDiscoveryFields,\n      protocolMode: this.protocolMode,\n    };\n\n    const checkUnsupportedFeature = <T extends object>(\n      obj: T,\n      ...fields: Array<keyof T>\n    ) => {\n      for (const field of fields) {\n        if (field in obj && obj[field] !== undefined) {\n          return this.toErrorResponse(\n            500,\n            `The code uses the new discovery feature '${String(\n              field\n            )}' but the runtime doesn't support it yet (discovery protocol negotiated version ${manifestVersion}). Either remove the usage of this feature, or upgrade the runtime.`\n          );\n        }\n      }\n      return;\n    };\n\n    // Verify none of the manifest v3 configuration options are used.\n    if (manifestVersion < 3) {\n      for (const service of discovery.services) {\n        const error = checkUnsupportedFeature(\n          service,\n          \"journalRetention\",\n          \"idempotencyRetention\",\n          \"inactivityTimeout\",\n          \"abortTimeout\",\n          \"enableLazyState\",\n          \"ingressPrivate\"\n        );\n        if (error !== undefined) {\n          return error;\n        }\n        for (const handler of service.handlers) {\n          const error = checkUnsupportedFeature(\n            handler,\n            \"journalRetention\",\n            \"idempotencyRetention\",\n            \"workflowCompletionRetention\",\n            \"inactivityTimeout\",\n            \"abortTimeout\",\n            \"enableLazyState\",\n            \"ingressPrivate\"\n          );\n          if (error !== undefined) {\n            return error;\n          }\n        }\n      }\n    }\n\n    if (manifestVersion < 4) {\n      // Blank the lambda compression field. No need to fail in this case.\n      discovery.lambdaCompression = undefined;\n      for (const service of discovery.services) {\n        const error = checkUnsupportedFeature(\n          service,\n          \"retryPolicyExponentiationFactor\",\n          \"retryPolicyInitialInterval\",\n          \"retryPolicyMaxAttempts\",\n          \"retryPolicyMaxInterval\",\n          \"retryPolicyOnMaxAttempts\"\n        );\n        if (error !== undefined) {\n          return error;\n        }\n        for (const handler of service.handlers) {\n          const error = checkUnsupportedFeature(\n            handler,\n            \"retryPolicyExponentiationFactor\",\n            \"retryPolicyInitialInterval\",\n            \"retryPolicyMaxAttempts\",\n            \"retryPolicyMaxInterval\",\n            \"retryPolicyOnMaxAttempts\"\n          );\n          if (error !== undefined) {\n            return error;\n          }\n        }\n      }\n    }\n\n    const body = JSON.stringify(discovery);\n    return {\n      headers: {\n        \"content-type\":\n          manifestVersion === 2\n            ? ENDPOINT_MANIFEST_V2\n            : manifestVersion === 3\n              ? ENDPOINT_MANIFEST_V3\n              : ENDPOINT_MANIFEST_V4,\n        \"x-restate-server\": X_RESTATE_SERVER,\n      },\n      statusCode: 200,\n      body: OnceStream(new TextEncoder().encode(body)),\n    };\n  }\n\n  private toErrorResponse(code: number, message: string): RestateResponse {\n    return {\n      headers: {\n        \"content-type\": \"application/json\",\n        \"x-restate-server\": X_RESTATE_SERVER,\n      },\n      statusCode: code,\n      body: OnceStream(new TextEncoder().encode(JSON.stringify({ message }))),\n    };\n  }\n}\n\n// See vm_log below for more details\nconst invocationLoggers: Map<number, Logger> = new Map<number, Logger>();\nconst logsTextDecoder = new TextDecoder(\"utf-8\", { fatal: false });\n\n/**\n * The shared core propagates logs to the SDK invoking this method.\n * When possible it provides an invocationId, which is used to access the registered invocationLoggers, that should contain the logger per invocation id.\n */\nexport function vm_log(\n  level: vm.LogLevel,\n  strBytes: Uint8Array,\n  loggerId?: number\n) {\n  try {\n    const logger = (loggerId && invocationLoggers.get(loggerId)) || undefined;\n    const str = logsTextDecoder.decode(strBytes);\n    if (logger !== undefined) {\n      logger.logForLevel(wasmLogLevelToRestateLogLevel(level), str);\n    } else {\n      defaultLoggerTransport(\n        {\n          level: wasmLogLevelToRestateLogLevel(level),\n          replaying: false,\n          source: LogSource.JOURNAL,\n        },\n        str\n      );\n    }\n  } catch (e) {\n    // This function CAN'T EVER propagate an error,\n    // because otherwise it will cause an awesome error in the shared core due to concurrent usage of it.\n    defaultLoggerTransport(\n      {\n        level: RestateLogLevel.ERROR,\n        replaying: false,\n        source: LogSource.SYSTEM,\n      },\n      \"Unexpected error thrown while trying to log: \" + e?.toString()\n    );\n  }\n}\n\nfunction wasmLogLevelToRestateLogLevel(level: vm.LogLevel): RestateLogLevel {\n  switch (level) {\n    case vm.LogLevel.TRACE:\n      return RestateLogLevel.TRACE;\n    case vm.LogLevel.DEBUG:\n      return RestateLogLevel.DEBUG;\n    case vm.LogLevel.INFO:\n      return RestateLogLevel.INFO;\n    case vm.LogLevel.WARN:\n      return RestateLogLevel.WARN;\n    case vm.LogLevel.ERROR:\n      return RestateLogLevel.ERROR;\n  }\n}\n\nfunction restateLogLevelToWasmLogLevel(level: RestateLogLevel): vm.LogLevel {\n  switch (level) {\n    case RestateLogLevel.TRACE:\n      return vm.LogLevel.TRACE;\n    case RestateLogLevel.DEBUG:\n      return vm.LogLevel.DEBUG;\n    case RestateLogLevel.INFO:\n      return vm.LogLevel.INFO;\n    case RestateLogLevel.WARN:\n      return vm.LogLevel.WARN;\n    case RestateLogLevel.ERROR:\n      return vm.LogLevel.ERROR;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AAkFA,MAAM,uBAAuB;AAC7B,MAAM,uBAAuB;AAC7B,MAAM,uBAAuB;AAE7B,SAAgB,0BACd,iBACA,KACA,SACA,mBACoB;AACpB,KAAI;EACF,MAAM,OAAO,IAAI,IAAI,KAAK,sBAAsB,CAAC;EACjD,MAAM,SAAS,mBAAmB,KAAK;AACvC,MAAI,OAAO,SAAS,SAClB;EAEF,MAAM,eAAe,wBAAwB,QAAQ;AACrD,SAAO,aACL,iBACA,UAAU,QACV,IAAI,cACF,cACA,OAAO,eACP,OAAO,aACP,QACA,QACA,kBACD,CACF;SACK;AACN;;;AAIJ,SAAS,wBAAwB,SAAkB;CACjD,MAAM,qBAAqB,QAAQ;AAOnC,QALE,OAAO,uBAAuB,WAC1B,qBACA,MAAM,QAAQ,mBAAmB,GAC9B,mBAAmB,MAAM,eAC1B;;;;;;;;;;;AAaV,IAAa,iBAAb,MAAsD;CACpD,AAAiB;CAEjB,YACE,AAASA,UACT,AAAiBC,cACjB,AAAiBC,2BACjB;EAHS;EACQ;EACA;AAGjB,MACE,KAAK,SAAS,WAAW,UACzB,KAAK,SAAS,OAAO,WAAW,EAEhC,MAAK,SAAS,KAAK,KACjB,8FACD;OACI;AACL,QAAK,SAAS,KAAK,KACjB,2CAA2C,KAAK,SAAS,OAAO,GACjE;AACD,QAAK,mBAAmB,IAAIC,qBAAwB,KAAK,SAAS,OAAO;;AAI3E,gBACE,8BAA8B,iCAAiC,CAChE;;CAIH,MAAa,OACX,SACA,SAC0B;AAC1B,MAAI;AACF,UAAO,MAAM,KAAK,QAAQ,SAAS,QAAQ;WACpC,GAAG;GACV,MAAM,QAAQ,YAAY,EAAE;AAC5B,IACE,0BACE,KAAK,SAAS,iBACd,QAAQ,KACR,QAAQ,QACT,IAAI,KAAK,SAAS,MACnB,MACA,oCAAoC,MAAM,SAAS,MAAM,SAC1D;AACD,UAAO,KAAK,gBACV,iBAAiB,eAAe,MAAM,OAAO,KAC7C,MAAM,QACP;;;CAIL,MAAc,QACZ,SACA,SAC0B;EAE1B,MAAM,OAAO,IAAI,IAAI,QAAQ,KAAK,sBAAsB,CAAC;EACzD,MAAM,SAAS,mBAAmB,KAAK;AAEvC,MAAI,OAAO,SAAS,WAAW;GAC7B,MAAM,MAAM,6FAA6F;AACzG,QAAK,SAAS,KAAK,MAAM,IAAI;AAC7B,UAAO,KAAK,gBAAgB,KAAK,IAAI;;AAGvC,MAAI,OAAO,SAAS,SAClB,QAAO;GACL,MAAM,WAAW,IAAI,aAAa,CAAC,OAAO,KAAK,CAAC;GAChD,SAAS;IACP,gBAAgB;IAChB,oBAAoB;IACrB;GACD,YAAY;GACb;EAGH,MAAM,QAAQ,KAAK,4BAA4B,MAAM,QAAQ,QAAQ;AACrE,MAAI,UAAU,KACZ,QAAO;AAET,MAAI,OAAO,SAAS,WAClB,QAAO,KAAK,gBAAgB,QAAQ,QAAQ,UAAU;AAGxD,MAAI,OADiC,QAAQ,QAAQ,oBACT,UAAU;GACpD,MAAM,eAAe;AACrB,QAAK,SAAS,KAAK,KAAK,aAAa;AACrC,UAAO,KAAK,gBAAgB,KAAK,aAAa;;EAEhD,MAAM,UAAU,KAAK,SAAS,WAAW,IAAI,OAAO,cAAc;AAClE,MAAI,CAAC,SAAS;GACZ,MAAM,MAAM,6BAA6B,KAAK,UAAU,OAAO;AAC/D,QAAK,SAAS,KAAK,MAAM,IAAI;AAC7B,UAAO,KAAK,gBAAgB,KAAK,IAAI;;EAEvC,MAAM,UAAU,SAAS,gBAAgB,OAAO;AAChD,MAAI,CAAC,SAAS;GACZ,MAAM,MAAM,6BAA6B,KAAK,UAAU,OAAO;AAC/D,QAAK,SAAS,KAAK,MAAM,IAAI;AAC7B,UAAO,KAAK,gBAAgB,KAAK,IAAI;;AAEvC,MAAI,CAAC,QAAQ,MAAM;GACjB,MAAM,MAAM;AACZ,QAAK,SAAS,KAAK,MAAM,IAAI;AAC7B,UAAO,KAAK,gBAAgB,KAAK,IAAI;;AAGvC,SAAO,KAAK,aACV,SACA,SACA,QAAQ,MACR,QAAQ,SACR,QAAQ,WACR,QAAQ,aACR,WAAW,EAAE,CACd;;CAGH,AAAQ,4BACN,MACA,SACwB;AACxB,MAAI,CAAC,KAAK,iBAER,QAAO;EAGT,MAAM,YAAY,OAAO,QAAQ,QAAQ,CACtC,QAAQ,GAAG,OAAO,MAAM,OAAU,CAClC,KACE,CAAC,GAAG,OACH,IAAIC,WAAc,GAAG,aAAa,QAAQ,EAAE,KAAO,EAAa,CACnE;AAEH,MAAI;AACF,QAAK,iBAAiB,gBAAgB,MAAM,UAAU;AACtD,UAAO;WACA,GAAG;AACV,QAAK,SAAS,KAAK,MAEjB,kDAAkD,IACnD;AACD,UAAO,KAAK,gBAAgB,KAAK,eAAe;;;CAIpD,MAAc,aACZ,SACA,SACA,MACA,SACA,WACA,aACA,mBAC0B;EAC1B,MAAMC,oBAAuC,KAAK,SAAS,oBACvD,MAAM,KAAK,SAAS,oBACpB;GACE,SAAS,UAAU;GACnB,SAAS,UAAU,QAAQ,QAAQ,MAAM;GAC1C;EACL,MAAM,WAAW,KAAK,MAAM,KAAK,QAAQ,GAAG,WAA6B;AAEzE,MAAI;GAEF,MAAM,YAAY,OAAO,QAAQ,QAAQ,CACtC,QAAQ,GAAG,OAAO,MAAM,OAAU,CAClC,KACE,CAAC,GAAG,OACH,IAAID,WAAc,GAAG,aAAa,QAAQ,EAAE,KAAO,EAAa,CACnE;GACH,MAAM,SAAS,IAAIE,OACjB,WACA,8BAA8B,iCAAiC,EAC/D,UACA,KAAK,SAAS,sBAAsB,OACrC;GACD,MAAM,eAAe,OAAO,mBAAmB;GAC/C,MAAM,kBAAkB,aAAa,QAAQ,QAC1C,WAAS,EAAE,KAAK,aAAa;KAC3B,MAAM;IACP,GAAGC;IACJ,GACD,EACE,oBAAoB,kBACrB,CACF;AAKD,qBAAkB,IAChB,UACA,aACE,KAAK,SAAS,iBACd,UAAU,SACV,IAAI,cACF,wBAAwB,QAAQ,EAChC,QAAQ,MAAM,EACd,QAAQ,MAAM,EACd,QACA,QACA,kBACD,CACF,CACF;GAED,MAAM,cAAc,KAAK,WAAW;AACpC,eAAY,iBACV,eACM;AACJ,sBAAkB,OAAO,SAAS;AAClC,IAAK,YAAY,QAAQ;MAE3B,EAAE,MAAM,MAAM,CACf;AAGD,UAAO,CAAC,OAAO,qBAAqB,EAAE;IACpC,MAAM,YAAY,MAAM,YAAY,MAAM;AAC1C,QAAI,UAAU,UAAU,OACtB,QAAO,aAAa,UAAU,MAAM;AAEtC,QAAI,UAAU,MAAM;AAClB,YAAO,qBAAqB;AAC5B;;;GAKJ,MAAM,QAAQ,OAAO,WAAW;GAEhC,MAAMC,oBAA6B;IACjC,IAAI,MAAM;IACV,SAAS,MAAM,QAAQ,QAAQ,WAAS,EAAE,KAAK,YAAY;AACzD,eAAQ,IAAI,KAAK,MAAM;AACvB,YAAOD;uBACN,IAAI,KAAK,CAAC;IACb,gBAAgB,OAAO,QAAQ,QAAQ,CAAC,QACrC,WAAS,CAAC,KAAK,WAAW;AACzB,SAAI,UAAU,OACZ,WAAQ,IAAI,KAAK,iBAAiB,QAAQ,MAAM,KAAK,MAAM;AAE7D,YAAOA;uBAET,IAAI,KAAK,CACV;IACD,MAAM,MAAM;IACZ;IACA,wBAAwB;IACzB;GAGD,MAAM,gBAAgB,IAAI,cACxB,MAAM,eACN,QAAQ,WAAW,CAAC,MAAM,EAC1B,QAAQ,MAAM,EACd,QAAQ,MAAM,KAAK,YAAY,UAAU,SAAY,MAAM,KAC3D,mBACA,kBACD;GACD,MAAM,YAAY,aAChB,KAAK,SAAS,iBACd,UAAU,MACV,qBACM,CAAC,OAAO,eAAe,CAC9B;GACD,MAAM,WAAW,aACf,KAAK,SAAS,iBACd,UAAU,SACV,cAED;AAED,qBAAkB,IAAI,UAAU,SAAS;AACzC,OAAI,CAAC,OAAO,eAAe,CACzB,UAAS,KAAK,wBAAwB;OAEtC,UAAS,KAAK,uBAAuB;GAQvC,MAAM,uBAAuB,IAAI,oBAA0B;GAG3D,MAAM,0BAA0B,IAAI,iBAA6B;GACjE,MAAM,eAAe,wBAAwB,SAAS,WAAW;GAGjE,MAAM,MAAM,IAAI,YACd,QACA,OACA,WACA,QAAQ,MAAM,EACd,UACA,mBACA,sBACA,aACA,cACA,mBACA,QAAQ,SAAS,OACjB,QAAQ,SAAS,gBAClB;AAED,qBACG,OAAO,MAAM,MAAM,CACnB,OAAO,MACN,QAAQ,OACN,IAAI,cACF,qDACE,YAAY,EAAE,CAAC,WAEjB,EACE,WAAW,KACZ,CACF,CACF,CACF,CACA,MAAM,iBAEL,QAAQ,OAAO,KAAK,aAAa,CAClC,CACA,MAAM,WAAW;AAEhB,WAAO,yBAAyB,kBAAkB,OAAO,OAAO,CAAC;AACjE,WAAO,SAAS;AAChB,aAAS,KAAK,qCAAqC;KACnD,CACD,OAAO,MAAM;IAEZ,MAAM,QAAQ,YAAY,GAAG,QAAQ,SAAS,gBAAgB;AAC9D,aAAS,UAAU,MAAM;AAIzB,QAAI,iBAAiB,eAAe;AAClC,YAAO,yBAAyB;MAC9B,MAAM,MAAM;MACZ,SAAS,MAAM;MAChB,CAAC;AACF,YAAO,SAAS;AAChB;;AAIF,UAAM;KACN,CACD,OAAO,MAAM;IAEZ,MAAM,QAAQ,YAAY,EAAE;AAC5B,QAAI,iBAAiB,eACnB,QAAO,iCACL,MAAM,SACN,MAAM,OACN,MAAM,eAAe,SACjB,OAAO,yBAAyB,MAAM,WAAW,CAAC,GAClD,OACL;QAED,QAAO,aAAa,MAAM,SAAS,MAAM,MAAM;KAEjD,CACD,cAAc;AACb,yBAAqB,SAAS;KAC9B;AAGJ,wBAAqB,QAClB,KAAK,YAAY;IAEhB,IAAI,aAAa,OAAO,aAAa;AAIrC,WAAO,eAAe,QAAQ,eAAe,QAAW;AACtD,WAAM,aAAa,MAAM,WAAW;AACpC,kBAAa,OAAO,aAAa;;AAEnC,UAAM,aAAa,OAAO;AAE1B,gBAAY,QAAQ,CAAC,YAAY,GAAG;KACpC,CACD,cAAc;AACb,sBAAkB,OAAO,SAAS;KAClC,CACD,YAAY,GAAG;AAElB,UAAO;IACL,SAAS;IACT,YAAY,aAAa;IACzB,MAAM,wBAAwB;IAC/B;WACM,OAAO;AACd,qBAAkB,OAAO,SAAS;AAClC,SAAM;;;CAIV,AAAQ,gBACN,sBACiB;AACjB,MAAI,OAAO,yBAAyB,UAAU;GAC5C,MAAM,eAAe;AACrB,QAAK,SAAS,KAAK,KAAK,aAAa;AACrC,UAAO,KAAK,gBAAgB,KAAK,aAAa;;EAIhD,IAAI;AACJ,MAAI,qBAAqB,SAAS,qBAAqB,CACrD,mBAAkB;WACT,qBAAqB,SAAS,qBAAqB,CAC5D,mBAAkB;WACT,qBAAqB,SAAS,qBAAqB,CAC5D,mBAAkB;OACb;GACL,MAAM,eAAe,mDAAmD,qBAAqB;AAC7F,QAAK,SAAS,KAAK,KAAK,aAAa;AACrC,UAAO,KAAK,gBAAgB,KAAK,aAAa;;EAGhD,MAAM,YAAY;GAChB,GAAG,KAAK,SAAS;GACjB,GAAG,KAAK;GACR,cAAc,KAAK;GACpB;EAED,MAAM,2BACJ,KACA,GAAG,WACA;AACH,QAAK,MAAM,SAAS,OAClB,KAAI,SAAS,OAAO,IAAI,WAAW,OACjC,QAAO,KAAK,gBACV,KACA,4CAA4C,OAC1C,MACD,CAAC,kFAAkF,gBAAgB,qEACrG;;AAOP,MAAI,kBAAkB,EACpB,MAAK,MAAM,WAAW,UAAU,UAAU;GACxC,MAAM,QAAQ,wBACZ,SACA,oBACA,wBACA,qBACA,gBACA,mBACA,iBACD;AACD,OAAI,UAAU,OACZ,QAAO;AAET,QAAK,MAAM,WAAW,QAAQ,UAAU;IACtC,MAAME,UAAQ,wBACZ,SACA,oBACA,wBACA,+BACA,qBACA,gBACA,mBACA,iBACD;AACD,QAAIA,YAAU,OACZ,QAAOA;;;AAMf,MAAI,kBAAkB,GAAG;AAEvB,aAAU,oBAAoB;AAC9B,QAAK,MAAM,WAAW,UAAU,UAAU;IACxC,MAAM,QAAQ,wBACZ,SACA,mCACA,8BACA,0BACA,0BACA,2BACD;AACD,QAAI,UAAU,OACZ,QAAO;AAET,SAAK,MAAM,WAAW,QAAQ,UAAU;KACtC,MAAMA,UAAQ,wBACZ,SACA,mCACA,8BACA,0BACA,0BACA,2BACD;AACD,SAAIA,YAAU,OACZ,QAAOA;;;;EAMf,MAAM,OAAO,KAAK,UAAU,UAAU;AACtC,SAAO;GACL,SAAS;IACP,gBACE,oBAAoB,IAChB,uBACA,oBAAoB,IAClB,uBACA;IACR,oBAAoB;IACrB;GACD,YAAY;GACZ,MAAM,WAAW,IAAI,aAAa,CAAC,OAAO,KAAK,CAAC;GACjD;;CAGH,AAAQ,gBAAgB,MAAc,SAAkC;AACtE,SAAO;GACL,SAAS;IACP,gBAAgB;IAChB,oBAAoB;IACrB;GACD,YAAY;GACZ,MAAM,WAAW,IAAI,aAAa,CAAC,OAAO,KAAK,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;GACxE;;;AAKL,MAAMC,oCAAyC,IAAI,KAAqB;AACxE,MAAM,kBAAkB,IAAI,YAAY,SAAS,EAAE,OAAO,OAAO,CAAC;;;;;AAMlE,SAAgB,OACd,OACA,UACA,UACA;AACA,KAAI;EACF,MAAM,SAAU,YAAY,kBAAkB,IAAI,SAAS,IAAK;EAChE,MAAM,MAAM,gBAAgB,OAAO,SAAS;AAC5C,MAAI,WAAW,OACb,QAAO,YAAY,8BAA8B,MAAM,EAAE,IAAI;MAE7D,wBACE;GACE,OAAO,8BAA8B,MAAM;GAC3C,WAAW;GACX,QAAQ,UAAU;GACnB,EACD,IACD;UAEI,GAAG;AAGV,yBACE;GACE,OAAO,gBAAgB;GACvB,WAAW;GACX,QAAQ,UAAU;GACnB,EACD,kDAAkD,GAAG,UAAU,CAChE;;;AAIL,SAAS,8BAA8B,OAAqC;AAC1E,SAAQ,OAAR;EACE,cAAiB,MACf,QAAO,gBAAgB;EACzB,cAAiB,MACf,QAAO,gBAAgB;EACzB,cAAiB,KACf,QAAO,gBAAgB;EACzB,cAAiB,KACf,QAAO,gBAAgB;EACzB,cAAiB,MACf,QAAO,gBAAgB;;;AAI7B,SAAS,8BAA8B,OAAqC;AAC1E,SAAQ,OAAR;EACE,KAAK,gBAAgB,MACnB,iBAAmB;EACrB,KAAK,gBAAgB,MACnB,iBAAmB;EACrB,KAAK,gBAAgB,KACnB,iBAAmB;EACrB,KAAK,gBAAgB,KACnB,iBAAmB;EACrB,KAAK,gBAAgB,MACnB,iBAAmB"}

package/dist/endpoint/handlers/lambda.cjs

@@ -0,0 +1,93 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const require_errors = require('../../types/errors.cjs');
+const require_user_agent = require('../../user_agent.cjs');
+const require_streams = require('../../utils/streams.cjs');
+const require_generic = require('./generic.cjs');
+let node_stream_web = require("node:stream/web");
+node_stream_web = require_rolldown_runtime.__toESM(node_stream_web);
+let node_buffer = require("node:buffer");
+node_buffer = require_rolldown_runtime.__toESM(node_buffer);
+let node_zlib = require("node:zlib");
+node_zlib = require_rolldown_runtime.__toESM(node_zlib);
+
+//#region src/endpoint/handlers/lambda.ts
+const RESPONSE_COMPRESSION_THRESHOLD = 3 * 1024 * 1024;
+var LambdaHandler = class {
+	constructor(handler, compressionSupported) {
+		this.handler = handler;
+		this.compressionSupported = compressionSupported;
+	}
+	async handleRequest(event, context) {
+		const path = "path" in event ? event.path : event.rawPath;
+		let requestContentEncoding;
+		let requestAcceptEncoding;
+		for (const [key, value] of Object.entries(event.headers)) if (key.localeCompare("content-encoding", void 0, { sensitivity: "accent" }) === 0) requestContentEncoding = value;
+		else if (key.localeCompare("accept-encoding", void 0, { sensitivity: "accent" }) === 0) requestAcceptEncoding = value;
+		let bodyStream;
+		if (!event.body) bodyStream = null;
+		else {
+			let bodyBuffer;
+			if (event.isBase64Encoded) bodyBuffer = node_buffer.Buffer.from(event.body, "base64");
+			else bodyBuffer = node_buffer.Buffer.from(new TextEncoder().encode(event.body));
+			if (requestContentEncoding && requestContentEncoding.includes("zstd")) {
+				if (!this.compressionSupported) throw new Error("The input is compressed using zstd, but this lambda deployment doesn't support compression. Make sure to deploy the Lambda using Node > 22");
+				bodyBuffer = node_zlib.zstdDecompressSync(bodyBuffer);
+			}
+			bodyStream = require_streams.OnceStream(bodyBuffer);
+		}
+		const abortController = new AbortController();
+		const request = {
+			body: bodyStream,
+			headers: event.headers,
+			url: path,
+			extraArgs: [context],
+			abortSignal: abortController.signal
+		};
+		let response;
+		try {
+			response = await this.handler.handle(request, { AWSRequestId: context.awsRequestId });
+		} catch (e) {
+			abortController.abort();
+			throw e;
+		}
+		const chunks = [];
+		try {
+			await response.body.pipeTo(new node_stream_web.WritableStream({ write: (chunk) => {
+				chunks.push(chunk);
+			} }));
+		} catch (e) {
+			const error = require_errors.ensureError(e);
+			(require_generic.tryCreateContextualLogger(this.handler.endpoint.loggerTransport, request.url, request.headers) ?? this.handler.endpoint.rlog).error("Error while collecting invocation response: " + (error.stack ?? error.message));
+			return {
+				headers: {
+					"content-type": "application/json",
+					"x-restate-server": require_user_agent.X_RESTATE_SERVER
+				},
+				statusCode: 500,
+				isBase64Encoded: false,
+				body: JSON.stringify({ message: error.message })
+			};
+		} finally {
+			abortController.abort();
+		}
+		const responseBodyBuffer = node_buffer.Buffer.concat(chunks);
+		let responseBody;
+		if (this.compressionSupported && responseBodyBuffer.length > RESPONSE_COMPRESSION_THRESHOLD && requestAcceptEncoding && requestAcceptEncoding.includes("zstd")) {
+			response.headers["content-encoding"] = "zstd";
+			responseBody = node_zlib.zstdCompressSync(responseBodyBuffer).toString("base64");
+		} else responseBody = responseBodyBuffer.toString("base64");
+		return {
+			headers: response.headers,
+			statusCode: response.statusCode,
+			isBase64Encoded: true,
+			body: responseBody
+		};
+	}
+};
+function isCompressionSupported() {
+	return "zstdDecompressSync" in node_zlib && "zstdCompressSync" in node_zlib;
+}
+
+//#endregion
+exports.LambdaHandler = LambdaHandler;
+exports.isCompressionSupported = isCompressionSupported;

package/dist/endpoint/handlers/lambda.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lambda.d.ts","sourceRoot":"","sources":["../../../src/endpoint/handlers/lambda.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EACV,oBAAoB,EACpB,sBAAsB,EACtB,qBAAqB,EACrB,iCAAiC,EACjC,OAAO,EACR,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EACV,cAAc,EAGf,MAAM,cAAc,CAAC;AAUtB,qBAAa,aAAa;IAEtB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,oBAAoB;gBADpB,OAAO,EAAE,cAAc,EACvB,oBAAoB,EAAE,OAAO;IAG1C,aAAa,CACjB,KAAK,EAAE,oBAAoB,GAAG,sBAAsB,EACpD,OAAO,EAAE,OAAO,GACf,OAAO,CAAC,qBAAqB,GAAG,iCAAiC,CAAC;CA8ItE;AAED,wBAAgB,sBAAsB,YAErC"}

package/dist/endpoint/handlers/lambda.js

@@ -0,0 +1,89 @@
+import { ensureError } from "../../types/errors.js";
+import { X_RESTATE_SERVER } from "../../user_agent.js";
+import { OnceStream } from "../../utils/streams.js";
+import { tryCreateContextualLogger } from "./generic.js";
+import { WritableStream } from "node:stream/web";
+import { Buffer } from "node:buffer";
+import * as zlib from "node:zlib";
+
+//#region src/endpoint/handlers/lambda.ts
+const RESPONSE_COMPRESSION_THRESHOLD = 3 * 1024 * 1024;
+var LambdaHandler = class {
+	constructor(handler, compressionSupported) {
+		this.handler = handler;
+		this.compressionSupported = compressionSupported;
+	}
+	async handleRequest(event, context) {
+		const path = "path" in event ? event.path : event.rawPath;
+		let requestContentEncoding;
+		let requestAcceptEncoding;
+		for (const [key, value] of Object.entries(event.headers)) if (key.localeCompare("content-encoding", void 0, { sensitivity: "accent" }) === 0) requestContentEncoding = value;
+		else if (key.localeCompare("accept-encoding", void 0, { sensitivity: "accent" }) === 0) requestAcceptEncoding = value;
+		let bodyStream;
+		if (!event.body) bodyStream = null;
+		else {
+			let bodyBuffer;
+			if (event.isBase64Encoded) bodyBuffer = Buffer.from(event.body, "base64");
+			else bodyBuffer = Buffer.from(new TextEncoder().encode(event.body));
+			if (requestContentEncoding && requestContentEncoding.includes("zstd")) {
+				if (!this.compressionSupported) throw new Error("The input is compressed using zstd, but this lambda deployment doesn't support compression. Make sure to deploy the Lambda using Node > 22");
+				bodyBuffer = zlib.zstdDecompressSync(bodyBuffer);
+			}
+			bodyStream = OnceStream(bodyBuffer);
+		}
+		const abortController = new AbortController();
+		const request = {
+			body: bodyStream,
+			headers: event.headers,
+			url: path,
+			extraArgs: [context],
+			abortSignal: abortController.signal
+		};
+		let response;
+		try {
+			response = await this.handler.handle(request, { AWSRequestId: context.awsRequestId });
+		} catch (e) {
+			abortController.abort();
+			throw e;
+		}
+		const chunks = [];
+		try {
+			await response.body.pipeTo(new WritableStream({ write: (chunk) => {
+				chunks.push(chunk);
+			} }));
+		} catch (e) {
+			const error = ensureError(e);
+			(tryCreateContextualLogger(this.handler.endpoint.loggerTransport, request.url, request.headers) ?? this.handler.endpoint.rlog).error("Error while collecting invocation response: " + (error.stack ?? error.message));
+			return {
+				headers: {
+					"content-type": "application/json",
+					"x-restate-server": X_RESTATE_SERVER
+				},
+				statusCode: 500,
+				isBase64Encoded: false,
+				body: JSON.stringify({ message: error.message })
+			};
+		} finally {
+			abortController.abort();
+		}
+		const responseBodyBuffer = Buffer.concat(chunks);
+		let responseBody;
+		if (this.compressionSupported && responseBodyBuffer.length > RESPONSE_COMPRESSION_THRESHOLD && requestAcceptEncoding && requestAcceptEncoding.includes("zstd")) {
+			response.headers["content-encoding"] = "zstd";
+			responseBody = zlib.zstdCompressSync(responseBodyBuffer).toString("base64");
+		} else responseBody = responseBodyBuffer.toString("base64");
+		return {
+			headers: response.headers,
+			statusCode: response.statusCode,
+			isBase64Encoded: true,
+			body: responseBody
+		};
+	}
+};
+function isCompressionSupported() {
+	return "zstdDecompressSync" in zlib && "zstdCompressSync" in zlib;
+}
+
+//#endregion
+export { LambdaHandler, isCompressionSupported };
+//# sourceMappingURL=lambda.js.map

package/dist/endpoint/handlers/lambda.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lambda.js","names":["handler: GenericHandler","compressionSupported: boolean","bodyStream: ReadableStream<Uint8Array> | null","bodyBuffer: Buffer | undefined","request: RestateRequest","response: RestateResponse","chunks: Uint8Array[]"],"sources":["../../../src/endpoint/handlers/lambda.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\nimport type {\n  APIGatewayProxyEvent,\n  APIGatewayProxyEventV2,\n  APIGatewayProxyResult,\n  APIGatewayProxyStructuredResultV2,\n  Context,\n} from \"aws-lambda\";\nimport { Buffer } from \"node:buffer\";\nimport type {\n  GenericHandler,\n  RestateRequest,\n  RestateResponse,\n} from \"./generic.js\";\nimport { tryCreateContextualLogger } from \"./generic.js\";\nimport { WritableStream, type ReadableStream } from \"node:stream/web\";\nimport { OnceStream } from \"../../utils/streams.js\";\nimport { X_RESTATE_SERVER } from \"../../user_agent.js\";\nimport { ensureError } from \"../../types/errors.js\";\nimport * as zlib from \"node:zlib\";\n\nconst RESPONSE_COMPRESSION_THRESHOLD = 3 * 1024 * 1024;\n\nexport class LambdaHandler {\n  constructor(\n    private readonly handler: GenericHandler,\n    private readonly compressionSupported: boolean\n  ) {}\n\n  async handleRequest(\n    event: APIGatewayProxyEvent | APIGatewayProxyEventV2,\n    context: Context\n  ): Promise<APIGatewayProxyResult | APIGatewayProxyStructuredResultV2> {\n    //\n    // Request path\n    //\n    const path = \"path\" in event ? event.path : event.rawPath;\n\n    // Deal with content-encoding\n    let requestContentEncoding;\n    let requestAcceptEncoding;\n    for (const [key, value] of Object.entries(event.headers)) {\n      if (\n        key.localeCompare(\"content-encoding\", undefined, {\n          sensitivity: \"accent\",\n        }) === 0\n      ) {\n        requestContentEncoding = value;\n      } else if (\n        key.localeCompare(\"accept-encoding\", undefined, {\n          sensitivity: \"accent\",\n        }) === 0\n      ) {\n        requestAcceptEncoding = value;\n      }\n    }\n\n    //\n    // Convert the request body to a Uint8Array stream\n    // Lambda functions receive the body as base64 encoded string\n    //\n    let bodyStream: ReadableStream<Uint8Array> | null;\n    if (!event.body) {\n      bodyStream = null;\n    } else {\n      let bodyBuffer: Buffer | undefined;\n      if (event.isBase64Encoded) {\n        bodyBuffer = Buffer.from(event.body, \"base64\");\n      } else {\n        bodyBuffer = Buffer.from(new TextEncoder().encode(event.body));\n      }\n\n      // Now decode if needed\n      if (requestContentEncoding && requestContentEncoding.includes(\"zstd\")) {\n        if (!this.compressionSupported) {\n          throw new Error(\n            \"The input is compressed using zstd, but this lambda deployment doesn't support compression. Make sure to deploy the Lambda using Node > 22\"\n          );\n        }\n\n        // Input encoded with zstd, let's decode it!\n        bodyBuffer = (\n          zlib as unknown as { zstdDecompressSync: (b: Buffer) => Buffer }\n        ).zstdDecompressSync(bodyBuffer);\n      }\n\n      // Prep the stream to pass through the endpoint handler\n      bodyStream = OnceStream(bodyBuffer);\n    }\n\n    const abortController = new AbortController();\n\n    const request: RestateRequest = {\n      body: bodyStream,\n      headers: event.headers,\n      url: path,\n      extraArgs: [context],\n      abortSignal: abortController.signal,\n    };\n\n    let response: RestateResponse;\n\n    try {\n      response = await this.handler.handle(request, {\n        AWSRequestId: context.awsRequestId,\n      });\n    } catch (e) {\n      abortController.abort();\n      throw e;\n    }\n\n    const chunks: Uint8Array[] = [];\n\n    try {\n      await response.body.pipeTo(\n        new WritableStream<Uint8Array>({\n          write: (chunk) => {\n            chunks.push(chunk);\n          },\n        })\n      );\n    } catch (e) {\n      // unlike in the streaming case, we can actually catch errors in the response body and form a nicer error\n      const error = ensureError(e);\n      (\n        tryCreateContextualLogger(\n          this.handler.endpoint.loggerTransport,\n          request.url,\n          request.headers\n        ) ?? this.handler.endpoint.rlog\n      ).error(\n        \"Error while collecting invocation response: \" +\n          (error.stack ?? error.message)\n      );\n      return {\n        headers: {\n          \"content-type\": \"application/json\",\n          \"x-restate-server\": X_RESTATE_SERVER,\n        },\n        statusCode: 500,\n        isBase64Encoded: false,\n        body: JSON.stringify({ message: error.message }),\n      };\n    } finally {\n      abortController.abort();\n    }\n\n    const responseBodyBuffer = Buffer.concat(chunks);\n    let responseBody;\n\n    // Now let's encode if we need to.\n    if (\n      this.compressionSupported &&\n      responseBodyBuffer.length > RESPONSE_COMPRESSION_THRESHOLD &&\n      requestAcceptEncoding &&\n      requestAcceptEncoding.includes(\"zstd\")\n    ) {\n      response.headers[\"content-encoding\"] = \"zstd\";\n\n      responseBody = (\n        zlib as unknown as { zstdCompressSync: (b: Buffer) => Buffer }\n      )\n        .zstdCompressSync(responseBodyBuffer)\n        .toString(\"base64\");\n    } else {\n      responseBody = responseBodyBuffer.toString(\"base64\");\n    }\n    return {\n      headers: response.headers,\n      statusCode: response.statusCode,\n      isBase64Encoded: true,\n      body: responseBody,\n    };\n  }\n}\n\nexport function isCompressionSupported() {\n  return \"zstdDecompressSync\" in zlib && \"zstdCompressSync\" in zlib;\n}\n"],"mappings":";;;;;;;;;AA+BA,MAAM,iCAAiC,IAAI,OAAO;AAElD,IAAa,gBAAb,MAA2B;CACzB,YACE,AAAiBA,SACjB,AAAiBC,sBACjB;EAFiB;EACA;;CAGnB,MAAM,cACJ,OACA,SACoE;EAIpE,MAAM,OAAO,UAAU,QAAQ,MAAM,OAAO,MAAM;EAGlD,IAAI;EACJ,IAAI;AACJ,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,MAAM,QAAQ,CACtD,KACE,IAAI,cAAc,oBAAoB,QAAW,EAC/C,aAAa,UACd,CAAC,KAAK,EAEP,0BAAyB;WAEzB,IAAI,cAAc,mBAAmB,QAAW,EAC9C,aAAa,UACd,CAAC,KAAK,EAEP,yBAAwB;EAQ5B,IAAIC;AACJ,MAAI,CAAC,MAAM,KACT,cAAa;OACR;GACL,IAAIC;AACJ,OAAI,MAAM,gBACR,cAAa,OAAO,KAAK,MAAM,MAAM,SAAS;OAE9C,cAAa,OAAO,KAAK,IAAI,aAAa,CAAC,OAAO,MAAM,KAAK,CAAC;AAIhE,OAAI,0BAA0B,uBAAuB,SAAS,OAAO,EAAE;AACrE,QAAI,CAAC,KAAK,qBACR,OAAM,IAAI,MACR,6IACD;AAIH,iBACE,KACA,mBAAmB,WAAW;;AAIlC,gBAAa,WAAW,WAAW;;EAGrC,MAAM,kBAAkB,IAAI,iBAAiB;EAE7C,MAAMC,UAA0B;GAC9B,MAAM;GACN,SAAS,MAAM;GACf,KAAK;GACL,WAAW,CAAC,QAAQ;GACpB,aAAa,gBAAgB;GAC9B;EAED,IAAIC;AAEJ,MAAI;AACF,cAAW,MAAM,KAAK,QAAQ,OAAO,SAAS,EAC5C,cAAc,QAAQ,cACvB,CAAC;WACK,GAAG;AACV,mBAAgB,OAAO;AACvB,SAAM;;EAGR,MAAMC,SAAuB,EAAE;AAE/B,MAAI;AACF,SAAM,SAAS,KAAK,OAClB,IAAI,eAA2B,EAC7B,QAAQ,UAAU;AAChB,WAAO,KAAK,MAAM;MAErB,CAAC,CACH;WACM,GAAG;GAEV,MAAM,QAAQ,YAAY,EAAE;AAC5B,IACE,0BACE,KAAK,QAAQ,SAAS,iBACtB,QAAQ,KACR,QAAQ,QACT,IAAI,KAAK,QAAQ,SAAS,MAC3B,MACA,kDACG,MAAM,SAAS,MAAM,SACzB;AACD,UAAO;IACL,SAAS;KACP,gBAAgB;KAChB,oBAAoB;KACrB;IACD,YAAY;IACZ,iBAAiB;IACjB,MAAM,KAAK,UAAU,EAAE,SAAS,MAAM,SAAS,CAAC;IACjD;YACO;AACR,mBAAgB,OAAO;;EAGzB,MAAM,qBAAqB,OAAO,OAAO,OAAO;EAChD,IAAI;AAGJ,MACE,KAAK,wBACL,mBAAmB,SAAS,kCAC5B,yBACA,sBAAsB,SAAS,OAAO,EACtC;AACA,YAAS,QAAQ,sBAAsB;AAEvC,kBACE,KAEC,iBAAiB,mBAAmB,CACpC,SAAS,SAAS;QAErB,gBAAe,mBAAmB,SAAS,SAAS;AAEtD,SAAO;GACL,SAAS,SAAS;GAClB,YAAY,SAAS;GACrB,iBAAiB;GACjB,MAAM;GACP;;;AAIL,SAAgB,yBAAyB;AACvC,QAAO,wBAAwB,QAAQ,sBAAsB"}