trpc-uwebsockets 0.8.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) Roman Volovoy
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,89 @@
+# trpc-uwebsockets
+
+[uWebSockets.js](https://github.com/uNetworking/uWebSockets.js) adapter for [tRPC](https://trpc.io/)
+
+# Installation
+
+```bash
+yarn install trpc-uwebsockets
+```
+
+or
+
+```bash
+yarn add trpc-uwebsockets
+```
+
+# Usage
+
+Import needed packages
+
+```typescript
+import { App } from 'uWebSockets.js';
+import * as trpc from '@trpc/server';
+```
+
+Define tRPC context and router
+
+```typescript
+type Context = {
+  user: {
+    name: string;
+  } | null;
+};
+
+const createContext: any = (opts: UWebSocketsContextOptions): Context => {
+  const getUser = () => {
+    if (opts.req.headers.authorization === 'meow') {
+      return {
+        name: 'KATT',
+      };
+    }
+    return null;
+  };
+
+  return {
+    user: getUser(),
+  };
+};
+
+const router = trpc.router<Context>().query('hello', {
+  input: z
+    .object({
+      who: z.string().nullish(),
+    })
+    .nullish(),
+  resolve({ input, ctx }) {
+    return {
+      text: `hello ${input?.who ?? ctx.user?.name ?? 'world'}`,
+    };
+  },
+});
+```
+
+Initialize uWebsockets server and attach tTRP router
+
+```typescript
+const app = App();
+
+createUWebSocketsHandler(app, '/trpc', {
+  router,
+  createContext,
+});
+
+app.listen('0.0.0.0', 8000, () => {
+  console.log('server listening on http://localhost:8000');
+});
+```
+
+# Testing
+
+```bash
+yarn t
+```
+
+or
+
+```bash
+yarn test:watch
+```

package/dist/index.js

@@ -0,0 +1,92 @@
+import { resolveHTTPResponse, } from '@trpc/server';
+import { readPostBody } from './utils';
+export * from './types';
+/**
+ *
+ * @param uWsApp uWebsockets server instance
+ * @param pathPrefix The path to endpoint without trailing slash (ex: "/trpc")
+ * @param opts router and createContext functions
+ */
+export function createUWebSocketsHandler(uWsApp, pathPrefix, opts) {
+    const prefixTrimLength = pathPrefix.length + 1; // remove /* from url
+    const handler = async (res, req) => {
+        const method = req.getMethod().toUpperCase();
+        if (method !== 'GET' && method !== 'POST') {
+            // handle only get and post requests, while the rest
+            // will not be captured and propagated further
+            req.setYield(true);
+            return;
+        }
+        const path = req.getUrl().substring(prefixTrimLength);
+        const query = new URLSearchParams(decodeURIComponent(req.getQuery()));
+        const headers = {};
+        req.forEach((key, value) => {
+            headers[key] = value;
+        });
+        // new request object needs to be created, because socket
+        // can only be accessed synchronously, after await it cannot be accessed
+        const requestObj = {
+            headers,
+            method,
+            query,
+            path,
+        };
+        const bodyResult = await readPostBody(method, res);
+        // req is no longer available!
+        const createContext = async function _() {
+            //res could be proxied here
+            return await opts.createContext?.({
+                // res,
+                req: requestObj,
+            });
+        };
+        const fakeReqObject = {
+            method,
+            headers,
+            query,
+            body: bodyResult.ok ? bodyResult.data : undefined,
+        };
+        // TODO batching, onError options need implementation.
+        // responseMeta is not applicable?
+        const result = await resolveHTTPResponse({
+            path,
+            createContext,
+            router: opts.router,
+            req: fakeReqObject,
+            error: bodyResult.ok ? null : bodyResult.error,
+        });
+        if ('status' in result) {
+            res.writeStatus(result.status.toString()); //temp
+        }
+        else {
+            // assume something went bad, should never happen?
+            // there is no way to know from res object that something was send to the socket
+            // can proxy it to detect res calls during createContext and resolve
+            // then will need to exit from here
+            res.cork(() => {
+                res.writeStatus('500 INTERNAL SERVER ERROR');
+                res.end();
+            });
+            return;
+        }
+        for (const [key, value] of Object.entries(result.headers ?? {})) {
+            if (typeof value === 'undefined') {
+                continue;
+            }
+            // FIX not sure why it could be an array. This code path is not tested
+            if (Array.isArray(value))
+                value.forEach((header) => {
+                    res.writeHeader(key, header);
+                });
+            else
+                res.writeHeader(key, value);
+        }
+        res.cork(() => {
+            if (result.body)
+                res.write(result.body);
+            res.end();
+        });
+    };
+    uWsApp.any(pathPrefix + '/*', handler);
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,mBAAmB,GACpB,MAAM,cAAc,CAAC;AAOtB,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,cAAc,SAAS,CAAC;AAExB;;;;;GAKG;AACH,MAAM,UAAU,wBAAwB,CACtC,MAAwB,EACxB,UAAkB,EAClB,IAAiD;IAEjD,MAAM,gBAAgB,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,qBAAqB;IAErE,MAAM,OAAO,GAAG,KAAK,EAAE,GAAqB,EAAE,GAAoB,EAAE,EAAE;QACpE,MAAM,MAAM,GAAG,GAAG,CAAC,SAAS,EAAE,CAAC,WAAW,EAAE,CAAC;QAC7C,IAAI,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,MAAM,EAAE;YACzC,oDAAoD;YACpD,8CAA8C;YAC9C,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACnB,OAAO;SACR;QACD,MAAM,IAAI,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,GAAG,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YACzB,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,yDAAyD;QACzD,wEAAwE;QACxE,MAAM,UAAU,GAA6B;YAC3C,OAAO;YACP,MAAM;YACN,KAAK;YACL,IAAI;SACL,CAAC;QAEF,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAEnD,8BAA8B;QAE9B,MAAM,aAAa,GAAG,KAAK,UAAU,CAAC;YAGpC,2BAA2B;YAC3B,OAAO,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBAChC,OAAO;gBACP,GAAG,EAAE,UAAU;aAChB,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,aAAa,GAAgB;YACjC,MAAM;YACN,OAAO;YACP,KAAK;YACL,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;SAClD,CAAC;QAEF,sDAAsD;QACtD,kCAAkC;QAClC,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC;YACvC,IAAI;YACJ,aAAa;YACb,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,GAAG,EAAE,aAAa;YAClB,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK;SAC/C,CAAC,CAAC;QAEH,IAAI,QAAQ,IAAI,MAAM,EAAE;YACtB,GAAG,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,MAAM;SAClD;aAAM;YACL,kDAAkD;YAClD,gFAAgF;YAChF,oEAAoE;YACpE,mCAAmC;YAEnC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE;gBACZ,GAAG,CAAC,WAAW,CAAC,2BAA2B,CAAC,CAAC;gBAC7C,GAAG,CAAC,GAAG,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YACH,OAAO;SACR;QAED,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE;YAC/D,IAAI,OAAO,KAAK,KAAK,WAAW,EAAE;gBAChC,SAAS;aACV;YACD,sEAAsE;YACtE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBACtB,KAAK,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;oBACvB,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;gBAC/B,CAAC,CAAC,CAAC;;gBACA,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;SAClC;QAED,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE;YACZ,IAAI,MAAM,CAAC,IAAI;gBAAE,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACxC,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,CAAC,GAAG,CAAC,UAAU,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;AACzC,CAAC"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/dist/utils.js

@@ -0,0 +1,47 @@
+import { TRPCError } from '@trpc/server';
+export function readPostBody(method, res) {
+    return new Promise((resolve) => {
+        if (method == 'GET') {
+            // no body in get request
+            resolve({
+                ok: true,
+                data: undefined,
+            });
+        }
+        let buffer;
+        res.onData((ab, isLast) => {
+            const chunk = Buffer.from(ab);
+            if (isLast) {
+                if (buffer) {
+                    // large request, with multiple chunks
+                    resolve({
+                        ok: true,
+                        data: buffer.toString(), // do i need utf8?
+                    });
+                }
+                else {
+                    // only a single chunk was recieved
+                    resolve({
+                        ok: true,
+                        data: chunk.toString(),
+                    });
+                }
+            }
+            else {
+                if (buffer) {
+                    buffer = Buffer.concat([buffer, chunk]);
+                }
+                else {
+                    buffer = Buffer.concat([chunk]);
+                }
+            }
+        });
+        res.onAborted(() => {
+            resolve({
+                ok: false,
+                error: new TRPCError({ code: 'CLIENT_CLOSED_REQUEST' }),
+            });
+        });
+    });
+}
+//# sourceMappingURL=utils.js.map

package/dist/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAGzC,MAAM,UAAU,YAAY,CAAC,MAAc,EAAE,GAAiB;IAC5D,OAAO,IAAI,OAAO,CAEhB,CAAC,OAAO,EAAE,EAAE;QACZ,IAAI,MAAM,IAAI,KAAK,EAAE;YACnB,yBAAyB;YACzB,OAAO,CAAC;gBACN,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,SAAS;aAChB,CAAC,CAAC;SACJ;QAED,IAAI,MAAc,CAAC;QACnB,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,MAAM,EAAE,EAAE;YACxB,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAE9B,IAAI,MAAM,EAAE;gBACV,IAAI,MAAM,EAAE;oBACV,sCAAsC;oBACtC,OAAO,CAAC;wBACN,EAAE,EAAE,IAAI;wBACR,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,kBAAkB;qBAC5C,CAAC,CAAC;iBACJ;qBAAM;oBACL,mCAAmC;oBACnC,OAAO,CAAC;wBACN,EAAE,EAAE,IAAI;wBACR,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE;qBACvB,CAAC,CAAC;iBACJ;aACF;iBAAM;gBACL,IAAI,MAAM,EAAE;oBACV,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;iBACzC;qBAAM;oBACL,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;iBACjC;aACF;QACH,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;YACjB,OAAO,CAAC;gBACN,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC;aACxD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/package.json

@@ -0,0 +1,71 @@
+{
+  "private": false,
+  "name": "trpc-uwebsockets",
+  "description": "uWebSockets adapter for tRPC",
+  "version": "0.8.0",
+  "module": "./dist/index.js",
+  "types": "./types/index.d.ts",
+  "scripts": {
+    "start": "jest --watch",
+    "build": "tsc",
+    "build:watch": "tsc --watch",
+    "test": "jest",
+    "test:watch": "jest --watch",
+    "test:coverage": "jest --coverage",
+    "test:publish": "yarn test && yarn build && yarn pack",
+    "prebuild": "yarn clean",
+    "prepublishOnly": "yarn prettier && yarn lint && yarn build",
+    "clean": "rimraf -rf ./dist && rimraf -rf ./types && rimraf -rf ./*-*.tgz",
+    "format": "yarn prettier && yarn lint",
+    "lint": "eslint . --ext .ts --fix",
+    "lint:dry": "eslint . --ext .ts",
+    "prettier": "prettier --config .prettierrc 'src/**/*.ts' --write",
+    "prettier:dry": "prettier --config .prettierrc 'src/**/*.ts'"
+  },
+  "dependencies": {
+    "@trpc/server": "^9.26.2",
+    "uWebSockets.js": "uNetworking/uWebSockets.js#v20.10.0"
+  },
+  "devDependencies": {
+    "@types/jest": "^28.1.6",
+    "babel-jest": "^28.1.3",
+    "jest": "^28.1.3",
+    "node-fetch": "2",
+    "@rollup/plugin-typescript": "^8.3.3",
+    "@swc/core": "^1.2.215",
+    "@swc/jest": "^0.2.21",
+    "@trpc/client": "^9.26.2",
+    "@types/node": "^18.0.6",
+    "@types/webrtc": "^0.0.32",
+    "@typescript-eslint/eslint-plugin": "^5.30.6",
+    "@typescript-eslint/parser": "^5.30.6",
+    "abort-controller": "^3.0.0",
+    "eslint": "^8.19.0",
+    "eslint-config-prettier": "^8.5.0",
+    "eslint-plugin-jest": "^26.6.0",
+    "eslint-plugin-prettier": "^4.2.1",
+    "jest-environment-jsdom": "^27.1.0",
+    "jest-environment-node": "^28.1.3",
+    "prettier": "^2.7.1",
+    "rimraf": "^3.0.2",
+    "rollup": "^2.77.0",
+    "rollup-plugin-swc": "^0.2.1",
+    "ts-jest": "^28.0.7",
+    "ts-node": "^10.9.1",
+    "typescript": "^4.4.4",
+    "zod": "^3.17.9"
+  },
+  "files": [
+    "dist",
+    "src",
+    "types",
+    "test"
+  ],
+  "license": "MIT",
+  "keywords": [],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/romanzy-1612/trpc-uwebsockets.git"
+  },
+  "author": "Roman Volovoy"
+}

package/src/index.ts

@@ -0,0 +1,118 @@
+import {
+  AnyRouter,
+  inferRouterContext,
+  resolveHTTPResponse,
+} from '@trpc/server';
+import { HTTPRequest } from '@trpc/server/dist/declarations/src/http/internals/types';
+import type * as uWs from 'uWebSockets.js';
+import {
+  UWebSocketsRegisterEndpointOptions,
+  UWebSocketsRequestObject,
+} from './types';
+import { readPostBody } from './utils';
+export * from './types';
+
+/**
+ *
+ * @param uWsApp uWebsockets server instance
+ * @param pathPrefix The path to endpoint without trailing slash (ex: "/trpc")
+ * @param opts router and createContext functions
+ */
+export function createUWebSocketsHandler<TRouter extends AnyRouter>(
+  uWsApp: uWs.TemplatedApp,
+  pathPrefix: string,
+  opts: UWebSocketsRegisterEndpointOptions<TRouter>
+) {
+  const prefixTrimLength = pathPrefix.length + 1; // remove /* from url
+
+  const handler = async (res: uWs.HttpResponse, req: uWs.HttpRequest) => {
+    const method = req.getMethod().toUpperCase();
+    if (method !== 'GET' && method !== 'POST') {
+      // handle only get and post requests, while the rest
+      // will not be captured and propagated further
+      req.setYield(true);
+      return;
+    }
+    const path = req.getUrl().substring(prefixTrimLength);
+    const query = new URLSearchParams(decodeURIComponent(req.getQuery()));
+
+    const headers: Record<string, string> = {};
+    req.forEach((key, value) => {
+      headers[key] = value;
+    });
+
+    // new request object needs to be created, because socket
+    // can only be accessed synchronously, after await it cannot be accessed
+    const requestObj: UWebSocketsRequestObject = {
+      headers,
+      method,
+      query,
+      path,
+    };
+
+    const bodyResult = await readPostBody(method, res);
+
+    // req is no longer available!
+
+    const createContext = async function _(): Promise<
+      inferRouterContext<TRouter>
+    > {
+      //res could be proxied here
+      return await opts.createContext?.({
+        // res,
+        req: requestObj,
+      });
+    };
+
+    const fakeReqObject: HTTPRequest = {
+      method,
+      headers,
+      query,
+      body: bodyResult.ok ? bodyResult.data : undefined,
+    };
+
+    // TODO batching, onError options need implementation.
+    // responseMeta is not applicable?
+    const result = await resolveHTTPResponse({
+      path,
+      createContext,
+      router: opts.router,
+      req: fakeReqObject,
+      error: bodyResult.ok ? null : bodyResult.error,
+    });
+
+    if ('status' in result) {
+      res.writeStatus(result.status.toString()); //temp
+    } else {
+      // assume something went bad, should never happen?
+      // there is no way to know from res object that something was send to the socket
+      // can proxy it to detect res calls during createContext and resolve
+      // then will need to exit from here
+
+      res.cork(() => {
+        res.writeStatus('500 INTERNAL SERVER ERROR');
+        res.end();
+      });
+      return;
+    }
+
+    for (const [key, value] of Object.entries(result.headers ?? {})) {
+      if (typeof value === 'undefined') {
+        continue;
+      }
+      // FIX not sure why it could be an array. This code path is not tested
+      if (Array.isArray(value))
+        value.forEach((header) => {
+          res.writeHeader(key, header);
+        });
+      else res.writeHeader(key, value);
+    }
+
+    res.cork(() => {
+      if (result.body) res.write(result.body);
+      res.end();
+    });
+  };
+
+  uWsApp.any(pathPrefix + '/*', handler);
+}

package/src/types.ts

@@ -0,0 +1,24 @@
+import { AnyRouter, inferRouterContext } from '@trpc/server';
+import { HttpResponse } from 'uWebSockets.js';
+
+export type UWebSocketsRegisterEndpointOptions<TRouter extends AnyRouter> = {
+  router: TRouter;
+  createContext?: (
+    opts: UWebSocketsCreateContextOptions
+  ) => Promise<inferRouterContext<TRouter>> | inferRouterContext<TRouter>;
+};
+
+export type UWebSocketsRequestObject = {
+  headers: Record<string, string>;
+  method: 'POST' | 'GET';
+  query: URLSearchParams;
+  path: string;
+};
+
+// if this to be used, it needs to be proxied
+export type UWebSocketsResponseObject = HttpResponse;
+
+export type UWebSocketsCreateContextOptions = {
+  req: UWebSocketsRequestObject;
+  // res: UWebSocketsResponseObject;
+};

package/src/utils.ts

@@ -0,0 +1,50 @@
+import { TRPCError } from '@trpc/server';
+import { HttpResponse } from 'uWebSockets.js';
+
+export function readPostBody(method: string, res: HttpResponse) {
+  return new Promise<
+    { ok: true; data: unknown } | { ok: false; error: TRPCError }
+  >((resolve) => {
+    if (method == 'GET') {
+      // no body in get request
+      resolve({
+        ok: true,
+        data: undefined,
+      });
+    }
+
+    let buffer: Buffer;
+    res.onData((ab, isLast) => {
+      const chunk = Buffer.from(ab);
+
+      if (isLast) {
+        if (buffer) {
+          // large request, with multiple chunks
+          resolve({
+            ok: true,
+            data: buffer.toString(), // do i need utf8?
+          });
+        } else {
+          // only a single chunk was recieved
+          resolve({
+            ok: true,
+            data: chunk.toString(),
+          });
+        }
+      } else {
+        if (buffer) {
+          buffer = Buffer.concat([buffer, chunk]);
+        } else {
+          buffer = Buffer.concat([chunk]);
+        }
+      }
+    });
+
+    res.onAborted(() => {
+      resolve({
+        ok: false,
+        error: new TRPCError({ code: 'CLIENT_CLOSED_REQUEST' }),
+      });
+    });
+  });
+}

package/test/index.spec.ts

@@ -0,0 +1,171 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import AbortController from 'abort-controller';
+import fetch from 'node-fetch';
+import { z } from 'zod';
+import { UWebSocketsCreateContextOptions } from '../src/types';
+import uWs from 'uWebSockets.js';
+
+import * as trpc from '@trpc/server';
+import { TRPCError } from '@trpc/server';
+import { createUWebSocketsHandler } from '../src/index';
+import { createTRPCClient } from '@trpc/client';
+
+const testPort = 8732;
+
+type Context = {
+  user: {
+    name: string;
+  } | null;
+};
+async function startServer() {
+  const createContext = (_opts: UWebSocketsCreateContextOptions): Context => {
+    const getUser = () => {
+      if (_opts.req.headers.authorization === 'meow') {
+        return {
+          name: 'KATT',
+        };
+      }
+      return null;
+    };
+
+    return {
+      user: getUser(),
+    };
+  };
+
+  const router = trpc
+    .router<Context>()
+    .query('hello', {
+      input: z
+        .object({
+          who: z.string().nullish(),
+        })
+        .nullish(),
+      resolve({ input, ctx }) {
+        return {
+          text: `hello ${input?.who ?? ctx.user?.name ?? 'world'}`,
+        };
+      },
+    })
+    .query('error', {
+      resolve() {
+        throw new TRPCError({
+          code: 'BAD_REQUEST',
+          message: 'error as expected',
+        });
+      },
+    })
+    .mutation('test', {
+      input: z.object({
+        value: z.string(),
+      }),
+      resolve({ input, ctx }) {
+        return {
+          originalValue: input.value,
+          user: ctx.user,
+        };
+      },
+    });
+
+  const app = uWs.App();
+
+  // Handle CORS
+  app.options('/trpc/*', (res) => {
+    res.writeHeader('Access-Control-Allow-Origin', '*');
+    res.writeStatus('200 OK');
+    res.end();
+  });
+
+  // need to register everything on the app object,
+  // as uWebSockets does not have middleware
+  createUWebSocketsHandler(app, '/trpc', {
+    router,
+    createContext,
+  });
+
+  const { server, socket } = await new Promise<{
+    server: uWs.TemplatedApp;
+    socket: uWs.us_listen_socket;
+  }>((resolve) => {
+    app.listen('0.0.0.0', testPort, (socket) => {
+      resolve({
+        server: app,
+        socket,
+      });
+    });
+  });
+
+  const client = createTRPCClient<typeof router>({
+    url: `http://localhost:${testPort}/trpc`,
+
+    AbortController: AbortController as any,
+    fetch: fetch as any,
+    headers: {
+      authorization: 'meow',
+    },
+  });
+
+  return {
+    close: () =>
+      new Promise<void>((resolve, reject) => {
+        try {
+          uWs.us_listen_socket_close(socket);
+        } catch (error) {
+          reject();
+        }
+        resolve();
+      }),
+    router,
+    client,
+  };
+}
+
+let t: trpc.inferAsyncReturnType<typeof startServer>;
+beforeEach(async () => {
+  t = await startServer();
+});
+afterEach(async () => {
+  await t.close();
+});
+
+test('simple query', async () => {
+  expect(
+    await t.client.query('hello', {
+      who: 'test',
+    })
+  ).toMatchInlineSnapshot(`
+    Object {
+      "text": "hello test",
+    }
+  `);
+
+  // t.client.runtime.headers()
+
+  expect(await t.client.query('hello')).toMatchInlineSnapshot(`
+    Object {
+      "text": "hello KATT",
+    }
+  `);
+});
+
+// Error status codes are correct
+test('error handling', async () => {
+  expect(t.client.query('error', null)).rejects.toThrowError(
+    'error as expected'
+  );
+});
+
+test('simple mutation', async () => {
+  expect(
+    await t.client.mutation('test', {
+      value: 'lala',
+    })
+  ).toMatchInlineSnapshot(`
+    Object {
+      "originalValue": "lala",
+      "user": Object {
+        "name": "KATT",
+      },
+    }
+  `);
+});

package/types/index.d.ts

@@ -0,0 +1,11 @@
+import { AnyRouter } from '@trpc/server';
+import type * as uWs from 'uWebSockets.js';
+import { UWebSocketsRegisterEndpointOptions } from './types';
+export * from './types';
+/**
+ *
+ * @param uWsApp uWebsockets server instance
+ * @param pathPrefix The path to endpoint without trailing slash (ex: "/trpc")
+ * @param opts router and createContext functions
+ */
+export declare function createUWebSocketsHandler<TRouter extends AnyRouter>(uWsApp: uWs.TemplatedApp, pathPrefix: string, opts: UWebSocketsRegisterEndpointOptions<TRouter>): void;

package/types/types.d.ts

@@ -0,0 +1,16 @@
+import { AnyRouter, inferRouterContext } from '@trpc/server';
+import { HttpResponse } from 'uWebSockets.js';
+export declare type UWebSocketsRegisterEndpointOptions<TRouter extends AnyRouter> = {
+    router: TRouter;
+    createContext?: (opts: UWebSocketsCreateContextOptions) => Promise<inferRouterContext<TRouter>> | inferRouterContext<TRouter>;
+};
+export declare type UWebSocketsRequestObject = {
+    headers: Record<string, string>;
+    method: 'POST' | 'GET';
+    query: URLSearchParams;
+    path: string;
+};
+export declare type UWebSocketsResponseObject = HttpResponse;
+export declare type UWebSocketsCreateContextOptions = {
+    req: UWebSocketsRequestObject;
+};

package/types/utils.d.ts

@@ -0,0 +1,9 @@
+import { TRPCError } from '@trpc/server';
+import { HttpResponse } from 'uWebSockets.js';
+export declare function readPostBody(method: string, res: HttpResponse): Promise<{
+    ok: true;
+    data: unknown;
+} | {
+    ok: false;
+    error: TRPCError;
+}>;