fix: add error handling

unkeyed · Sep 20, 2024 · db49abd · db49abd
1 parent caf3f87
commit db49abd
Showing 1 changed file with 20 additions and 6 deletions.
diff --git a/apps/dashboard/lib/trpc/routers/key/create.ts b/apps/dashboard/lib/trpc/routers/key/create.ts
@@ -154,16 +154,30 @@ async function upsertIdentity(
       set: {
         updatedAt: Date.now(),
       },
+    })
+    .catch((_err) => {
+      throw new TRPCError({
+        code: "INTERNAL_SERVER_ERROR",
+        message: "Failed to insert identity",
+      });
+    });
+
+  identity = await db.query.identities
+    .findFirst({
+      where: (table, { and, eq }) =>
+        and(eq(table.workspaceId, workspaceId), eq(table.externalId, externalId)),
+    })
+    .catch((_err) => {
+      throw new TRPCError({
+        code: "INTERNAL_SERVER_ERROR",
+        message: "Failed to read identity after upsert",
+      });
     });
 
-  identity = await db.query.identities.findFirst({
-    where: (table, { and, eq }) =>
-      and(eq(table.workspaceId, workspaceId), eq(table.externalId, externalId)),
-  });
   if (!identity) {
     throw new TRPCError({
-      code: "INTERNAL_SERVER_ERROR",
-      message: "Failed to read identity after upsert",
+      code: "NOT_FOUND",
+      message: "No identity present!",
     });
   }
   return identity;