From 34eee6d956b6ac67745652203b3424de2f7d194f Mon Sep 17 00:00:00 2001
From: DavidSpek <vanderspek.david@gmail.com>
Date: Tue, 2 Feb 2021 10:47:36 +0100
Subject: [PATCH] check if snapshots are available frontend

---
 labextension/src/widget.tsx | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/labextension/src/widget.tsx b/labextension/src/widget.tsx
index 65649b3b2..c9dcb06f8 100644
--- a/labextension/src/widget.tsx
+++ b/labextension/src/widget.tsx
@@ -84,6 +84,7 @@ async function activate(
   //  env we are in (like Local Laptop, MiniKF, GCP, UI without Kale, ...)
   const backend = await getBackend(kernel);
   let rokError: IRPCError = null;
+  let snapshotError: IRPCError = null;
   if (backend) {
     try {
       await executeRpc(kernel, 'log.setup_logging');
@@ -111,6 +112,26 @@ async function activate(
         throw error;
       }
     }
+
+    try {
+      await executeRpc(kernel, 'snapshot.check_snapshot_availability');
+    } catch (error) {
+      const unexpectedErrorCodes = [
+        RPC_CALL_STATUS.EncodingError,
+        RPC_CALL_STATUS.ImportError,
+        RPC_CALL_STATUS.UnhandledError,
+      ];
+      if (
+        error instanceof RPCError &&
+        !unexpectedErrorCodes.includes(error.error.code)
+      ) {
+        snapshotError = error.error;
+        console.warn('Snapshots are not available', snapshotError);
+      } else {
+        globalUnhandledRejection({ reason: error });
+        throw error;
+      }
+    }
   } else {
     rokError = {
       rpc: 'rok.check_rok_availability',
@@ -175,6 +196,7 @@ async function activate(
         backend={backend}
         kernel={kernel}
         rokError={rokError}
+        snapshotError={snapshotError}
       />,
     );
     widget.id = 'kubeflow-kale/kubeflowDeployment';