isolate middleware functions from trpc configuration

Author: GustavoOS

worker/trpc/middlewares/auth.ts

@@ -1,20 +1,8 @@
 import { createClerkClient, type User } from "@clerk/backend";
-import { TRPCError } from "@trpc/server";
-import { t } from "@worker/trpc";
 import type { Context } from "../context";
 
 export type AuthenticatedContext = Context & { user: User };
 
-export const authentication = t.middleware(async ({ next, ctx }) => {
-  try {
-    const user = await authFunction(ctx);
-    return next({ ctx: { ...ctx, user } });
-  } catch (error) {
-    console.error(error);
-    throw new TRPCError({ message: "Unable to get auth", cause: error, code: "UNAUTHORIZED" });
-  }
-});
-
 export async function authFunction(ctx: Context) {
   const clerk = createClerkClient({ secretKey: ctx.env.CLERK_SECRET_KEY, publishableKey: ctx.env.CLERK_PUBLISHABLE_KEY });
   const requestState = await clerk.authenticateRequest(ctx.req);

worker/trpc/middlewares/index.ts

@@ -0,0 +1,19 @@
+import { TRPCError } from "@trpc/server";
+import { t } from "@worker/trpc";
+import { authFunction } from "@worker/trpc/middlewares/auth";
+import { rateLimitFunction } from "@worker/trpc/middlewares/rate-limit";
+
+export const authentication = t.middleware(async ({ next, ctx }) => {
+  try {
+    const user = await authFunction(ctx);
+    return next({ ctx: { ...ctx, user } });
+  } catch (error) {
+    console.error(error);
+    throw new TRPCError({ message: "Unable to get auth", cause: error, code: "UNAUTHORIZED" });
+  }
+});
+
+export const rateLimit = t.middleware(async ({ next, ctx }) => {
+  await rateLimitFunction(ctx)
+  return next({ ctx });
+});

worker/trpc/middlewares/rate-limit.ts

@@ -1,6 +1,5 @@
 import { TRPCError } from "@trpc/server";
 import { Ratelimit } from "@upstash/ratelimit";
-import { t } from "@worker/trpc";
 import type { Context } from "@worker/trpc/context";
 import { isLocal } from "@worker/trpc/util/local";
 
@@ -32,8 +31,3 @@ export async function rateLimitFunction(ctx: Context) {
     });
   }
 }
-
-export const rateLimit = t.middleware(async ({ next, ctx }) => {
-  await rateLimitFunction(ctx)
-  return next({ ctx });
-});

worker/trpc/procedures.ts

@@ -1,6 +1,5 @@
 import { t } from "@worker/trpc";
-import { authentication } from "@worker/trpc/middlewares/auth";
-import { rateLimit } from "@worker/trpc/middlewares/rate-limit";
+import { rateLimit, authentication } from "@worker/trpc/middlewares";
 
 export const publicProcedure = t.procedure.use(rateLimit);
 export const privateProcedure = t.procedure.use(rateLimit).use(authentication);