@restatedev/restate-sdk 1.9.0 → 1.10.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (419) hide show
  1. package/LICENSE +21 -0
  2. package/dist/_virtual/rolldown_runtime.cjs +25 -0
  3. package/dist/common_api.cjs +25 -0
  4. package/dist/common_api.d.cts +30 -0
  5. package/dist/common_api.d.cts.map +1 -0
  6. package/dist/common_api.d.ts +30 -0
  7. package/dist/common_api.d.ts.map +1 -0
  8. package/dist/common_api.js +19 -0
  9. package/dist/common_api.js.map +1 -0
  10. package/dist/context.cjs +29 -0
  11. package/dist/context.d.cts +699 -0
  12. package/dist/context.d.cts.map +1 -0
  13. package/dist/context.d.ts +699 -0
  14. package/dist/context.d.ts.map +1 -0
  15. package/dist/context.js +29 -0
  16. package/dist/context.js.map +1 -0
  17. package/dist/context_impl.cjs +441 -0
  18. package/dist/{cjs/src/context_impl.d.ts → context_impl.d.ts} +2 -1
  19. package/dist/context_impl.d.ts.map +1 -0
  20. package/dist/context_impl.js +440 -0
  21. package/dist/context_impl.js.map +1 -0
  22. package/dist/endpoint/components.cjs +257 -0
  23. package/dist/endpoint/components.d.ts.map +1 -0
  24. package/dist/endpoint/components.js +253 -0
  25. package/dist/endpoint/components.js.map +1 -0
  26. package/dist/endpoint/discovery.d.ts.map +1 -0
  27. package/dist/{esm/src/endpoint → endpoint}/discovery.js +0 -1
  28. package/dist/endpoint/discovery.js.map +1 -0
  29. package/dist/endpoint/endpoint.cjs +117 -0
  30. package/dist/endpoint/endpoint.d.ts.map +1 -0
  31. package/dist/endpoint/endpoint.js +118 -0
  32. package/dist/endpoint/endpoint.js.map +1 -0
  33. package/dist/endpoint/fetch_endpoint.cjs +41 -0
  34. package/dist/endpoint/fetch_endpoint.d.cts +39 -0
  35. package/dist/endpoint/fetch_endpoint.d.cts.map +1 -0
  36. package/dist/endpoint/fetch_endpoint.d.ts +39 -0
  37. package/dist/endpoint/fetch_endpoint.d.ts.map +1 -0
  38. package/dist/endpoint/fetch_endpoint.js +42 -0
  39. package/dist/endpoint/fetch_endpoint.js.map +1 -0
  40. package/dist/endpoint/handlers/fetch.cjs +21 -0
  41. package/dist/endpoint/handlers/fetch.d.ts.map +1 -0
  42. package/dist/endpoint/handlers/fetch.js +21 -0
  43. package/dist/endpoint/handlers/fetch.js.map +1 -0
  44. package/dist/endpoint/handlers/generic.cjs +328 -0
  45. package/dist/{esm/src/endpoint → endpoint}/handlers/generic.d.ts +6 -1
  46. package/dist/endpoint/handlers/generic.d.ts.map +1 -0
  47. package/dist/endpoint/handlers/generic.js +324 -0
  48. package/dist/endpoint/handlers/generic.js.map +1 -0
  49. package/dist/endpoint/handlers/lambda.cjs +93 -0
  50. package/dist/endpoint/handlers/lambda.d.ts.map +1 -0
  51. package/dist/endpoint/handlers/lambda.js +89 -0
  52. package/dist/endpoint/handlers/lambda.js.map +1 -0
  53. package/dist/{cjs/src/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.js → endpoint/handlers/vm/sdk_shared_core_wasm_bindings.cjs} +1431 -1416
  54. package/dist/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.d.ts.map +1 -0
  55. package/dist/{esm/src/endpoint → endpoint}/handlers/vm/sdk_shared_core_wasm_bindings.js +1427 -1405
  56. package/dist/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.js.map +1 -0
  57. package/dist/endpoint/lambda_endpoint.cjs +36 -0
  58. package/dist/endpoint/lambda_endpoint.d.cts +26 -0
  59. package/dist/endpoint/lambda_endpoint.d.cts.map +1 -0
  60. package/dist/endpoint/lambda_endpoint.d.ts +26 -0
  61. package/dist/endpoint/lambda_endpoint.d.ts.map +1 -0
  62. package/dist/endpoint/lambda_endpoint.js +37 -0
  63. package/dist/endpoint/lambda_endpoint.js.map +1 -0
  64. package/dist/endpoint/node_endpoint.cjs +101 -0
  65. package/dist/{esm/src/endpoint → endpoint}/node_endpoint.d.ts +1 -1
  66. package/dist/endpoint/node_endpoint.d.ts.map +1 -0
  67. package/dist/endpoint/node_endpoint.js +99 -0
  68. package/dist/endpoint/node_endpoint.js.map +1 -0
  69. package/dist/endpoint/types.d.cts +61 -0
  70. package/dist/endpoint/types.d.cts.map +1 -0
  71. package/dist/endpoint/types.d.ts +61 -0
  72. package/dist/endpoint/types.d.ts.map +1 -0
  73. package/dist/{cjs/src/endpoint → endpoint}/types.js.map +1 -1
  74. package/dist/endpoint/withOptions.cjs +15 -0
  75. package/dist/endpoint/withOptions.d.ts.map +1 -0
  76. package/dist/endpoint/withOptions.js +15 -0
  77. package/dist/endpoint/withOptions.js.map +1 -0
  78. package/dist/endpoint.d.cts +112 -0
  79. package/dist/endpoint.d.cts.map +1 -0
  80. package/dist/endpoint.d.ts +112 -0
  81. package/dist/endpoint.d.ts.map +1 -0
  82. package/dist/endpoint.js.map +1 -0
  83. package/dist/fetch.cjs +83 -0
  84. package/dist/fetch.d.cts +52 -0
  85. package/dist/fetch.d.cts.map +1 -0
  86. package/dist/fetch.d.ts +52 -0
  87. package/dist/fetch.d.ts.map +1 -0
  88. package/dist/fetch.js +44 -0
  89. package/dist/fetch.js.map +1 -0
  90. package/dist/index.cjs +48 -0
  91. package/dist/index.d.cts +9 -0
  92. package/dist/index.d.ts +9 -0
  93. package/dist/index.d.ts.map +1 -0
  94. package/dist/index.js +7 -0
  95. package/dist/index.js.map +1 -0
  96. package/dist/io.cjs +48 -0
  97. package/dist/io.d.ts.map +1 -0
  98. package/dist/io.js +48 -0
  99. package/dist/io.js.map +1 -0
  100. package/dist/lambda.cjs +74 -0
  101. package/dist/lambda.d.cts +32 -0
  102. package/dist/lambda.d.cts.map +1 -0
  103. package/dist/lambda.d.ts +32 -0
  104. package/dist/lambda.d.ts.map +1 -0
  105. package/dist/lambda.js +35 -0
  106. package/dist/lambda.js.map +1 -0
  107. package/dist/logging/console_logger_transport.cjs +54 -0
  108. package/dist/logging/console_logger_transport.d.ts.map +1 -0
  109. package/dist/logging/console_logger_transport.js +54 -0
  110. package/dist/logging/console_logger_transport.js.map +1 -0
  111. package/dist/logging/logger.cjs +43 -0
  112. package/dist/logging/logger.d.ts.map +1 -0
  113. package/dist/logging/logger.js +44 -0
  114. package/dist/logging/logger.js.map +1 -0
  115. package/dist/logging/logger_transport.cjs +42 -0
  116. package/dist/logging/logger_transport.d.cts +57 -0
  117. package/dist/logging/logger_transport.d.cts.map +1 -0
  118. package/dist/logging/logger_transport.d.ts +57 -0
  119. package/dist/logging/logger_transport.d.ts.map +1 -0
  120. package/dist/logging/logger_transport.js +40 -0
  121. package/dist/logging/logger_transport.js.map +1 -0
  122. package/dist/node.cjs +90 -0
  123. package/dist/node.d.cts +51 -0
  124. package/dist/node.d.cts.map +1 -0
  125. package/dist/node.d.ts +51 -0
  126. package/dist/node.d.ts.map +1 -0
  127. package/dist/node.js +50 -0
  128. package/dist/node.js.map +1 -0
  129. package/dist/package.cjs +11 -0
  130. package/dist/package.js +6 -0
  131. package/dist/package.js.map +1 -0
  132. package/dist/promises.cjs +234 -0
  133. package/dist/{esm/src/promises.d.ts → promises.d.ts} +6 -6
  134. package/dist/promises.d.ts.map +1 -0
  135. package/dist/promises.js +226 -0
  136. package/dist/promises.js.map +1 -0
  137. package/dist/types/errors.cjs +110 -0
  138. package/dist/types/errors.d.cts +71 -0
  139. package/dist/types/errors.d.cts.map +1 -0
  140. package/dist/types/errors.d.ts +71 -0
  141. package/dist/types/errors.d.ts.map +1 -0
  142. package/dist/types/errors.js +102 -0
  143. package/dist/types/errors.js.map +1 -0
  144. package/dist/types/rpc.cjs +425 -0
  145. package/dist/types/rpc.d.cts +580 -0
  146. package/dist/types/rpc.d.cts.map +1 -0
  147. package/dist/types/rpc.d.ts +580 -0
  148. package/dist/types/rpc.d.ts.map +1 -0
  149. package/dist/types/rpc.js +406 -0
  150. package/dist/types/rpc.js.map +1 -0
  151. package/dist/user_agent.cjs +7 -0
  152. package/dist/user_agent.d.ts +2 -0
  153. package/dist/user_agent.d.ts.map +1 -0
  154. package/dist/user_agent.js +8 -0
  155. package/dist/user_agent.js.map +1 -0
  156. package/dist/utils/completable_promise.cjs +22 -0
  157. package/dist/utils/completable_promise.d.ts.map +1 -0
  158. package/dist/utils/completable_promise.js +22 -0
  159. package/dist/utils/completable_promise.js.map +1 -0
  160. package/dist/utils/rand.cjs +72 -0
  161. package/dist/utils/rand.d.ts.map +1 -0
  162. package/dist/utils/rand.js +71 -0
  163. package/dist/utils/rand.js.map +1 -0
  164. package/dist/utils/streams.cjs +14 -0
  165. package/dist/utils/streams.d.ts.map +1 -0
  166. package/dist/utils/streams.js +13 -0
  167. package/dist/utils/streams.js.map +1 -0
  168. package/package.json +49 -80
  169. package/dist/cjs/package.json +0 -1
  170. package/dist/cjs/src/common_api.d.ts +0 -37
  171. package/dist/cjs/src/common_api.d.ts.map +0 -1
  172. package/dist/cjs/src/common_api.js +0 -49
  173. package/dist/cjs/src/common_api.js.map +0 -1
  174. package/dist/cjs/src/context.d.ts +0 -695
  175. package/dist/cjs/src/context.d.ts.map +0 -1
  176. package/dist/cjs/src/context.js +0 -90
  177. package/dist/cjs/src/context.js.map +0 -1
  178. package/dist/cjs/src/context_impl.d.ts.map +0 -1
  179. package/dist/cjs/src/context_impl.js +0 -597
  180. package/dist/cjs/src/context_impl.js.map +0 -1
  181. package/dist/cjs/src/endpoint/components.d.ts.map +0 -1
  182. package/dist/cjs/src/endpoint/components.js +0 -353
  183. package/dist/cjs/src/endpoint/components.js.map +0 -1
  184. package/dist/cjs/src/endpoint/discovery.d.ts.map +0 -1
  185. package/dist/cjs/src/endpoint/discovery.js +0 -9
  186. package/dist/cjs/src/endpoint/discovery.js.map +0 -1
  187. package/dist/cjs/src/endpoint/endpoint.d.ts.map +0 -1
  188. package/dist/cjs/src/endpoint/endpoint.js +0 -159
  189. package/dist/cjs/src/endpoint/endpoint.js.map +0 -1
  190. package/dist/cjs/src/endpoint/fetch_endpoint.d.ts +0 -49
  191. package/dist/cjs/src/endpoint/fetch_endpoint.d.ts.map +0 -1
  192. package/dist/cjs/src/endpoint/fetch_endpoint.js +0 -53
  193. package/dist/cjs/src/endpoint/fetch_endpoint.js.map +0 -1
  194. package/dist/cjs/src/endpoint/handlers/fetch.d.ts.map +0 -1
  195. package/dist/cjs/src/endpoint/handlers/fetch.js +0 -34
  196. package/dist/cjs/src/endpoint/handlers/fetch.js.map +0 -1
  197. package/dist/cjs/src/endpoint/handlers/generic.d.ts +0 -56
  198. package/dist/cjs/src/endpoint/handlers/generic.d.ts.map +0 -1
  199. package/dist/cjs/src/endpoint/handlers/generic.js +0 -485
  200. package/dist/cjs/src/endpoint/handlers/generic.js.map +0 -1
  201. package/dist/cjs/src/endpoint/handlers/lambda.d.ts.map +0 -1
  202. package/dist/cjs/src/endpoint/handlers/lambda.js +0 -180
  203. package/dist/cjs/src/endpoint/handlers/lambda.js.map +0 -1
  204. package/dist/cjs/src/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.d.ts.map +0 -1
  205. package/dist/cjs/src/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.js.map +0 -1
  206. package/dist/cjs/src/endpoint/lambda_endpoint.d.ts +0 -30
  207. package/dist/cjs/src/endpoint/lambda_endpoint.d.ts.map +0 -1
  208. package/dist/cjs/src/endpoint/lambda_endpoint.js +0 -52
  209. package/dist/cjs/src/endpoint/lambda_endpoint.js.map +0 -1
  210. package/dist/cjs/src/endpoint/node_endpoint.d.ts +0 -16
  211. package/dist/cjs/src/endpoint/node_endpoint.d.ts.map +0 -1
  212. package/dist/cjs/src/endpoint/node_endpoint.js +0 -148
  213. package/dist/cjs/src/endpoint/node_endpoint.js.map +0 -1
  214. package/dist/cjs/src/endpoint/types.d.ts +0 -55
  215. package/dist/cjs/src/endpoint/types.d.ts.map +0 -1
  216. package/dist/cjs/src/endpoint/types.js +0 -3
  217. package/dist/cjs/src/endpoint/withOptions.d.ts.map +0 -1
  218. package/dist/cjs/src/endpoint/withOptions.js +0 -22
  219. package/dist/cjs/src/endpoint/withOptions.js.map +0 -1
  220. package/dist/cjs/src/endpoint.d.ts +0 -108
  221. package/dist/cjs/src/endpoint.d.ts.map +0 -1
  222. package/dist/cjs/src/endpoint.js +0 -13
  223. package/dist/cjs/src/endpoint.js.map +0 -1
  224. package/dist/cjs/src/fetch.d.ts +0 -43
  225. package/dist/cjs/src/fetch.d.ts.map +0 -1
  226. package/dist/cjs/src/fetch.js +0 -64
  227. package/dist/cjs/src/fetch.js.map +0 -1
  228. package/dist/cjs/src/generated/version.d.ts +0 -2
  229. package/dist/cjs/src/generated/version.d.ts.map +0 -1
  230. package/dist/cjs/src/generated/version.js +0 -5
  231. package/dist/cjs/src/generated/version.js.map +0 -1
  232. package/dist/cjs/src/io.d.ts.map +0 -1
  233. package/dist/cjs/src/io.js +0 -78
  234. package/dist/cjs/src/io.js.map +0 -1
  235. package/dist/cjs/src/lambda.d.ts +0 -23
  236. package/dist/cjs/src/lambda.d.ts.map +0 -1
  237. package/dist/cjs/src/lambda.js +0 -55
  238. package/dist/cjs/src/lambda.js.map +0 -1
  239. package/dist/cjs/src/logging/console_logger_transport.d.ts.map +0 -1
  240. package/dist/cjs/src/logging/console_logger_transport.js +0 -98
  241. package/dist/cjs/src/logging/console_logger_transport.js.map +0 -1
  242. package/dist/cjs/src/logging/logger.d.ts.map +0 -1
  243. package/dist/cjs/src/logging/logger.js +0 -60
  244. package/dist/cjs/src/logging/logger.js.map +0 -1
  245. package/dist/cjs/src/logging/logger_transport.d.ts +0 -52
  246. package/dist/cjs/src/logging/logger_transport.d.ts.map +0 -1
  247. package/dist/cjs/src/logging/logger_transport.js +0 -59
  248. package/dist/cjs/src/logging/logger_transport.js.map +0 -1
  249. package/dist/cjs/src/node.d.ts +0 -39
  250. package/dist/cjs/src/node.d.ts.map +0 -1
  251. package/dist/cjs/src/node.js +0 -71
  252. package/dist/cjs/src/node.js.map +0 -1
  253. package/dist/cjs/src/promises.d.ts +0 -111
  254. package/dist/cjs/src/promises.d.ts.map +0 -1
  255. package/dist/cjs/src/promises.js +0 -326
  256. package/dist/cjs/src/promises.js.map +0 -1
  257. package/dist/cjs/src/public_api.d.ts +0 -2
  258. package/dist/cjs/src/public_api.d.ts.map +0 -1
  259. package/dist/cjs/src/public_api.js +0 -28
  260. package/dist/cjs/src/public_api.js.map +0 -1
  261. package/dist/cjs/src/types/errors.d.ts +0 -74
  262. package/dist/cjs/src/types/errors.d.ts.map +0 -1
  263. package/dist/cjs/src/types/errors.js +0 -138
  264. package/dist/cjs/src/types/errors.js.map +0 -1
  265. package/dist/cjs/src/types/rpc.d.ts +0 -620
  266. package/dist/cjs/src/types/rpc.d.ts.map +0 -1
  267. package/dist/cjs/src/types/rpc.js +0 -589
  268. package/dist/cjs/src/types/rpc.js.map +0 -1
  269. package/dist/cjs/src/user_agent.d.ts +0 -2
  270. package/dist/cjs/src/user_agent.d.ts.map +0 -1
  271. package/dist/cjs/src/user_agent.js +0 -16
  272. package/dist/cjs/src/user_agent.js.map +0 -1
  273. package/dist/cjs/src/utils/completable_promise.d.ts.map +0 -1
  274. package/dist/cjs/src/utils/completable_promise.js +0 -34
  275. package/dist/cjs/src/utils/completable_promise.js.map +0 -1
  276. package/dist/cjs/src/utils/rand.d.ts.map +0 -1
  277. package/dist/cjs/src/utils/rand.js +0 -113
  278. package/dist/cjs/src/utils/rand.js.map +0 -1
  279. package/dist/cjs/src/utils/streams.d.ts.map +0 -1
  280. package/dist/cjs/src/utils/streams.js +0 -23
  281. package/dist/cjs/src/utils/streams.js.map +0 -1
  282. package/dist/cjs/tsconfig.tsbuildinfo +0 -1
  283. package/dist/esm/src/common_api.d.ts +0 -37
  284. package/dist/esm/src/common_api.d.ts.map +0 -1
  285. package/dist/esm/src/common_api.js +0 -32
  286. package/dist/esm/src/common_api.js.map +0 -1
  287. package/dist/esm/src/context.d.ts +0 -695
  288. package/dist/esm/src/context.d.ts.map +0 -1
  289. package/dist/esm/src/context.js +0 -87
  290. package/dist/esm/src/context.js.map +0 -1
  291. package/dist/esm/src/context_impl.d.ts +0 -69
  292. package/dist/esm/src/context_impl.d.ts.map +0 -1
  293. package/dist/esm/src/context_impl.js +0 -592
  294. package/dist/esm/src/context_impl.js.map +0 -1
  295. package/dist/esm/src/endpoint/components.d.ts +0 -97
  296. package/dist/esm/src/endpoint/components.d.ts.map +0 -1
  297. package/dist/esm/src/endpoint/components.js +0 -343
  298. package/dist/esm/src/endpoint/components.js.map +0 -1
  299. package/dist/esm/src/endpoint/discovery.d.ts +0 -184
  300. package/dist/esm/src/endpoint/discovery.d.ts.map +0 -1
  301. package/dist/esm/src/endpoint/discovery.js.map +0 -1
  302. package/dist/esm/src/endpoint/endpoint.d.ts +0 -39
  303. package/dist/esm/src/endpoint/endpoint.d.ts.map +0 -1
  304. package/dist/esm/src/endpoint/endpoint.js +0 -155
  305. package/dist/esm/src/endpoint/endpoint.js.map +0 -1
  306. package/dist/esm/src/endpoint/fetch_endpoint.d.ts +0 -49
  307. package/dist/esm/src/endpoint/fetch_endpoint.d.ts.map +0 -1
  308. package/dist/esm/src/endpoint/fetch_endpoint.js +0 -49
  309. package/dist/esm/src/endpoint/fetch_endpoint.js.map +0 -1
  310. package/dist/esm/src/endpoint/handlers/fetch.d.ts +0 -5
  311. package/dist/esm/src/endpoint/handlers/fetch.d.ts.map +0 -1
  312. package/dist/esm/src/endpoint/handlers/fetch.js +0 -31
  313. package/dist/esm/src/endpoint/handlers/fetch.js.map +0 -1
  314. package/dist/esm/src/endpoint/handlers/generic.d.ts.map +0 -1
  315. package/dist/esm/src/endpoint/handlers/generic.js +0 -447
  316. package/dist/esm/src/endpoint/handlers/generic.js.map +0 -1
  317. package/dist/esm/src/endpoint/handlers/lambda.d.ts +0 -10
  318. package/dist/esm/src/endpoint/handlers/lambda.d.ts.map +0 -1
  319. package/dist/esm/src/endpoint/handlers/lambda.js +0 -142
  320. package/dist/esm/src/endpoint/handlers/lambda.js.map +0 -1
  321. package/dist/esm/src/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.d.ts +0 -367
  322. package/dist/esm/src/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.d.ts.map +0 -1
  323. package/dist/esm/src/endpoint/handlers/vm/sdk_shared_core_wasm_bindings.js.map +0 -1
  324. package/dist/esm/src/endpoint/lambda_endpoint.d.ts +0 -30
  325. package/dist/esm/src/endpoint/lambda_endpoint.d.ts.map +0 -1
  326. package/dist/esm/src/endpoint/lambda_endpoint.js +0 -48
  327. package/dist/esm/src/endpoint/lambda_endpoint.js.map +0 -1
  328. package/dist/esm/src/endpoint/node_endpoint.d.ts.map +0 -1
  329. package/dist/esm/src/endpoint/node_endpoint.js +0 -111
  330. package/dist/esm/src/endpoint/node_endpoint.js.map +0 -1
  331. package/dist/esm/src/endpoint/types.d.ts +0 -55
  332. package/dist/esm/src/endpoint/types.d.ts.map +0 -1
  333. package/dist/esm/src/endpoint/types.js.map +0 -1
  334. package/dist/esm/src/endpoint/withOptions.d.ts +0 -4
  335. package/dist/esm/src/endpoint/withOptions.d.ts.map +0 -1
  336. package/dist/esm/src/endpoint/withOptions.js +0 -19
  337. package/dist/esm/src/endpoint/withOptions.js.map +0 -1
  338. package/dist/esm/src/endpoint.d.ts +0 -108
  339. package/dist/esm/src/endpoint.d.ts.map +0 -1
  340. package/dist/esm/src/endpoint.js.map +0 -1
  341. package/dist/esm/src/fetch.d.ts +0 -43
  342. package/dist/esm/src/fetch.d.ts.map +0 -1
  343. package/dist/esm/src/fetch.js +0 -46
  344. package/dist/esm/src/fetch.js.map +0 -1
  345. package/dist/esm/src/generated/version.d.ts +0 -2
  346. package/dist/esm/src/generated/version.d.ts.map +0 -1
  347. package/dist/esm/src/generated/version.js +0 -2
  348. package/dist/esm/src/generated/version.js.map +0 -1
  349. package/dist/esm/src/io.d.ts +0 -24
  350. package/dist/esm/src/io.d.ts.map +0 -1
  351. package/dist/esm/src/io.js +0 -73
  352. package/dist/esm/src/io.js.map +0 -1
  353. package/dist/esm/src/lambda.d.ts +0 -23
  354. package/dist/esm/src/lambda.d.ts.map +0 -1
  355. package/dist/esm/src/lambda.js +0 -37
  356. package/dist/esm/src/lambda.js.map +0 -1
  357. package/dist/esm/src/logging/console_logger_transport.d.ts +0 -5
  358. package/dist/esm/src/logging/console_logger_transport.d.ts.map +0 -1
  359. package/dist/esm/src/logging/console_logger_transport.js +0 -94
  360. package/dist/esm/src/logging/console_logger_transport.js.map +0 -1
  361. package/dist/esm/src/logging/logger.d.ts +0 -10
  362. package/dist/esm/src/logging/logger.d.ts.map +0 -1
  363. package/dist/esm/src/logging/logger.js +0 -57
  364. package/dist/esm/src/logging/logger.js.map +0 -1
  365. package/dist/esm/src/logging/logger_transport.d.ts +0 -52
  366. package/dist/esm/src/logging/logger_transport.d.ts.map +0 -1
  367. package/dist/esm/src/logging/logger_transport.js +0 -55
  368. package/dist/esm/src/logging/logger_transport.js.map +0 -1
  369. package/dist/esm/src/node.d.ts +0 -39
  370. package/dist/esm/src/node.d.ts.map +0 -1
  371. package/dist/esm/src/node.js +0 -52
  372. package/dist/esm/src/node.js.map +0 -1
  373. package/dist/esm/src/promises.d.ts.map +0 -1
  374. package/dist/esm/src/promises.js +0 -314
  375. package/dist/esm/src/promises.js.map +0 -1
  376. package/dist/esm/src/public_api.d.ts +0 -2
  377. package/dist/esm/src/public_api.d.ts.map +0 -1
  378. package/dist/esm/src/public_api.js +0 -12
  379. package/dist/esm/src/public_api.js.map +0 -1
  380. package/dist/esm/src/types/errors.d.ts +0 -74
  381. package/dist/esm/src/types/errors.d.ts.map +0 -1
  382. package/dist/esm/src/types/errors.js +0 -128
  383. package/dist/esm/src/types/errors.js.map +0 -1
  384. package/dist/esm/src/types/rpc.d.ts +0 -620
  385. package/dist/esm/src/types/rpc.d.ts.map +0 -1
  386. package/dist/esm/src/types/rpc.js +0 -577
  387. package/dist/esm/src/types/rpc.js.map +0 -1
  388. package/dist/esm/src/user_agent.d.ts +0 -2
  389. package/dist/esm/src/user_agent.d.ts.map +0 -1
  390. package/dist/esm/src/user_agent.js +0 -13
  391. package/dist/esm/src/user_agent.js.map +0 -1
  392. package/dist/esm/src/utils/completable_promise.d.ts +0 -9
  393. package/dist/esm/src/utils/completable_promise.d.ts.map +0 -1
  394. package/dist/esm/src/utils/completable_promise.js +0 -30
  395. package/dist/esm/src/utils/completable_promise.js.map +0 -1
  396. package/dist/esm/src/utils/rand.d.ts +0 -13
  397. package/dist/esm/src/utils/rand.d.ts.map +0 -1
  398. package/dist/esm/src/utils/rand.js +0 -109
  399. package/dist/esm/src/utils/rand.js.map +0 -1
  400. package/dist/esm/src/utils/streams.d.ts +0 -3
  401. package/dist/esm/src/utils/streams.d.ts.map +0 -1
  402. package/dist/esm/src/utils/streams.js +0 -20
  403. package/dist/esm/src/utils/streams.js.map +0 -1
  404. package/dist/esm/tsconfig.tsbuildinfo +0 -1
  405. /package/dist/{cjs/src/endpoint → endpoint}/components.d.ts +0 -0
  406. /package/dist/{cjs/src/endpoint → endpoint}/discovery.d.ts +0 -0
  407. /package/dist/{cjs/src/endpoint → endpoint}/endpoint.d.ts +0 -0
  408. /package/dist/{cjs/src/endpoint → endpoint}/handlers/fetch.d.ts +0 -0
  409. /package/dist/{cjs/src/endpoint → endpoint}/handlers/lambda.d.ts +0 -0
  410. /package/dist/{cjs/src/endpoint → endpoint}/handlers/vm/sdk_shared_core_wasm_bindings.d.ts +0 -0
  411. /package/dist/{esm/src/endpoint → endpoint}/types.js +0 -0
  412. /package/dist/{cjs/src/endpoint → endpoint}/withOptions.d.ts +0 -0
  413. /package/dist/{esm/src/endpoint.js → endpoint.js} +0 -0
  414. /package/dist/{cjs/src/io.d.ts → io.d.ts} +0 -0
  415. /package/dist/{cjs/src/logging → logging}/console_logger_transport.d.ts +0 -0
  416. /package/dist/{cjs/src/logging → logging}/logger.d.ts +0 -0
  417. /package/dist/{cjs/src/utils → utils}/completable_promise.d.ts +0 -0
  418. /package/dist/{cjs/src/utils → utils}/rand.d.ts +0 -0
  419. /package/dist/{cjs/src/utils → utils}/streams.d.ts +0 -0
@@ -0,0 +1,324 @@
1
+ import { RestateError, RetryableError, TerminalError, ensureError, logError } from "../../types/errors.js";
2
+ import { HandlerKind } from "../../types/rpc.js";
3
+ import { parseUrlComponents } from "../components.js";
4
+ import { X_RESTATE_SERVER } from "../../user_agent.js";
5
+ import { OnceStream } from "../../utils/streams.js";
6
+ import { CompletablePromise } from "../../utils/completable_promise.js";
7
+ import { LogSource, LoggerContext, RestateLogLevel } from "../../logging/logger_transport.js";
8
+ import { createLogger } from "../../logging/logger.js";
9
+ import { DEFAULT_CONSOLE_LOGGER_LOG_LEVEL, defaultLoggerTransport } from "../../logging/console_logger_transport.js";
10
+ import { LogLevel, WasmHeader, WasmIdentityVerifier, WasmVM, set_log_level } from "./vm/sdk_shared_core_wasm_bindings.js";
11
+ import { ContextImpl } from "../../context_impl.js";
12
+ import { millisOrDurationToMillis } from "@restatedev/restate-sdk-core";
13
+ import { TransformStream } from "node:stream/web";
14
+
15
+ //#region src/endpoint/handlers/generic.ts
16
+ const ENDPOINT_MANIFEST_V2 = "application/vnd.restate.endpointmanifest.v2+json";
17
+ const ENDPOINT_MANIFEST_V3 = "application/vnd.restate.endpointmanifest.v3+json";
18
+ const ENDPOINT_MANIFEST_V4 = "application/vnd.restate.endpointmanifest.v4+json";
19
+ function tryCreateContextualLogger(loggerTransport, url, headers, additionalContext) {
20
+ try {
21
+ const path = new URL(url, "https://example.com").pathname;
22
+ const parsed = parseUrlComponents(path);
23
+ if (parsed.type !== "invoke") return;
24
+ const invocationId = invocationIdFromHeaders(headers);
25
+ return createLogger(loggerTransport, LogSource.SYSTEM, new LoggerContext(invocationId, parsed.componentName, parsed.handlerName, void 0, void 0, additionalContext));
26
+ } catch {
27
+ return;
28
+ }
29
+ }
30
+ function invocationIdFromHeaders(headers) {
31
+ const invocationIdHeader = headers["x-restate-invocation-id"];
32
+ return typeof invocationIdHeader === "string" ? invocationIdHeader : Array.isArray(invocationIdHeader) ? invocationIdHeader[0] ?? "unknown id" : "unknown id";
33
+ }
34
+ /**
35
+ * This is an internal API to support 'fetch' like handlers.
36
+ * It supports both request-reply mode and bidirectional streaming mode.
37
+ *
38
+ * An individual handler will have to convert the shape of the incoming request
39
+ * to a RestateRequest, and then pass it to this handler, and eventually convert back
40
+ * the response.
41
+ * Different runtimes have slightly different shapes of the incoming request, and responses.
42
+ */
43
+ var GenericHandler = class {
44
+ identityVerifier;
45
+ constructor(endpoint, protocolMode, additionalDiscoveryFields) {
46
+ this.endpoint = endpoint;
47
+ this.protocolMode = protocolMode;
48
+ this.additionalDiscoveryFields = additionalDiscoveryFields;
49
+ if (this.endpoint.keySet === void 0 || this.endpoint.keySet.length === 0) this.endpoint.rlog.warn(`Accepting requests without validating request signatures; handler access must be restricted`);
50
+ else {
51
+ this.endpoint.rlog.info(`Validating requests using signing keys [${this.endpoint.keySet}]`);
52
+ this.identityVerifier = new WasmIdentityVerifier(this.endpoint.keySet);
53
+ }
54
+ set_log_level(restateLogLevelToWasmLogLevel(DEFAULT_CONSOLE_LOGGER_LOG_LEVEL));
55
+ }
56
+ async handle(request, context) {
57
+ try {
58
+ return await this._handle(request, context);
59
+ } catch (e) {
60
+ const error = ensureError(e);
61
+ (tryCreateContextualLogger(this.endpoint.loggerTransport, request.url, request.headers) ?? this.endpoint.rlog).error("Error while handling request: " + (error.stack ?? error.message));
62
+ return this.toErrorResponse(error instanceof RestateError ? error.code : 500, error.message);
63
+ }
64
+ }
65
+ async _handle(request, context) {
66
+ const path = new URL(request.url, "https://example.com").pathname;
67
+ const parsed = parseUrlComponents(path);
68
+ if (parsed.type === "unknown") {
69
+ const msg = `Invalid path. Allowed are /health, or /discover, or /invoke/SvcName/handlerName, but was: ${path}`;
70
+ this.endpoint.rlog.trace(msg);
71
+ return this.toErrorResponse(404, msg);
72
+ }
73
+ if (parsed.type === "health") return {
74
+ body: OnceStream(new TextEncoder().encode("OK")),
75
+ headers: {
76
+ "content-type": "application/text",
77
+ "x-restate-server": X_RESTATE_SERVER
78
+ },
79
+ statusCode: 200
80
+ };
81
+ const error = this.validateConnectionSignature(path, request.headers);
82
+ if (error !== null) return error;
83
+ if (parsed.type === "discover") return this.handleDiscovery(request.headers["accept"]);
84
+ if (typeof request.headers["content-type"] !== "string") {
85
+ const errorMessage = "Missing content-type header";
86
+ this.endpoint.rlog.warn(errorMessage);
87
+ return this.toErrorResponse(415, errorMessage);
88
+ }
89
+ const service = this.endpoint.components.get(parsed.componentName);
90
+ if (!service) {
91
+ const msg = `No service found for URL: ${JSON.stringify(parsed)}`;
92
+ this.endpoint.rlog.error(msg);
93
+ return this.toErrorResponse(404, msg);
94
+ }
95
+ const handler = service?.handlerMatching(parsed);
96
+ if (!handler) {
97
+ const msg = `No service found for URL: ${JSON.stringify(parsed)}`;
98
+ this.endpoint.rlog.error(msg);
99
+ return this.toErrorResponse(404, msg);
100
+ }
101
+ if (!request.body) {
102
+ const msg = "The incoming message body was null";
103
+ this.endpoint.rlog.error(msg);
104
+ return this.toErrorResponse(400, msg);
105
+ }
106
+ return this.handleInvoke(service, handler, request.body, request.headers, request.extraArgs, request.abortSignal, context ?? {});
107
+ }
108
+ validateConnectionSignature(path, headers) {
109
+ if (!this.identityVerifier) return null;
110
+ const vmHeaders = Object.entries(headers).filter(([, v]) => v !== void 0).map(([k, v]) => new WasmHeader(k, v instanceof Array ? v[0] : v));
111
+ try {
112
+ this.identityVerifier.verify_identity(path, vmHeaders);
113
+ return null;
114
+ } catch (e) {
115
+ this.endpoint.rlog.error(`Rejecting request as its JWT did not validate: ${e}`);
116
+ return this.toErrorResponse(401, "Unauthorized");
117
+ }
118
+ }
119
+ async handleInvoke(service, handler, body, headers, extraArgs, abortSignal, additionalContext) {
120
+ const journalValueCodec = this.endpoint.journalValueCodec ? await this.endpoint.journalValueCodec : {
121
+ encode: (entry) => entry,
122
+ decode: (entry) => Promise.resolve(entry)
123
+ };
124
+ const loggerId = Math.floor(Math.random() * 4294967295);
125
+ try {
126
+ const vmHeaders = Object.entries(headers).filter(([, v]) => v !== void 0).map(([k, v]) => new WasmHeader(k, v instanceof Array ? v[0] : v));
127
+ const coreVm = new WasmVM(vmHeaders, restateLogLevelToWasmLogLevel(DEFAULT_CONSOLE_LOGGER_LOG_LEVEL), loggerId, this.endpoint.journalValueCodec !== void 0);
128
+ const responseHead = coreVm.get_response_head();
129
+ const responseHeaders = responseHead.headers.reduce((headers$1, { key, value }) => ({
130
+ [key]: value,
131
+ ...headers$1
132
+ }), { "x-restate-server": X_RESTATE_SERVER });
133
+ invocationLoggers.set(loggerId, createLogger(this.endpoint.loggerTransport, LogSource.JOURNAL, new LoggerContext(invocationIdFromHeaders(headers), service.name(), handler.name(), void 0, void 0, additionalContext)));
134
+ const inputReader = body.getReader();
135
+ abortSignal.addEventListener("abort", () => {
136
+ invocationLoggers.delete(loggerId);
137
+ inputReader.cancel();
138
+ }, { once: true });
139
+ while (!coreVm.is_ready_to_execute()) {
140
+ const nextValue = await inputReader.read();
141
+ if (nextValue.value !== void 0) coreVm.notify_input(nextValue.value);
142
+ if (nextValue.done) {
143
+ coreVm.notify_input_closed();
144
+ break;
145
+ }
146
+ }
147
+ const input = coreVm.sys_input();
148
+ const invocationRequest = {
149
+ id: input.invocation_id,
150
+ headers: input.headers.reduce((headers$1, { key, value }) => {
151
+ headers$1.set(key, value);
152
+ return headers$1;
153
+ }, /* @__PURE__ */ new Map()),
154
+ attemptHeaders: Object.entries(headers).reduce((headers$1, [key, value]) => {
155
+ if (value !== void 0) headers$1.set(key, value instanceof Array ? value[0] : value);
156
+ return headers$1;
157
+ }, /* @__PURE__ */ new Map()),
158
+ body: input.input,
159
+ extraArgs,
160
+ attemptCompletedSignal: abortSignal
161
+ };
162
+ const loggerContext = new LoggerContext(input.invocation_id, handler.component().name(), handler.name(), handler.kind() === HandlerKind.SERVICE ? void 0 : input.key, invocationRequest, additionalContext);
163
+ const ctxLogger = createLogger(this.endpoint.loggerTransport, LogSource.USER, loggerContext, () => !coreVm.is_processing());
164
+ const vmLogger = createLogger(this.endpoint.loggerTransport, LogSource.JOURNAL, loggerContext);
165
+ invocationLoggers.set(loggerId, vmLogger);
166
+ if (!coreVm.is_processing()) vmLogger.info("Replaying invocation.");
167
+ else vmLogger.info("Starting invocation.");
168
+ const invocationEndPromise = new CompletablePromise();
169
+ const responseTransformStream = new TransformStream();
170
+ const outputWriter = responseTransformStream.writable.getWriter();
171
+ const ctx = new ContextImpl(coreVm, input, ctxLogger, handler.kind(), vmLogger, invocationRequest, invocationEndPromise, inputReader, outputWriter, journalValueCodec, service.options?.serde, service.options?.asTerminalError);
172
+ journalValueCodec.decode(input.input).catch((e) => Promise.reject(new TerminalError(`Failed to decode input using journal value codec: ${ensureError(e).message}`, { errorCode: 400 }))).then((decodedInput) => handler.invoke(ctx, decodedInput)).then((output) => {
173
+ coreVm.sys_write_output_success(journalValueCodec.encode(output));
174
+ coreVm.sys_end();
175
+ vmLogger.info("Invocation completed successfully.");
176
+ }).catch((e) => {
177
+ const error = ensureError(e, service.options?.asTerminalError);
178
+ logError(vmLogger, error);
179
+ if (error instanceof TerminalError) {
180
+ coreVm.sys_write_output_failure({
181
+ code: error.code,
182
+ message: error.message
183
+ });
184
+ coreVm.sys_end();
185
+ return;
186
+ }
187
+ throw error;
188
+ }).catch((e) => {
189
+ const error = ensureError(e);
190
+ if (error instanceof RetryableError) coreVm.notify_error_with_delay_override(error.message, error.stack, error.retryAfter !== void 0 ? BigInt(millisOrDurationToMillis(error.retryAfter)) : void 0);
191
+ else coreVm.notify_error(error.message, error.stack);
192
+ }).finally(() => {
193
+ invocationEndPromise.resolve();
194
+ });
195
+ invocationEndPromise.promise.then(async () => {
196
+ let nextOutput = coreVm.take_output();
197
+ while (nextOutput !== null && nextOutput !== void 0) {
198
+ await outputWriter.write(nextOutput);
199
+ nextOutput = coreVm.take_output();
200
+ }
201
+ await outputWriter.close();
202
+ inputReader.cancel().catch(() => {});
203
+ }).finally(() => {
204
+ invocationLoggers.delete(loggerId);
205
+ }).catch(() => {});
206
+ return {
207
+ headers: responseHeaders,
208
+ statusCode: responseHead.status_code,
209
+ body: responseTransformStream.readable
210
+ };
211
+ } catch (error) {
212
+ invocationLoggers.delete(loggerId);
213
+ throw error;
214
+ }
215
+ }
216
+ handleDiscovery(acceptVersionsString) {
217
+ if (typeof acceptVersionsString !== "string") {
218
+ const errorMessage = "Missing accept header";
219
+ this.endpoint.rlog.warn(errorMessage);
220
+ return this.toErrorResponse(415, errorMessage);
221
+ }
222
+ let manifestVersion;
223
+ if (acceptVersionsString.includes(ENDPOINT_MANIFEST_V4)) manifestVersion = 4;
224
+ else if (acceptVersionsString.includes(ENDPOINT_MANIFEST_V3)) manifestVersion = 3;
225
+ else if (acceptVersionsString.includes(ENDPOINT_MANIFEST_V2)) manifestVersion = 2;
226
+ else {
227
+ const errorMessage = `Unsupported service discovery protocol version '${acceptVersionsString}'`;
228
+ this.endpoint.rlog.warn(errorMessage);
229
+ return this.toErrorResponse(415, errorMessage);
230
+ }
231
+ const discovery = {
232
+ ...this.endpoint.discoveryMetadata,
233
+ ...this.additionalDiscoveryFields,
234
+ protocolMode: this.protocolMode
235
+ };
236
+ const checkUnsupportedFeature = (obj, ...fields) => {
237
+ for (const field of fields) if (field in obj && obj[field] !== void 0) return this.toErrorResponse(500, `The code uses the new discovery feature '${String(field)}' but the runtime doesn't support it yet (discovery protocol negotiated version ${manifestVersion}). Either remove the usage of this feature, or upgrade the runtime.`);
238
+ };
239
+ if (manifestVersion < 3) for (const service of discovery.services) {
240
+ const error = checkUnsupportedFeature(service, "journalRetention", "idempotencyRetention", "inactivityTimeout", "abortTimeout", "enableLazyState", "ingressPrivate");
241
+ if (error !== void 0) return error;
242
+ for (const handler of service.handlers) {
243
+ const error$1 = checkUnsupportedFeature(handler, "journalRetention", "idempotencyRetention", "workflowCompletionRetention", "inactivityTimeout", "abortTimeout", "enableLazyState", "ingressPrivate");
244
+ if (error$1 !== void 0) return error$1;
245
+ }
246
+ }
247
+ if (manifestVersion < 4) {
248
+ discovery.lambdaCompression = void 0;
249
+ for (const service of discovery.services) {
250
+ const error = checkUnsupportedFeature(service, "retryPolicyExponentiationFactor", "retryPolicyInitialInterval", "retryPolicyMaxAttempts", "retryPolicyMaxInterval", "retryPolicyOnMaxAttempts");
251
+ if (error !== void 0) return error;
252
+ for (const handler of service.handlers) {
253
+ const error$1 = checkUnsupportedFeature(handler, "retryPolicyExponentiationFactor", "retryPolicyInitialInterval", "retryPolicyMaxAttempts", "retryPolicyMaxInterval", "retryPolicyOnMaxAttempts");
254
+ if (error$1 !== void 0) return error$1;
255
+ }
256
+ }
257
+ }
258
+ const body = JSON.stringify(discovery);
259
+ return {
260
+ headers: {
261
+ "content-type": manifestVersion === 2 ? ENDPOINT_MANIFEST_V2 : manifestVersion === 3 ? ENDPOINT_MANIFEST_V3 : ENDPOINT_MANIFEST_V4,
262
+ "x-restate-server": X_RESTATE_SERVER
263
+ },
264
+ statusCode: 200,
265
+ body: OnceStream(new TextEncoder().encode(body))
266
+ };
267
+ }
268
+ toErrorResponse(code, message) {
269
+ return {
270
+ headers: {
271
+ "content-type": "application/json",
272
+ "x-restate-server": X_RESTATE_SERVER
273
+ },
274
+ statusCode: code,
275
+ body: OnceStream(new TextEncoder().encode(JSON.stringify({ message })))
276
+ };
277
+ }
278
+ };
279
+ const invocationLoggers = /* @__PURE__ */ new Map();
280
+ const logsTextDecoder = new TextDecoder("utf-8", { fatal: false });
281
+ /**
282
+ * The shared core propagates logs to the SDK invoking this method.
283
+ * When possible it provides an invocationId, which is used to access the registered invocationLoggers, that should contain the logger per invocation id.
284
+ */
285
+ function vm_log(level, strBytes, loggerId) {
286
+ try {
287
+ const logger = loggerId && invocationLoggers.get(loggerId) || void 0;
288
+ const str = logsTextDecoder.decode(strBytes);
289
+ if (logger !== void 0) logger.logForLevel(wasmLogLevelToRestateLogLevel(level), str);
290
+ else defaultLoggerTransport({
291
+ level: wasmLogLevelToRestateLogLevel(level),
292
+ replaying: false,
293
+ source: LogSource.JOURNAL
294
+ }, str);
295
+ } catch (e) {
296
+ defaultLoggerTransport({
297
+ level: RestateLogLevel.ERROR,
298
+ replaying: false,
299
+ source: LogSource.SYSTEM
300
+ }, "Unexpected error thrown while trying to log: " + e?.toString());
301
+ }
302
+ }
303
+ function wasmLogLevelToRestateLogLevel(level) {
304
+ switch (level) {
305
+ case LogLevel.TRACE: return RestateLogLevel.TRACE;
306
+ case LogLevel.DEBUG: return RestateLogLevel.DEBUG;
307
+ case LogLevel.INFO: return RestateLogLevel.INFO;
308
+ case LogLevel.WARN: return RestateLogLevel.WARN;
309
+ case LogLevel.ERROR: return RestateLogLevel.ERROR;
310
+ }
311
+ }
312
+ function restateLogLevelToWasmLogLevel(level) {
313
+ switch (level) {
314
+ case RestateLogLevel.TRACE: return LogLevel.TRACE;
315
+ case RestateLogLevel.DEBUG: return LogLevel.DEBUG;
316
+ case RestateLogLevel.INFO: return LogLevel.INFO;
317
+ case RestateLogLevel.WARN: return LogLevel.WARN;
318
+ case RestateLogLevel.ERROR: return LogLevel.ERROR;
319
+ }
320
+ }
321
+
322
+ //#endregion
323
+ export { GenericHandler, tryCreateContextualLogger, vm_log };
324
+ //# sourceMappingURL=generic.js.map
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,93 @@
1
+ const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
2
+ const require_errors = require('../../types/errors.cjs');
3
+ const require_user_agent = require('../../user_agent.cjs');
4
+ const require_streams = require('../../utils/streams.cjs');
5
+ const require_generic = require('./generic.cjs');
6
+ let node_stream_web = require("node:stream/web");
7
+ node_stream_web = require_rolldown_runtime.__toESM(node_stream_web);
8
+ let node_buffer = require("node:buffer");
9
+ node_buffer = require_rolldown_runtime.__toESM(node_buffer);
10
+ let node_zlib = require("node:zlib");
11
+ node_zlib = require_rolldown_runtime.__toESM(node_zlib);
12
+
13
+ //#region src/endpoint/handlers/lambda.ts
14
+ const RESPONSE_COMPRESSION_THRESHOLD = 3 * 1024 * 1024;
15
+ var LambdaHandler = class {
16
+ constructor(handler, compressionSupported) {
17
+ this.handler = handler;
18
+ this.compressionSupported = compressionSupported;
19
+ }
20
+ async handleRequest(event, context) {
21
+ const path = "path" in event ? event.path : event.rawPath;
22
+ let requestContentEncoding;
23
+ let requestAcceptEncoding;
24
+ for (const [key, value] of Object.entries(event.headers)) if (key.localeCompare("content-encoding", void 0, { sensitivity: "accent" }) === 0) requestContentEncoding = value;
25
+ else if (key.localeCompare("accept-encoding", void 0, { sensitivity: "accent" }) === 0) requestAcceptEncoding = value;
26
+ let bodyStream;
27
+ if (!event.body) bodyStream = null;
28
+ else {
29
+ let bodyBuffer;
30
+ if (event.isBase64Encoded) bodyBuffer = node_buffer.Buffer.from(event.body, "base64");
31
+ else bodyBuffer = node_buffer.Buffer.from(new TextEncoder().encode(event.body));
32
+ if (requestContentEncoding && requestContentEncoding.includes("zstd")) {
33
+ 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");
34
+ bodyBuffer = node_zlib.zstdDecompressSync(bodyBuffer);
35
+ }
36
+ bodyStream = require_streams.OnceStream(bodyBuffer);
37
+ }
38
+ const abortController = new AbortController();
39
+ const request = {
40
+ body: bodyStream,
41
+ headers: event.headers,
42
+ url: path,
43
+ extraArgs: [context],
44
+ abortSignal: abortController.signal
45
+ };
46
+ let response;
47
+ try {
48
+ response = await this.handler.handle(request, { AWSRequestId: context.awsRequestId });
49
+ } catch (e) {
50
+ abortController.abort();
51
+ throw e;
52
+ }
53
+ const chunks = [];
54
+ try {
55
+ await response.body.pipeTo(new node_stream_web.WritableStream({ write: (chunk) => {
56
+ chunks.push(chunk);
57
+ } }));
58
+ } catch (e) {
59
+ const error = require_errors.ensureError(e);
60
+ (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));
61
+ return {
62
+ headers: {
63
+ "content-type": "application/json",
64
+ "x-restate-server": require_user_agent.X_RESTATE_SERVER
65
+ },
66
+ statusCode: 500,
67
+ isBase64Encoded: false,
68
+ body: JSON.stringify({ message: error.message })
69
+ };
70
+ } finally {
71
+ abortController.abort();
72
+ }
73
+ const responseBodyBuffer = node_buffer.Buffer.concat(chunks);
74
+ let responseBody;
75
+ if (this.compressionSupported && responseBodyBuffer.length > RESPONSE_COMPRESSION_THRESHOLD && requestAcceptEncoding && requestAcceptEncoding.includes("zstd")) {
76
+ response.headers["content-encoding"] = "zstd";
77
+ responseBody = node_zlib.zstdCompressSync(responseBodyBuffer).toString("base64");
78
+ } else responseBody = responseBodyBuffer.toString("base64");
79
+ return {
80
+ headers: response.headers,
81
+ statusCode: response.statusCode,
82
+ isBase64Encoded: true,
83
+ body: responseBody
84
+ };
85
+ }
86
+ };
87
+ function isCompressionSupported() {
88
+ return "zstdDecompressSync" in node_zlib && "zstdCompressSync" in node_zlib;
89
+ }
90
+
91
+ //#endregion
92
+ exports.LambdaHandler = LambdaHandler;
93
+ exports.isCompressionSupported = isCompressionSupported;
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,89 @@
1
+ import { ensureError } from "../../types/errors.js";
2
+ import { X_RESTATE_SERVER } from "../../user_agent.js";
3
+ import { OnceStream } from "../../utils/streams.js";
4
+ import { tryCreateContextualLogger } from "./generic.js";
5
+ import { WritableStream } from "node:stream/web";
6
+ import { Buffer } from "node:buffer";
7
+ import * as zlib from "node:zlib";
8
+
9
+ //#region src/endpoint/handlers/lambda.ts
10
+ const RESPONSE_COMPRESSION_THRESHOLD = 3 * 1024 * 1024;
11
+ var LambdaHandler = class {
12
+ constructor(handler, compressionSupported) {
13
+ this.handler = handler;
14
+ this.compressionSupported = compressionSupported;
15
+ }
16
+ async handleRequest(event, context) {
17
+ const path = "path" in event ? event.path : event.rawPath;
18
+ let requestContentEncoding;
19
+ let requestAcceptEncoding;
20
+ for (const [key, value] of Object.entries(event.headers)) if (key.localeCompare("content-encoding", void 0, { sensitivity: "accent" }) === 0) requestContentEncoding = value;
21
+ else if (key.localeCompare("accept-encoding", void 0, { sensitivity: "accent" }) === 0) requestAcceptEncoding = value;
22
+ let bodyStream;
23
+ if (!event.body) bodyStream = null;
24
+ else {
25
+ let bodyBuffer;
26
+ if (event.isBase64Encoded) bodyBuffer = Buffer.from(event.body, "base64");
27
+ else bodyBuffer = Buffer.from(new TextEncoder().encode(event.body));
28
+ if (requestContentEncoding && requestContentEncoding.includes("zstd")) {
29
+ 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");
30
+ bodyBuffer = zlib.zstdDecompressSync(bodyBuffer);
31
+ }
32
+ bodyStream = OnceStream(bodyBuffer);
33
+ }
34
+ const abortController = new AbortController();
35
+ const request = {
36
+ body: bodyStream,
37
+ headers: event.headers,
38
+ url: path,
39
+ extraArgs: [context],
40
+ abortSignal: abortController.signal
41
+ };
42
+ let response;
43
+ try {
44
+ response = await this.handler.handle(request, { AWSRequestId: context.awsRequestId });
45
+ } catch (e) {
46
+ abortController.abort();
47
+ throw e;
48
+ }
49
+ const chunks = [];
50
+ try {
51
+ await response.body.pipeTo(new WritableStream({ write: (chunk) => {
52
+ chunks.push(chunk);
53
+ } }));
54
+ } catch (e) {
55
+ const error = ensureError(e);
56
+ (tryCreateContextualLogger(this.handler.endpoint.loggerTransport, request.url, request.headers) ?? this.handler.endpoint.rlog).error("Error while collecting invocation response: " + (error.stack ?? error.message));
57
+ return {
58
+ headers: {
59
+ "content-type": "application/json",
60
+ "x-restate-server": X_RESTATE_SERVER
61
+ },
62
+ statusCode: 500,
63
+ isBase64Encoded: false,
64
+ body: JSON.stringify({ message: error.message })
65
+ };
66
+ } finally {
67
+ abortController.abort();
68
+ }
69
+ const responseBodyBuffer = Buffer.concat(chunks);
70
+ let responseBody;
71
+ if (this.compressionSupported && responseBodyBuffer.length > RESPONSE_COMPRESSION_THRESHOLD && requestAcceptEncoding && requestAcceptEncoding.includes("zstd")) {
72
+ response.headers["content-encoding"] = "zstd";
73
+ responseBody = zlib.zstdCompressSync(responseBodyBuffer).toString("base64");
74
+ } else responseBody = responseBodyBuffer.toString("base64");
75
+ return {
76
+ headers: response.headers,
77
+ statusCode: response.statusCode,
78
+ isBase64Encoded: true,
79
+ body: responseBody
80
+ };
81
+ }
82
+ };
83
+ function isCompressionSupported() {
84
+ return "zstdDecompressSync" in zlib && "zstdCompressSync" in zlib;
85
+ }
86
+
87
+ //#endregion
88
+ export { LambdaHandler, isCompressionSupported };
89
+ //# sourceMappingURL=lambda.js.map
@@ -0,0 +1 @@
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"}