From 350dbd04a5778b5592982a7200585bbf9b573777 Mon Sep 17 00:00:00 2001
From: Matt Hill <matthewonthemoon@gmail.com>
Date: Sat, 17 Dec 2022 22:08:45 -0700
Subject: [PATCH] auto reconnect to logs websocket and hopefully fix scrolling
 issues

---
 frontend/projects/shared/styles/shared.scss   |  20 ++-
 .../app/components/logs/logs.component.html   |  20 ++-
 .../src/app/components/logs/logs.component.ts | 144 ++++++++++++------
 .../apps-routes/app-logs/app-logs.page.html   |   2 +-
 .../kernel-logs/kernel-logs.page.html         |   4 +-
 .../server-logs/server-logs.page.html         |   4 +-
 .../services/api/embassy-mock-api.service.ts  |   1 +
 frontend/projects/ui/src/styles.scss          |  18 ---
 8 files changed, 133 insertions(+), 80 deletions(-)

diff --git a/frontend/projects/shared/styles/shared.scss b/frontend/projects/shared/styles/shared.scss
index 59f8f5ca98..62d1571ceb 100644
--- a/frontend/projects/shared/styles/shared.scss
+++ b/frontend/projects/shared/styles/shared.scss
@@ -120,4 +120,22 @@ ion-modal {
 
 .color-primary-shade {
   color: var(--ion-color-primary-shade)
-}
\ No newline at end of file
+}
+
+@keyframes ellipsis-dot {
+  25% {
+    content: '';
+  }
+
+  50% {
+    content: '.';
+  }
+
+  75% {
+    content: '..';
+  }
+
+  100% {
+    content: '...';
+  }
+}
diff --git a/frontend/projects/ui/src/app/components/logs/logs.component.html b/frontend/projects/ui/src/app/components/logs/logs.component.html
index 9040d4108e..231f0347aa 100644
--- a/frontend/projects/ui/src/app/components/logs/logs.component.html
+++ b/frontend/projects/ui/src/app/components/logs/logs.component.html
@@ -3,7 +3,7 @@
     <ion-buttons slot="start">
       <ion-back-button [defaultHref]="defaultBack"></ion-back-button>
     </ion-buttons>
-    <ion-title>{{ title }}</ion-title>
+    <ion-title>{{ pageTitle }}</ion-title>
   </ion-toolbar>
 </ion-header>
 
@@ -15,7 +15,7 @@
 >
   <ion-infinite-scroll
     id="scroller"
-    [disabled]="!needInfinite"
+    [disabled]="infiniteStatus !== 1"
     position="top"
     threshold="1000"
     (ionInfinite)="doInfinite($event)"
@@ -33,10 +33,18 @@
 
   <ng-container *ngIf="!loading">
     <div id="bottom-div"></div>
-    <div *ngIf="websocketFail" class="ion-text-center ion-padding">
-      <ion-text color="warning"> Websocket failed.... </ion-text>
-    </div>
-
+    <p
+      *ngIf="websocketStatus === 'reconnecting'"
+      class="ion-text-center loading-dots"
+    >
+      <ion-text color="success">Reconnecting</ion-text>
+    </p>
+    <p
+      *ngIf="websocketStatus === 'disconnected'"
+      class="ion-text-center loading-dots"
+    >
+      <ion-text color="warning">Waiting for network connectivity</ion-text>
+    </p>
     <div
       [ngStyle]="{
         position: 'fixed',
diff --git a/frontend/projects/ui/src/app/components/logs/logs.component.ts b/frontend/projects/ui/src/app/components/logs/logs.component.ts
index ede7a11650..26acb5182a 100644
--- a/frontend/projects/ui/src/app/components/logs/logs.component.ts
+++ b/frontend/projects/ui/src/app/components/logs/logs.component.ts
@@ -1,6 +1,16 @@
 import { Component, Input, ViewChild } from '@angular/core'
 import { IonContent, LoadingController } from '@ionic/angular'
-import { bufferTime, takeUntil, tap } from 'rxjs'
+import {
+  bufferTime,
+  catchError,
+  filter,
+  finalize,
+  from,
+  Observable,
+  switchMap,
+  takeUntil,
+  tap,
+} from 'rxjs'
 import { WebSocketSubjectConfig } from 'rxjs/webSocket'
 import {
   LogsRes,
@@ -13,6 +23,7 @@ import {
 } from '@start9labs/shared'
 import { RR } from 'src/app/services/api/api.types'
 import { ApiService } from 'src/app/services/api/embassy-api.service'
+import { ConnectionService } from 'src/app/services/connection.service'
 
 var Convert = require('ansi-to-html')
 var convert = new Convert({
@@ -40,15 +51,20 @@ export class LogsComponent {
   @Input() fetchLogs!: (params: ServerLogsReq) => Promise<LogsRes>
   @Input() context!: string
   @Input() defaultBack!: string
-  @Input() title!: string
+  @Input() pageTitle!: string
 
   loading = true
-  needInfinite = false
+  infiniteStatus: 0 | 1 | 2 = 0
   startCursor?: string
   isOnBottom = true
   autoScroll = true
-  websocketFail = false
+  websocketStatus:
+    | 'connecting'
+    | 'connected'
+    | 'reconnecting'
+    | 'disconnected' = 'connecting'
   limit = 400
+  count = 0
 
   constructor(
     private readonly errToast: ErrorToastService,
@@ -56,50 +72,20 @@ export class LogsComponent {
     private readonly api: ApiService,
     private readonly loadingCtrl: LoadingController,
     private readonly downloadHtml: DownloadHTMLService,
+    private readonly connectionService: ConnectionService,
   ) {}
 
   async ngOnInit() {
-    try {
-      const { 'start-cursor': startCursor, guid } = await this.followLogs({
-        limit: this.limit,
-      })
-
-      this.startCursor = startCursor
-
-      const config: WebSocketSubjectConfig<Log> = {
-        url: `/rpc/${guid}`,
-        openObserver: {
-          next: () => {
-            this.websocketFail = false
-          },
-        },
-      }
-
-      let totalLogs = 0
-
-      this.api
-        .openLogsWebsocket$(config)
-        .pipe(
-          tap(_ => {
-            totalLogs++
-            if (totalLogs === this.limit) this.needInfinite = true
-          }),
-          bufferTime(500),
-          tap(msgs => {
-            this.loading = false
-            this.processRes({ entries: msgs })
-          }),
-          takeUntil(this.destroy$),
-        )
-        .subscribe({
-          error: () => {
-            this.websocketFail = true
-            if (this.isOnBottom) this.scrollToBottom()
-          },
-        })
-    } catch (e: any) {
-      this.errToast.present(e)
-    }
+    from(this.followLogs({ limit: this.limit }))
+      .pipe(
+        switchMap(({ 'start-cursor': startCursor, guid }) => {
+          this.startCursor = startCursor
+          return this.connect$(guid)
+        }),
+        takeUntil(this.destroy$),
+        finalize(() => console.log('CLOSING')),
+      )
+      .subscribe()
   }
 
   async doInfinite(e: any): Promise<void> {
@@ -119,7 +105,7 @@ export class LogsComponent {
   }
 
   handleScroll(e: any) {
-    if (e.detail.deltaY < 0) this.autoScroll = false
+    if (e.detail.deltaY < -50) this.autoScroll = false
   }
 
   handleScrollEnd() {
@@ -130,7 +116,7 @@ export class LogsComponent {
   }
 
   scrollToBottom() {
-    this.content?.scrollToBottom(250)
+    this.content?.scrollToBottom(200)
   }
 
   async download() {
@@ -160,6 +146,65 @@ export class LogsComponent {
     }
   }
 
+  private reconnect$(): Observable<Log[]> {
+    return from(this.followLogs({})).pipe(
+      tap(_ => this.recordConnectionChange()),
+      switchMap(({ guid }) => this.connect$(guid, true)),
+    )
+  }
+
+  private connect$(guid: string, reconnect = false) {
+    const config: WebSocketSubjectConfig<Log> = {
+      url: `/rpc/${guid}`,
+      openObserver: {
+        next: () => {
+          this.websocketStatus = 'connected'
+        },
+      },
+    }
+
+    return this.api.openLogsWebsocket$(config).pipe(
+      tap(_ => this.count++),
+      bufferTime(1000),
+      tap(msgs => {
+        this.loading = false
+        this.processRes({ entries: msgs })
+        if (this.infiniteStatus === 0 && this.count >= this.limit)
+          this.infiniteStatus = 1
+      }),
+      catchError(() => {
+        this.recordConnectionChange(false)
+        return this.connectionService.connected$.pipe(
+          tap(
+            connected =>
+              (this.websocketStatus = connected
+                ? 'reconnecting'
+                : 'disconnected'),
+          ),
+          filter(Boolean),
+          switchMap(() => this.reconnect$()),
+        )
+      }),
+    )
+  }
+
+  private recordConnectionChange(success = true) {
+    const container = document.getElementById('container')
+    const elem = document.getElementById('template')?.cloneNode()
+    if (!(elem instanceof HTMLElement)) return
+    elem.innerHTML = `<div style="padding: ${
+      success ? '36px 0' : '36px 0 0 0'
+    }; color: ${success ? '#2fdf75' : '#ff4961'}; text-align: center;">${
+      success ? 'Reconnected' : 'Disconnected'
+    } at ${toLocalIsoString(new Date())}</div>`
+    container?.append(elem)
+    if (this.isOnBottom) {
+      setTimeout(() => {
+        this.scrollToBottom()
+      }, 25)
+    }
+  }
+
   private processRes(res: LogsRes) {
     const { entries, 'start-cursor': startCursor } = res
 
@@ -180,7 +225,7 @@ export class LogsComponent {
       container?.prepend(newLogs)
       const afterContainerHeight = container?.scrollHeight || 0
 
-      // scroll down
+      // maintain scroll height
       setTimeout(() => {
         this.content?.scrollToPoint(
           0,
@@ -189,12 +234,11 @@ export class LogsComponent {
       }, 25)
 
       if (entries.length < this.limit) {
-        this.needInfinite = false
+        this.infiniteStatus = 2
       }
     } else {
       container?.append(newLogs)
       if (this.autoScroll) {
-        // scroll to bottom
         setTimeout(() => {
           this.scrollToBottom()
         }, 25)
diff --git a/frontend/projects/ui/src/app/pages/apps-routes/app-logs/app-logs.page.html b/frontend/projects/ui/src/app/pages/apps-routes/app-logs/app-logs.page.html
index 30bdea9646..4617646f50 100644
--- a/frontend/projects/ui/src/app/pages/apps-routes/app-logs/app-logs.page.html
+++ b/frontend/projects/ui/src/app/pages/apps-routes/app-logs/app-logs.page.html
@@ -3,6 +3,6 @@
   [followLogs]="followLogs()"
   [defaultBack]="'/services/' + pkgId"
   [context]="pkgId"
-  title="Service Logs"
+  pageTitle="Service Logs"
   class="ion-page"
 ></logs>
diff --git a/frontend/projects/ui/src/app/pages/server-routes/kernel-logs/kernel-logs.page.html b/frontend/projects/ui/src/app/pages/server-routes/kernel-logs/kernel-logs.page.html
index f37d9c886e..e744ec68f8 100644
--- a/frontend/projects/ui/src/app/pages/server-routes/kernel-logs/kernel-logs.page.html
+++ b/frontend/projects/ui/src/app/pages/server-routes/kernel-logs/kernel-logs.page.html
@@ -2,7 +2,7 @@
   [fetchLogs]="fetchLogs()"
   [followLogs]="followLogs()"
   context="kernel"
-  defaultBack="embassy"
-  title="Kernel Logs"
+  defaultBack="system"
+  pageTitle="Kernel Logs"
   class="ion-page"
 ></logs>
diff --git a/frontend/projects/ui/src/app/pages/server-routes/server-logs/server-logs.page.html b/frontend/projects/ui/src/app/pages/server-routes/server-logs/server-logs.page.html
index 3d84f59680..e9501c2b9a 100644
--- a/frontend/projects/ui/src/app/pages/server-routes/server-logs/server-logs.page.html
+++ b/frontend/projects/ui/src/app/pages/server-routes/server-logs/server-logs.page.html
@@ -2,7 +2,7 @@
   [fetchLogs]="fetchLogs()"
   [followLogs]="followLogs()"
   context="eos"
-  defaultBack="embassy"
-  title="OS Logs"
+  defaultBack="system"
+  pageTitle="OS Logs"
   class="ion-page"
 ></logs>
diff --git a/frontend/projects/ui/src/app/services/api/embassy-mock-api.service.ts b/frontend/projects/ui/src/app/services/api/embassy-mock-api.service.ts
index d511c8a60c..558dc71e28 100644
--- a/frontend/projects/ui/src/app/services/api/embassy-mock-api.service.ts
+++ b/frontend/projects/ui/src/app/services/api/embassy-mock-api.service.ts
@@ -171,6 +171,7 @@ export class MockApiService extends ApiService {
       map((_, index) => {
         // mock fire open observer
         if (index === 0) config.openObserver?.next(new Event(''))
+        if (index === 100) throw new Error('HAAHHA')
         return Mock.ServerLogs[0]
       }),
     )
diff --git a/frontend/projects/ui/src/styles.scss b/frontend/projects/ui/src/styles.scss
index e7bdfb8fcd..b11b0bdcd4 100644
--- a/frontend/projects/ui/src/styles.scss
+++ b/frontend/projects/ui/src/styles.scss
@@ -295,24 +295,6 @@ h2 {
   line-height: unset;
 }
 
-@keyframes ellipsis-dot {
-  25% {
-    content: '';
-  }
-
-  50% {
-    content: '.';
-  }
-
-  75% {
-    content: '..';
-  }
-
-  100% {
-    content: '...';
-  }
-}
-
 @keyframes flickerAnimation {
   0% {
     opacity: 1;