Add methods to handle subscriptions for Daemon's AppUpgradeEvent stream

1 files changed, 34 insertions, 0 deletions

diff --git a/desktop/packages/mullvad-vpn/src/main/daemon-rpc.ts b/desktop/packages/mullvad-vpn/src/main/daemon-rpc.ts
index 8169688fcf..a8675f6d24 100644
--- a/desktop/packages/mullvad-vpn/src/main/daemon-rpc.ts
+++ b/desktop/packages/mullvad-vpn/src/main/daemon-rpc.ts
@@ -12,6 +12,7 @@ import {
   BridgeState,
   CustomListError,
   CustomProxy,
+  DaemonAppUpgradeEvent,
   DaemonEvent,
   DeviceState,
   IAppVersionInfo,
@@ -31,6 +32,7 @@ import {
 import { ConnectionObserver, GrpcClient, noConnectionError } from './grpc-client';
 import {
   convertFromApiAccessMethodSetting,
+  convertFromAppUpgradeEvent,
   convertFromDaemonEvent,
   convertFromDevice,
   convertFromDeviceState,
@@ -91,6 +93,38 @@ export class DaemonRpc extends GrpcClient {
     super.disconnect();
   }
 
+  public subscribeAppUpgradeEventListener(listener: SubscriptionListener<DaemonAppUpgradeEvent>) {
+    const call = this.isConnected && this.client.appUpgradeEventsListen(new Empty());
+    if (!call) {
+      throw noConnectionError;
+    }
+    const subscriptionId = this.subscriptionId();
+    listener.subscriptionId = subscriptionId;
+    this.subscriptions.set(subscriptionId, call);
+
+    call.on('data', (data: grpcTypes.AppUpgradeEvent) => {
+      try {
+        const appUpgradeEvent = convertFromAppUpgradeEvent(data);
+        listener.onEvent(appUpgradeEvent);
+      } catch (e) {
+        const error = e as Error;
+        listener.onError(error);
+      }
+    });
+
+    call.on('error', (error) => {
+      listener.onError(error);
+      this.removeSubscription(subscriptionId);
+    });
+  }
+
+  public unsubscribeAppUpgradeEventListener(listener: SubscriptionListener<DaemonAppUpgradeEvent>) {
+    const id = listener.subscriptionId;
+    if (id !== undefined) {
+      this.removeSubscription(id);
+    }
+  }
+
   public subscribeDaemonEventListener(listener: SubscriptionListener<DaemonEvent>) {
     const call = this.isConnected && this.client.eventsListen(new Empty());
     if (!call) {