From 9e0d332a1a19ef4ccd40d62dd557edb73516fb24 Mon Sep 17 00:00:00 2001
From: Geomitron <22552797+Geomitron@users.noreply.github.com>
Date: Thu, 21 May 2020 14:17:51 -0400
Subject: [PATCH] Added update ipc handler

---
 electron-builder.json                         |  1 -
 package.json                                  |  2 +-
 .../components/toolbar/toolbar.component.html |  2 +-
 .../components/toolbar/toolbar.component.ts   | 22 +----
 src/app/core/services/download.service.ts     |  1 -
 src/electron/ipc/UpdateHandler.ipc.ts         | 99 +++++++++++++++++++
 src/electron/shared/IPCHandler.ts             | 28 ++++--
 7 files changed, 128 insertions(+), 27 deletions(-)
 create mode 100644 src/electron/ipc/UpdateHandler.ipc.ts

diff --git a/electron-builder.json b/electron-builder.json
index 24ff8c3..a2bcf3a 100644
--- a/electron-builder.json
+++ b/electron-builder.json
@@ -8,7 +8,6 @@
   ],
   "publish": {
     "provider": "github",
-    "private": true,
     "releaseType": "release"
   },
   "nsis": {
diff --git a/package.json b/package.json
index da0bb09..459e90f 100644
--- a/package.json
+++ b/package.json
@@ -10,7 +10,7 @@
   },
   "repository": {
     "type": "git",
-    "url": "https://github.com/Geomitron/Bridge"
+    "url": "https://github.com/Geomitron/Bridge.git"
   },
   "main": "dist/electron/main.js",
   "private": true,
diff --git a/src/app/components/toolbar/toolbar.component.html b/src/app/components/toolbar/toolbar.component.html
index 50c49c3..af50ed0 100644
--- a/src/app/components/toolbar/toolbar.component.html
+++ b/src/app/components/toolbar/toolbar.component.html
@@ -2,7 +2,7 @@
   <a class="item" routerLinkActive="active" routerLink="/browse">Browse</a>
   <a class="item" routerLinkActive="active" routerLink="/library">Library</a>
   <a class="item" routerLinkActive="active" routerLink="/settings">Settings</a>
-  <a class="item" routerLinkActive="active" routerLink="/about">About</a>
+  <a class="item" routerLinkActive="active" routerLink="/about"><i class="teal small circle icon"></i>About</a>
 
   <div class="right menu">
     <a class="item traffic-light" (click)="minimize()"><i class="minus icon"></i></a>
diff --git a/src/app/components/toolbar/toolbar.component.ts b/src/app/components/toolbar/toolbar.component.ts
index 6c05db0..567eabe 100644
--- a/src/app/components/toolbar/toolbar.component.ts
+++ b/src/app/components/toolbar/toolbar.component.ts
@@ -1,22 +1,5 @@
 import { Component, OnInit, ChangeDetectorRef } from '@angular/core'
 import { ElectronService } from '../../core/services/electron.service'
-// import { autoUpdater, UpdateInfo } from 'electron-updater'
-// autoUpdater.autoDownload = false
-// autoUpdater.on('error', (err) => {})
-// autoUpdater.on('checking-for-update', () => {})
-// autoUpdater.on('update-available', (info: UpdateInfo) => {})
-// autoUpdater.on('update-not-available', () => {})
-// autoUpdater.on('download-progress', (progress) => {
-//   console.log(progress.bytesPerSecond, progress.percent, progress.transferred, progress.total)
-// })
-// autoUpdater.on('update-downloaded', (info: UpdateInfo) => {})
-// autoUpdater.currentVersion // TODO: display this in the corner or on the about page?
-// autoUpdater.logger = null
-// autoUpdater.checkForUpdates()
-// autoUpdater.downloadUpdate()
-// autoUpdater.quitAndInstall(false) // By default; autoUpdater installs a downloaded update on the next program restart
-// TODO: check for updates on initialization; show a button indicating a new version can be downloaded
-
 
 @Component({
   selector: 'app-toolbar',
@@ -26,6 +9,7 @@ import { ElectronService } from '../../core/services/electron.service'
 export class ToolbarComponent implements OnInit {
 
   isMaximized: boolean
+  updateAvailable = false
 
   constructor(private electronService: ElectronService, private ref: ChangeDetectorRef) { }
 
@@ -39,6 +23,10 @@ export class ToolbarComponent implements OnInit {
       this.isMaximized = true
       this.ref.detectChanges()
     })
+
+    this.electronService.receiveIPC('update-available', () => {
+      this.updateAvailable = true
+    })
   }
 
   minimize() {
diff --git a/src/app/core/services/download.service.ts b/src/app/core/services/download.service.ts
index 29358ca..8d38ff9 100644
--- a/src/app/core/services/download.service.ts
+++ b/src/app/core/services/download.service.ts
@@ -33,7 +33,6 @@ export class DownloadService {
         const thisDownloadIndex = this.downloads.findIndex(download => download.versionID == result.versionID)
         if (thisDownloadIndex == -1) {
           this.downloads.push(result)
-          // TODO: this.downloads.sort(downloadSorter)
         } else {
           this.downloads[thisDownloadIndex] = result
         }
diff --git a/src/electron/ipc/UpdateHandler.ipc.ts b/src/electron/ipc/UpdateHandler.ipc.ts
new file mode 100644
index 0000000..05c9e41
--- /dev/null
+++ b/src/electron/ipc/UpdateHandler.ipc.ts
@@ -0,0 +1,99 @@
+import { IPCEmitHandler, IPCInvokeHandler } from '../shared/IPCHandler'
+import { autoUpdater, UpdateInfo } from 'electron-updater'
+import { emitIPCEvent } from '../main'
+
+export interface UpdateProgress {
+  bytesPerSecond: number
+  percent: number
+  transferred: number
+  total: number
+}
+
+/**
+ * Checks for updates when the program is launched.
+ */
+class UpdateChecker {
+
+  constructor() {
+    autoUpdater.autoDownload = false
+    autoUpdater.logger = null
+    this.registerUpdaterListeners()
+    autoUpdater.checkForUpdates()
+  }
+
+  private registerUpdaterListeners() {
+    autoUpdater.on('error', (err: Error) => {
+      console.log('token:', process.env.GH_TOKEN)
+      console.log('error callback', err)
+      emitIPCEvent('update-error', err)
+    })
+
+    autoUpdater.on('update-available', (info: UpdateInfo) => {
+      console.log('update available callback', info)
+      emitIPCEvent('update-available', info)
+    })
+  }
+}
+
+new UpdateChecker()
+
+/**
+ * Handles the 'get-current-version' event.
+ */
+class GetCurrentVersionHandler implements IPCInvokeHandler<'get-current-version'> {
+  event: 'get-current-version' = 'get-current-version'
+
+  /**
+   * @returns the current version of Bridge.
+   */
+  handler() {
+    console.log('Printing version:', autoUpdater.currentVersion.raw)
+    return autoUpdater.currentVersion.raw // TODO: display this on the about page
+  }
+}
+
+export const getCurrentVersionHandler = new GetCurrentVersionHandler()
+
+/**
+ * Handles the 'download-update' event.
+ */
+class DownloadUpdateHandler implements IPCEmitHandler<'download-update'> {
+  event: 'download-update' = 'download-update'
+  downloading = false
+
+  /**
+   * Begins the process of downloading the latest update.
+   */
+  handler() {
+    if (this.downloading) { return }
+    this.downloading = true
+
+    autoUpdater.on('download-progress', (updateProgress: UpdateProgress) => {
+      emitIPCEvent('update-progress', updateProgress)
+    })
+
+    autoUpdater.on('update-downloaded', () => {
+      emitIPCEvent('update-downloaded', undefined)
+    })
+
+    autoUpdater.downloadUpdate()
+  }
+}
+
+export const downloadUpdateHandler = new DownloadUpdateHandler()
+
+/**
+ * Handles the 'quit-and-install' event.
+ */
+class QuitAndInstallHandler implements IPCEmitHandler<'quit-and-install'> {
+  event: 'quit-and-install' = 'quit-and-install'
+
+  /**
+   * Immediately closes the application and installs the update.
+   */
+  handler() {
+    autoUpdater.quitAndInstall() // autoUpdater installs a downloaded update on the next program restart by default
+  }
+}
+
+export const quitAndInstallHandler = new QuitAndInstallHandler()
\ No newline at end of file
diff --git a/src/electron/shared/IPCHandler.ts b/src/electron/shared/IPCHandler.ts
index 548b4cd..c4a3539 100644
--- a/src/electron/shared/IPCHandler.ts
+++ b/src/electron/shared/IPCHandler.ts
@@ -1,13 +1,15 @@
 import { SongSearch, SongResult } from './interfaces/search.interface'
 import { VersionResult, AlbumArtResult } from './interfaces/songDetails.interface'
-import { searchHandler } from '../ipc/SearchHandler.ipc'
-import { songDetailsHandler } from '../ipc/SongDetailsHandler.ipc'
-import { albumArtHandler } from '../ipc/AlbumArtHandler.ipc'
+import { searchHandler } from '../ipc/browse/SearchHandler.ipc'
+import { songDetailsHandler } from '../ipc/browse/SongDetailsHandler.ipc'
+import { albumArtHandler } from '../ipc/browse/AlbumArtHandler.ipc'
 import { Download, DownloadProgress } from './interfaces/download.interface'
 import { downloadHandler } from '../ipc/download/DownloadHandler'
 import { Settings } from './Settings'
-import { batchSongDetailsHandler } from '../ipc/BatchSongDetailsHandler.ipc'
+import { batchSongDetailsHandler } from '../ipc/browse/BatchSongDetailsHandler.ipc'
 import { getSettingsHandler, setSettingsHandler } from '../ipc/SettingsHandler.ipc'
+import { UpdateProgress, getCurrentVersionHandler, downloadUpdateHandler, quitAndInstallHandler } from '../ipc/UpdateHandler.ipc'
+import { UpdateInfo } from 'electron-updater'
 
 /**
  * To add a new IPC listener:
@@ -23,7 +25,8 @@ export function getIPCInvokeHandlers(): IPCInvokeHandler<keyof IPCInvokeEvents>[
     searchHandler,
     songDetailsHandler,
     batchSongDetailsHandler,
-    albumArtHandler
+    albumArtHandler,
+    getCurrentVersionHandler
   ]
 }
 
@@ -51,6 +54,10 @@ export type IPCInvokeEvents = {
     input: number[]
     output: VersionResult[]
   }
+  'get-current-version': {
+    input: undefined
+    output: string
+  }
 }
 
 /**
@@ -65,7 +72,9 @@ export interface IPCInvokeHandler<E extends keyof IPCInvokeEvents> {
 export function getIPCEmitHandlers(): IPCEmitHandler<keyof IPCEmitEvents>[] {
   return [
     downloadHandler,
-    setSettingsHandler
+    setSettingsHandler,
+    downloadUpdateHandler,
+    quitAndInstallHandler
   ]
 }
 
@@ -77,6 +86,13 @@ export type IPCEmitEvents = {
   'download-updated': DownloadProgress
   'set-settings': Settings
   'queue-updated': number[]
+
+  'update-error': Error
+  'update-available': UpdateInfo
+  'update-progress': UpdateProgress
+  'update-downloaded': undefined
+  'download-update': undefined
+  'quit-and-install': undefined
 }
 
 /**