@restatedev/restate-sdk-cloudflare-workers 1.11.1 → 1.13.0

package/dist/endpoint/components.js.map

@@ -1 +1 @@
-{"version":3,"file":"components.js","names":["serde","componentName: string","description?: string","metadata?: Record<string, string>","options?: ServiceOptions","ServiceHandler","handlers: d.Handler[]","handlerName: string","handlerWrapper: HandlerWrapper","parent: ServiceComponent","options?: ObjectOptions","parent: VirtualObjectComponent","options?: WorkflowOptions","WorkflowHandler","parent: WorkflowComponent"],"sources":["../../src/endpoint/components.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\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\nimport type * as d from \"./discovery.js\";\nimport type { ContextImpl } from \"../context_impl.js\";\nimport type {\n  HandlerWrapper,\n  ObjectOptions,\n  ServiceOptions,\n  WorkflowOptions,\n} from \"../types/rpc.js\";\nimport { HandlerKind } from \"../types/rpc.js\";\nimport type { Serde } from \"@restatedev/restate-sdk-core\";\nimport { millisOrDurationToMillis, serde } from \"@restatedev/restate-sdk-core\";\n\n//\n// Interfaces\n//\nexport interface Component {\n  name(): string;\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined;\n  discovery(): d.Service;\n  options?: ServiceOptions | ObjectOptions | WorkflowOptions;\n}\n\nexport interface ComponentHandler {\n  name(): string;\n  component(): Component;\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array>;\n  kind(): HandlerKind;\n}\n\n//\n// Service\n//\n\nfunction handlerInputDiscovery(\n  handler: HandlerWrapper,\n  defaultSerde: Serde<any>\n): d.InputPayload {\n  const serde = handler.inputSerde ?? defaultSerde;\n\n  let contentType = undefined;\n  let jsonSchema = undefined;\n\n  if (serde.jsonSchema) {\n    jsonSchema = serde.jsonSchema;\n    contentType = handler.accept ?? serde.contentType;\n  } else if (handler.accept) {\n    contentType = handler.accept;\n  } else if (serde.contentType) {\n    contentType = serde.contentType;\n  } else {\n    // no input information\n    return {};\n  }\n\n  return {\n    required: false,\n    contentType,\n    jsonSchema,\n  };\n}\n\nfunction handlerOutputDiscovery(\n  handler: HandlerWrapper,\n  defaultSerde: Serde<any>\n): d.OutputPayload {\n  const serde = handler.outputSerde ?? defaultSerde;\n\n  let contentType = undefined;\n  let jsonSchema = undefined;\n\n  if (serde.jsonSchema) {\n    jsonSchema = serde.jsonSchema;\n    contentType = serde.contentType ?? \"application/json\";\n  } else if (serde.contentType) {\n    contentType = serde.contentType;\n  } else {\n    // no input information\n    return { setContentTypeIfEmpty: false };\n  }\n\n  return {\n    setContentTypeIfEmpty: false,\n    jsonSchema,\n    contentType,\n  };\n}\n\nexport class ServiceComponent implements Component {\n  private readonly handlers: Map<string, ServiceHandler> = new Map();\n\n  constructor(\n    private readonly componentName: string,\n    public readonly description?: string,\n    public readonly metadata?: Record<string, string>,\n    public readonly options?: ServiceOptions\n  ) {}\n\n  name(): string {\n    return this.componentName;\n  }\n\n  add(name: string, handlerWrapper: HandlerWrapper) {\n    this.handlers.set(name, new ServiceHandler(name, handlerWrapper, this));\n  }\n\n  discovery(): d.Service {\n    const handlers: d.Handler[] = [...this.handlers.entries()].map(\n      ([name, handler]) => {\n        return {\n          name,\n          ...commonHandlerOptions(\n            handler.handlerWrapper,\n            this.options?.serde ?? serde.json\n          ),\n        } satisfies d.Handler;\n      }\n    );\n\n    return {\n      name: this.componentName,\n      ty: \"SERVICE\",\n      handlers,\n      documentation: this.description,\n      metadata: this.metadata,\n      ...commonServiceOptions(this.options),\n    } satisfies d.Service;\n  }\n\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined {\n    return this.handlers.get(url.handlerName);\n  }\n}\n\nexport class ServiceHandler implements ComponentHandler {\n  constructor(\n    private readonly handlerName: string,\n    public readonly handlerWrapper: HandlerWrapper,\n    private readonly parent: ServiceComponent\n  ) {}\n\n  name(): string {\n    return this.handlerName;\n  }\n\n  component(): Component {\n    return this.parent;\n  }\n\n  kind(): HandlerKind {\n    return this.handlerWrapper.kind;\n  }\n\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array> {\n    return this.handlerWrapper.invoke(context, input);\n  }\n}\n\n//\n// Virtual Object\n//\n\nexport class VirtualObjectComponent implements Component {\n  private readonly handlers: Map<string, VirtualObjectHandler> = new Map();\n\n  constructor(\n    public readonly componentName: string,\n    public readonly description?: string,\n    public readonly metadata?: Record<string, string>,\n    public readonly options?: ObjectOptions\n  ) {}\n\n  name(): string {\n    return this.componentName;\n  }\n\n  add(name: string, wrapper: HandlerWrapper) {\n    this.handlers.set(name, new VirtualObjectHandler(name, wrapper, this));\n  }\n\n  discovery(): d.Service {\n    const handlers: d.Handler[] = [...this.handlers.entries()].map(\n      ([name, handler]) => {\n        return {\n          name,\n          ty: handler.kind() === HandlerKind.EXCLUSIVE ? \"EXCLUSIVE\" : \"SHARED\",\n          ...commonHandlerOptions(\n            handler.handlerWrapper,\n            this.options?.serde ?? serde.json\n          ),\n        } satisfies d.Handler;\n      }\n    );\n\n    return {\n      name: this.componentName,\n      ty: \"VIRTUAL_OBJECT\",\n      handlers,\n      documentation: this.description,\n      metadata: this.metadata,\n      ...commonServiceOptions(this.options),\n    } satisfies d.Service;\n  }\n\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined {\n    return this.handlers.get(url.handlerName);\n  }\n}\n\nexport class VirtualObjectHandler implements ComponentHandler {\n  constructor(\n    private readonly handlerName: string,\n    public readonly handlerWrapper: HandlerWrapper,\n    private readonly parent: VirtualObjectComponent\n  ) {}\n\n  name(): string {\n    return this.handlerName;\n  }\n\n  component(): Component {\n    return this.parent;\n  }\n\n  kind(): HandlerKind {\n    return this.handlerWrapper.kind;\n  }\n\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array> {\n    return this.handlerWrapper.invoke(context, input);\n  }\n}\n\n// Workflow\n\nexport class WorkflowComponent implements Component {\n  private readonly handlers: Map<string, WorkflowHandler> = new Map();\n\n  constructor(\n    public readonly componentName: string,\n    public readonly description?: string,\n    public readonly metadata?: Record<string, string>,\n    public readonly options?: WorkflowOptions\n  ) {}\n\n  name(): string {\n    return this.componentName;\n  }\n\n  add(name: string, wrapper: HandlerWrapper) {\n    this.handlers.set(name, new WorkflowHandler(name, wrapper, this));\n  }\n\n  discovery(): d.Service {\n    const handlers: d.Handler[] = [...this.handlers.entries()].map(\n      ([name, handler]) => {\n        return {\n          name,\n          ty: handler.kind() === HandlerKind.WORKFLOW ? \"WORKFLOW\" : \"SHARED\",\n          workflowCompletionRetention:\n            handler.kind() === HandlerKind.WORKFLOW &&\n            this.options?.workflowRetention !== undefined\n              ? millisOrDurationToMillis(this.options?.workflowRetention)\n              : undefined,\n          ...commonHandlerOptions(\n            handler.handlerWrapper,\n            this.options?.serde ?? serde.json\n          ),\n        } satisfies d.Handler;\n      }\n    );\n\n    return {\n      name: this.componentName,\n      ty: \"WORKFLOW\",\n      handlers,\n      documentation: this.description,\n      metadata: this.metadata,\n      ...commonServiceOptions(this.options),\n    } satisfies d.Service;\n  }\n\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined {\n    return this.handlers.get(url.handlerName);\n  }\n}\n\nexport class WorkflowHandler implements ComponentHandler {\n  constructor(\n    private readonly handlerName: string,\n    public readonly handlerWrapper: HandlerWrapper,\n    private readonly parent: WorkflowComponent\n  ) {}\n\n  name(): string {\n    return this.handlerName;\n  }\n  component(): Component {\n    return this.parent;\n  }\n\n  kind(): HandlerKind {\n    return this.handlerWrapper.kind;\n  }\n\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array> {\n    return this.handlerWrapper.invoke(context, input);\n  }\n}\n\nexport type PathComponents =\n  | InvokePathComponents\n  | { type: \"discover\" }\n  | { type: \"health\" }\n  | { type: \"unknown\"; path: string };\n\nexport type InvokePathComponents = {\n  type: \"invoke\";\n  componentName: string;\n  handlerName: string;\n};\n\nexport function parseUrlComponents(urlPath?: string): PathComponents {\n  if (!urlPath) {\n    return { type: \"unknown\", path: \"\" };\n  }\n  const fragments = urlPath.split(\"/\");\n  if (fragments.length >= 3 && fragments[fragments.length - 3] === \"invoke\") {\n    return {\n      type: \"invoke\",\n      componentName: fragments[fragments.length - 2]!,\n      handlerName: fragments[fragments.length - 1]!,\n    };\n  }\n  if (fragments.length > 0 && fragments[fragments.length - 1] === \"discover\") {\n    return { type: \"discover\" };\n  }\n  if (fragments.length > 0 && fragments[fragments.length - 1] === \"health\") {\n    return { type: \"health\" };\n  }\n  return { type: \"unknown\", path: urlPath };\n}\n\nfunction commonServiceOptions(\n  options?: ServiceOptions | ObjectOptions | WorkflowOptions\n): Partial<d.Service> {\n  return {\n    journalRetention:\n      options?.journalRetention !== undefined\n        ? millisOrDurationToMillis(options.journalRetention)\n        : undefined,\n    idempotencyRetention:\n      options?.idempotencyRetention !== undefined\n        ? millisOrDurationToMillis(options.idempotencyRetention)\n        : undefined,\n    inactivityTimeout:\n      options?.inactivityTimeout !== undefined\n        ? millisOrDurationToMillis(options.inactivityTimeout)\n        : undefined,\n    abortTimeout:\n      options?.abortTimeout !== undefined\n        ? millisOrDurationToMillis(options.abortTimeout)\n        : undefined,\n    ingressPrivate: options?.ingressPrivate,\n    enableLazyState:\n      options !== undefined && \"enableLazyState\" in options\n        ? options.enableLazyState\n        : undefined,\n    retryPolicyExponentiationFactor: options?.retryPolicy?.exponentiationFactor,\n    retryPolicyInitialInterval:\n      options?.retryPolicy?.initialInterval !== undefined\n        ? millisOrDurationToMillis(options?.retryPolicy?.initialInterval)\n        : undefined,\n    retryPolicyMaxInterval:\n      options?.retryPolicy?.maxInterval !== undefined\n        ? millisOrDurationToMillis(options?.retryPolicy?.maxInterval)\n        : undefined,\n    retryPolicyMaxAttempts: options?.retryPolicy?.maxAttempts,\n    retryPolicyOnMaxAttempts: (options?.retryPolicy?.onMaxAttempts === \"kill\"\n      ? \"KILL\"\n      : options?.retryPolicy?.onMaxAttempts === \"pause\"\n        ? \"PAUSE\"\n        : undefined) as d.RetryPolicyOnMaxAttempts,\n  };\n}\n\nfunction commonHandlerOptions(\n  wrapper: HandlerWrapper,\n  defaultSerde: Serde<any>\n) {\n  return {\n    input: handlerInputDiscovery(wrapper, defaultSerde),\n    output: handlerOutputDiscovery(wrapper, defaultSerde),\n    journalRetention:\n      wrapper.journalRetention !== undefined\n        ? millisOrDurationToMillis(wrapper.journalRetention)\n        : undefined,\n    idempotencyRetention:\n      wrapper.idempotencyRetention !== undefined\n        ? millisOrDurationToMillis(wrapper.idempotencyRetention)\n        : undefined,\n    inactivityTimeout:\n      wrapper.inactivityTimeout !== undefined\n        ? millisOrDurationToMillis(wrapper.inactivityTimeout)\n        : undefined,\n    abortTimeout:\n      wrapper.abortTimeout !== undefined\n        ? millisOrDurationToMillis(wrapper.abortTimeout)\n        : undefined,\n    ingressPrivate: wrapper.ingressPrivate,\n    enableLazyState: wrapper.enableLazyState,\n    retryPolicyExponentiationFactor: wrapper.retryPolicy?.exponentiationFactor,\n    retryPolicyInitialInterval:\n      wrapper.retryPolicy?.initialInterval !== undefined\n        ? millisOrDurationToMillis(wrapper.retryPolicy?.initialInterval)\n        : undefined,\n    retryPolicyMaxInterval:\n      wrapper.retryPolicy?.maxInterval !== undefined\n        ? millisOrDurationToMillis(wrapper.retryPolicy?.maxInterval)\n        : undefined,\n    retryPolicyMaxAttempts: wrapper.retryPolicy?.maxAttempts,\n    retryPolicyOnMaxAttempts: (wrapper.retryPolicy?.onMaxAttempts === \"kill\"\n      ? \"KILL\"\n      : wrapper.retryPolicy?.onMaxAttempts === \"pause\"\n        ? \"PAUSE\"\n        : undefined) as d.RetryPolicyOnMaxAttempts1,\n\n    documentation: wrapper.description,\n    metadata: wrapper.metadata,\n  };\n}\n"],"mappings":";;;;AA8CA,SAAS,sBACP,SACA,cACgB;CAChB,MAAMA,UAAQ,QAAQ,cAAc;CAEpC,IAAI,cAAc;CAClB,IAAI,aAAa;AAEjB,KAAIA,QAAM,YAAY;AACpB,eAAaA,QAAM;AACnB,gBAAc,QAAQ,UAAUA,QAAM;YAC7B,QAAQ,OACjB,eAAc,QAAQ;UACbA,QAAM,YACf,eAAcA,QAAM;KAGpB,QAAO,EAAE;AAGX,QAAO;EACL,UAAU;EACV;EACA;EACD;;AAGH,SAAS,uBACP,SACA,cACiB;CACjB,MAAMA,UAAQ,QAAQ,eAAe;CAErC,IAAI,cAAc;CAClB,IAAI,aAAa;AAEjB,KAAIA,QAAM,YAAY;AACpB,eAAaA,QAAM;AACnB,gBAAcA,QAAM,eAAe;YAC1BA,QAAM,YACf,eAAcA,QAAM;KAGpB,QAAO,EAAE,uBAAuB,OAAO;AAGzC,QAAO;EACL,uBAAuB;EACvB;EACA;EACD;;AAGH,IAAa,mBAAb,MAAmD;CACjD,AAAiB,2BAAwC,IAAI,KAAK;CAElE,YACE,AAAiBC,eACjB,AAAgBC,aAChB,AAAgBC,UAChB,AAAgBC,SAChB;EAJiB;EACD;EACA;EACA;;CAGlB,OAAe;AACb,SAAO,KAAK;;CAGd,IAAI,MAAc,gBAAgC;AAChD,OAAK,SAAS,IAAI,MAAM,IAAIC,iBAAe,MAAM,gBAAgB,KAAK,CAAC;;CAGzE,YAAuB;EACrB,MAAMC,WAAwB,CAAC,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,KACxD,CAAC,MAAM,aAAa;AACnB,UAAO;IACL;IACA,GAAG,qBACD,QAAQ,gBACR,KAAK,SAAS,SAAS,MAAM,KAC9B;IACF;IAEJ;AAED,SAAO;GACL,MAAM,KAAK;GACX,IAAI;GACJ;GACA,eAAe,KAAK;GACpB,UAAU,KAAK;GACf,GAAG,qBAAqB,KAAK,QAAQ;GACtC;;CAGH,gBAAgB,KAAyD;AACvE,SAAO,KAAK,SAAS,IAAI,IAAI,YAAY;;;AAI7C,IAAaD,mBAAb,MAAwD;CACtD,YACE,AAAiBE,aACjB,AAAgBC,gBAChB,AAAiBC,QACjB;EAHiB;EACD;EACC;;CAGnB,OAAe;AACb,SAAO,KAAK;;CAGd,YAAuB;AACrB,SAAO,KAAK;;CAGd,OAAoB;AAClB,SAAO,KAAK,eAAe;;CAG7B,OAAO,SAAsB,OAAwC;AACnE,SAAO,KAAK,eAAe,OAAO,SAAS,MAAM;;;AAQrD,IAAa,yBAAb,MAAyD;CACvD,AAAiB,2BAA8C,IAAI,KAAK;CAExE,YACE,AAAgBR,eAChB,AAAgBC,aAChB,AAAgBC,UAChB,AAAgBO,SAChB;EAJgB;EACA;EACA;EACA;;CAGlB,OAAe;AACb,SAAO,KAAK;;CAGd,IAAI,MAAc,SAAyB;AACzC,OAAK,SAAS,IAAI,MAAM,IAAI,qBAAqB,MAAM,SAAS,KAAK,CAAC;;CAGxE,YAAuB;EACrB,MAAMJ,WAAwB,CAAC,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,KACxD,CAAC,MAAM,aAAa;AACnB,UAAO;IACL;IACA,IAAI,QAAQ,MAAM,KAAK,YAAY,YAAY,cAAc;IAC7D,GAAG,qBACD,QAAQ,gBACR,KAAK,SAAS,SAAS,MAAM,KAC9B;IACF;IAEJ;AAED,SAAO;GACL,MAAM,KAAK;GACX,IAAI;GACJ;GACA,eAAe,KAAK;GACpB,UAAU,KAAK;GACf,GAAG,qBAAqB,KAAK,QAAQ;GACtC;;CAGH,gBAAgB,KAAyD;AACvE,SAAO,KAAK,SAAS,IAAI,IAAI,YAAY;;;AAI7C,IAAa,uBAAb,MAA8D;CAC5D,YACE,AAAiBC,aACjB,AAAgBC,gBAChB,AAAiBG,QACjB;EAHiB;EACD;EACC;;CAGnB,OAAe;AACb,SAAO,KAAK;;CAGd,YAAuB;AACrB,SAAO,KAAK;;CAGd,OAAoB;AAClB,SAAO,KAAK,eAAe;;CAG7B,OAAO,SAAsB,OAAwC;AACnE,SAAO,KAAK,eAAe,OAAO,SAAS,MAAM;;;AAMrD,IAAa,oBAAb,MAAoD;CAClD,AAAiB,2BAAyC,IAAI,KAAK;CAEnE,YACE,AAAgBV,eAChB,AAAgBC,aAChB,AAAgBC,UAChB,AAAgBS,SAChB;EAJgB;EACA;EACA;EACA;;CAGlB,OAAe;AACb,SAAO,KAAK;;CAGd,IAAI,MAAc,SAAyB;AACzC,OAAK,SAAS,IAAI,MAAM,IAAIC,kBAAgB,MAAM,SAAS,KAAK,CAAC;;CAGnE,YAAuB;EACrB,MAAMP,WAAwB,CAAC,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,KACxD,CAAC,MAAM,aAAa;AACnB,UAAO;IACL;IACA,IAAI,QAAQ,MAAM,KAAK,YAAY,WAAW,aAAa;IAC3D,6BACE,QAAQ,MAAM,KAAK,YAAY,YAC/B,KAAK,SAAS,sBAAsB,SAChC,yBAAyB,KAAK,SAAS,kBAAkB,GACzD;IACN,GAAG,qBACD,QAAQ,gBACR,KAAK,SAAS,SAAS,MAAM,KAC9B;IACF;IAEJ;AAED,SAAO;GACL,MAAM,KAAK;GACX,IAAI;GACJ;GACA,eAAe,KAAK;GACpB,UAAU,KAAK;GACf,GAAG,qBAAqB,KAAK,QAAQ;GACtC;;CAGH,gBAAgB,KAAyD;AACvE,SAAO,KAAK,SAAS,IAAI,IAAI,YAAY;;;AAI7C,IAAaO,oBAAb,MAAyD;CACvD,YACE,AAAiBN,aACjB,AAAgBC,gBAChB,AAAiBM,QACjB;EAHiB;EACD;EACC;;CAGnB,OAAe;AACb,SAAO,KAAK;;CAEd,YAAuB;AACrB,SAAO,KAAK;;CAGd,OAAoB;AAClB,SAAO,KAAK,eAAe;;CAG7B,OAAO,SAAsB,OAAwC;AACnE,SAAO,KAAK,eAAe,OAAO,SAAS,MAAM;;;AAgBrD,SAAgB,mBAAmB,SAAkC;AACnE,KAAI,CAAC,QACH,QAAO;EAAE,MAAM;EAAW,MAAM;EAAI;CAEtC,MAAM,YAAY,QAAQ,MAAM,IAAI;AACpC,KAAI,UAAU,UAAU,KAAK,UAAU,UAAU,SAAS,OAAO,SAC/D,QAAO;EACL,MAAM;EACN,eAAe,UAAU,UAAU,SAAS;EAC5C,aAAa,UAAU,UAAU,SAAS;EAC3C;AAEH,KAAI,UAAU,SAAS,KAAK,UAAU,UAAU,SAAS,OAAO,WAC9D,QAAO,EAAE,MAAM,YAAY;AAE7B,KAAI,UAAU,SAAS,KAAK,UAAU,UAAU,SAAS,OAAO,SAC9D,QAAO,EAAE,MAAM,UAAU;AAE3B,QAAO;EAAE,MAAM;EAAW,MAAM;EAAS;;AAG3C,SAAS,qBACP,SACoB;AACpB,QAAO;EACL,kBACE,SAAS,qBAAqB,SAC1B,yBAAyB,QAAQ,iBAAiB,GAClD;EACN,sBACE,SAAS,yBAAyB,SAC9B,yBAAyB,QAAQ,qBAAqB,GACtD;EACN,mBACE,SAAS,sBAAsB,SAC3B,yBAAyB,QAAQ,kBAAkB,GACnD;EACN,cACE,SAAS,iBAAiB,SACtB,yBAAyB,QAAQ,aAAa,GAC9C;EACN,gBAAgB,SAAS;EACzB,iBACE,YAAY,UAAa,qBAAqB,UAC1C,QAAQ,kBACR;EACN,iCAAiC,SAAS,aAAa;EACvD,4BACE,SAAS,aAAa,oBAAoB,SACtC,yBAAyB,SAAS,aAAa,gBAAgB,GAC/D;EACN,wBACE,SAAS,aAAa,gBAAgB,SAClC,yBAAyB,SAAS,aAAa,YAAY,GAC3D;EACN,wBAAwB,SAAS,aAAa;EAC9C,0BAA2B,SAAS,aAAa,kBAAkB,SAC/D,SACA,SAAS,aAAa,kBAAkB,UACtC,UACA;EACP;;AAGH,SAAS,qBACP,SACA,cACA;AACA,QAAO;EACL,OAAO,sBAAsB,SAAS,aAAa;EACnD,QAAQ,uBAAuB,SAAS,aAAa;EACrD,kBACE,QAAQ,qBAAqB,SACzB,yBAAyB,QAAQ,iBAAiB,GAClD;EACN,sBACE,QAAQ,yBAAyB,SAC7B,yBAAyB,QAAQ,qBAAqB,GACtD;EACN,mBACE,QAAQ,sBAAsB,SAC1B,yBAAyB,QAAQ,kBAAkB,GACnD;EACN,cACE,QAAQ,iBAAiB,SACrB,yBAAyB,QAAQ,aAAa,GAC9C;EACN,gBAAgB,QAAQ;EACxB,iBAAiB,QAAQ;EACzB,iCAAiC,QAAQ,aAAa;EACtD,4BACE,QAAQ,aAAa,oBAAoB,SACrC,yBAAyB,QAAQ,aAAa,gBAAgB,GAC9D;EACN,wBACE,QAAQ,aAAa,gBAAgB,SACjC,yBAAyB,QAAQ,aAAa,YAAY,GAC1D;EACN,wBAAwB,QAAQ,aAAa;EAC7C,0BAA2B,QAAQ,aAAa,kBAAkB,SAC9D,SACA,QAAQ,aAAa,kBAAkB,UACrC,UACA;EAEN,eAAe,QAAQ;EACvB,UAAU,QAAQ;EACnB"}
+{"version":3,"file":"components.js","names":["serde","componentName: string","description?: string","metadata?: Record<string, string>","options?: ServiceOptions","ServiceHandler","handlers: d.Handler[]","handlerName: string","handlerWrapper: HandlerWrapper","parent: ServiceComponent","options?: ObjectOptions","parent: VirtualObjectComponent","options?: WorkflowOptions","WorkflowHandler","parent: WorkflowComponent"],"sources":["../../src/endpoint/components.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\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\nimport type * as d from \"./discovery.js\";\nimport type { ContextImpl } from \"../context_impl.js\";\nimport type {\n  HandlerWrapper,\n  ObjectOptions,\n  ServiceHandlerOpts,\n  ServiceOptions,\n  WorkflowOptions,\n} from \"../types/rpc.js\";\nimport { HandlerKind } from \"../types/rpc.js\";\nimport type { Serde } from \"@restatedev/restate-sdk-core\";\nimport { millisOrDurationToMillis, serde } from \"@restatedev/restate-sdk-core\";\nimport type { HooksProvider } from \"../hooks.js\";\nimport type { TerminalError } from \"../types/errors.js\";\n\n//\n// Interfaces\n//\nexport interface Component {\n  name(): string;\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined;\n  discovery(): d.Service;\n}\n\n/**\n * Execution-related options\n */\nexport interface ExecutionOptions {\n  asTerminalError?: (error: any) => TerminalError | undefined;\n  /**\n   * Default serde to use for requests, responses, state, side effects, awakeables, promises. Used when no other serde is specified.\n   */\n  defaultSerde?: Serde<any>;\n  hooks?: HooksProvider[];\n  explicitCancellation?: boolean;\n}\n\nexport interface ComponentHandler {\n  name(): string;\n  component(): Component;\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array>;\n  kind(): HandlerKind;\n\n  /**\n   * Returns the execution options, already merged with different layers (endpoint -> service -> handler)\n   */\n  executionOptions: ExecutionOptions;\n}\n\n//\n// Service\n//\n\nfunction handlerInputDiscovery(\n  handler: HandlerWrapper,\n  defaultSerde: Serde<any>\n): d.InputPayload {\n  const serde = handler.options?.input ?? defaultSerde;\n\n  let contentType = undefined;\n  let jsonSchema = undefined;\n\n  if (serde.jsonSchema) {\n    jsonSchema = serde.jsonSchema;\n    contentType = handler.options?.accept ?? serde.contentType;\n  } else if (handler.options?.accept) {\n    contentType = handler.options?.accept;\n  } else if (serde.contentType) {\n    contentType = serde.contentType;\n  } else {\n    // no input information\n    return {};\n  }\n\n  return {\n    required: false,\n    contentType,\n    jsonSchema,\n  };\n}\n\nfunction handlerOutputDiscovery(\n  handler: HandlerWrapper,\n  defaultSerde: Serde<any>\n): d.OutputPayload {\n  const serde = handler.options?.output ?? defaultSerde;\n  const setContentTypeIfEmpty =\n    handler.options?.setOutputContentTypeIfEmpty ?? false;\n\n  let contentType = undefined;\n  let jsonSchema = undefined;\n\n  if (serde.jsonSchema) {\n    jsonSchema = serde.jsonSchema;\n    contentType = serde.contentType ?? \"application/json\";\n  } else if (serde.contentType) {\n    contentType = serde.contentType;\n  } else {\n    // no input information\n    return { setContentTypeIfEmpty };\n  }\n\n  return {\n    setContentTypeIfEmpty,\n    jsonSchema,\n    contentType,\n  };\n}\n\nfunction createExecutionOptions(\n  serviceOptions?: ServiceOptions,\n  handlerOptions?: ServiceHandlerOpts<unknown, unknown>\n): ExecutionOptions {\n  // Service-level hooks run outermost, handler-level hooks run innermost.\n  // Both are merged into a single list: service hooks first, then handler hooks.\n  const hooks = [\n    ...(serviceOptions?.hooks ?? []),\n    ...(handlerOptions?.hooks ?? []),\n  ];\n  return {\n    defaultSerde: handlerOptions?.serde ?? serviceOptions?.serde,\n    asTerminalError:\n      handlerOptions?.asTerminalError ?? serviceOptions?.asTerminalError,\n    hooks: hooks.length > 0 ? hooks : undefined,\n    explicitCancellation:\n      handlerOptions?.explicitCancellation ??\n      serviceOptions?.explicitCancellation,\n  };\n}\n\nexport class ServiceComponent implements Component {\n  private readonly handlers: Map<string, ServiceHandler> = new Map();\n\n  constructor(\n    private readonly componentName: string,\n    public readonly description?: string,\n    public readonly metadata?: Record<string, string>,\n    public readonly options?: ServiceOptions\n  ) {}\n\n  name(): string {\n    return this.componentName;\n  }\n\n  add(name: string, handlerWrapper: HandlerWrapper) {\n    this.handlers.set(name, new ServiceHandler(name, handlerWrapper, this));\n  }\n\n  discovery(): d.Service {\n    const handlers: d.Handler[] = [...this.handlers.entries()].map(\n      ([name, handler]) => {\n        return {\n          name,\n          ...commonHandlerOptions(\n            handler.handlerWrapper,\n            handler.executionOptions.defaultSerde ?? serde.json\n          ),\n        } satisfies d.Handler;\n      }\n    );\n\n    return {\n      name: this.componentName,\n      ty: \"SERVICE\",\n      handlers,\n      documentation: this.description,\n      metadata: this.metadata,\n      ...commonServiceOptions(this.options),\n    } satisfies d.Service;\n  }\n\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined {\n    return this.handlers.get(url.handlerName);\n  }\n}\n\nexport class ServiceHandler implements ComponentHandler {\n  readonly executionOptions: ExecutionOptions;\n\n  constructor(\n    private readonly handlerName: string,\n    public readonly handlerWrapper: HandlerWrapper,\n    private readonly parent: ServiceComponent\n  ) {\n    this.executionOptions = createExecutionOptions(\n      this.parent.options,\n      handlerWrapper.options\n    );\n  }\n\n  name(): string {\n    return this.handlerName;\n  }\n\n  component(): Component {\n    return this.parent;\n  }\n\n  kind(): HandlerKind {\n    return this.handlerWrapper.kind;\n  }\n\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array> {\n    return this.handlerWrapper.invoke(context, input);\n  }\n}\n\n//\n// Virtual Object\n//\n\nexport class VirtualObjectComponent implements Component {\n  private readonly handlers: Map<string, VirtualObjectHandler> = new Map();\n\n  constructor(\n    public readonly componentName: string,\n    public readonly description?: string,\n    public readonly metadata?: Record<string, string>,\n    public readonly options?: ObjectOptions\n  ) {}\n\n  name(): string {\n    return this.componentName;\n  }\n\n  add(name: string, wrapper: HandlerWrapper) {\n    this.handlers.set(name, new VirtualObjectHandler(name, wrapper, this));\n  }\n\n  discovery(): d.Service {\n    const handlers: d.Handler[] = [...this.handlers.entries()].map(\n      ([name, handler]) => {\n        return {\n          name,\n          ty: handler.kind() === HandlerKind.EXCLUSIVE ? \"EXCLUSIVE\" : \"SHARED\",\n          ...commonHandlerOptions(\n            handler.handlerWrapper,\n            handler.executionOptions.defaultSerde ?? serde.json\n          ),\n        } satisfies d.Handler;\n      }\n    );\n\n    return {\n      name: this.componentName,\n      ty: \"VIRTUAL_OBJECT\",\n      handlers,\n      documentation: this.description,\n      metadata: this.metadata,\n      ...commonServiceOptions(this.options),\n    } satisfies d.Service;\n  }\n\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined {\n    return this.handlers.get(url.handlerName);\n  }\n}\n\nexport class VirtualObjectHandler implements ComponentHandler {\n  readonly executionOptions: ExecutionOptions;\n\n  constructor(\n    private readonly handlerName: string,\n    public readonly handlerWrapper: HandlerWrapper,\n    private readonly parent: VirtualObjectComponent\n  ) {\n    this.executionOptions = createExecutionOptions(\n      this.parent.options,\n      handlerWrapper.options\n    );\n  }\n\n  name(): string {\n    return this.handlerName;\n  }\n\n  component(): Component {\n    return this.parent;\n  }\n\n  kind(): HandlerKind {\n    return this.handlerWrapper.kind;\n  }\n\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array> {\n    return this.handlerWrapper.invoke(context, input);\n  }\n}\n\n// Workflow\n\nexport class WorkflowComponent implements Component {\n  private readonly handlers: Map<string, WorkflowHandler> = new Map();\n\n  constructor(\n    public readonly componentName: string,\n    public readonly description?: string,\n    public readonly metadata?: Record<string, string>,\n    public readonly options?: WorkflowOptions\n  ) {}\n\n  name(): string {\n    return this.componentName;\n  }\n\n  add(name: string, wrapper: HandlerWrapper) {\n    this.handlers.set(name, new WorkflowHandler(name, wrapper, this));\n  }\n\n  discovery(): d.Service {\n    const handlers: d.Handler[] = [...this.handlers.entries()].map(\n      ([name, handler]) => {\n        return {\n          name,\n          ty: handler.kind() === HandlerKind.WORKFLOW ? \"WORKFLOW\" : \"SHARED\",\n          workflowCompletionRetention:\n            handler.kind() === HandlerKind.WORKFLOW &&\n            this.options?.workflowRetention !== undefined\n              ? millisOrDurationToMillis(this.options?.workflowRetention)\n              : undefined,\n          ...commonHandlerOptions(\n            handler.handlerWrapper,\n            handler.executionOptions.defaultSerde ?? serde.json\n          ),\n        } satisfies d.Handler;\n      }\n    );\n\n    return {\n      name: this.componentName,\n      ty: \"WORKFLOW\",\n      handlers,\n      documentation: this.description,\n      metadata: this.metadata,\n      ...commonServiceOptions(this.options),\n    } satisfies d.Service;\n  }\n\n  handlerMatching(url: InvokePathComponents): ComponentHandler | undefined {\n    return this.handlers.get(url.handlerName);\n  }\n}\n\nexport class WorkflowHandler implements ComponentHandler {\n  readonly executionOptions: ExecutionOptions;\n\n  constructor(\n    private readonly handlerName: string,\n    public readonly handlerWrapper: HandlerWrapper,\n    private readonly parent: WorkflowComponent\n  ) {\n    this.executionOptions = createExecutionOptions(\n      this.parent.options,\n      handlerWrapper.options\n    );\n  }\n\n  name(): string {\n    return this.handlerName;\n  }\n  component(): Component {\n    return this.parent;\n  }\n\n  kind(): HandlerKind {\n    return this.handlerWrapper.kind;\n  }\n\n  invoke(context: ContextImpl, input: Uint8Array): Promise<Uint8Array> {\n    return this.handlerWrapper.invoke(context, input);\n  }\n}\n\nexport type PathComponents =\n  | InvokePathComponents\n  | { type: \"discover\" }\n  | { type: \"health\" }\n  | { type: \"unknown\"; path: string };\n\nexport type InvokePathComponents = {\n  type: \"invoke\";\n  componentName: string;\n  handlerName: string;\n};\n\nexport function parseUrlComponents(urlPath?: string): PathComponents {\n  if (!urlPath) {\n    return { type: \"unknown\", path: \"\" };\n  }\n  const fragments = urlPath.split(\"/\");\n  if (fragments.length >= 3 && fragments[fragments.length - 3] === \"invoke\") {\n    return {\n      type: \"invoke\",\n      componentName: fragments[fragments.length - 2]!,\n      handlerName: fragments[fragments.length - 1]!,\n    };\n  }\n  if (fragments.length > 0 && fragments[fragments.length - 1] === \"discover\") {\n    return { type: \"discover\" };\n  }\n  if (fragments.length > 0 && fragments[fragments.length - 1] === \"health\") {\n    return { type: \"health\" };\n  }\n  return { type: \"unknown\", path: urlPath };\n}\n\nfunction commonServiceOptions(\n  options?: ServiceOptions | ObjectOptions | WorkflowOptions\n): Partial<d.Service> {\n  return {\n    journalRetention:\n      options?.journalRetention !== undefined\n        ? millisOrDurationToMillis(options.journalRetention)\n        : undefined,\n    idempotencyRetention:\n      options?.idempotencyRetention !== undefined\n        ? millisOrDurationToMillis(options.idempotencyRetention)\n        : undefined,\n    inactivityTimeout:\n      options?.inactivityTimeout !== undefined\n        ? millisOrDurationToMillis(options.inactivityTimeout)\n        : undefined,\n    abortTimeout:\n      options?.abortTimeout !== undefined\n        ? millisOrDurationToMillis(options.abortTimeout)\n        : undefined,\n    ingressPrivate: options?.ingressPrivate,\n    enableLazyState:\n      options !== undefined && \"enableLazyState\" in options\n        ? options.enableLazyState\n        : undefined,\n    retryPolicyExponentiationFactor: options?.retryPolicy?.exponentiationFactor,\n    retryPolicyInitialInterval:\n      options?.retryPolicy?.initialInterval !== undefined\n        ? millisOrDurationToMillis(options?.retryPolicy?.initialInterval)\n        : undefined,\n    retryPolicyMaxInterval:\n      options?.retryPolicy?.maxInterval !== undefined\n        ? millisOrDurationToMillis(options?.retryPolicy?.maxInterval)\n        : undefined,\n    retryPolicyMaxAttempts: options?.retryPolicy?.maxAttempts,\n    retryPolicyOnMaxAttempts: (options?.retryPolicy?.onMaxAttempts === \"kill\"\n      ? \"KILL\"\n      : options?.retryPolicy?.onMaxAttempts === \"pause\"\n        ? \"PAUSE\"\n        : undefined) as d.RetryPolicyOnMaxAttempts,\n  };\n}\n\nfunction commonHandlerOptions(\n  wrapper: HandlerWrapper,\n  defaultSerde: Serde<any>\n) {\n  return {\n    input: handlerInputDiscovery(wrapper, defaultSerde),\n    output: handlerOutputDiscovery(wrapper, defaultSerde),\n    journalRetention:\n      wrapper.options?.journalRetention !== undefined\n        ? millisOrDurationToMillis(wrapper.options?.journalRetention)\n        : undefined,\n    idempotencyRetention:\n      wrapper.options?.idempotencyRetention !== undefined\n        ? millisOrDurationToMillis(wrapper.options?.idempotencyRetention)\n        : undefined,\n    inactivityTimeout:\n      wrapper.options?.inactivityTimeout !== undefined\n        ? millisOrDurationToMillis(wrapper.options?.inactivityTimeout)\n        : undefined,\n    abortTimeout:\n      wrapper.options?.abortTimeout !== undefined\n        ? millisOrDurationToMillis(wrapper.options?.abortTimeout)\n        : undefined,\n    ingressPrivate: wrapper.options?.ingressPrivate,\n    enableLazyState:\n      wrapper.options !== undefined && \"enableLazyState\" in wrapper.options\n        ? wrapper.options?.enableLazyState\n        : undefined,\n    retryPolicyExponentiationFactor:\n      wrapper.options?.retryPolicy?.exponentiationFactor,\n    retryPolicyInitialInterval:\n      wrapper.options?.retryPolicy?.initialInterval !== undefined\n        ? millisOrDurationToMillis(\n            wrapper.options?.retryPolicy?.initialInterval\n          )\n        : undefined,\n    retryPolicyMaxInterval:\n      wrapper.options?.retryPolicy?.maxInterval !== undefined\n        ? millisOrDurationToMillis(wrapper.options?.retryPolicy?.maxInterval)\n        : undefined,\n    retryPolicyMaxAttempts: wrapper.options?.retryPolicy?.maxAttempts,\n    retryPolicyOnMaxAttempts: (wrapper.options?.retryPolicy?.onMaxAttempts ===\n    \"kill\"\n      ? \"KILL\"\n      : wrapper.options?.retryPolicy?.onMaxAttempts === \"pause\"\n        ? \"PAUSE\"\n        : undefined) as d.RetryPolicyOnMaxAttempts1,\n\n    documentation: wrapper.options?.description,\n    metadata: wrapper.options?.metadata,\n  };\n}\n"],"mappings":";;;;AAkEA,SAAS,sBACP,SACA,cACgB;CAChB,MAAMA,UAAQ,QAAQ,SAAS,SAAS;CAExC,IAAI,cAAc;CAClB,IAAI,aAAa;AAEjB,KAAIA,QAAM,YAAY;AACpB,eAAaA,QAAM;AACnB,gBAAc,QAAQ,SAAS,UAAUA,QAAM;YACtC,QAAQ,SAAS,OAC1B,eAAc,QAAQ,SAAS;UACtBA,QAAM,YACf,eAAcA,QAAM;KAGpB,QAAO,EAAE;AAGX,QAAO;EACL,UAAU;EACV;EACA;EACD;;AAGH,SAAS,uBACP,SACA,cACiB;CACjB,MAAMA,UAAQ,QAAQ,SAAS,UAAU;CACzC,MAAM,wBACJ,QAAQ,SAAS,+BAA+B;CAElD,IAAI,cAAc;CAClB,IAAI,aAAa;AAEjB,KAAIA,QAAM,YAAY;AACpB,eAAaA,QAAM;AACnB,gBAAcA,QAAM,eAAe;YAC1BA,QAAM,YACf,eAAcA,QAAM;KAGpB,QAAO,EAAE,uBAAuB;AAGlC,QAAO;EACL;EACA;EACA;EACD;;AAGH,SAAS,uBACP,gBACA,gBACkB;CAGlB,MAAM,QAAQ,CACZ,GAAI,gBAAgB,SAAS,EAAE,EAC/B,GAAI,gBAAgB,SAAS,EAAE,CAChC;AACD,QAAO;EACL,cAAc,gBAAgB,SAAS,gBAAgB;EACvD,iBACE,gBAAgB,mBAAmB,gBAAgB;EACrD,OAAO,MAAM,SAAS,IAAI,QAAQ;EAClC,sBACE,gBAAgB,wBAChB,gBAAgB;EACnB;;AAGH,IAAa,mBAAb,MAAmD;CACjD,AAAiB,2BAAwC,IAAI,KAAK;CAElE,YACE,AAAiBC,eACjB,AAAgBC,aAChB,AAAgBC,UAChB,AAAgBC,SAChB;EAJiB;EACD;EACA;EACA;;CAGlB,OAAe;AACb,SAAO,KAAK;;CAGd,IAAI,MAAc,gBAAgC;AAChD,OAAK,SAAS,IAAI,MAAM,IAAIC,iBAAe,MAAM,gBAAgB,KAAK,CAAC;;CAGzE,YAAuB;EACrB,MAAMC,WAAwB,CAAC,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,KACxD,CAAC,MAAM,aAAa;AACnB,UAAO;IACL;IACA,GAAG,qBACD,QAAQ,gBACR,QAAQ,iBAAiB,gBAAgB,MAAM,KAChD;IACF;IAEJ;AAED,SAAO;GACL,MAAM,KAAK;GACX,IAAI;GACJ;GACA,eAAe,KAAK;GACpB,UAAU,KAAK;GACf,GAAG,qBAAqB,KAAK,QAAQ;GACtC;;CAGH,gBAAgB,KAAyD;AACvE,SAAO,KAAK,SAAS,IAAI,IAAI,YAAY;;;AAI7C,IAAaD,mBAAb,MAAwD;CACtD,AAAS;CAET,YACE,AAAiBE,aACjB,AAAgBC,gBAChB,AAAiBC,QACjB;EAHiB;EACD;EACC;AAEjB,OAAK,mBAAmB,uBACtB,KAAK,OAAO,SACZ,eAAe,QAChB;;CAGH,OAAe;AACb,SAAO,KAAK;;CAGd,YAAuB;AACrB,SAAO,KAAK;;CAGd,OAAoB;AAClB,SAAO,KAAK,eAAe;;CAG7B,OAAO,SAAsB,OAAwC;AACnE,SAAO,KAAK,eAAe,OAAO,SAAS,MAAM;;;AAQrD,IAAa,yBAAb,MAAyD;CACvD,AAAiB,2BAA8C,IAAI,KAAK;CAExE,YACE,AAAgBR,eAChB,AAAgBC,aAChB,AAAgBC,UAChB,AAAgBO,SAChB;EAJgB;EACA;EACA;EACA;;CAGlB,OAAe;AACb,SAAO,KAAK;;CAGd,IAAI,MAAc,SAAyB;AACzC,OAAK,SAAS,IAAI,MAAM,IAAI,qBAAqB,MAAM,SAAS,KAAK,CAAC;;CAGxE,YAAuB;EACrB,MAAMJ,WAAwB,CAAC,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,KACxD,CAAC,MAAM,aAAa;AACnB,UAAO;IACL;IACA,IAAI,QAAQ,MAAM,KAAK,YAAY,YAAY,cAAc;IAC7D,GAAG,qBACD,QAAQ,gBACR,QAAQ,iBAAiB,gBAAgB,MAAM,KAChD;IACF;IAEJ;AAED,SAAO;GACL,MAAM,KAAK;GACX,IAAI;GACJ;GACA,eAAe,KAAK;GACpB,UAAU,KAAK;GACf,GAAG,qBAAqB,KAAK,QAAQ;GACtC;;CAGH,gBAAgB,KAAyD;AACvE,SAAO,KAAK,SAAS,IAAI,IAAI,YAAY;;;AAI7C,IAAa,uBAAb,MAA8D;CAC5D,AAAS;CAET,YACE,AAAiBC,aACjB,AAAgBC,gBAChB,AAAiBG,QACjB;EAHiB;EACD;EACC;AAEjB,OAAK,mBAAmB,uBACtB,KAAK,OAAO,SACZ,eAAe,QAChB;;CAGH,OAAe;AACb,SAAO,KAAK;;CAGd,YAAuB;AACrB,SAAO,KAAK;;CAGd,OAAoB;AAClB,SAAO,KAAK,eAAe;;CAG7B,OAAO,SAAsB,OAAwC;AACnE,SAAO,KAAK,eAAe,OAAO,SAAS,MAAM;;;AAMrD,IAAa,oBAAb,MAAoD;CAClD,AAAiB,2BAAyC,IAAI,KAAK;CAEnE,YACE,AAAgBV,eAChB,AAAgBC,aAChB,AAAgBC,UAChB,AAAgBS,SAChB;EAJgB;EACA;EACA;EACA;;CAGlB,OAAe;AACb,SAAO,KAAK;;CAGd,IAAI,MAAc,SAAyB;AACzC,OAAK,SAAS,IAAI,MAAM,IAAIC,kBAAgB,MAAM,SAAS,KAAK,CAAC;;CAGnE,YAAuB;EACrB,MAAMP,WAAwB,CAAC,GAAG,KAAK,SAAS,SAAS,CAAC,CAAC,KACxD,CAAC,MAAM,aAAa;AACnB,UAAO;IACL;IACA,IAAI,QAAQ,MAAM,KAAK,YAAY,WAAW,aAAa;IAC3D,6BACE,QAAQ,MAAM,KAAK,YAAY,YAC/B,KAAK,SAAS,sBAAsB,SAChC,yBAAyB,KAAK,SAAS,kBAAkB,GACzD;IACN,GAAG,qBACD,QAAQ,gBACR,QAAQ,iBAAiB,gBAAgB,MAAM,KAChD;IACF;IAEJ;AAED,SAAO;GACL,MAAM,KAAK;GACX,IAAI;GACJ;GACA,eAAe,KAAK;GACpB,UAAU,KAAK;GACf,GAAG,qBAAqB,KAAK,QAAQ;GACtC;;CAGH,gBAAgB,KAAyD;AACvE,SAAO,KAAK,SAAS,IAAI,IAAI,YAAY;;;AAI7C,IAAaO,oBAAb,MAAyD;CACvD,AAAS;CAET,YACE,AAAiBN,aACjB,AAAgBC,gBAChB,AAAiBM,QACjB;EAHiB;EACD;EACC;AAEjB,OAAK,mBAAmB,uBACtB,KAAK,OAAO,SACZ,eAAe,QAChB;;CAGH,OAAe;AACb,SAAO,KAAK;;CAEd,YAAuB;AACrB,SAAO,KAAK;;CAGd,OAAoB;AAClB,SAAO,KAAK,eAAe;;CAG7B,OAAO,SAAsB,OAAwC;AACnE,SAAO,KAAK,eAAe,OAAO,SAAS,MAAM;;;AAgBrD,SAAgB,mBAAmB,SAAkC;AACnE,KAAI,CAAC,QACH,QAAO;EAAE,MAAM;EAAW,MAAM;EAAI;CAEtC,MAAM,YAAY,QAAQ,MAAM,IAAI;AACpC,KAAI,UAAU,UAAU,KAAK,UAAU,UAAU,SAAS,OAAO,SAC/D,QAAO;EACL,MAAM;EACN,eAAe,UAAU,UAAU,SAAS;EAC5C,aAAa,UAAU,UAAU,SAAS;EAC3C;AAEH,KAAI,UAAU,SAAS,KAAK,UAAU,UAAU,SAAS,OAAO,WAC9D,QAAO,EAAE,MAAM,YAAY;AAE7B,KAAI,UAAU,SAAS,KAAK,UAAU,UAAU,SAAS,OAAO,SAC9D,QAAO,EAAE,MAAM,UAAU;AAE3B,QAAO;EAAE,MAAM;EAAW,MAAM;EAAS;;AAG3C,SAAS,qBACP,SACoB;AACpB,QAAO;EACL,kBACE,SAAS,qBAAqB,SAC1B,yBAAyB,QAAQ,iBAAiB,GAClD;EACN,sBACE,SAAS,yBAAyB,SAC9B,yBAAyB,QAAQ,qBAAqB,GACtD;EACN,mBACE,SAAS,sBAAsB,SAC3B,yBAAyB,QAAQ,kBAAkB,GACnD;EACN,cACE,SAAS,iBAAiB,SACtB,yBAAyB,QAAQ,aAAa,GAC9C;EACN,gBAAgB,SAAS;EACzB,iBACE,YAAY,UAAa,qBAAqB,UAC1C,QAAQ,kBACR;EACN,iCAAiC,SAAS,aAAa;EACvD,4BACE,SAAS,aAAa,oBAAoB,SACtC,yBAAyB,SAAS,aAAa,gBAAgB,GAC/D;EACN,wBACE,SAAS,aAAa,gBAAgB,SAClC,yBAAyB,SAAS,aAAa,YAAY,GAC3D;EACN,wBAAwB,SAAS,aAAa;EAC9C,0BAA2B,SAAS,aAAa,kBAAkB,SAC/D,SACA,SAAS,aAAa,kBAAkB,UACtC,UACA;EACP;;AAGH,SAAS,qBACP,SACA,cACA;AACA,QAAO;EACL,OAAO,sBAAsB,SAAS,aAAa;EACnD,QAAQ,uBAAuB,SAAS,aAAa;EACrD,kBACE,QAAQ,SAAS,qBAAqB,SAClC,yBAAyB,QAAQ,SAAS,iBAAiB,GAC3D;EACN,sBACE,QAAQ,SAAS,yBAAyB,SACtC,yBAAyB,QAAQ,SAAS,qBAAqB,GAC/D;EACN,mBACE,QAAQ,SAAS,sBAAsB,SACnC,yBAAyB,QAAQ,SAAS,kBAAkB,GAC5D;EACN,cACE,QAAQ,SAAS,iBAAiB,SAC9B,yBAAyB,QAAQ,SAAS,aAAa,GACvD;EACN,gBAAgB,QAAQ,SAAS;EACjC,iBACE,QAAQ,YAAY,UAAa,qBAAqB,QAAQ,UAC1D,QAAQ,SAAS,kBACjB;EACN,iCACE,QAAQ,SAAS,aAAa;EAChC,4BACE,QAAQ,SAAS,aAAa,oBAAoB,SAC9C,yBACE,QAAQ,SAAS,aAAa,gBAC/B,GACD;EACN,wBACE,QAAQ,SAAS,aAAa,gBAAgB,SAC1C,yBAAyB,QAAQ,SAAS,aAAa,YAAY,GACnE;EACN,wBAAwB,QAAQ,SAAS,aAAa;EACtD,0BAA2B,QAAQ,SAAS,aAAa,kBACzD,SACI,SACA,QAAQ,SAAS,aAAa,kBAAkB,UAC9C,UACA;EAEN,eAAe,QAAQ,SAAS;EAChC,UAAU,QAAQ,SAAS;EAC5B"}

package/dist/endpoint/endpoint.cjs

@@ -1,7 +1,7 @@
-const require_logger_transport = require('../logging/logger_transport.cjs');
-const require_console_logger_transport = require('../logging/console_logger_transport.cjs');
 const require_rpc = require('../types/rpc.cjs');
 const require_components = require('./components.cjs');
+const require_logger_transport = require('../logging/logger_transport.cjs');
+const require_console_logger_transport = require('../logging/console_logger_transport.cjs');
 const require_logger = require('../logging/logger.cjs');
 
 //#region src/endpoint/endpoint.ts

package/dist/endpoint/endpoint.d.cts

@@ -0,0 +1,5 @@
+import "./components.cjs";
+import "../logging/logger_transport.cjs";
+import "../logging/logger.cjs";
+import "../endpoint.cjs";
+import { JournalValueCodec, ServiceDefinition, VirtualObjectDefinition, WorkflowDefinition } from "@restatedev/restate-sdk-core";

package/dist/endpoint/endpoint.d.ts

@@ -0,0 +1,5 @@
+import "./components.js";
+import "../logging/logger_transport.js";
+import "../logging/logger.js";
+import "../endpoint.js";
+import { JournalValueCodec, ServiceDefinition, VirtualObjectDefinition, WorkflowDefinition } from "@restatedev/restate-sdk-core";

package/dist/endpoint/endpoint.js

@@ -1,7 +1,7 @@
-import { LogSource } from "../logging/logger_transport.js";
-import { defaultLoggerTransport } from "../logging/console_logger_transport.js";
 import { HandlerWrapper } from "../types/rpc.js";
 import { ServiceComponent, VirtualObjectComponent, WorkflowComponent } from "./components.js";
+import { LogSource } from "../logging/logger_transport.js";
+import { defaultLoggerTransport } from "../logging/console_logger_transport.js";
 import { createLogger } from "../logging/logger.js";
 
 //#region src/endpoint/endpoint.ts

package/dist/endpoint/endpoint.js.map

@@ -1 +1 @@
-{"version":3,"file":"endpoint.js","names":["name"],"sources":["../../src/endpoint/endpoint.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\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\nimport type {\n  JournalValueCodec,\n  ServiceDefinition,\n  VirtualObjectDefinition,\n  WorkflowDefinition,\n} from \"@restatedev/restate-sdk-core\";\n\nimport type {\n  ObjectOptions,\n  ServiceOptions,\n  WorkflowOptions,\n} from \"../types/rpc.js\";\nimport { HandlerWrapper } from \"../types/rpc.js\";\nimport type { Component } from \"./components.js\";\nimport {\n  ServiceComponent,\n  VirtualObjectComponent,\n  WorkflowComponent,\n} from \"./components.js\";\nimport type * as discovery from \"./discovery.js\";\nimport { defaultLoggerTransport } from \"../logging/console_logger_transport.js\";\nimport {\n  type LoggerTransport,\n  LogSource,\n} from \"../logging/logger_transport.js\";\nimport type { Logger } from \"../logging/logger.js\";\nimport { createLogger } from \"../logging/logger.js\";\nimport type { DefaultServiceOptions } from \"../endpoint.js\";\n\nfunction isServiceDefinition<P extends string, M>(\n  m: Record<string, any>\n): m is ServiceDefinition<P, M> & { service: M } {\n  return m && m.service !== undefined;\n}\n\nfunction isObjectDefinition<P extends string, M>(\n  m: Record<string, any>\n): m is VirtualObjectDefinition<P, M> & { object: M } {\n  return m && m.object !== undefined;\n}\n\nfunction isWorkflowDefinition<P extends string, M>(\n  m: Record<string, any>\n): m is WorkflowDefinition<P, M> & { workflow: M } {\n  return m && m.workflow !== undefined;\n}\n\n/**\n * Services can have additional information that is not part of the definition.\n * For example a description or metadata.\n */\ntype ServiceAuxInfo = {\n  description?: string;\n  metadata?: Record<string, any>;\n  options?: ServiceOptions | ObjectOptions | WorkflowOptions;\n};\n\nexport type Endpoint = {\n  loggerTransport: LoggerTransport;\n  components: Map<string, Component>;\n  keySet: string[];\n  /**\n   * This is a simple console without contextual info.\n   *\n   * This should be used only in cases where no contextual info is available.\n   */\n  rlog: Logger;\n  /**\n   * All discovery metadata, except protocol mode provided by the node/fetch/lambda endpoint implementations\n   */\n  discoveryMetadata: Omit<discovery.Endpoint, \"protocolMode\">;\n\n  /**\n   * Codec provider to use for journal values.\n   */\n  journalValueCodec?: Promise<JournalValueCodec>;\n};\n\nexport class EndpointBuilder {\n  private readonly serviceDefinitions: Map<\n    string,\n    | ServiceDefinition<string, any>\n    | VirtualObjectDefinition<string, any>\n    | WorkflowDefinition<string, any>\n  > = new Map();\n  private loggerTransport: LoggerTransport = defaultLoggerTransport;\n  private keySet: string[] = [];\n  private defaultServiceOptions: DefaultServiceOptions = {};\n  private journalValueCodecProvider?: () => Promise<JournalValueCodec>;\n\n  public bind<P extends string, M>(\n    definition:\n      | ServiceDefinition<P, M>\n      | VirtualObjectDefinition<P, M>\n      | WorkflowDefinition<P, M>\n  ) {\n    // Validate service name\n    if (definition.name.indexOf(\"/\") !== -1) {\n      throw new Error(\"service name must not contain any slash '/'\");\n    }\n\n    this.serviceDefinitions.set(definition.name, definition);\n  }\n\n  public addIdentityKeys(...keys: string[]) {\n    this.keySet.push(...keys);\n  }\n\n  public setDefaultServiceOptions(options: DefaultServiceOptions) {\n    this.defaultServiceOptions = options;\n  }\n\n  public setLogger(newLogger: LoggerTransport) {\n    this.loggerTransport = newLogger;\n  }\n\n  public setJournalValueCodecProvider(\n    codecProvider: () => Promise<JournalValueCodec>\n  ) {\n    this.journalValueCodecProvider = codecProvider;\n  }\n\n  public build(): Endpoint {\n    const rlog = createLogger(this.loggerTransport, LogSource.SYSTEM);\n\n    // Build the components\n    const components = new Map<string, Component>();\n    for (const [name, definition] of this.serviceDefinitions) {\n      if (isServiceDefinition(definition)) {\n        const { name, service } = definition;\n        if (!service) {\n          throw new TypeError(`no service implementation found.`);\n        }\n        components.set(\n          name,\n          buildServiceComponent(\n            name,\n            service,\n            definition as ServiceAuxInfo,\n            this.defaultServiceOptions\n          )\n        );\n      } else if (isObjectDefinition(definition)) {\n        const { name, object } = definition;\n        if (!object) {\n          throw new TypeError(`no object implementation found.`);\n        }\n        components.set(\n          name,\n          buildVirtualObjectComponent(\n            name,\n            object,\n            definition as ServiceAuxInfo,\n            this.defaultServiceOptions\n          )\n        );\n      } else if (isWorkflowDefinition(definition)) {\n        const { name, workflow } = definition;\n        if (!workflow) {\n          throw new TypeError(`no workflow implementation found.`);\n        }\n        components.set(\n          name,\n          buildWorkflowComponent(\n            name,\n            workflow,\n            definition as ServiceAuxInfo,\n            this.defaultServiceOptions\n          )\n        );\n      } else {\n        throw new TypeError(\n          `cannot bind ${name}, can only bind a service or a virtual object or a workflow definition`\n        );\n      }\n    }\n\n    // Compute discovery metadata\n    const discoveryMetadata = computeDiscovery(components);\n\n    return {\n      keySet: this.keySet,\n      loggerTransport: this.loggerTransport,\n      rlog,\n      components,\n      discoveryMetadata,\n      journalValueCodec: this.journalValueCodecProvider\n        ? this.journalValueCodecProvider()\n        : undefined,\n    };\n  }\n}\n\nfunction computeDiscovery(\n  components: Map<string, Component>\n): discovery.Endpoint {\n  return {\n    minProtocolVersion: 5,\n    maxProtocolVersion: 6,\n    services: [...components.values()].map((c) => c.discovery()),\n  };\n}\n\nfunction buildServiceComponent(\n  name: string,\n  router: any,\n  definition: ServiceAuxInfo,\n  defaultServiceOptions: DefaultServiceOptions\n): ServiceComponent {\n  if (name.indexOf(\"/\") !== -1) {\n    throw new Error(\"service name must not contain any slash '/'\");\n  }\n  const component = new ServiceComponent(\n    name,\n    definition.description,\n    definition.metadata,\n    {\n      ...defaultServiceOptions,\n      ...(definition?.options as ServiceOptions),\n    }\n  );\n\n  for (const [route, handler] of Object.entries(\n    router as { [s: string]: any }\n  )) {\n    const wrapper = HandlerWrapper.fromHandler(handler);\n    if (!wrapper) {\n      throw new TypeError(`${route} is not a restate handler.`);\n    }\n    wrapper.bindInstance(router);\n    component.add(route, wrapper);\n  }\n\n  return component;\n}\n\nfunction buildVirtualObjectComponent(\n  name: string,\n  router: any,\n  definition: ServiceAuxInfo,\n  defaultServiceOptions: DefaultServiceOptions\n): VirtualObjectComponent {\n  if (name.indexOf(\"/\") !== -1) {\n    throw new Error(\"service name must not contain any slash '/'\");\n  }\n  const component = new VirtualObjectComponent(\n    name,\n    definition.description,\n    definition.metadata,\n    {\n      ...defaultServiceOptions,\n      ...(definition?.options as ObjectOptions),\n    }\n  );\n\n  for (const [route, handler] of Object.entries(\n    router as { [s: string]: any }\n  )) {\n    const wrapper = HandlerWrapper.fromHandler(handler);\n    if (!wrapper) {\n      throw new TypeError(`${route} is not a restate handler.`);\n    }\n    wrapper.bindInstance(router);\n    component.add(route, wrapper);\n  }\n  return component;\n}\n\nfunction buildWorkflowComponent(\n  name: string,\n  workflow: any,\n  definition: ServiceAuxInfo,\n  defaultServiceOptions: DefaultServiceOptions\n): WorkflowComponent {\n  if (name.indexOf(\"/\") !== -1) {\n    throw new Error(\"service name must not contain any slash '/'\");\n  }\n  const component = new WorkflowComponent(\n    name,\n    definition.description,\n    definition.metadata,\n    {\n      ...defaultServiceOptions,\n      ...(definition?.options as WorkflowOptions),\n    }\n  );\n\n  for (const [route, handler] of Object.entries(\n    workflow as { [s: string]: any }\n  )) {\n    const wrapper = HandlerWrapper.fromHandler(handler);\n    if (!wrapper) {\n      throw new TypeError(`${route} is not a restate handler.`);\n    }\n    wrapper.bindInstance(workflow);\n    component.add(route, wrapper);\n  }\n  return component;\n}\n"],"mappings":";;;;;;;AA0CA,SAAS,oBACP,GAC+C;AAC/C,QAAO,KAAK,EAAE,YAAY;;AAG5B,SAAS,mBACP,GACoD;AACpD,QAAO,KAAK,EAAE,WAAW;;AAG3B,SAAS,qBACP,GACiD;AACjD,QAAO,KAAK,EAAE,aAAa;;AAkC7B,IAAa,kBAAb,MAA6B;CAC3B,AAAiB,qCAKb,IAAI,KAAK;CACb,AAAQ,kBAAmC;CAC3C,AAAQ,SAAmB,EAAE;CAC7B,AAAQ,wBAA+C,EAAE;CACzD,AAAQ;CAER,AAAO,KACL,YAIA;AAEA,MAAI,WAAW,KAAK,QAAQ,IAAI,KAAK,GACnC,OAAM,IAAI,MAAM,8CAA8C;AAGhE,OAAK,mBAAmB,IAAI,WAAW,MAAM,WAAW;;CAG1D,AAAO,gBAAgB,GAAG,MAAgB;AACxC,OAAK,OAAO,KAAK,GAAG,KAAK;;CAG3B,AAAO,yBAAyB,SAAgC;AAC9D,OAAK,wBAAwB;;CAG/B,AAAO,UAAU,WAA4B;AAC3C,OAAK,kBAAkB;;CAGzB,AAAO,6BACL,eACA;AACA,OAAK,4BAA4B;;CAGnC,AAAO,QAAkB;EACvB,MAAM,OAAO,aAAa,KAAK,iBAAiB,UAAU,OAAO;EAGjE,MAAM,6BAAa,IAAI,KAAwB;AAC/C,OAAK,MAAM,CAAC,MAAM,eAAe,KAAK,mBACpC,KAAI,oBAAoB,WAAW,EAAE;GACnC,MAAM,EAAE,cAAM,YAAY;AAC1B,OAAI,CAAC,QACH,OAAM,IAAI,UAAU,mCAAmC;AAEzD,cAAW,IACTA,QACA,sBACEA,QACA,SACA,YACA,KAAK,sBACN,CACF;aACQ,mBAAmB,WAAW,EAAE;GACzC,MAAM,EAAE,cAAM,WAAW;AACzB,OAAI,CAAC,OACH,OAAM,IAAI,UAAU,kCAAkC;AAExD,cAAW,IACTA,QACA,4BACEA,QACA,QACA,YACA,KAAK,sBACN,CACF;aACQ,qBAAqB,WAAW,EAAE;GAC3C,MAAM,EAAE,cAAM,aAAa;AAC3B,OAAI,CAAC,SACH,OAAM,IAAI,UAAU,oCAAoC;AAE1D,cAAW,IACTA,QACA,uBACEA,QACA,UACA,YACA,KAAK,sBACN,CACF;QAED,OAAM,IAAI,UACR,eAAe,KAAK,wEACrB;EAKL,MAAM,oBAAoB,iBAAiB,WAAW;AAEtD,SAAO;GACL,QAAQ,KAAK;GACb,iBAAiB,KAAK;GACtB;GACA;GACA;GACA,mBAAmB,KAAK,4BACpB,KAAK,2BAA2B,GAChC;GACL;;;AAIL,SAAS,iBACP,YACoB;AACpB,QAAO;EACL,oBAAoB;EACpB,oBAAoB;EACpB,UAAU,CAAC,GAAG,WAAW,QAAQ,CAAC,CAAC,KAAK,MAAM,EAAE,WAAW,CAAC;EAC7D;;AAGH,SAAS,sBACP,MACA,QACA,YACA,uBACkB;AAClB,KAAI,KAAK,QAAQ,IAAI,KAAK,GACxB,OAAM,IAAI,MAAM,8CAA8C;CAEhE,MAAM,YAAY,IAAI,iBACpB,MACA,WAAW,aACX,WAAW,UACX;EACE,GAAG;EACH,GAAI,YAAY;EACjB,CACF;AAED,MAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QACpC,OACD,EAAE;EACD,MAAM,UAAU,eAAe,YAAY,QAAQ;AACnD,MAAI,CAAC,QACH,OAAM,IAAI,UAAU,GAAG,MAAM,4BAA4B;AAE3D,UAAQ,aAAa,OAAO;AAC5B,YAAU,IAAI,OAAO,QAAQ;;AAG/B,QAAO;;AAGT,SAAS,4BACP,MACA,QACA,YACA,uBACwB;AACxB,KAAI,KAAK,QAAQ,IAAI,KAAK,GACxB,OAAM,IAAI,MAAM,8CAA8C;CAEhE,MAAM,YAAY,IAAI,uBACpB,MACA,WAAW,aACX,WAAW,UACX;EACE,GAAG;EACH,GAAI,YAAY;EACjB,CACF;AAED,MAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QACpC,OACD,EAAE;EACD,MAAM,UAAU,eAAe,YAAY,QAAQ;AACnD,MAAI,CAAC,QACH,OAAM,IAAI,UAAU,GAAG,MAAM,4BAA4B;AAE3D,UAAQ,aAAa,OAAO;AAC5B,YAAU,IAAI,OAAO,QAAQ;;AAE/B,QAAO;;AAGT,SAAS,uBACP,MACA,UACA,YACA,uBACmB;AACnB,KAAI,KAAK,QAAQ,IAAI,KAAK,GACxB,OAAM,IAAI,MAAM,8CAA8C;CAEhE,MAAM,YAAY,IAAI,kBACpB,MACA,WAAW,aACX,WAAW,UACX;EACE,GAAG;EACH,GAAI,YAAY;EACjB,CACF;AAED,MAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QACpC,SACD,EAAE;EACD,MAAM,UAAU,eAAe,YAAY,QAAQ;AACnD,MAAI,CAAC,QACH,OAAM,IAAI,UAAU,GAAG,MAAM,4BAA4B;AAE3D,UAAQ,aAAa,SAAS;AAC9B,YAAU,IAAI,OAAO,QAAQ;;AAE/B,QAAO"}
+{"version":3,"file":"endpoint.js","names":["name"],"sources":["../../src/endpoint/endpoint.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\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\nimport type {\n  JournalValueCodec,\n  ServiceDefinition,\n  VirtualObjectDefinition,\n  WorkflowDefinition,\n} from \"@restatedev/restate-sdk-core\";\n\nimport type {\n  ObjectOptions,\n  ServiceOptions,\n  WorkflowOptions,\n} from \"../types/rpc.js\";\nimport { HandlerWrapper } from \"../types/rpc.js\";\nimport type { Component } from \"./components.js\";\nimport {\n  ServiceComponent,\n  VirtualObjectComponent,\n  WorkflowComponent,\n} from \"./components.js\";\nimport type * as discovery from \"./discovery.js\";\nimport { defaultLoggerTransport } from \"../logging/console_logger_transport.js\";\nimport {\n  type LoggerTransport,\n  LogSource,\n} from \"../logging/logger_transport.js\";\nimport type { Logger } from \"../logging/logger.js\";\nimport { createLogger } from \"../logging/logger.js\";\nimport type { DefaultServiceOptions } from \"../endpoint.js\";\n\nfunction isServiceDefinition<P extends string, M>(\n  m: Record<string, any>\n): m is ServiceDefinition<P, M> & { service: M } {\n  return m && m.service !== undefined;\n}\n\nfunction isObjectDefinition<P extends string, M>(\n  m: Record<string, any>\n): m is VirtualObjectDefinition<P, M> & { object: M } {\n  return m && m.object !== undefined;\n}\n\nfunction isWorkflowDefinition<P extends string, M>(\n  m: Record<string, any>\n): m is WorkflowDefinition<P, M> & { workflow: M } {\n  return m && m.workflow !== undefined;\n}\n\n/**\n * Services can have additional information that is not part of the definition.\n * For example a description or metadata.\n */\ntype ServiceAuxInfo = {\n  description?: string;\n  metadata?: Record<string, any>;\n  options?: ServiceOptions | ObjectOptions | WorkflowOptions;\n};\n\nexport type Endpoint = {\n  loggerTransport: LoggerTransport;\n  components: Map<string, Component>;\n  keySet: string[];\n  /**\n   * This is a simple console without contextual info.\n   *\n   * This should be used only in cases where no contextual info is available.\n   */\n  rlog: Logger;\n  /**\n   * All discovery metadata, except protocol mode provided by the node/fetch/lambda endpoint implementations\n   */\n  discoveryMetadata: Omit<discovery.Endpoint, \"protocolMode\">;\n\n  /**\n   * Codec provider to use for journal values.\n   */\n  journalValueCodec?: Promise<JournalValueCodec>;\n};\n\nexport class EndpointBuilder {\n  private readonly serviceDefinitions: Map<\n    string,\n    | ServiceDefinition<string, any>\n    | VirtualObjectDefinition<string, any>\n    | WorkflowDefinition<string, any>\n  > = new Map();\n  private loggerTransport: LoggerTransport = defaultLoggerTransport;\n  private keySet: string[] = [];\n  private defaultServiceOptions: DefaultServiceOptions = {};\n  private journalValueCodecProvider?: () => Promise<JournalValueCodec>;\n\n  public bind<P extends string, M>(\n    definition:\n      | ServiceDefinition<P, M>\n      | VirtualObjectDefinition<P, M>\n      | WorkflowDefinition<P, M>\n  ) {\n    // Validate service name\n    if (definition.name.indexOf(\"/\") !== -1) {\n      throw new Error(\"service name must not contain any slash '/'\");\n    }\n\n    this.serviceDefinitions.set(definition.name, definition);\n  }\n\n  public addIdentityKeys(...keys: string[]) {\n    this.keySet.push(...keys);\n  }\n\n  public setDefaultServiceOptions(options: DefaultServiceOptions) {\n    this.defaultServiceOptions = options;\n  }\n\n  public setLogger(newLogger: LoggerTransport) {\n    this.loggerTransport = newLogger;\n  }\n\n  public setJournalValueCodecProvider(\n    codecProvider: () => Promise<JournalValueCodec>\n  ) {\n    this.journalValueCodecProvider = codecProvider;\n  }\n\n  public build(): Endpoint {\n    const rlog = createLogger(this.loggerTransport, LogSource.SYSTEM);\n\n    // Build the components\n    const components = new Map<string, Component>();\n    for (const [name, definition] of this.serviceDefinitions) {\n      if (isServiceDefinition(definition)) {\n        const { name, service } = definition;\n        if (!service) {\n          throw new TypeError(`no service implementation found.`);\n        }\n        components.set(\n          name,\n          buildServiceComponent(\n            name,\n            service,\n            definition as ServiceAuxInfo,\n            this.defaultServiceOptions\n          )\n        );\n      } else if (isObjectDefinition(definition)) {\n        const { name, object } = definition;\n        if (!object) {\n          throw new TypeError(`no object implementation found.`);\n        }\n        components.set(\n          name,\n          buildVirtualObjectComponent(\n            name,\n            object,\n            definition as ServiceAuxInfo,\n            this.defaultServiceOptions\n          )\n        );\n      } else if (isWorkflowDefinition(definition)) {\n        const { name, workflow } = definition;\n        if (!workflow) {\n          throw new TypeError(`no workflow implementation found.`);\n        }\n        components.set(\n          name,\n          buildWorkflowComponent(\n            name,\n            workflow,\n            definition as ServiceAuxInfo,\n            this.defaultServiceOptions\n          )\n        );\n      } else {\n        throw new TypeError(\n          `cannot bind ${name}, can only bind a service or a virtual object or a workflow definition`\n        );\n      }\n    }\n\n    // Compute discovery metadata\n    const discoveryMetadata = computeDiscovery(components);\n\n    return {\n      keySet: this.keySet,\n      loggerTransport: this.loggerTransport,\n      rlog,\n      components,\n      discoveryMetadata,\n      journalValueCodec: this.journalValueCodecProvider\n        ? this.journalValueCodecProvider()\n        : undefined,\n    };\n  }\n}\n\nfunction computeDiscovery(\n  components: Map<string, Component>\n): discovery.Endpoint {\n  return {\n    minProtocolVersion: 5,\n    maxProtocolVersion: 6,\n    services: [...components.values()].map((c) => c.discovery()),\n  };\n}\n\nfunction buildServiceComponent(\n  name: string,\n  router: any,\n  definition: ServiceAuxInfo,\n  defaultServiceOptions: DefaultServiceOptions\n): ServiceComponent {\n  if (name.indexOf(\"/\") !== -1) {\n    throw new Error(\"service name must not contain any slash '/'\");\n  }\n  const component = new ServiceComponent(\n    name,\n    definition.description,\n    definition.metadata,\n    {\n      ...defaultServiceOptions,\n      ...(definition?.options as ServiceOptions | undefined),\n    }\n  );\n\n  for (const [route, handler] of Object.entries(\n    router as { [s: string]: any }\n  )) {\n    const wrapper = HandlerWrapper.fromHandler(handler);\n    if (!wrapper) {\n      throw new TypeError(`${route} is not a restate handler.`);\n    }\n    wrapper.bindInstance(router);\n    component.add(route, wrapper);\n  }\n\n  return component;\n}\n\nfunction buildVirtualObjectComponent(\n  name: string,\n  router: any,\n  definition: ServiceAuxInfo,\n  defaultServiceOptions: DefaultServiceOptions\n): VirtualObjectComponent {\n  if (name.indexOf(\"/\") !== -1) {\n    throw new Error(\"service name must not contain any slash '/'\");\n  }\n  const component = new VirtualObjectComponent(\n    name,\n    definition.description,\n    definition.metadata,\n    {\n      ...defaultServiceOptions,\n      ...(definition?.options as ObjectOptions | undefined),\n    }\n  );\n\n  for (const [route, handler] of Object.entries(\n    router as { [s: string]: any }\n  )) {\n    const wrapper = HandlerWrapper.fromHandler(handler);\n    if (!wrapper) {\n      throw new TypeError(`${route} is not a restate handler.`);\n    }\n    wrapper.bindInstance(router);\n    component.add(route, wrapper);\n  }\n  return component;\n}\n\nfunction buildWorkflowComponent(\n  name: string,\n  workflow: any,\n  definition: ServiceAuxInfo,\n  defaultServiceOptions: DefaultServiceOptions\n): WorkflowComponent {\n  if (name.indexOf(\"/\") !== -1) {\n    throw new Error(\"service name must not contain any slash '/'\");\n  }\n  const component = new WorkflowComponent(\n    name,\n    definition.description,\n    definition.metadata,\n    {\n      ...defaultServiceOptions,\n      ...(definition?.options as WorkflowOptions | undefined),\n    }\n  );\n\n  for (const [route, handler] of Object.entries(\n    workflow as { [s: string]: any }\n  )) {\n    const wrapper = HandlerWrapper.fromHandler(handler);\n    if (!wrapper) {\n      throw new TypeError(`${route} is not a restate handler.`);\n    }\n    wrapper.bindInstance(workflow);\n    component.add(route, wrapper);\n  }\n  return component;\n}\n"],"mappings":";;;;;;;AA0CA,SAAS,oBACP,GAC+C;AAC/C,QAAO,KAAK,EAAE,YAAY;;AAG5B,SAAS,mBACP,GACoD;AACpD,QAAO,KAAK,EAAE,WAAW;;AAG3B,SAAS,qBACP,GACiD;AACjD,QAAO,KAAK,EAAE,aAAa;;AAkC7B,IAAa,kBAAb,MAA6B;CAC3B,AAAiB,qCAKb,IAAI,KAAK;CACb,AAAQ,kBAAmC;CAC3C,AAAQ,SAAmB,EAAE;CAC7B,AAAQ,wBAA+C,EAAE;CACzD,AAAQ;CAER,AAAO,KACL,YAIA;AAEA,MAAI,WAAW,KAAK,QAAQ,IAAI,KAAK,GACnC,OAAM,IAAI,MAAM,8CAA8C;AAGhE,OAAK,mBAAmB,IAAI,WAAW,MAAM,WAAW;;CAG1D,AAAO,gBAAgB,GAAG,MAAgB;AACxC,OAAK,OAAO,KAAK,GAAG,KAAK;;CAG3B,AAAO,yBAAyB,SAAgC;AAC9D,OAAK,wBAAwB;;CAG/B,AAAO,UAAU,WAA4B;AAC3C,OAAK,kBAAkB;;CAGzB,AAAO,6BACL,eACA;AACA,OAAK,4BAA4B;;CAGnC,AAAO,QAAkB;EACvB,MAAM,OAAO,aAAa,KAAK,iBAAiB,UAAU,OAAO;EAGjE,MAAM,6BAAa,IAAI,KAAwB;AAC/C,OAAK,MAAM,CAAC,MAAM,eAAe,KAAK,mBACpC,KAAI,oBAAoB,WAAW,EAAE;GACnC,MAAM,EAAE,cAAM,YAAY;AAC1B,OAAI,CAAC,QACH,OAAM,IAAI,UAAU,mCAAmC;AAEzD,cAAW,IACTA,QACA,sBACEA,QACA,SACA,YACA,KAAK,sBACN,CACF;aACQ,mBAAmB,WAAW,EAAE;GACzC,MAAM,EAAE,cAAM,WAAW;AACzB,OAAI,CAAC,OACH,OAAM,IAAI,UAAU,kCAAkC;AAExD,cAAW,IACTA,QACA,4BACEA,QACA,QACA,YACA,KAAK,sBACN,CACF;aACQ,qBAAqB,WAAW,EAAE;GAC3C,MAAM,EAAE,cAAM,aAAa;AAC3B,OAAI,CAAC,SACH,OAAM,IAAI,UAAU,oCAAoC;AAE1D,cAAW,IACTA,QACA,uBACEA,QACA,UACA,YACA,KAAK,sBACN,CACF;QAED,OAAM,IAAI,UACR,eAAe,KAAK,wEACrB;EAKL,MAAM,oBAAoB,iBAAiB,WAAW;AAEtD,SAAO;GACL,QAAQ,KAAK;GACb,iBAAiB,KAAK;GACtB;GACA;GACA;GACA,mBAAmB,KAAK,4BACpB,KAAK,2BAA2B,GAChC;GACL;;;AAIL,SAAS,iBACP,YACoB;AACpB,QAAO;EACL,oBAAoB;EACpB,oBAAoB;EACpB,UAAU,CAAC,GAAG,WAAW,QAAQ,CAAC,CAAC,KAAK,MAAM,EAAE,WAAW,CAAC;EAC7D;;AAGH,SAAS,sBACP,MACA,QACA,YACA,uBACkB;AAClB,KAAI,KAAK,QAAQ,IAAI,KAAK,GACxB,OAAM,IAAI,MAAM,8CAA8C;CAEhE,MAAM,YAAY,IAAI,iBACpB,MACA,WAAW,aACX,WAAW,UACX;EACE,GAAG;EACH,GAAI,YAAY;EACjB,CACF;AAED,MAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QACpC,OACD,EAAE;EACD,MAAM,UAAU,eAAe,YAAY,QAAQ;AACnD,MAAI,CAAC,QACH,OAAM,IAAI,UAAU,GAAG,MAAM,4BAA4B;AAE3D,UAAQ,aAAa,OAAO;AAC5B,YAAU,IAAI,OAAO,QAAQ;;AAG/B,QAAO;;AAGT,SAAS,4BACP,MACA,QACA,YACA,uBACwB;AACxB,KAAI,KAAK,QAAQ,IAAI,KAAK,GACxB,OAAM,IAAI,MAAM,8CAA8C;CAEhE,MAAM,YAAY,IAAI,uBACpB,MACA,WAAW,aACX,WAAW,UACX;EACE,GAAG;EACH,GAAI,YAAY;EACjB,CACF;AAED,MAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QACpC,OACD,EAAE;EACD,MAAM,UAAU,eAAe,YAAY,QAAQ;AACnD,MAAI,CAAC,QACH,OAAM,IAAI,UAAU,GAAG,MAAM,4BAA4B;AAE3D,UAAQ,aAAa,OAAO;AAC5B,YAAU,IAAI,OAAO,QAAQ;;AAE/B,QAAO;;AAGT,SAAS,uBACP,MACA,UACA,YACA,uBACmB;AACnB,KAAI,KAAK,QAAQ,IAAI,KAAK,GACxB,OAAM,IAAI,MAAM,8CAA8C;CAEhE,MAAM,YAAY,IAAI,kBACpB,MACA,WAAW,aACX,WAAW,UACX;EACE,GAAG;EACH,GAAI,YAAY;EACjB,CACF;AAED,MAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QACpC,SACD,EAAE;EACD,MAAM,UAAU,eAAe,YAAY,QAAQ;AACnD,MAAI,CAAC,QACH,OAAM,IAAI,UAAU,GAAG,MAAM,4BAA4B;AAE3D,UAAQ,aAAa,SAAS;AAC9B,YAAU,IAAI,OAAO,QAAQ;;AAE/B,QAAO"}

package/dist/endpoint/handlers/generic.cjs

@@ -1,21 +1,23 @@
 const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-const require_logger_transport = require('../../logging/logger_transport.cjs');
-const require_console_logger_transport = require('../../logging/console_logger_transport.cjs');
-const require_core_logging = require('./core_logging.cjs');
-const require_sdk_shared_core_wasm_bindings = require('./vm/sdk_shared_core_wasm_bindings.cjs');
 const require_errors = require('../../types/errors.cjs');
-const require_rpc = require('../../types/rpc.cjs');
 const require_completable_promise = require('../../utils/completable_promise.cjs');
-const require_context_impl = require('../../context_impl.cjs');
+const require_rpc = require('../../types/rpc.cjs');
 const require_components = require('../components.cjs');
+const require_logger_transport = require('../../logging/logger_transport.cjs');
+const require_console_logger_transport = require('../../logging/console_logger_transport.cjs');
 const require_logger = require('../../logging/logger.cjs');
 const require_user_agent = require('../../user_agent.cjs');
+const require_core_logging = require('./core_logging.cjs');
+const require_sdk_shared_core_wasm_bindings = require('./vm/sdk_shared_core_wasm_bindings.cjs');
+const require_context_impl = require('../../context_impl.cjs');
+const require_error_sanitization = require('../../error_sanitization.cjs');
 const require_utils = require('./utils.cjs');
 const require_discovery = require('./discovery.cjs');
 let __restatedev_restate_sdk_core = require("@restatedev/restate-sdk-core");
 __restatedev_restate_sdk_core = require_rolldown_runtime.__toESM(__restatedev_restate_sdk_core);
 
 //#region src/endpoint/handlers/generic.ts
+const HOOK_CONTEXT_IS_PROCESSING_SYMBOL = Symbol.for("@restatedev/restate-sdk/hooks.isProcessing");
 function createRestateHandler(endpoint, protocolMode, additionalDiscoveryFields) {
 	return new RestateHandlerImpl(endpoint, protocolMode, additionalDiscoveryFields);
 }
@@ -110,7 +112,7 @@ var RestateInvokeResponse = class {
 		this.loggerId = Math.floor(Math.random() * 4294967295);
 		const isJournalCodecDefined = this.journalValueCodecInit !== void 0;
 		const vmHeaders = Object.entries(this.attemptHeaders).filter(([, v]) => v !== void 0).map(([k, v]) => new require_sdk_shared_core_wasm_bindings.WasmHeader(k, v instanceof Array ? v[0] : v));
-		this.coreVm = new require_sdk_shared_core_wasm_bindings.WasmVM(vmHeaders, restateLogLevelToWasmLogLevel(require_console_logger_transport.DEFAULT_CONSOLE_LOGGER_LOG_LEVEL), this.loggerId, isJournalCodecDefined);
+		this.coreVm = new require_sdk_shared_core_wasm_bindings.WasmVM(vmHeaders, restateLogLevelToWasmLogLevel(require_console_logger_transport.DEFAULT_CONSOLE_LOGGER_LOG_LEVEL), this.loggerId, isJournalCodecDefined, handler.executionOptions.explicitCancellation ?? false);
 		const responseHead = this.coreVm.get_response_head();
 		this.statusCode = responseHead.status_code;
 		this.headers = responseHead.headers.reduce((headers, { key, value }) => ({
@@ -122,6 +124,10 @@ var RestateInvokeResponse = class {
 	async process({ inputReader, outputWriter, abortSignal }) {
 		abortSignal.addEventListener("abort", () => {
 			require_core_logging.destroyLogger(this.loggerId);
+			setImmediate(() => {
+				const msg = "Connection closed";
+				this.coreVm.notify_error(msg, msg);
+			});
 		}, { once: true });
 		require_core_logging.registerLogger(this.loggerId, this.vmLogger);
 		const journalValueCodec = this.journalValueCodecInit ? await this.journalValueCodecInit : {
@@ -134,6 +140,14 @@ var RestateInvokeResponse = class {
 			await bufferJournalReplayInCoreVm(this.coreVm, inputReader);
 			const input = this.coreVm.sys_input();
 			const invocationRequest = {
+				target: {
+					service: this.service.name(),
+					handler: this.handler.name(),
+					key: input.key || void 0,
+					toString() {
+						return this.key !== void 0 ? `${this.service}/${this.key}/${this.handler}` : `${this.service}/${this.handler}`;
+					}
+				},
 				id: input.invocation_id,
 				headers: input.headers.reduce((headers, { key, value }) => {
 					headers.set(key, value);
@@ -153,18 +167,20 @@ var RestateInvokeResponse = class {
 			require_core_logging.registerLogger(this.loggerId, this.vmLogger);
 			if (!this.coreVm.is_processing()) this.vmLogger.info("Replaying invocation.");
 			else this.vmLogger.info("Starting invocation.");
-			ctx = new require_context_impl.ContextImpl(this.coreVm, input, ctxLogger, this.handler.kind(), this.vmLogger, invocationRequest, invocationEndPromise, inputReader, outputWriter, journalValueCodec, this.service.options?.serde, this.service.options?.asTerminalError);
+			ctx = new require_context_impl.ContextImpl(this.coreVm, input, ctxLogger, this.handler.kind(), this.vmLogger, invocationRequest, invocationEndPromise, inputReader, outputWriter, journalValueCodec, this.handler.executionOptions);
 		} catch (e) {
 			const error = require_errors.ensureError(e);
 			this.coreVm.notify_error(error.message, error.message);
 			await flushAndClose(this.coreVm, this.vmLogger, inputReader, outputWriter);
 			return;
 		}
-		startUserHandler(ctx, this.service, this.handler, journalValueCodec).finally(() => {
-			invocationEndPromise.resolve();
-		});
-		await invocationEndPromise.promise;
-		await flushAndClose(this.coreVm, this.vmLogger, inputReader, outputWriter);
+		try {
+			await startUserHandler(ctx, this.service, this.handler, journalValueCodec);
+		} catch (e) {
+			notifyError(e, ctx, this.handler.executionOptions.asTerminalError);
+		} finally {
+			await flushAndClose(this.coreVm, this.vmLogger, inputReader, outputWriter);
+		}
 	}
 };
 async function bufferJournalReplayInCoreVm(coreVm, inputReader) {
@@ -178,35 +194,57 @@ async function bufferJournalReplayInCoreVm(coreVm, inputReader) {
 	}
 }
 async function startUserHandler(ctx, service, handler, journalValueCodec) {
+	const hooks = [];
+	for (const provider of handler.executionOptions.hooks ?? []) {
+		const hookContext = { request: ctx.request() };
+		Object.defineProperty(hookContext, HOOK_CONTEXT_IS_PROCESSING_SYMBOL, {
+			value: () => ctx.isProcessing(),
+			enumerable: false
+		});
+		hooks.push(provider(hookContext));
+	}
+	const handlerInterceptor = composeInterceptors(hooks.map((h) => h.interceptor?.handler).filter(isDefined));
+	ctx.setRunInterceptor(composeInterceptors(hooks.map((h) => h.interceptor?.run).filter(isDefined)));
+	let encodedOutput;
+	await raceWithAttemptEnd(ctx, handlerInterceptor)(async () => {
+		const decodedInput = await journalValueCodec.decode(ctx.request().body).catch((e) => Promise.reject(new require_errors.TerminalError(`Failed to decode input using journal value codec: ${require_errors.ensureError(e).message}`, { errorCode: 400 })));
+		const output = await handler.invoke(ctx, decodedInput);
+		encodedOutput = journalValueCodec.encode(output);
+	});
+	ctx.coreVm.sys_write_output_success(encodedOutput);
+	ctx.coreVm.sys_end();
+	ctx.vmLogger.info("Invocation completed successfully.");
+}
+/**
+* Classifies the error and notifies the VM. Called from the process() catch
+* block — the single place that decides how to report errors to the VM.
+*/
+function notifyError(e, ctx, asTerminalError) {
+	if (e instanceof require_context_impl.CommandError) {
+		const cause = require_errors.ensureError(e.cause);
+		require_errors.logError(ctx.vmLogger, cause);
+		if (e.hasCommandIndex) ctx.coreVm.notify_error_for_specific_command(cause.message, cause.stack, e.commandType, e.commandIndex, null);
+		else ctx.coreVm.notify_error_for_next_command(cause.message, cause.stack, e.commandType);
+		return;
+	}
+	const error = require_errors.ensureError(e, asTerminalError);
+	require_errors.logError(ctx.vmLogger, error);
 	try {
-		try {
-			const decodedInput = await journalValueCodec.decode(ctx.request().body).catch((e) => Promise.reject(new require_errors.TerminalError(`Failed to decode input using journal value codec: ${require_errors.ensureError(e).message}`, { errorCode: 400 })));
-			const output = await handler.invoke(ctx, decodedInput);
-			const encodedOutput = journalValueCodec.encode(output);
-			ctx.coreVm.sys_write_output_success(encodedOutput);
+		if (error instanceof require_errors.TerminalError) {
+			ctx.coreVm.sys_write_output_failure({
+				code: error.code,
+				message: error.message,
+				metadata: Object.entries(error.metadata ?? {}).map(([key, value]) => ({
+					key,
+					value
+				}))
+			});
 			ctx.coreVm.sys_end();
-			ctx.vmLogger.info("Invocation completed successfully.");
-		} catch (e) {
-			const error = require_errors.ensureError(e, service.options?.asTerminalError);
-			require_errors.logError(ctx.vmLogger, error);
-			if (error instanceof require_errors.TerminalError) {
-				ctx.coreVm.sys_write_output_failure({
-					code: error.code,
-					message: error.message,
-					metadata: Object.entries(error.metadata ?? {}).map(([key, value]) => ({
-						key,
-						value
-					}))
-				});
-				ctx.coreVm.sys_end();
-				return;
-			}
-			throw error;
-		}
-	} catch (e) {
-		const error = require_errors.ensureError(e);
-		if (error instanceof require_errors.RetryableError) ctx.coreVm.notify_error_with_delay_override(error.message, error.stack, error.retryAfter !== void 0 ? BigInt((0, __restatedev_restate_sdk_core.millisOrDurationToMillis)(error.retryAfter)) : void 0);
+		} else if (error instanceof require_errors.RetryableError) ctx.coreVm.notify_error_with_delay_override(error.message, error.stack, error.retryAfter !== void 0 ? BigInt((0, __restatedev_restate_sdk_core.millisOrDurationToMillis)(error.retryAfter)) : void 0);
 		else ctx.coreVm.notify_error(error.message, error.stack);
+	} catch (vmError) {
+		const inner = require_errors.ensureError(vmError);
+		ctx.coreVm.notify_error(inner.message, inner.stack);
 	}
 }
 async function flushAndClose(coreVm, vmLogger, inputReader, outputWriter) {
@@ -234,6 +272,42 @@ async function flushAndClose(coreVm, vmLogger, inputReader, outputWriter) {
 function isAbortErrorOnWrite(error) {
 	return error.name === "AbortError" || error.message === "Invalid state: WritableStream is closed" || error.code === "ERR_HTTP2_INVALID_STREAM";
 }
+function composeInterceptors(interceptors) {
+	return interceptors.reduceRight((innerInterceptor, interceptor) => (...args) => {
+		const context = args.slice(0, -1);
+		const callback = args.at(-1);
+		return interceptor(...context, () => innerInterceptor(...context, callback));
+	}, (...args) => args.at(-1)());
+}
+/**
+* Wraps an interceptor so that both `next()` and the interceptor body race
+* against `invocationEndPromise`. When the attempt ends (suspension, retryable
+* error, etc.), the promise rejects and the interceptor chain unwinds through
+* catch/finally blocks — preventing interceptors from hanging on a `next()`
+* that will never settle.
+*
+* SDK-internal metadata (CommandError, retryAfter) is stripped before the
+* interceptor sees the error and restored after the chain exits.
+*/
+function raceWithAttemptEnd(ctx, interceptor) {
+	return (...args) => {
+		let originalError;
+		const signal = ctx.invocationEndPromise.promise.catch((e) => {
+			originalError = e;
+			throw require_error_sanitization.sanitizeError(e);
+		});
+		const originalNext = args.at(-1);
+		const racingNext = () => Promise.race([originalNext(), signal]);
+		const newArgs = [...args.slice(0, -1), racingNext];
+		return Promise.race([interceptor(...newArgs), signal]).catch((e) => {
+			if (originalError !== void 0) throw require_error_sanitization.restoreError(e, originalError);
+			throw e;
+		});
+	};
+}
+function isDefined(value) {
+	return value != null;
+}
 function restateLogLevelToWasmLogLevel(level) {
 	switch (level) {
 		case require_logger_transport.RestateLogLevel.TRACE: return require_sdk_shared_core_wasm_bindings.LogLevel.TRACE;

package/dist/endpoint/handlers/generic.js

@@ -1,19 +1,21 @@
-import { LogSource, LoggerContext, RestateLogLevel } from "../../logging/logger_transport.js";
-import { DEFAULT_CONSOLE_LOGGER_LOG_LEVEL } from "../../logging/console_logger_transport.js";
-import { destroyLogger, registerLogger } from "./core_logging.js";
-import { LogLevel, WasmHeader, WasmIdentityVerifier, WasmVM, set_log_level } from "./vm/sdk_shared_core_wasm_bindings.js";
 import { RestateError, RetryableError, TerminalError, ensureError, logError } from "../../types/errors.js";
-import { HandlerKind } from "../../types/rpc.js";
 import { CompletablePromise } from "../../utils/completable_promise.js";
-import { ContextImpl } from "../../context_impl.js";
+import { HandlerKind } from "../../types/rpc.js";
 import { parseUrlComponents } from "../components.js";
+import { LogSource, LoggerContext, RestateLogLevel } from "../../logging/logger_transport.js";
+import { DEFAULT_CONSOLE_LOGGER_LOG_LEVEL } from "../../logging/console_logger_transport.js";
 import { createLogger } from "../../logging/logger.js";
 import { X_RESTATE_SERVER } from "../../user_agent.js";
+import { destroyLogger, registerLogger } from "./core_logging.js";
+import { LogLevel, WasmHeader, WasmIdentityVerifier, WasmVM, set_log_level } from "./vm/sdk_shared_core_wasm_bindings.js";
+import { CommandError, ContextImpl } from "../../context_impl.js";
+import { restoreError, sanitizeError } from "../../error_sanitization.js";
 import { errorResponse, invocationIdFromHeaders, simpleResponse, tryCreateContextualLogger } from "./utils.js";
 import { handleDiscovery } from "./discovery.js";
 import { millisOrDurationToMillis } from "@restatedev/restate-sdk-core";
 
 //#region src/endpoint/handlers/generic.ts
+const HOOK_CONTEXT_IS_PROCESSING_SYMBOL = Symbol.for("@restatedev/restate-sdk/hooks.isProcessing");
 function createRestateHandler(endpoint, protocolMode, additionalDiscoveryFields) {
 	return new RestateHandlerImpl(endpoint, protocolMode, additionalDiscoveryFields);
 }
@@ -108,7 +110,7 @@ var RestateInvokeResponse = class {
 		this.loggerId = Math.floor(Math.random() * 4294967295);
 		const isJournalCodecDefined = this.journalValueCodecInit !== void 0;
 		const vmHeaders = Object.entries(this.attemptHeaders).filter(([, v]) => v !== void 0).map(([k, v]) => new WasmHeader(k, v instanceof Array ? v[0] : v));
-		this.coreVm = new WasmVM(vmHeaders, restateLogLevelToWasmLogLevel(DEFAULT_CONSOLE_LOGGER_LOG_LEVEL), this.loggerId, isJournalCodecDefined);
+		this.coreVm = new WasmVM(vmHeaders, restateLogLevelToWasmLogLevel(DEFAULT_CONSOLE_LOGGER_LOG_LEVEL), this.loggerId, isJournalCodecDefined, handler.executionOptions.explicitCancellation ?? false);
 		const responseHead = this.coreVm.get_response_head();
 		this.statusCode = responseHead.status_code;
 		this.headers = responseHead.headers.reduce((headers, { key, value }) => ({
@@ -120,6 +122,10 @@ var RestateInvokeResponse = class {
 	async process({ inputReader, outputWriter, abortSignal }) {
 		abortSignal.addEventListener("abort", () => {
 			destroyLogger(this.loggerId);
+			setImmediate(() => {
+				const msg = "Connection closed";
+				this.coreVm.notify_error(msg, msg);
+			});
 		}, { once: true });
 		registerLogger(this.loggerId, this.vmLogger);
 		const journalValueCodec = this.journalValueCodecInit ? await this.journalValueCodecInit : {
@@ -132,6 +138,14 @@ var RestateInvokeResponse = class {
 			await bufferJournalReplayInCoreVm(this.coreVm, inputReader);
 			const input = this.coreVm.sys_input();
 			const invocationRequest = {
+				target: {
+					service: this.service.name(),
+					handler: this.handler.name(),
+					key: input.key || void 0,
+					toString() {
+						return this.key !== void 0 ? `${this.service}/${this.key}/${this.handler}` : `${this.service}/${this.handler}`;
+					}
+				},
 				id: input.invocation_id,
 				headers: input.headers.reduce((headers, { key, value }) => {
 					headers.set(key, value);
@@ -151,18 +165,20 @@ var RestateInvokeResponse = class {
 			registerLogger(this.loggerId, this.vmLogger);
 			if (!this.coreVm.is_processing()) this.vmLogger.info("Replaying invocation.");
 			else this.vmLogger.info("Starting invocation.");
-			ctx = new ContextImpl(this.coreVm, input, ctxLogger, this.handler.kind(), this.vmLogger, invocationRequest, invocationEndPromise, inputReader, outputWriter, journalValueCodec, this.service.options?.serde, this.service.options?.asTerminalError);
+			ctx = new ContextImpl(this.coreVm, input, ctxLogger, this.handler.kind(), this.vmLogger, invocationRequest, invocationEndPromise, inputReader, outputWriter, journalValueCodec, this.handler.executionOptions);
 		} catch (e) {
 			const error = ensureError(e);
 			this.coreVm.notify_error(error.message, error.message);
 			await flushAndClose(this.coreVm, this.vmLogger, inputReader, outputWriter);
 			return;
 		}
-		startUserHandler(ctx, this.service, this.handler, journalValueCodec).finally(() => {
-			invocationEndPromise.resolve();
-		});
-		await invocationEndPromise.promise;
-		await flushAndClose(this.coreVm, this.vmLogger, inputReader, outputWriter);
+		try {
+			await startUserHandler(ctx, this.service, this.handler, journalValueCodec);
+		} catch (e) {
+			notifyError(e, ctx, this.handler.executionOptions.asTerminalError);
+		} finally {
+			await flushAndClose(this.coreVm, this.vmLogger, inputReader, outputWriter);
+		}
 	}
 };
 async function bufferJournalReplayInCoreVm(coreVm, inputReader) {
@@ -176,35 +192,57 @@ async function bufferJournalReplayInCoreVm(coreVm, inputReader) {
 	}
 }
 async function startUserHandler(ctx, service, handler, journalValueCodec) {
+	const hooks = [];
+	for (const provider of handler.executionOptions.hooks ?? []) {
+		const hookContext = { request: ctx.request() };
+		Object.defineProperty(hookContext, HOOK_CONTEXT_IS_PROCESSING_SYMBOL, {
+			value: () => ctx.isProcessing(),
+			enumerable: false
+		});
+		hooks.push(provider(hookContext));
+	}
+	const handlerInterceptor = composeInterceptors(hooks.map((h) => h.interceptor?.handler).filter(isDefined));
+	ctx.setRunInterceptor(composeInterceptors(hooks.map((h) => h.interceptor?.run).filter(isDefined)));
+	let encodedOutput;
+	await raceWithAttemptEnd(ctx, handlerInterceptor)(async () => {
+		const decodedInput = await journalValueCodec.decode(ctx.request().body).catch((e) => Promise.reject(new TerminalError(`Failed to decode input using journal value codec: ${ensureError(e).message}`, { errorCode: 400 })));
+		const output = await handler.invoke(ctx, decodedInput);
+		encodedOutput = journalValueCodec.encode(output);
+	});
+	ctx.coreVm.sys_write_output_success(encodedOutput);
+	ctx.coreVm.sys_end();
+	ctx.vmLogger.info("Invocation completed successfully.");
+}
+/**
+* Classifies the error and notifies the VM. Called from the process() catch
+* block — the single place that decides how to report errors to the VM.
+*/
+function notifyError(e, ctx, asTerminalError) {
+	if (e instanceof CommandError) {
+		const cause = ensureError(e.cause);
+		logError(ctx.vmLogger, cause);
+		if (e.hasCommandIndex) ctx.coreVm.notify_error_for_specific_command(cause.message, cause.stack, e.commandType, e.commandIndex, null);
+		else ctx.coreVm.notify_error_for_next_command(cause.message, cause.stack, e.commandType);
+		return;
+	}
+	const error = ensureError(e, asTerminalError);
+	logError(ctx.vmLogger, error);
 	try {
-		try {
-			const decodedInput = await journalValueCodec.decode(ctx.request().body).catch((e) => Promise.reject(new TerminalError(`Failed to decode input using journal value codec: ${ensureError(e).message}`, { errorCode: 400 })));
-			const output = await handler.invoke(ctx, decodedInput);
-			const encodedOutput = journalValueCodec.encode(output);
-			ctx.coreVm.sys_write_output_success(encodedOutput);
+		if (error instanceof TerminalError) {
+			ctx.coreVm.sys_write_output_failure({
+				code: error.code,
+				message: error.message,
+				metadata: Object.entries(error.metadata ?? {}).map(([key, value]) => ({
+					key,
+					value
+				}))
+			});
 			ctx.coreVm.sys_end();
-			ctx.vmLogger.info("Invocation completed successfully.");
-		} catch (e) {
-			const error = ensureError(e, service.options?.asTerminalError);
-			logError(ctx.vmLogger, error);
-			if (error instanceof TerminalError) {
-				ctx.coreVm.sys_write_output_failure({
-					code: error.code,
-					message: error.message,
-					metadata: Object.entries(error.metadata ?? {}).map(([key, value]) => ({
-						key,
-						value
-					}))
-				});
-				ctx.coreVm.sys_end();
-				return;
-			}
-			throw error;
-		}
-	} catch (e) {
-		const error = ensureError(e);
-		if (error instanceof RetryableError) ctx.coreVm.notify_error_with_delay_override(error.message, error.stack, error.retryAfter !== void 0 ? BigInt(millisOrDurationToMillis(error.retryAfter)) : void 0);
+		} else if (error instanceof RetryableError) ctx.coreVm.notify_error_with_delay_override(error.message, error.stack, error.retryAfter !== void 0 ? BigInt(millisOrDurationToMillis(error.retryAfter)) : void 0);
 		else ctx.coreVm.notify_error(error.message, error.stack);
+	} catch (vmError) {
+		const inner = ensureError(vmError);
+		ctx.coreVm.notify_error(inner.message, inner.stack);
 	}
 }
 async function flushAndClose(coreVm, vmLogger, inputReader, outputWriter) {
@@ -232,6 +270,42 @@ async function flushAndClose(coreVm, vmLogger, inputReader, outputWriter) {
 function isAbortErrorOnWrite(error) {
 	return error.name === "AbortError" || error.message === "Invalid state: WritableStream is closed" || error.code === "ERR_HTTP2_INVALID_STREAM";
 }
+function composeInterceptors(interceptors) {
+	return interceptors.reduceRight((innerInterceptor, interceptor) => (...args) => {
+		const context = args.slice(0, -1);
+		const callback = args.at(-1);
+		return interceptor(...context, () => innerInterceptor(...context, callback));
+	}, (...args) => args.at(-1)());
+}
+/**
+* Wraps an interceptor so that both `next()` and the interceptor body race
+* against `invocationEndPromise`. When the attempt ends (suspension, retryable
+* error, etc.), the promise rejects and the interceptor chain unwinds through
+* catch/finally blocks — preventing interceptors from hanging on a `next()`
+* that will never settle.
+*
+* SDK-internal metadata (CommandError, retryAfter) is stripped before the
+* interceptor sees the error and restored after the chain exits.
+*/
+function raceWithAttemptEnd(ctx, interceptor) {
+	return (...args) => {
+		let originalError;
+		const signal = ctx.invocationEndPromise.promise.catch((e) => {
+			originalError = e;
+			throw sanitizeError(e);
+		});
+		const originalNext = args.at(-1);
+		const racingNext = () => Promise.race([originalNext(), signal]);
+		const newArgs = [...args.slice(0, -1), racingNext];
+		return Promise.race([interceptor(...newArgs), signal]).catch((e) => {
+			if (originalError !== void 0) throw restoreError(e, originalError);
+			throw e;
+		});
+	};
+}
+function isDefined(value) {
+	return value != null;
+}
 function restateLogLevelToWasmLogLevel(level) {
 	switch (level) {
 		case RestateLogLevel.TRACE: return LogLevel.TRACE;