@orpc/server 0.0.0-next.a6dec3e → 0.0.0-next.a764ee3

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 (69) hide show
  1. package/README.md +20 -23
  2. package/dist/adapters/aws-lambda/index.d.mts +46 -0
  3. package/dist/adapters/aws-lambda/index.d.ts +46 -0
  4. package/dist/adapters/aws-lambda/index.mjs +42 -0
  5. package/dist/adapters/bun-ws/index.d.mts +36 -0
  6. package/dist/adapters/bun-ws/index.d.ts +36 -0
  7. package/dist/adapters/bun-ws/index.mjs +47 -0
  8. package/dist/adapters/crossws/index.d.mts +33 -0
  9. package/dist/adapters/crossws/index.d.ts +33 -0
  10. package/dist/adapters/crossws/index.mjs +45 -0
  11. package/dist/adapters/fetch/index.d.mts +74 -8
  12. package/dist/adapters/fetch/index.d.ts +74 -8
  13. package/dist/adapters/fetch/index.mjs +176 -8
  14. package/dist/adapters/message-port/index.d.mts +31 -0
  15. package/dist/adapters/message-port/index.d.ts +31 -0
  16. package/dist/adapters/message-port/index.mjs +39 -0
  17. package/dist/adapters/node/index.d.mts +50 -8
  18. package/dist/adapters/node/index.d.ts +50 -8
  19. package/dist/adapters/node/index.mjs +76 -14
  20. package/dist/adapters/standard/index.d.mts +8 -13
  21. package/dist/adapters/standard/index.d.ts +8 -13
  22. package/dist/adapters/standard/index.mjs +5 -5
  23. package/dist/adapters/standard-peer/index.d.mts +18 -0
  24. package/dist/adapters/standard-peer/index.d.ts +18 -0
  25. package/dist/adapters/standard-peer/index.mjs +7 -0
  26. package/dist/adapters/websocket/index.d.mts +54 -0
  27. package/dist/adapters/websocket/index.d.ts +54 -0
  28. package/dist/adapters/websocket/index.mjs +67 -0
  29. package/dist/adapters/ws/index.d.mts +31 -0
  30. package/dist/adapters/ws/index.d.ts +31 -0
  31. package/dist/adapters/ws/index.mjs +37 -0
  32. package/dist/helpers/index.d.mts +134 -0
  33. package/dist/helpers/index.d.ts +134 -0
  34. package/dist/helpers/index.mjs +188 -0
  35. package/dist/hibernation/index.d.mts +44 -0
  36. package/dist/hibernation/index.d.ts +44 -0
  37. package/dist/hibernation/index.mjs +65 -0
  38. package/dist/index.d.mts +562 -50
  39. package/dist/index.d.ts +562 -50
  40. package/dist/index.mjs +138 -12
  41. package/dist/plugins/index.d.mts +57 -12
  42. package/dist/plugins/index.d.ts +57 -12
  43. package/dist/plugins/index.mjs +99 -58
  44. package/dist/shared/{server.C37gDhSZ.mjs → server.7jWaIryJ.mjs} +109 -57
  45. package/dist/shared/server.B7b2w3_i.d.ts +12 -0
  46. package/dist/shared/server.BEFBl-Cb.d.mts +12 -0
  47. package/dist/shared/{server.DFuJLDuo.mjs → server.BHZCyRuJ.mjs} +80 -51
  48. package/dist/shared/server.BU4WI18A.d.mts +32 -0
  49. package/dist/shared/{server.DFFT_EZo.d.ts → server.Bmh5xd4n.d.ts} +18 -17
  50. package/dist/shared/{server.DLt5njUb.d.mts → server.CYNGeoCm.d.mts} +60 -9
  51. package/dist/shared/{server.DLt5njUb.d.ts → server.CYNGeoCm.d.ts} +60 -9
  52. package/dist/shared/server.D0H-iaY3.d.ts +32 -0
  53. package/dist/shared/{server.BVwwTHyO.mjs → server.DZ5BIITo.mjs} +1 -1
  54. package/dist/shared/server.DhJj-1X9.d.mts +42 -0
  55. package/dist/shared/server.UVMTOWrk.mjs +26 -0
  56. package/dist/shared/{server.DOYDVeMX.d.mts → server.gqRxT-yN.d.mts} +18 -17
  57. package/dist/shared/server.jMTkVNIb.d.ts +42 -0
  58. package/package.json +71 -20
  59. package/dist/adapters/hono/index.d.mts +0 -22
  60. package/dist/adapters/hono/index.d.ts +0 -22
  61. package/dist/adapters/hono/index.mjs +0 -35
  62. package/dist/adapters/next/index.d.mts +0 -29
  63. package/dist/adapters/next/index.d.ts +0 -29
  64. package/dist/adapters/next/index.mjs +0 -32
  65. package/dist/shared/server.Bm0UqHzd.mjs +0 -103
  66. package/dist/shared/server.C8NkqxHo.d.ts +0 -17
  67. package/dist/shared/server.CGCwEAt_.d.mts +0 -10
  68. package/dist/shared/server.DCQgF_JR.d.mts +0 -17
  69. package/dist/shared/server._2UufoXA.d.ts +0 -10
@@ -1,5 +1,7 @@
1
- import { value } from '@orpc/shared';
1
+ import { runWithSpan, value, setSpanError, isAsyncIteratorObject, clone } from '@orpc/shared';
2
+ import { flattenHeader } from '@orpc/standard-server';
2
3
  import { parseBatchRequest, toBatchResponse } from '@orpc/standard-server/batch';
4
+ import { toFetchHeaders } from '@orpc/standard-server-fetch';
3
5
  import { ORPCError } from '@orpc/client';
4
6
  export { S as StrictGetMethodPlugin } from '../shared/server.BW-nUGgA.mjs';
5
7
  import '@orpc/contract';
@@ -25,61 +27,80 @@ class BatchHandlerPlugin {
25
27
  init(options) {
26
28
  options.rootInterceptors ??= [];
27
29
  options.rootInterceptors.unshift(async (options2) => {
28
- if (options2.request.headers["x-orpc-batch"] !== "1") {
30
+ const xHeader = flattenHeader(options2.request.headers["x-orpc-batch"]);
31
+ if (xHeader === void 0) {
29
32
  return options2.next();
30
33
  }
31
34
  let isParsing = false;
32
35
  try {
33
- isParsing = true;
34
- const parsed = parseBatchRequest({ ...options2.request, body: await options2.request.body() });
35
- isParsing = false;
36
- const maxSize = await value(this.maxSize, options2);
37
- if (parsed.length > maxSize) {
38
- return {
39
- matched: true,
40
- response: {
41
- status: 413,
42
- headers: {},
43
- body: "Batch request size exceeds the maximum allowed size"
44
- }
45
- };
46
- }
47
- const responses = parsed.map(
48
- (request, index) => {
49
- const mapped = this.mapRequestItem(request, options2);
50
- return options2.next({ ...options2, request: { ...mapped, body: () => Promise.resolve(mapped.body) } }).then(({ response: response2, matched }) => {
51
- if (matched) {
52
- return { ...response2, index };
36
+ return await runWithSpan({ name: "handle_batch_request" }, async (span) => {
37
+ const mode = xHeader === "buffered" ? "buffered" : "streaming";
38
+ isParsing = true;
39
+ const parsed = parseBatchRequest({ ...options2.request, body: await options2.request.body() });
40
+ isParsing = false;
41
+ span?.setAttribute("batch.mode", mode);
42
+ span?.setAttribute("batch.size", parsed.length);
43
+ const maxSize = await value(this.maxSize, options2);
44
+ if (parsed.length > maxSize) {
45
+ const message = "Batch request size exceeds the maximum allowed size";
46
+ setSpanError(span, message);
47
+ return {
48
+ matched: true,
49
+ response: {
50
+ status: 413,
51
+ headers: {},
52
+ body: message
53
53
  }
54
- return { index, status: 404, headers: {}, body: "No procedure matched" };
55
- }).catch(() => {
56
- return { index, status: 500, headers: {}, body: "Internal server error" };
57
- });
54
+ };
58
55
  }
59
- );
60
- await Promise.race(responses);
61
- const status = await value(this.successStatus, responses, options2);
62
- const headers = await value(this.headers, responses, options2);
63
- const response = toBatchResponse({
64
- status,
65
- headers,
66
- body: async function* () {
67
- const promises = [...responses];
68
- while (true) {
69
- const handling = promises.filter((p) => p !== void 0);
70
- if (handling.length === 0) {
71
- return;
72
- }
73
- const result = await Promise.race(handling);
74
- promises[result.index] = void 0;
75
- yield result;
56
+ const responses = parsed.map(
57
+ (request, index) => {
58
+ const mapped = this.mapRequestItem(request, options2);
59
+ return options2.next({ ...options2, request: { ...mapped, body: () => Promise.resolve(mapped.body) } }).then(({ response: response2, matched }) => {
60
+ span?.addEvent(`response.${index}.${matched ? "success" : "not_matched"}`);
61
+ if (matched) {
62
+ if (response2.body instanceof Blob || response2.body instanceof FormData || isAsyncIteratorObject(response2.body)) {
63
+ return {
64
+ index,
65
+ status: 500,
66
+ headers: {},
67
+ body: "Batch responses do not support file/blob, or event-iterator. Please call this procedure separately outside of the batch request."
68
+ };
69
+ }
70
+ return { ...response2, index };
71
+ }
72
+ return { index, status: 404, headers: {}, body: "No procedure matched" };
73
+ }).catch((err) => {
74
+ Promise.reject(err);
75
+ return { index, status: 500, headers: {}, body: "Internal server error" };
76
+ });
76
77
  }
77
- }()
78
+ );
79
+ await Promise.race(responses);
80
+ const status = await value(this.successStatus, responses, options2);
81
+ const headers = await value(this.headers, responses, options2);
82
+ const response = await toBatchResponse({
83
+ status,
84
+ headers,
85
+ mode,
86
+ body: (async function* () {
87
+ const promises = [...responses];
88
+ while (true) {
89
+ const handling = promises.filter((p) => p !== void 0);
90
+ if (handling.length === 0) {
91
+ return;
92
+ }
93
+ const result = await Promise.race(handling);
94
+ promises[result.index] = void 0;
95
+ yield result;
96
+ }
97
+ })()
98
+ });
99
+ return {
100
+ matched: true,
101
+ response
102
+ };
78
103
  });
79
- return {
80
- matched: true,
81
- response
82
- };
83
104
  } catch (cause) {
84
105
  if (isParsing) {
85
106
  return {
@@ -115,13 +136,11 @@ class CORSPlugin {
115
136
  resHeaders["access-control-max-age"] = this.options.maxAge.toString();
116
137
  }
117
138
  if (this.options.allowMethods?.length) {
118
- resHeaders["access-control-allow-methods"] = this.options.allowMethods.join(",");
139
+ resHeaders["access-control-allow-methods"] = flattenHeader(this.options.allowMethods);
119
140
  }
120
141
  const allowHeaders = this.options.allowHeaders ?? interceptorOptions.request.headers["access-control-request-headers"];
121
- if (Array.isArray(allowHeaders) && allowHeaders.length) {
122
- resHeaders["access-control-allow-headers"] = allowHeaders.join(",");
123
- } else if (typeof allowHeaders === "string") {
124
- resHeaders["access-control-allow-headers"] = allowHeaders;
142
+ if (typeof allowHeaders === "string" || allowHeaders?.length) {
143
+ resHeaders["access-control-allow-headers"] = flattenHeader(allowHeaders);
125
144
  }
126
145
  return {
127
146
  matched: true,
@@ -139,7 +158,7 @@ class CORSPlugin {
139
158
  if (!result.matched) {
140
159
  return result;
141
160
  }
142
- const origin = Array.isArray(interceptorOptions.request.headers.origin) ? interceptorOptions.request.headers.origin.join(",") : interceptorOptions.request.headers.origin || "";
161
+ const origin = flattenHeader(interceptorOptions.request.headers.origin) ?? "";
143
162
  const allowedOrigin = await value(this.options.origin, origin, interceptorOptions);
144
163
  const allowedOriginArr = Array.isArray(allowedOrigin) ? allowedOrigin : [allowedOrigin];
145
164
  if (allowedOriginArr.includes("*")) {
@@ -161,13 +180,29 @@ class CORSPlugin {
161
180
  result.response.headers["access-control-allow-credentials"] = "true";
162
181
  }
163
182
  if (this.options.exposeHeaders?.length) {
164
- result.response.headers["access-control-expose-headers"] = this.options.exposeHeaders.join(",");
183
+ result.response.headers["access-control-expose-headers"] = flattenHeader(this.options.exposeHeaders);
165
184
  }
166
185
  return result;
167
186
  });
168
187
  }
169
188
  }
170
189
 
190
+ class RequestHeadersPlugin {
191
+ init(options) {
192
+ options.rootInterceptors ??= [];
193
+ options.rootInterceptors.push((interceptorOptions) => {
194
+ const reqHeaders = interceptorOptions.context.reqHeaders ?? toFetchHeaders(interceptorOptions.request.headers);
195
+ return interceptorOptions.next({
196
+ ...interceptorOptions,
197
+ context: {
198
+ ...interceptorOptions.context,
199
+ reqHeaders
200
+ }
201
+ });
202
+ });
203
+ }
204
+ }
205
+
171
206
  class ResponseHeadersPlugin {
172
207
  init(options) {
173
208
  options.rootInterceptors ??= [];
@@ -183,7 +218,7 @@ class ResponseHeadersPlugin {
183
218
  if (!result.matched) {
184
219
  return result;
185
220
  }
186
- const responseHeaders = result.response.headers;
221
+ const responseHeaders = clone(result.response.headers);
187
222
  for (const [key, value] of resHeaders) {
188
223
  if (Array.isArray(responseHeaders[key])) {
189
224
  responseHeaders[key].push(value);
@@ -193,7 +228,13 @@ class ResponseHeadersPlugin {
193
228
  responseHeaders[key] = value;
194
229
  }
195
230
  }
196
- return result;
231
+ return {
232
+ ...result,
233
+ response: {
234
+ ...result.response,
235
+ headers: responseHeaders
236
+ }
237
+ };
197
238
  });
198
239
  }
199
240
  }
@@ -241,4 +282,4 @@ class SimpleCsrfProtectionHandlerPlugin {
241
282
  }
242
283
  }
243
284
 
244
- export { BatchHandlerPlugin, CORSPlugin, ResponseHeadersPlugin, SimpleCsrfProtectionHandlerPlugin };
285
+ export { BatchHandlerPlugin, CORSPlugin, RequestHeadersPlugin, ResponseHeadersPlugin, SimpleCsrfProtectionHandlerPlugin };
@@ -1,6 +1,7 @@
1
1
  import { isContractProcedure, ValidationError, mergePrefix, mergeErrorMap, enhanceRoute } from '@orpc/contract';
2
- import { fallbackORPCErrorStatus, ORPCError } from '@orpc/client';
3
- import { value, intercept } from '@orpc/shared';
2
+ import { resolveMaybeOptionalOptions, toArray, value, runWithSpan, intercept, isAsyncIteratorObject, asyncIteratorWithSpan } from '@orpc/shared';
3
+ import { ORPCError, fallbackORPCErrorStatus, mapEventIterator } from '@orpc/client';
4
+ import { HibernationEventIterator } from '@orpc/standard-server';
4
5
 
5
6
  const LAZY_SYMBOL = Symbol("ORPC_LAZY_SYMBOL");
6
7
  function lazy(loader, meta = {}) {
@@ -46,6 +47,9 @@ function addMiddleware(middlewares, addition) {
46
47
  }
47
48
 
48
49
  class Procedure {
50
+ /**
51
+ * This property holds the defined options.
52
+ */
49
53
  "~orpc";
50
54
  constructor(def) {
51
55
  this["~orpc"] = def;
@@ -68,14 +72,15 @@ function createORPCErrorConstructorMap(errors) {
68
72
  if (typeof code !== "string") {
69
73
  return Reflect.get(target, code);
70
74
  }
71
- const item = (...[options]) => {
75
+ const item = (...rest) => {
76
+ const options = resolveMaybeOptionalOptions(rest);
72
77
  const config = errors[code];
73
78
  return new ORPCError(code, {
74
79
  defined: Boolean(config),
75
80
  status: config?.status,
76
- message: options?.message ?? config?.message,
77
- data: options?.data,
78
- cause: options?.cause
81
+ message: options.message ?? config?.message,
82
+ data: options.data,
83
+ cause: options.cause
79
84
  });
80
85
  };
81
86
  return item;
@@ -103,34 +108,59 @@ function middlewareOutputFn(output) {
103
108
  return { output, context: {} };
104
109
  }
105
110
 
106
- function createProcedureClient(lazyableProcedure, ...[options]) {
111
+ function createProcedureClient(lazyableProcedure, ...rest) {
112
+ const options = resolveMaybeOptionalOptions(rest);
107
113
  return async (...[input, callerOptions]) => {
108
- const path = options?.path ?? [];
114
+ const path = toArray(options.path);
109
115
  const { default: procedure } = await unlazy(lazyableProcedure);
110
116
  const clientContext = callerOptions?.context ?? {};
111
- const context = await value(options?.context ?? {}, clientContext);
117
+ const context = await value(options.context ?? {}, clientContext);
112
118
  const errors = createORPCErrorConstructorMap(procedure["~orpc"].errorMap);
119
+ const validateError = async (e) => {
120
+ if (e instanceof ORPCError) {
121
+ return await validateORPCError(procedure["~orpc"].errorMap, e);
122
+ }
123
+ return e;
124
+ };
113
125
  try {
114
- return await intercept(
115
- options?.interceptors ?? [],
116
- {
117
- context,
118
- input,
119
- // input only optional when it undefinable so we can safely cast it
120
- errors,
121
- path,
122
- procedure,
123
- signal: callerOptions?.signal,
124
- lastEventId: callerOptions?.lastEventId
125
- },
126
- (interceptorOptions) => executeProcedureInternal(interceptorOptions.procedure, interceptorOptions)
126
+ const output = await runWithSpan(
127
+ { name: "call_procedure", signal: callerOptions?.signal },
128
+ (span) => {
129
+ span?.setAttribute("procedure.path", [...path]);
130
+ return intercept(
131
+ toArray(options.interceptors),
132
+ {
133
+ context,
134
+ input,
135
+ // input only optional when it undefinable so we can safely cast it
136
+ errors,
137
+ path,
138
+ procedure,
139
+ signal: callerOptions?.signal,
140
+ lastEventId: callerOptions?.lastEventId
141
+ },
142
+ (interceptorOptions) => executeProcedureInternal(interceptorOptions.procedure, interceptorOptions)
143
+ );
144
+ }
127
145
  );
128
- } catch (e) {
129
- if (!(e instanceof ORPCError)) {
130
- throw e;
146
+ if (isAsyncIteratorObject(output)) {
147
+ if (output instanceof HibernationEventIterator) {
148
+ return output;
149
+ }
150
+ return mapEventIterator(
151
+ asyncIteratorWithSpan(
152
+ { name: "consume_event_iterator_output", signal: callerOptions?.signal },
153
+ output
154
+ ),
155
+ {
156
+ value: (v) => v,
157
+ error: (e) => validateError(e)
158
+ }
159
+ );
131
160
  }
132
- const validated = await validateORPCError(procedure["~orpc"].errorMap, e);
133
- throw validated;
161
+ return output;
162
+ } catch (e) {
163
+ throw await validateError(e);
134
164
  }
135
165
  };
136
166
  }
@@ -139,31 +169,41 @@ async function validateInput(procedure, input) {
139
169
  if (!schema) {
140
170
  return input;
141
171
  }
142
- const result = await schema["~standard"].validate(input);
143
- if (result.issues) {
144
- throw new ORPCError("BAD_REQUEST", {
145
- message: "Input validation failed",
146
- data: {
147
- issues: result.issues
148
- },
149
- cause: new ValidationError({ message: "Input validation failed", issues: result.issues })
150
- });
151
- }
152
- return result.value;
172
+ return runWithSpan(
173
+ { name: "validate_input" },
174
+ async () => {
175
+ const result = await schema["~standard"].validate(input);
176
+ if (result.issues) {
177
+ throw new ORPCError("BAD_REQUEST", {
178
+ message: "Input validation failed",
179
+ data: {
180
+ issues: result.issues
181
+ },
182
+ cause: new ValidationError({ message: "Input validation failed", issues: result.issues })
183
+ });
184
+ }
185
+ return result.value;
186
+ }
187
+ );
153
188
  }
154
189
  async function validateOutput(procedure, output) {
155
190
  const schema = procedure["~orpc"].outputSchema;
156
191
  if (!schema) {
157
192
  return output;
158
193
  }
159
- const result = await schema["~standard"].validate(output);
160
- if (result.issues) {
161
- throw new ORPCError("INTERNAL_SERVER_ERROR", {
162
- message: "Output validation failed",
163
- cause: new ValidationError({ message: "Output validation failed", issues: result.issues })
164
- });
165
- }
166
- return result.value;
194
+ return runWithSpan(
195
+ { name: "validate_output" },
196
+ async () => {
197
+ const result = await schema["~standard"].validate(output);
198
+ if (result.issues) {
199
+ throw new ORPCError("INTERNAL_SERVER_ERROR", {
200
+ message: "Output validation failed",
201
+ cause: new ValidationError({ message: "Output validation failed", issues: result.issues })
202
+ });
203
+ }
204
+ return result.value;
205
+ }
206
+ );
167
207
  }
168
208
  async function executeProcedureInternal(procedure, options) {
169
209
  const middlewares = procedure["~orpc"].middlewares;
@@ -175,17 +215,28 @@ async function executeProcedureInternal(procedure, options) {
175
215
  currentInput = await validateInput(procedure, currentInput);
176
216
  }
177
217
  const mid = middlewares[index];
178
- const output = mid ? (await mid({
179
- ...options,
180
- context,
181
- next: async (...[nextOptions]) => {
182
- const nextContext = nextOptions?.context ?? {};
183
- return {
184
- output: await next(index + 1, mergeCurrentContext(context, nextContext), currentInput),
185
- context: nextContext
186
- };
218
+ const output = mid ? await runWithSpan(
219
+ { name: `middleware.${mid.name}`, signal: options.signal },
220
+ async (span) => {
221
+ span?.setAttribute("middleware.index", index);
222
+ span?.setAttribute("middleware.name", mid.name);
223
+ const result = await mid({
224
+ ...options,
225
+ context,
226
+ next: async (...[nextOptions]) => {
227
+ const nextContext = nextOptions?.context ?? {};
228
+ return {
229
+ output: await next(index + 1, mergeCurrentContext(context, nextContext), currentInput),
230
+ context: nextContext
231
+ };
232
+ }
233
+ }, currentInput, middlewareOutputFn);
234
+ return result.output;
187
235
  }
188
- }, currentInput, middlewareOutputFn)).output : await procedure["~orpc"].handler({ ...options, context, input: currentInput });
236
+ ) : await runWithSpan(
237
+ { name: "handler", signal: options.signal },
238
+ () => procedure["~orpc"].handler({ ...options, context, input: currentInput })
239
+ );
189
240
  if (index === outputValidationIndex) {
190
241
  return await validateOutput(procedure, output);
191
242
  }
@@ -358,7 +409,8 @@ function createContractedProcedure(procedure, contract) {
358
409
  });
359
410
  }
360
411
  function call(procedure, input, ...rest) {
361
- return createProcedureClient(procedure, ...rest)(input);
412
+ const options = resolveMaybeOptionalOptions(rest);
413
+ return createProcedureClient(procedure, options)(input, options);
362
414
  }
363
415
 
364
416
  export { LAZY_SYMBOL as L, Procedure as P, createContractedProcedure as a, addMiddleware as b, createProcedureClient as c, isLazy as d, enhanceRouter as e, createAssertedLazyProcedure as f, getRouter as g, createORPCErrorConstructorMap as h, isProcedure as i, getLazyMeta as j, middlewareOutputFn as k, lazy as l, mergeCurrentContext as m, isStartWithMiddlewares as n, mergeMiddlewares as o, call as p, getHiddenRouterContract as q, createAccessibleLazyRouter as r, setHiddenRouterContract as s, traverseContractProcedures as t, unlazy as u, validateORPCError as v, resolveContractProcedures as w, unlazyRouter as x };
@@ -0,0 +1,12 @@
1
+ import { C as Context } from './server.CYNGeoCm.js';
2
+ import { b as StandardHandleOptions } from './server.Bmh5xd4n.js';
3
+
4
+ type FriendlyStandardHandleOptions<T extends Context> = Omit<StandardHandleOptions<T>, 'context'> & (Record<never, never> extends T ? {
5
+ context?: T;
6
+ } : {
7
+ context: T;
8
+ });
9
+ declare function resolveFriendlyStandardHandleOptions<T extends Context>(options: FriendlyStandardHandleOptions<T>): StandardHandleOptions<T>;
10
+
11
+ export { resolveFriendlyStandardHandleOptions as r };
12
+ export type { FriendlyStandardHandleOptions as F };
@@ -0,0 +1,12 @@
1
+ import { C as Context } from './server.CYNGeoCm.mjs';
2
+ import { b as StandardHandleOptions } from './server.gqRxT-yN.mjs';
3
+
4
+ type FriendlyStandardHandleOptions<T extends Context> = Omit<StandardHandleOptions<T>, 'context'> & (Record<never, never> extends T ? {
5
+ context?: T;
6
+ } : {
7
+ context: T;
8
+ });
9
+ declare function resolveFriendlyStandardHandleOptions<T extends Context>(options: FriendlyStandardHandleOptions<T>): StandardHandleOptions<T>;
10
+
11
+ export { resolveFriendlyStandardHandleOptions as r };
12
+ export type { FriendlyStandardHandleOptions as F };
@@ -1,18 +1,17 @@
1
1
  import { toHttpPath, StandardRPCJsonSerializer, StandardRPCSerializer } from '@orpc/client/standard';
2
- import { toArray, intercept, parseEmptyableJSON } from '@orpc/shared';
3
- import '@orpc/standard-server/batch';
4
2
  import { ORPCError, toORPCError } from '@orpc/client';
5
- import { S as StrictGetMethodPlugin } from './server.BW-nUGgA.mjs';
6
- import { c as createProcedureClient, t as traverseContractProcedures, i as isProcedure, u as unlazy, g as getRouter, a as createContractedProcedure } from './server.C37gDhSZ.mjs';
3
+ import { toArray, intercept, runWithSpan, ORPC_NAME, isAsyncIteratorObject, asyncIteratorWithSpan, setSpanError, parseEmptyableJSON, NullProtoObj, value } from '@orpc/shared';
4
+ import { flattenHeader } from '@orpc/standard-server';
5
+ import { c as createProcedureClient, t as traverseContractProcedures, i as isProcedure, u as unlazy, g as getRouter, a as createContractedProcedure } from './server.7jWaIryJ.mjs';
7
6
 
8
7
  class CompositeStandardHandlerPlugin {
9
8
  plugins;
10
9
  constructor(plugins = []) {
11
10
  this.plugins = [...plugins].sort((a, b) => (a.order ?? 0) - (b.order ?? 0));
12
11
  }
13
- init(options) {
12
+ init(options, router) {
14
13
  for (const plugin of this.plugins) {
15
- plugin.init?.(options);
14
+ plugin.init?.(options, router);
16
15
  }
17
16
  }
18
17
  }
@@ -22,7 +21,7 @@ class StandardHandler {
22
21
  this.matcher = matcher;
23
22
  this.codec = codec;
24
23
  const plugins = new CompositeStandardHandlerPlugin(options.plugins);
25
- plugins.init(options);
24
+ plugins.init(options, router);
26
25
  this.interceptors = toArray(options.interceptors);
27
26
  this.clientInterceptors = toArray(options.clientInterceptors);
28
27
  this.rootInterceptors = toArray(options.rootInterceptors);
@@ -40,47 +39,74 @@ class StandardHandler {
40
39
  this.rootInterceptors,
41
40
  { ...options, request, prefix },
42
41
  async (interceptorOptions) => {
43
- let isDecoding = false;
44
- try {
45
- return await intercept(
46
- this.interceptors,
47
- interceptorOptions,
48
- async ({ request: request2, context, prefix: prefix2 }) => {
49
- const method = request2.method;
50
- const url = request2.url;
51
- const pathname = prefix2 ? url.pathname.replace(prefix2, "") : url.pathname;
52
- const match = await this.matcher.match(method, `/${pathname.replace(/^\/|\/$/g, "")}`);
53
- if (!match) {
54
- return { matched: false, response: void 0 };
42
+ return runWithSpan(
43
+ { name: `${request.method} ${request.url.pathname}` },
44
+ async (span) => {
45
+ let step;
46
+ try {
47
+ return await intercept(
48
+ this.interceptors,
49
+ interceptorOptions,
50
+ async ({ request: request2, context, prefix: prefix2 }) => {
51
+ const method = request2.method;
52
+ const url = request2.url;
53
+ const pathname = prefix2 ? url.pathname.replace(prefix2, "") : url.pathname;
54
+ const match = await runWithSpan(
55
+ { name: "find_procedure" },
56
+ () => this.matcher.match(method, `/${pathname.replace(/^\/|\/$/g, "")}`)
57
+ );
58
+ if (!match) {
59
+ return { matched: false, response: void 0 };
60
+ }
61
+ span?.updateName(`${ORPC_NAME}.${match.path.join("/")}`);
62
+ span?.setAttribute("rpc.system", ORPC_NAME);
63
+ span?.setAttribute("rpc.method", match.path.join("."));
64
+ step = "decode_input";
65
+ let input = await runWithSpan(
66
+ { name: "decode_input" },
67
+ () => this.codec.decode(request2, match.params, match.procedure)
68
+ );
69
+ step = void 0;
70
+ if (isAsyncIteratorObject(input)) {
71
+ input = asyncIteratorWithSpan(
72
+ { name: "consume_event_iterator_input", signal: request2.signal },
73
+ input
74
+ );
75
+ }
76
+ const client = createProcedureClient(match.procedure, {
77
+ context,
78
+ path: match.path,
79
+ interceptors: this.clientInterceptors
80
+ });
81
+ step = "call_procedure";
82
+ const output = await client(input, {
83
+ signal: request2.signal,
84
+ lastEventId: flattenHeader(request2.headers["last-event-id"])
85
+ });
86
+ step = void 0;
87
+ const response = this.codec.encode(output, match.procedure);
88
+ return {
89
+ matched: true,
90
+ response
91
+ };
92
+ }
93
+ );
94
+ } catch (e) {
95
+ if (step !== "call_procedure") {
96
+ setSpanError(span, e);
55
97
  }
56
- const client = createProcedureClient(match.procedure, {
57
- context,
58
- path: match.path,
59
- interceptors: this.clientInterceptors
60
- });
61
- isDecoding = true;
62
- const input = await this.codec.decode(request2, match.params, match.procedure);
63
- isDecoding = false;
64
- const lastEventId = Array.isArray(request2.headers["last-event-id"]) ? request2.headers["last-event-id"].at(-1) : request2.headers["last-event-id"];
65
- const output = await client(input, { signal: request2.signal, lastEventId });
66
- const response = this.codec.encode(output, match.procedure);
98
+ const error = step === "decode_input" && !(e instanceof ORPCError) ? new ORPCError("BAD_REQUEST", {
99
+ message: `Malformed request. Ensure the request body is properly formatted and the 'Content-Type' header is set correctly.`,
100
+ cause: e
101
+ }) : toORPCError(e);
102
+ const response = this.codec.encodeError(error);
67
103
  return {
68
104
  matched: true,
69
105
  response
70
106
  };
71
107
  }
72
- );
73
- } catch (e) {
74
- const error = isDecoding && !(e instanceof ORPCError) ? new ORPCError("BAD_REQUEST", {
75
- message: `Malformed request. Ensure the request body is properly formatted and the 'Content-Type' header is set correctly.`,
76
- cause: e
77
- }) : toORPCError(e);
78
- const response = this.codec.encodeError(error);
79
- return {
80
- matched: true,
81
- response
82
- };
83
- }
108
+ }
109
+ );
84
110
  }
85
111
  );
86
112
  }
@@ -111,10 +137,18 @@ class StandardRPCCodec {
111
137
  }
112
138
 
113
139
  class StandardRPCMatcher {
114
- tree = {};
140
+ filter;
141
+ tree = new NullProtoObj();
115
142
  pendingRouters = [];
143
+ constructor(options = {}) {
144
+ this.filter = options.filter ?? true;
145
+ }
116
146
  init(router, path = []) {
117
- const laziedOptions = traverseContractProcedures({ router, path }, ({ path: path2, contract }) => {
147
+ const laziedOptions = traverseContractProcedures({ router, path }, (traverseOptions) => {
148
+ if (!value(this.filter, traverseOptions)) {
149
+ return;
150
+ }
151
+ const { path: path2, contract } = traverseOptions;
118
152
  const httpPath = toHttpPath(path2);
119
153
  if (isProcedure(contract)) {
120
154
  this.tree[httpPath] = {
@@ -173,15 +207,10 @@ class StandardRPCMatcher {
173
207
  }
174
208
 
175
209
  class StandardRPCHandler extends StandardHandler {
176
- constructor(router, options) {
177
- options.plugins ??= [];
178
- const strictGetMethodPluginEnabled = options.strictGetMethodPluginEnabled ?? true;
179
- if (strictGetMethodPluginEnabled) {
180
- options.plugins.push(new StrictGetMethodPlugin());
181
- }
210
+ constructor(router, options = {}) {
182
211
  const jsonSerializer = new StandardRPCJsonSerializer(options);
183
212
  const serializer = new StandardRPCSerializer(jsonSerializer);
184
- const matcher = new StandardRPCMatcher();
213
+ const matcher = new StandardRPCMatcher(options);
185
214
  const codec = new StandardRPCCodec(serializer);
186
215
  super(router, matcher, codec, options);
187
216
  }