feat: ✨ throw not found error when has no user shelter

emiliosheinz · May 20, 2024 · 051b6dc · 051b6dc
1 parent 123e82a
commit 051b6dc
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/src/server/api/routers/shelter.ts b/src/server/api/routers/shelter.ts
@@ -1,5 +1,6 @@
-import { type z, z as zod } from "zod";
+import { z } from "zod";
 import { shelterSchema } from "~/schemas/shelter";
+import { TRPCError } from "@trpc/server";
 
 import {
   createTRPCRouter,
@@ -14,8 +15,8 @@ export const shelterRouter = createTRPCRouter({
   }),
   findById: publicProcedure
     .input(
-      zod.object({
-        id: zod.number(),
+      z.object({
+        id: z.number(),
       }),
     )
     .query(async (opts) => {
@@ -28,7 +29,10 @@ export const shelterRouter = createTRPCRouter({
       });
 
       if (!result) {
-        throw new Error("Shelter not found");
+        throw new TRPCError({
+          code: "NOT_FOUND",
+          message: "The server cannot find the requested resource.",
+        });
       }
 
       return {
@@ -56,8 +60,8 @@ export const shelterRouter = createTRPCRouter({
     }),
   findUserShelterById: protectedProcedure
     .input(
-      zod.object({
-        id: zod.number(),
+      z.object({
+        id: z.number(),
       }),
     )
     .query(async ({ ctx, input }) => {
@@ -69,7 +73,10 @@ export const shelterRouter = createTRPCRouter({
       });
 
       if (!result) {
-        throw new Error("Shelter not found");
+        throw new TRPCError({
+          code: "NOT_FOUND",
+          message: "The server cannot find the requested resource.",
+        });
       }
 
       return {
@@ -159,8 +166,8 @@ export const shelterRouter = createTRPCRouter({
     }),
   delete: protectedProcedure
     .input(
-      zod.object({
-        id: zod.number(),
+      z.object({
+        id: z.number(),
       }),
     )
     .mutation(async ({ ctx, input }) => {