From 6fbe207a8c4d84135d30faee6b60c93957974328 Mon Sep 17 00:00:00 2001
From: Drew Ansbacher <drew.ansbacher@spiredigital.com>
Date: Thu, 10 Feb 2022 16:49:56 -0700
Subject: [PATCH] update done fix

---
 .../src/app/services/api/mock-api.service.ts  | 34 +++++++++----------
 frontend/projects/ui/src/app/app.component.ts |  1 +
 .../server-show/server-show.page.html         | 33 ++++++++++++------
 .../server-show/server-show.page.ts           | 17 ++++++++--
 4 files changed, 55 insertions(+), 30 deletions(-)

diff --git a/frontend/projects/setup-wizard/src/app/services/api/mock-api.service.ts b/frontend/projects/setup-wizard/src/app/services/api/mock-api.service.ts
index a73b2edff..78a216a64 100644
--- a/frontend/projects/setup-wizard/src/app/services/api/mock-api.service.ts
+++ b/frontend/projects/setup-wizard/src/app/services/api/mock-api.service.ts
@@ -8,14 +8,13 @@ let tries = 0
   providedIn: 'root',
 })
 export class MockApiService extends ApiService {
-
-  constructor () {
+  constructor() {
     super()
   }
 
   // ** UNENCRYPTED **
 
-  async getStatus () {
+  async getStatus() {
     await pauseFor(1000)
     return {
       'product-key': true,
@@ -23,7 +22,7 @@ export class MockApiService extends ApiService {
     }
   }
 
-  async getDrives () {
+  async getDrives() {
     await pauseFor(1000)
     return {
       disks: [
@@ -43,22 +42,22 @@ export class MockApiService extends ApiService {
                 'password-hash': null,
                 'wrapped-key': null,
               },
-            }
+            },
           ],
           capacity: 123456789123,
           guid: 'uuid-uuid-uuid-uuid',
-        }
+        },
       ],
       reconnect: [],
     }
   }
 
-  async set02XDrive () {
+  async set02XDrive() {
     await pauseFor(1000)
     return
   }
 
-  async getRecoveryStatus () {
+  async getRecoveryStatus() {
     tries = Math.min(tries + 1, 4)
     return {
       'bytes-transferred': tries,
@@ -69,38 +68,38 @@ export class MockApiService extends ApiService {
 
   // ** ENCRYPTED **
 
-  async verifyCifs (params: CifsRecoverySource) {
+  async verifyCifs(params: CifsRecoverySource) {
     await pauseFor(1000)
     return {
       version: '0.3.0',
       full: true,
-      'password-hash': '$argon2d$v=19$m=1024,t=1,p=1$YXNkZmFzZGZhc2RmYXNkZg$Ceev1I901G6UwU+hY0sHrFZ56D+o+LNJ',
+      'password-hash':
+        '$argon2d$v=19$m=1024,t=1,p=1$YXNkZmFzZGZhc2RmYXNkZg$Ceev1I901G6UwU+hY0sHrFZ56D+o+LNJ',
       'wrapped-key': '',
     }
   }
 
-  async verifyProductKey () {
+  async verifyProductKey() {
     await pauseFor(1000)
     return
   }
 
-  async importDrive (guid: string) {
+  async importDrive(guid: string) {
     await pauseFor(3000)
     return setupRes
   }
 
-  async setupEmbassy (setupInfo: SetupEmbassyReq) {
+  async setupEmbassy(setupInfo: SetupEmbassyReq) {
     await pauseFor(3000)
     return setupRes
   }
 
-  async setupComplete () {
+  async setupComplete() {
     await pauseFor(1000)
   }
 }
 
-const rootCA =
-  `-----BEGIN CERTIFICATE-----
+const rootCA = `-----BEGIN CERTIFICATE-----
 MIIDpzCCAo+gAwIBAgIRAIIuOarlQETlUQEOZJGZYdIwDQYJKoZIhvcNAQELBQAw
 bTELMAkGA1UEBhMCVVMxFTATBgNVBAoMDEV4YW1wbGUgQ29ycDEOMAwGA1UECwwF
 U2FsZXMxCzAJBgNVBAgMAldBMRgwFgYDVQQDDA93d3cuZXhhbXBsZS5jb20xEDAO
@@ -188,7 +187,8 @@ const disks = [
           version: '0.3.6',
           full: true,
           // password is 'asdfasdf'
-          'password-hash': '$argon2d$v=19$m=1024,t=1,p=1$YXNkZmFzZGZhc2RmYXNkZg$Ceev1I901G6UwU+hY0sHrFZ56D+o+LNJ',
+          'password-hash':
+            '$argon2d$v=19$m=1024,t=1,p=1$YXNkZmFzZGZhc2RmYXNkZg$Ceev1I901G6UwU+hY0sHrFZ56D+o+LNJ',
           'wrapped-key': '',
         },
       },
diff --git a/frontend/projects/ui/src/app/app.component.ts b/frontend/projects/ui/src/app/app.component.ts
index 23e6edde7..d57618727 100644
--- a/frontend/projects/ui/src/app/app.component.ts
+++ b/frontend/projects/ui/src/app/app.component.ts
@@ -298,6 +298,7 @@ export class AppComponent {
         }
       })
   }
+  m
 
   private watchUpdateProgress (): Subscription {
     return this.patch
diff --git a/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.html b/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.html
index e75b85fc6..8c0a87054 100644
--- a/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.html
+++ b/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.html
@@ -63,19 +63,30 @@
             </p>
             <!-- "Software Update" button only -->
             <p *ngIf="button.title === 'Software Update'">
-              <ng-container
-                *ngIf="eosService.updateAvailable$ | async; else check"
-              >
-                <ion-text class="inline" color="success">
-                  <ion-icon name="rocket-outline"></ion-icon>
-                  Update Available
+              <ng-container *ngIf="button.disabled | async; else enabled">
+                <ion-text
+                  *ngIf="patch.data['server-info']['status-info'].updated"
+                  class="inline"
+                  color="warning"
+                >
+                  Update Complete, Restart to apply changes
                 </ion-text>
               </ng-container>
-              <ng-template #check>
-                <ion-text class="inline" color="dark">
-                  <ion-icon name="refresh"></ion-icon>
-                  Check for updates
-                </ion-text>
+              <ng-template #enabled>
+                <ng-container
+                  *ngIf="eosService.updateAvailable$ | async; else check"
+                >
+                  <ion-text class="inline" color="success">
+                    <ion-icon name="rocket-outline"></ion-icon>
+                    Update Available
+                  </ion-text>
+                </ng-container>
+                <ng-template #check>
+                  <ion-text class="inline" color="dark">
+                    <ion-icon name="refresh"></ion-icon>
+                    Check for updates
+                  </ion-text>
+                </ng-template>
               </ng-template>
             </p>
           </ion-label>
diff --git a/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.ts b/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.ts
index d65e28c07..ead88c75d 100644
--- a/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.ts
+++ b/frontend/projects/ui/src/app/pages/server-routes/server-show/server-show.page.ts
@@ -236,7 +236,10 @@ export class ServerShowPage {
         disabled: this.patch
           .watch$('server-info', 'status-info')
           .pipe(
-            map(status => status['backing-up'] || !!status['update-progress']),
+            map(
+              status =>
+                status && (status['backing-up'] || !!status['update-progress']),
+            ),
           ),
       },
     ],
@@ -250,7 +253,17 @@ export class ServerShowPage {
             ? this.updateEos()
             : this.checkForEosUpdate(),
         detail: false,
-        disabled: of(false),
+        disabled: this.patch
+          .watch$('server-info', 'status-info')
+          .pipe(
+            map(
+              status =>
+                status &&
+                (status['backing-up'] ||
+                  !!status['update-progress'] ||
+                  status.updated),
+            ),
+          ),
       },
       {
         title: 'Preferences',