fix: 404 template shows error page (#6441)

Emyrk · web-flow · commit 260b2b233373 · 2023-03-06T13:58:36.000Z
diff --git a/site/src/components/TemplateLayout/TemplateLayout.tsx b/site/src/components/TemplateLayout/TemplateLayout.tsx
@@ -14,6 +14,7 @@ import { Permissions } from "xServices/auth/authXService"
 import { Loader } from "components/Loader/Loader"
 import { usePermissions } from "hooks/usePermissions"
 import { TemplatePageHeader } from "./TemplatePageHeader"
+import { AlertBanner } from "components/AlertBanner/AlertBanner"
 
 const useTemplateName = () => {
   const { template } = useParams()
@@ -57,9 +58,21 @@ export const TemplateLayout: FC<{ children?: JSX.Element }> = ({
       organizationId,
     },
   })
-  const { template, permissions: templatePermissions } = templateState.context
+  const {
+    template,
+    permissions: templatePermissions,
+    getTemplateError,
+  } = templateState.context
   const permissions = usePermissions()
 
+  if (getTemplateError) {
+    return (
+      <div className={styles.error}>
+        <AlertBanner severity="error" error={getTemplateError} />
+      </div>
+    )
+  }
+
   if (!template || !templatePermissions) {
     return <Loader />
   }
@@ -117,6 +130,9 @@ export const TemplateLayout: FC<{ children?: JSX.Element }> = ({
 
 export const useStyles = makeStyles((theme) => {
   return {
+    error: {
+      margin: theme.spacing(2),
+    },
     tabs: {
       borderBottom: `1px solid ${theme.palette.divider}`,
       marginBottom: theme.spacing(5),
