trpc-openapi/dist/cjs/adapters/node-http/core.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.createOpenApiNodeHttpHandler = void 0;
const server_1 = require("@trpc/server");
const unstable_core_do_not_import_1 = require("@trpc/server/unstable-core-do-not-import");
const zod_1 = require("zod");
const generator_1 = require("../../generator");
const utils_1 = require("../../utils");
const errors_1 = require("./errors");
const input_1 = require("./input");
const procedures_1 = require("./procedures");
const createOpenApiNodeHttpHandler = (opts) => {
    const router = Object.assign({}, opts.router);
    // Validate router
    if (process.env.NODE_ENV !== 'production') {
        (0, generator_1.generateOpenApiDocument)(router, { title: '', version: '', baseUrl: '' });
    }
    const { createContext, responseMeta, onError, maxBodySize } = opts;
    const getProcedure = (0, procedures_1.createProcedureCache)(router);
    return async (req, res, next) => {
        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l;
        const sendResponse = (statusCode, headers, body) => {
            res.statusCode = statusCode;
            res.setHeader('Content-Type', 'application/json');
            for (const [key, value] of Object.entries(headers)) {
                if (typeof value !== 'undefined') {
                    res.setHeader(key, value);
                }
            }
            res.end(JSON.stringify(body));
        };
        const method = req.method;
        const reqUrl = req.url;
        const url = new URL(reqUrl.startsWith('/') ? `http://127.0.0.1${reqUrl}` : reqUrl);
        const path = (0, utils_1.normalizePath)(url.pathname);
        let input = undefined;
        let ctx = undefined;
        let info = undefined;
        let data = undefined;
        const { procedure, pathInput } = (_a = getProcedure(method, path)) !== null && _a !== void 0 ? _a : {};
        try {
            if (!procedure) {
                if (next) {
                    return next();
                }
                // Can be used for warmup
                if (method === 'HEAD') {
                    sendResponse(204, {}, undefined);
                    return;
                }
                throw new server_1.TRPCError({
                    message: 'Not found',
                    code: 'NOT_FOUND',
                });
            }
            const contentType = (0, utils_1.getContentType)(req);
            const useBody = (0, utils_1.acceptsRequestBody)(method);
            if (useBody && !(contentType === null || contentType === void 0 ? void 0 : contentType.startsWith('application/json'))) {
                throw new server_1.TRPCError({
                    code: 'UNSUPPORTED_MEDIA_TYPE',
                    message: contentType
                        ? `Unsupported content-type "${contentType}`
                        : 'Missing content-type header',
                });
            }
            const { inputParser } = (0, utils_1.getInputOutputParsers)(procedure.procedure);
            const unwrappedSchema = (0, utils_1.unwrapZodType)(inputParser, true);
            // input should stay undefined if z.void()
            if (!(0, utils_1.instanceofZodTypeLikeVoid)(unwrappedSchema)) {
                input = Object.assign(Object.assign({}, (useBody ? await (0, input_1.getBody)(req, maxBodySize) : (0, input_1.getQuery)(req, url))), pathInput);
            }
            // if supported, coerce all string values to correct types
            if (utils_1.zodSupportsCoerce && (0, utils_1.instanceofZodTypeObject)(unwrappedSchema)) {
                if (!useBody) {
                    for (const [key, shape] of Object.entries(unwrappedSchema.shape)) {
                        let isArray = false;
                        // Check if it's a direct array
                        if (shape instanceof zod_1.ZodArray) {
                            isArray = true;
                        }
                        // Check if it's an optional array
                        else if ((0, utils_1.instanceofZodTypeOptional)(shape)) {
                            const innerType = shape.unwrap();
                            if (innerType instanceof zod_1.ZodArray) {
                                isArray = true;
                            }
                        }
                        if (isArray && input[key] !== undefined && !Array.isArray(input[key])) {
                            input[key] = [input[key]];
                        }
                    }
                }
                (0, utils_1.coerceSchema)(unwrappedSchema);
            }
            info = {
                isBatchCall: false,
                accept: null,
                calls: [],
                type: procedure.type,
                connectionParams: null,
                signal: (0, utils_1.getRequestSignal)(req, res, maxBodySize),
                url,
            };
            ctx = await (createContext === null || createContext === void 0 ? void 0 : createContext({ req, res, info }));
            const caller = router.createCaller(ctx);
            const segments = procedure.path.split('.');
            const procedureFn = segments.reduce((acc, curr) => acc[curr], caller);
            data = await procedureFn(input);
            const meta = responseMeta === null || responseMeta === void 0 ? void 0 : responseMeta({
                type: procedure.type,
                paths: [procedure.path],
                ctx,
                data: [data],
                errors: [],
                info,
                eagerGeneration: true,
            });
            const statusCode = (_b = meta === null || meta === void 0 ? void 0 : meta.status) !== null && _b !== void 0 ? _b : 200;
            const headers = (_c = meta === null || meta === void 0 ? void 0 : meta.headers) !== null && _c !== void 0 ? _c : {};
            const body = data;
            sendResponse(statusCode, headers, body);
        }
        catch (cause) {
            const error = (0, errors_1.getErrorFromUnknown)(cause);
            onError === null || onError === void 0 ? void 0 : onError({
                error,
                type: (_d = procedure === null || procedure === void 0 ? void 0 : procedure.type) !== null && _d !== void 0 ? _d : 'unknown',
                path: procedure === null || procedure === void 0 ? void 0 : procedure.path,
                input,
                ctx,
                req,
            });
            const meta = responseMeta === null || responseMeta === void 0 ? void 0 : responseMeta({
                type: (_e = procedure === null || procedure === void 0 ? void 0 : procedure.type) !== null && _e !== void 0 ? _e : 'unknown',
                paths: (procedure === null || procedure === void 0 ? void 0 : procedure.path) ? [procedure === null || procedure === void 0 ? void 0 : procedure.path] : undefined,
                ctx,
                data: [data],
                errors: [error],
                info,
                eagerGeneration: true,
            });
            const errorShape = (0, unstable_core_do_not_import_1.getErrorShape)({
                config: router._def._config,
                error,
                type: (_f = procedure === null || procedure === void 0 ? void 0 : procedure.type) !== null && _f !== void 0 ? _f : 'unknown',
                path: procedure === null || procedure === void 0 ? void 0 : procedure.path,
                input,
                ctx,
            });
            const isInputValidationError = error.code === 'BAD_REQUEST' &&
                error.cause instanceof Error &&
                error.cause.name === 'ZodError';
            const statusCode = (_h = (_g = meta === null || meta === void 0 ? void 0 : meta.status) !== null && _g !== void 0 ? _g : errors_1.TRPC_ERROR_CODE_HTTP_STATUS[error.code]) !== null && _h !== void 0 ? _h : 500;
            const headers = (_j = meta === null || meta === void 0 ? void 0 : meta.headers) !== null && _j !== void 0 ? _j : {};
            const body = Object.assign(Object.assign({}, errorShape), { message: isInputValidationError
                    ? 'Input validation failed'
                    : ((_l = (_k = errorShape === null || errorShape === void 0 ? void 0 : errorShape.message) !== null && _k !== void 0 ? _k : error.message) !== null && _l !== void 0 ? _l : 'An error occurred'), code: error.code, issues: isInputValidationError ? error.cause.issues : undefined });
            sendResponse(statusCode, headers, body);
        }
    };
};
exports.createOpenApiNodeHttpHandler = createOpenApiNodeHttpHandler;
//# sourceMappingURL=core.js.map