Fix production mode detection for dataset view (#73)

* Fix production mode detection for dataset view * Run linters --------- Co-authored-by: Lint Bot <[email protected]>
kuzudb · Feb 8, 2024 · 2fc201e · 2fc201e
1 parent 7bf8bb7
commit 2fc201e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/src/components/DatasetView/DatasetMainView.vue b/src/components/DatasetView/DatasetMainView.vue
@@ -156,18 +156,19 @@ export default {
   },
   methods: {
     fetchDatasets() {
+      Axios.get('/api/mode').then((response) => {
+        const isProduction = response.mode.isProduction;
+        this.isProduction = isProduction;
+      }).catch((error) => {
+        console.error(error);
+        this.isProduction = true;
+      });
       Axios.get('/api/datasets')
         .then((response) => {
           this.allDatasets = response.data;
           if (!this.selectedDataset && this.allDatasets.length > 0) {
             this.selectedDataset = this.allDatasets[0];
           }
-          for (const dataset of this.allDatasets) {
-            if (!dataset.isProduction) {
-              this.isProduction = false;
-              break;
-            }
-          }
         })
         .catch((error) => {
           console.error(error);

diff --git a/src/server/Mode.js b/src/server/Mode.js
@@ -5,7 +5,10 @@ const database = require("./utils/Database");
 router.get("/", async (_, res) => {
   try {
     const mode = database.getAccessModeString();
-    res.send({ mode });
+    res.send({
+      mode,
+      isProduction: process.env.NODE_ENV === "production",
+  });
   } catch (err) {
     return res.status(400).send({ error: err.message });
   }