diff options
Diffstat (limited to 'packages/api/middlewares')
| -rw-r--r-- | packages/api/middlewares/auth.ts | 22 | ||||
| -rw-r--r-- | packages/api/middlewares/trpcAdapter.ts | 41 |
2 files changed, 63 insertions, 0 deletions
diff --git a/packages/api/middlewares/auth.ts b/packages/api/middlewares/auth.ts new file mode 100644 index 00000000..7f39a6f9 --- /dev/null +++ b/packages/api/middlewares/auth.ts @@ -0,0 +1,22 @@ +import { createMiddleware } from "hono/factory"; +import { HTTPException } from "hono/http-exception"; + +import { AuthedContext, createCallerFactory } from "@karakeep/trpc"; +import { appRouter } from "@karakeep/trpc/routers/_app"; + +const createCaller = createCallerFactory(appRouter); + +export const authMiddleware = createMiddleware<{ + Variables: { + ctx: AuthedContext; + api: ReturnType<typeof createCaller>; + }; +}>(async (c, next) => { + if (!c.var.ctx || !c.var.ctx.user || c.var.ctx.user === null) { + throw new HTTPException(401, { + message: "Unauthorized", + }); + } + c.set("api", createCaller(c.get("ctx"))); + await next(); +}); diff --git a/packages/api/middlewares/trpcAdapter.ts b/packages/api/middlewares/trpcAdapter.ts new file mode 100644 index 00000000..6bb4a790 --- /dev/null +++ b/packages/api/middlewares/trpcAdapter.ts @@ -0,0 +1,41 @@ +import { TRPCError } from "@trpc/server"; +import { createMiddleware } from "hono/factory"; +import { HTTPException } from "hono/http-exception"; + +function trpcCodeToHttpCode(code: TRPCError["code"]) { + switch (code) { + case "BAD_REQUEST": + case "PARSE_ERROR": + return 400; + case "UNAUTHORIZED": + return 401; + case "FORBIDDEN": + return 403; + case "NOT_FOUND": + return 404; + case "METHOD_NOT_SUPPORTED": + return 405; + case "TIMEOUT": + return 408; + case "PAYLOAD_TOO_LARGE": + return 413; + case "INTERNAL_SERVER_ERROR": + return 500; + default: + return 500; + } +} + +const trpcAdapter = createMiddleware(async (c, next) => { + await next(); + const e = c.error; + if (e instanceof TRPCError) { + const code = trpcCodeToHttpCode(e.code); + throw new HTTPException(code, { + message: e.message, + cause: e.cause, + }); + } +}); + +export default trpcAdapter; |
