From 87f6d095c4b772d8e7a7f618a72fe31ca6d81df4 Mon Sep 17 00:00:00 2001
From: CarinaWolli <wollencarina@gmail.com>
Date: Wed, 20 Jul 2022 18:34:42 -0400
Subject: [PATCH] show error message when wrong workflow id

---
 apps/web/ee/pages/workflows/[workflow].tsx   | 24 +++++++++++++-------
 apps/web/server/routers/viewer/workflows.tsx |  5 ++++
 2 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/apps/web/ee/pages/workflows/[workflow].tsx b/apps/web/ee/pages/workflows/[workflow].tsx
index 7128dd08b92ee0..8da46fe82abf8a 100644
--- a/apps/web/ee/pages/workflows/[workflow].tsx
+++ b/apps/web/ee/pages/workflows/[workflow].tsx
@@ -78,6 +78,8 @@ function WorkflowPage() {
 
   const {
     data: workflow,
+    isError,
+    error,
     isLoading,
     dataUpdatedAt,
   } = trpc.useQuery([
@@ -153,15 +155,21 @@ function WorkflowPage() {
           <Alert className="border " severity="warning" title={t("pro_feature_workflows")} />
         ) : (
           <>
-            {isAllDataLoaded ? (
-              <WorkflowDetailsPage
-                form={form}
-                workflowId={+workflowId}
-                selectedEventTypes={selectedEventTypes}
-                setSelectedEventTypes={setSelectedEventTypes}
-              />
+            {!isError ? (
+              <>
+                {isAllDataLoaded ? (
+                  <WorkflowDetailsPage
+                    form={form}
+                    workflowId={+workflowId}
+                    selectedEventTypes={selectedEventTypes}
+                    setSelectedEventTypes={setSelectedEventTypes}
+                  />
+                ) : (
+                  <Loader />
+                )}
+              </>
             ) : (
-              <Loader />
+              <Alert severity="error" title="Something went wrong" message={error.message} />
             )}
           </>
         )}
diff --git a/apps/web/server/routers/viewer/workflows.tsx b/apps/web/server/routers/viewer/workflows.tsx
index 1319b6f047cf84..b62793aa23d369 100644
--- a/apps/web/server/routers/viewer/workflows.tsx
+++ b/apps/web/server/routers/viewer/workflows.tsx
@@ -82,6 +82,11 @@ export const workflowsRouter = createProtectedRouter()
           },
         },
       });
+      if (!workflow) {
+        throw new TRPCError({
+          code: "UNAUTHORIZED",
+        });
+      }
       return workflow;
     },
   })