1 files changed, 38 insertions, 3 deletions

diff --git a/packages/main/src/stores/Service.ts b/packages/main/src/stores/Service.ts
index d8f3166..1d46dc9 100644
--- a/packages/main/src/stores/Service.ts
+++ b/packages/main/src/stores/Service.ts
@@ -20,12 +20,13 @@
 
 import type { UnreadCount } from '@sophie/service-shared';
 import {
-  CertificateSnapshotIn,
+  type Certificate,
+  type CertificateSnapshotIn,
   defineServiceModel,
   ServiceAction,
-  ServiceStateSnapshotIn,
+  type ServiceStateSnapshotIn,
 } from '@sophie/shared';
-import { Instance, getSnapshot, cast } from 'mobx-state-tree';
+import { type Instance, getSnapshot, cast, flow } from 'mobx-state-tree';
 
 import type { ServiceView } from '../infrastructure/electron/types';
 import { getLogger } from '../utils/log';
@@ -129,6 +130,35 @@ const Service = defineServiceModel(ServiceSettings)
     toggleDeveloperTools(): void {
       self.serviceView?.toggleDeveloperTools();
     },
+    downloadCertificate: flow(function* downloadCertificate(
+      fingerprint: string,
+    ) {
+      const { state } = self;
+      if (state.type !== 'certificateError') {
+        log.warn(
+          'Tried to save certificate',
+          fingerprint,
+          'when there is no certificate error',
+        );
+        return;
+      }
+      let { certificate } = state;
+      while (
+        certificate !== undefined &&
+        certificate.fingerprint !== fingerprint
+      ) {
+        certificate = certificate.issuerCert as Certificate;
+      }
+      if (certificate === undefined) {
+        log.warn(
+          'Tried to save certificate',
+          fingerprint,
+          'which is not part of the current certificate chain',
+        );
+        return;
+      }
+      yield getEnv(self).saveTextFile('certificate.pem', certificate.data);
+    }),
   }))
   .actions((self) => {
     function setState(state: ServiceStateSnapshotIn): void {
@@ -272,6 +302,11 @@ const Service = defineServiceModel(ServiceSettings)
         case 'dismiss-all-popups':
           self.dismissAllPopups();
           break;
+        case 'download-certificate':
+          self.downloadCertificate(action.fingerprint).catch((error) => {
+            log.error('Error while saving certificate', error);
+          });
+          break;
         default:
           log.error('Unknown action to dispatch', action);
           break;