From 89a984ef62e6126debbdbea56ed3d30284c15366 Mon Sep 17 00:00:00 2001
From: wh1te909 <dcparsi@gmail.com>
Date: Mon, 14 Sep 2020 06:02:13 +0000
Subject: [PATCH] alert if mesh agent not uploaded when trying to generate an
 installer

---
 src/components/modals/agents/InstallAgent.vue | 53 +++++++++++++++----
 1 file changed, 44 insertions(+), 9 deletions(-)

diff --git a/src/components/modals/agents/InstallAgent.vue b/src/components/modals/agents/InstallAgent.vue
index fc9b1a3..bdf8e2c 100644
--- a/src/components/modals/agents/InstallAgent.vue
+++ b/src/components/modals/agents/InstallAgent.vue
@@ -149,14 +149,30 @@ export default {
       };
 
       if (this.installMethod === "manual") {
-        axios.post("/agents/installagent/", data).then(r => {
-          this.info = {
-            expires: this.expires,
-            data: r.data,
-            arch: this.arch,
-          };
-          this.showAgentDownload = true;
-        });
+        this.$axios
+          .post("/agents/installagent/", data)
+          .then(r => {
+            this.info = {
+              expires: this.expires,
+              data: r.data,
+              arch: this.arch,
+            };
+            this.showAgentDownload = true;
+          })
+          .catch(e => {
+            let err;
+            switch (e.response.status) {
+              case 406:
+                err = "Missing 64 bit meshagent.exe. Upload it from File > Upload Mesh Agent";
+                break;
+              case 415:
+                err = "Missing 32 bit meshagent-x86.exe. Upload it from File > Upload Mesh Agent";
+                break;
+              default:
+                err = "Something went wrong";
+            }
+            this.notifyError(err, 4000);
+          });
       } else if (this.installMethod === "exe") {
         this.$q.loading.show({ message: "Generating executable..." });
 
@@ -184,6 +200,12 @@ export default {
               case 412:
                 err = "Golang build failed. Check debug log for the error message";
                 break;
+              case 406:
+                err = "Missing 64 bit meshagent.exe. Upload it from File > Upload Mesh Agent";
+                break;
+              case 415:
+                err = "Missing 32 bit meshagent-x86.exe. Upload it from File > Upload Mesh Agent";
+                break;
               default:
                 err = "Something went wrong";
             }
@@ -202,7 +224,20 @@ export default {
             link.click();
             this.showDLMessage();
           })
-          .catch(e => this.notifyError("Something went wrong"));
+          .catch(e => {
+            let err;
+            switch (e.response.status) {
+              case 406:
+                err = "Missing 64 bit meshagent.exe. Upload it from File > Upload Mesh Agent";
+                break;
+              case 415:
+                err = "Missing 32 bit meshagent-x86.exe. Upload it from File > Upload Mesh Agent";
+                break;
+              default:
+                err = "Something went wrong";
+            }
+            this.notifyError(err, 4000);
+          });
       }
     },
     showDLMessage() {