@trpc/server 10.28.2 → 10.29.0

package/dist/{nodeHTTPRequestHandler-5bbf93f1.js → nodeHTTPRequestHandler-e637755a.js}

@@ -1,6 +1,8 @@
 'use strict';
 
-var resolveHTTPResponse = require('./resolveHTTPResponse-9523b4e3.js');
+require('./index-4a823936.js');
+var resolveHTTPResponse = require('./resolveHTTPResponse-2c307e04.js');
+var batchStreamFormatter = require('./batchStreamFormatter-93cdcdd4.js');
 var adapters_nodeHttp_contentType_json_index = require('./adapters/node-http/content-type/json/index.js');
 
 const defaultJSONContentTypeHandler = adapters_nodeHttp_contentType_json_index.nodeHTTPJSONContentTypeHandler();
@@ -31,7 +33,37 @@ async function nodeHTTPRequestHandler(opts) {
             query,
             body: bodyResult.ok ? bodyResult.data : undefined
         };
-        const result = await resolveHTTPResponse.resolveHTTPResponse({
+        const onHead = (head)=>{
+            if ('status' in head && (!opts.res.statusCode || opts.res.statusCode === 200)) {
+                opts.res.statusCode = head.status;
+            }
+            for (const [key, value] of Object.entries(head.headers ?? {})){
+                /* istanbul ignore if -- @preserve */ if (typeof value === 'undefined') {
+                    continue;
+                }
+                opts.res.setHeader(key, value);
+            }
+        };
+        const formatter = batchStreamFormatter.getBatchStreamFormatter();
+        let isStream = false;
+        const onChunk = ([index, string])=>{
+            if (index === -1) {
+                /**
+         * Full response, no streaming. This can happen
+         * - if the response is an error
+         * - if response is empty (HEAD request)
+         */ opts.res.end(string);
+                return;
+            }
+            if (!isStream) {
+                opts.res.setHeader('Transfer-Encoding', 'chunked');
+                const vary = opts.res.getHeader('Vary');
+                opts.res.setHeader('Vary', vary ? 'trpc-batch-mode, ' + vary : 'trpc-batch-mode');
+                isStream = true;
+            }
+            opts.res.write(formatter(index, string));
+        };
+        await resolveHTTPResponse.resolveHTTPResponse({
             batching: opts.batching,
             responseMeta: opts.responseMeta,
             path: opts.path,
@@ -46,19 +78,16 @@ async function nodeHTTPRequestHandler(opts) {
                     req: opts.req
                 });
             },
-            contentTypeHandler
+            contentTypeHandler,
+            onHead,
+            onChunk
         });
-        const { res  } = opts;
-        if ('status' in result && (!res.statusCode || res.statusCode === 200)) {
-            res.statusCode = result.status;
-        }
-        for (const [key, value] of Object.entries(result.headers ?? {})){
-            if (typeof value === 'undefined') {
-                continue;
-            }
-            res.setHeader(key, value);
+        if (!isStream) {
+            return opts.res;
         }
-        res.end(result.body);
+        opts.res.write(formatter.end());
+        opts.res.end();
+        return opts.res;
     });
 }
 

package/dist/resolveHTTPResponse-2bdf2cd7.js

@@ -0,0 +1,256 @@
+import { b as callProcedure } from './config-54786004.js';
+import { T as TRPCError, a as getTRPCErrorFromUnknown } from './TRPCError-7c3909ea.js';
+import { t as transformTRPCResponse, g as getErrorShape } from './transformTRPCResponse-7fcbe472.js';
+import { g as getJsonContentTypeInputs } from './contentType-7354963c.js';
+import { b as getHTTPStatusCode } from './index-ca9a128e.js';
+
+const HTTP_METHOD_PROCEDURE_TYPE_MAP = {
+    GET: 'query',
+    POST: 'mutation',
+};
+const fallbackContentTypeHandler = {
+    getInputs: getJsonContentTypeInputs,
+};
+function initResponse(initOpts) {
+    const { ctx, paths, type, responseMeta, untransformedJSON, errors = [], } = initOpts;
+    let status = untransformedJSON ? getHTTPStatusCode(untransformedJSON) : 200;
+    const headers = {
+        'Content-Type': 'application/json',
+    };
+    const eagerGeneration = !untransformedJSON;
+    const data = eagerGeneration
+        ? []
+        : Array.isArray(untransformedJSON)
+            ? untransformedJSON
+            : [untransformedJSON];
+    const meta = responseMeta?.({
+        ctx,
+        paths,
+        type,
+        data,
+        errors,
+        eagerGeneration,
+    }) ?? {};
+    for (const [key, value] of Object.entries(meta.headers ?? {})) {
+        headers[key] = value;
+    }
+    if (meta.status) {
+        status = meta.status;
+    }
+    return {
+        status,
+        headers,
+    };
+}
+async function inputToProcedureCall(procedureOpts) {
+    const { opts, ctx, type, input, path } = procedureOpts;
+    try {
+        const data = await callProcedure({
+            procedures: opts.router._def.procedures,
+            path,
+            rawInput: input,
+            ctx,
+            type,
+        });
+        return {
+            result: {
+                data,
+            },
+        };
+    }
+    catch (cause) {
+        const error = getTRPCErrorFromUnknown(cause);
+        opts.onError?.({ error, path, input, ctx, type: type, req: opts.req });
+        return {
+            error: getErrorShape({
+                config: opts.router._def._config,
+                error,
+                type,
+                path,
+                input,
+                ctx,
+            }),
+        };
+    }
+}
+function caughtErrorToData(cause, errorOpts) {
+    const { router, req, onError } = errorOpts.opts;
+    const error = getTRPCErrorFromUnknown(cause);
+    onError?.({
+        error,
+        path: errorOpts.path,
+        input: errorOpts.input,
+        ctx: errorOpts.ctx,
+        type: errorOpts.type,
+        req,
+    });
+    const untransformedJSON = {
+        error: getErrorShape({
+            config: router._def._config,
+            error,
+            type: errorOpts.type,
+            path: errorOpts.path,
+            input: errorOpts.input,
+            ctx: errorOpts.ctx,
+        }),
+    };
+    const transformedJSON = transformTRPCResponse(router._def._config, untransformedJSON);
+    const body = JSON.stringify(transformedJSON);
+    return {
+        error,
+        untransformedJSON,
+        body,
+    };
+}
+// implementation
+async function resolveHTTPResponse(opts) {
+    const { router, req, onHead, onChunk } = opts;
+    if (req.method === 'HEAD') {
+        // can be used for lambda warmup
+        const headResponse = {
+            status: 204,
+        };
+        onHead?.(headResponse);
+        onChunk?.([-1, '']);
+        return headResponse;
+    }
+    const contentTypeHandler = opts.contentTypeHandler ?? fallbackContentTypeHandler;
+    const batchingEnabled = opts.batching?.enabled ?? true;
+    const type = HTTP_METHOD_PROCEDURE_TYPE_MAP[req.method] ?? 'unknown';
+    let ctx = undefined;
+    let paths;
+    const isBatchCall = !!req.query.get('batch');
+    const isStreamCall = isBatchCall &&
+        onHead &&
+        onChunk &&
+        req.headers['trpc-batch-mode'] === 'stream';
+    try {
+        if (opts.error) {
+            throw opts.error;
+        }
+        if (isBatchCall && !batchingEnabled) {
+            throw new Error(`Batching is not enabled on the server`);
+        }
+        /* istanbul ignore if -- @preserve */
+        if (type === 'subscription') {
+            throw new TRPCError({
+                message: 'Subscriptions should use wsLink',
+                code: 'METHOD_NOT_SUPPORTED',
+            });
+        }
+        if (type === 'unknown') {
+            throw new TRPCError({
+                message: `Unexpected request method ${req.method}`,
+                code: 'METHOD_NOT_SUPPORTED',
+            });
+        }
+        const inputs = await contentTypeHandler.getInputs({
+            isBatchCall,
+            req,
+            router,
+            preprocessedBody: opts.preprocessedBody ?? false,
+        });
+        paths = isBatchCall ? opts.path.split(',') : [opts.path];
+        ctx = await opts.createContext();
+        const promises = paths.map((path, index) => inputToProcedureCall({ opts, ctx, type, input: inputs[index], path }));
+        if (!isStreamCall) {
+            /**
+             * Non-streaming response:
+             * - await all responses in parallel, blocking on the slowest one
+             * - create headers with known response body
+             * - return a complete HTTPResponse
+             */
+            const untransformedJSON = await Promise.all(promises);
+            const errors = untransformedJSON.flatMap((response) => 'error' in response ? [response.error] : []);
+            const headResponse = initResponse({
+                ctx,
+                paths,
+                type,
+                responseMeta: opts.responseMeta,
+                untransformedJSON,
+                errors,
+            });
+            onHead?.(headResponse);
+            // return body stuff
+            const result = isBatchCall ? untransformedJSON : untransformedJSON[0]; // eslint-disable-line @typescript-eslint/no-non-null-assertion -- `untransformedJSON` should be the length of `paths` which should be at least 1 otherwise there wouldn't be a request at all
+            const transformedJSON = transformTRPCResponse(router._def._config, result);
+            const body = JSON.stringify(transformedJSON);
+            onChunk?.([-1, body]);
+            return {
+                status: headResponse.status,
+                headers: headResponse.headers,
+                body,
+            };
+        }
+        /**
+         * Streaming response:
+         * - block on none, call `onChunk` as soon as each response is ready
+         * - create headers with minimal data (cannot know the response body in advance)
+         * - return void
+         */
+        const headResponse = initResponse({
+            ctx,
+            paths,
+            type,
+            responseMeta: opts.responseMeta,
+        });
+        onHead(headResponse);
+        const indexedPromises = new Map(promises.map((promise, index) => [
+            index,
+            promise.then((r) => [index, r]),
+        ]));
+        for (let i = 0; i < paths.length; i++) {
+            const [index, untransformedJSON] = await Promise.race(indexedPromises.values());
+            indexedPromises.delete(index);
+            try {
+                const transformedJSON = transformTRPCResponse(router._def._config, untransformedJSON);
+                const body = JSON.stringify(transformedJSON);
+                onChunk([index, body]);
+            }
+            catch (cause) {
+                const path = paths[index];
+                const input = inputs[index];
+                const { body } = caughtErrorToData(cause, {
+                    opts,
+                    ctx,
+                    type,
+                    path,
+                    input,
+                });
+                onChunk([index, body]);
+            }
+        }
+        return;
+    }
+    catch (cause) {
+        // we get here if
+        // - batching is called when it's not enabled
+        // - `createContext()` throws
+        // - `router._def._config.transformer.output.serialize()` throws
+        // - post body is too large
+        // - input deserialization fails
+        // - `errorFormatter` return value is malformed
+        const { error, untransformedJSON, body } = caughtErrorToData(cause, {
+            opts,
+            ctx,
+            type,
+        });
+        const headResponse = initResponse({
+            ctx,
+            paths,
+            type,
+            responseMeta: opts.responseMeta,
+            untransformedJSON,
+            errors: [error],
+        });
+        onHead?.(headResponse);
+        onChunk?.([-1, body]);
+        return {
+            status: headResponse.status,
+            headers: headResponse.headers,
+            body,
+        };
+    }
+}
+
+export { resolveHTTPResponse as r };

package/dist/resolveHTTPResponse-2c307e04.js

@@ -0,0 +1,284 @@
+'use strict';
+
+var config = require('./config-f3ce0144.js');
+var TRPCError = require('./TRPCError-16b2e52d.js');
+var transformTRPCResponse = require('./transformTRPCResponse-48d70ef6.js');
+var contentType = require('./contentType-8356d528.js');
+var index = require('./index-4a823936.js');
+
+const HTTP_METHOD_PROCEDURE_TYPE_MAP = {
+    GET: 'query',
+    POST: 'mutation'
+};
+const fallbackContentTypeHandler = {
+    getInputs: contentType.getJsonContentTypeInputs
+};
+function initResponse(initOpts) {
+    const { ctx , paths , type , responseMeta , untransformedJSON , errors =[] ,  } = initOpts;
+    let status = untransformedJSON ? index.getHTTPStatusCode(untransformedJSON) : 200;
+    const headers = {
+        'Content-Type': 'application/json'
+    };
+    const eagerGeneration = !untransformedJSON;
+    const data = eagerGeneration ? [] : Array.isArray(untransformedJSON) ? untransformedJSON : [
+        untransformedJSON
+    ];
+    const meta = responseMeta?.({
+        ctx,
+        paths,
+        type,
+        data,
+        errors,
+        eagerGeneration
+    }) ?? {};
+    for (const [key, value] of Object.entries(meta.headers ?? {})){
+        headers[key] = value;
+    }
+    if (meta.status) {
+        status = meta.status;
+    }
+    return {
+        status,
+        headers
+    };
+}
+async function inputToProcedureCall(procedureOpts) {
+    const { opts , ctx , type , input , path  } = procedureOpts;
+    try {
+        const data = await config.callProcedure({
+            procedures: opts.router._def.procedures,
+            path,
+            rawInput: input,
+            ctx,
+            type
+        });
+        return {
+            result: {
+                data
+            }
+        };
+    } catch (cause) {
+        const error = TRPCError.getTRPCErrorFromUnknown(cause);
+        opts.onError?.({
+            error,
+            path,
+            input,
+            ctx,
+            type: type,
+            req: opts.req
+        });
+        return {
+            error: transformTRPCResponse.getErrorShape({
+                config: opts.router._def._config,
+                error,
+                type,
+                path,
+                input,
+                ctx
+            })
+        };
+    }
+}
+function caughtErrorToData(cause, errorOpts) {
+    const { router , req , onError  } = errorOpts.opts;
+    const error = TRPCError.getTRPCErrorFromUnknown(cause);
+    onError?.({
+        error,
+        path: errorOpts.path,
+        input: errorOpts.input,
+        ctx: errorOpts.ctx,
+        type: errorOpts.type,
+        req
+    });
+    const untransformedJSON = {
+        error: transformTRPCResponse.getErrorShape({
+            config: router._def._config,
+            error,
+            type: errorOpts.type,
+            path: errorOpts.path,
+            input: errorOpts.input,
+            ctx: errorOpts.ctx
+        })
+    };
+    const transformedJSON = transformTRPCResponse.transformTRPCResponse(router._def._config, untransformedJSON);
+    const body = JSON.stringify(transformedJSON);
+    return {
+        error,
+        untransformedJSON,
+        body
+    };
+}
+// implementation
+async function resolveHTTPResponse(opts) {
+    const { router , req , onHead , onChunk  } = opts;
+    if (req.method === 'HEAD') {
+        // can be used for lambda warmup
+        const headResponse = {
+            status: 204
+        };
+        onHead?.(headResponse);
+        onChunk?.([
+            -1,
+            ''
+        ]);
+        return headResponse;
+    }
+    const contentTypeHandler = opts.contentTypeHandler ?? fallbackContentTypeHandler;
+    const batchingEnabled = opts.batching?.enabled ?? true;
+    const type = HTTP_METHOD_PROCEDURE_TYPE_MAP[req.method] ?? 'unknown';
+    let ctx = undefined;
+    let paths;
+    const isBatchCall = !!req.query.get('batch');
+    const isStreamCall = isBatchCall && onHead && onChunk && req.headers['trpc-batch-mode'] === 'stream';
+    try {
+        if (opts.error) {
+            throw opts.error;
+        }
+        if (isBatchCall && !batchingEnabled) {
+            throw new Error(`Batching is not enabled on the server`);
+        }
+        /* istanbul ignore if -- @preserve */ if (type === 'subscription') {
+            throw new TRPCError.TRPCError({
+                message: 'Subscriptions should use wsLink',
+                code: 'METHOD_NOT_SUPPORTED'
+            });
+        }
+        if (type === 'unknown') {
+            throw new TRPCError.TRPCError({
+                message: `Unexpected request method ${req.method}`,
+                code: 'METHOD_NOT_SUPPORTED'
+            });
+        }
+        const inputs = await contentTypeHandler.getInputs({
+            isBatchCall,
+            req,
+            router,
+            preprocessedBody: opts.preprocessedBody ?? false
+        });
+        paths = isBatchCall ? opts.path.split(',') : [
+            opts.path
+        ];
+        ctx = await opts.createContext();
+        const promises = paths.map((path, index)=>inputToProcedureCall({
+                opts,
+                ctx,
+                type,
+                input: inputs[index],
+                path
+            }));
+        if (!isStreamCall) {
+            /**
+       * Non-streaming response:
+       * - await all responses in parallel, blocking on the slowest one
+       * - create headers with known response body
+       * - return a complete HTTPResponse
+       */ const untransformedJSON = await Promise.all(promises);
+            const errors = untransformedJSON.flatMap((response)=>'error' in response ? [
+                    response.error
+                ] : []);
+            const headResponse1 = initResponse({
+                ctx,
+                paths,
+                type,
+                responseMeta: opts.responseMeta,
+                untransformedJSON,
+                errors
+            });
+            onHead?.(headResponse1);
+            // return body stuff
+            const result = isBatchCall ? untransformedJSON : untransformedJSON[0]; // eslint-disable-line @typescript-eslint/no-non-null-assertion -- `untransformedJSON` should be the length of `paths` which should be at least 1 otherwise there wouldn't be a request at all
+            const transformedJSON = transformTRPCResponse.transformTRPCResponse(router._def._config, result);
+            const body = JSON.stringify(transformedJSON);
+            onChunk?.([
+                -1,
+                body
+            ]);
+            return {
+                status: headResponse1.status,
+                headers: headResponse1.headers,
+                body
+            };
+        }
+        /**
+     * Streaming response:
+     * - block on none, call `onChunk` as soon as each response is ready
+     * - create headers with minimal data (cannot know the response body in advance)
+     * - return void
+     */ const headResponse2 = initResponse({
+            ctx,
+            paths,
+            type,
+            responseMeta: opts.responseMeta
+        });
+        onHead(headResponse2);
+        const indexedPromises = new Map(promises.map((promise, index)=>[
+                index,
+                promise.then((r)=>[
+                        index,
+                        r
+                    ])
+            ]));
+        for(let i = 0; i < paths.length; i++){
+            const [index, untransformedJSON1] = await Promise.race(indexedPromises.values());
+            indexedPromises.delete(index);
+            try {
+                const transformedJSON1 = transformTRPCResponse.transformTRPCResponse(router._def._config, untransformedJSON1);
+                const body1 = JSON.stringify(transformedJSON1);
+                onChunk([
+                    index,
+                    body1
+                ]);
+            } catch (cause) {
+                const path = paths[index];
+                const input = inputs[index];
+                const { body: body2  } = caughtErrorToData(cause, {
+                    opts,
+                    ctx,
+                    type,
+                    path,
+                    input
+                });
+                onChunk([
+                    index,
+                    body2
+                ]);
+            }
+        }
+        return;
+    } catch (cause1) {
+        // we get here if
+        // - batching is called when it's not enabled
+        // - `createContext()` throws
+        // - `router._def._config.transformer.output.serialize()` throws
+        // - post body is too large
+        // - input deserialization fails
+        // - `errorFormatter` return value is malformed
+        const { error , untransformedJSON: untransformedJSON2 , body: body3  } = caughtErrorToData(cause1, {
+            opts,
+            ctx,
+            type
+        });
+        const headResponse3 = initResponse({
+            ctx,
+            paths,
+            type,
+            responseMeta: opts.responseMeta,
+            untransformedJSON: untransformedJSON2,
+            errors: [
+                error
+            ]
+        });
+        onHead?.(headResponse3);
+        onChunk?.([
+            -1,
+            body3
+        ]);
+        return {
+            status: headResponse3.status,
+            headers: headResponse3.headers,
+            body: body3
+        };
+    }
+}
+
+exports.resolveHTTPResponse = resolveHTTPResponse;