fix: when grace period is over select the enrollment type based on ce…

…rtificate availability (#16599)
wireapp · Jan 23, 2024 · 86903da · 86903da
1 parent 75cb794
commit 86903da
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/script/E2EIdentity/E2EIdentityEnrollment.ts b/src/script/E2EIdentity/E2EIdentityEnrollment.ts
@@ -128,8 +128,8 @@ export class E2EIHandler extends TypedEventEmitter<Events> {
       gracePeriodInMs,
       timer: new SnoozableTimer({
         gracePeriodInMS: gracePeriodInMs,
-        onGracePeriodExpired: () => this.startEnrollment(ModalType.ENROLL),
-        onSnoozeExpired: () => this.startEnrollment(ModalType.ENROLL),
+        onGracePeriodExpired: () => this.processEnrollmentUponExpiry(),
+        onSnoozeExpired: () => this.processEnrollmentUponExpiry(),
       }),
     };
 
@@ -138,6 +138,12 @@ export class E2EIHandler extends TypedEventEmitter<Events> {
     return this;
   }
 
+  private async processEnrollmentUponExpiry() {
+    const hasCertificate = await hasActiveCertificate();
+    const enrollmentType = hasCertificate ? ModalType.CERTIFICATE_RENEWAL : ModalType.ENROLL;
+    await this.startEnrollment(enrollmentType);
+  }
+
   public async attemptEnrollment(): Promise<void> {
     const hasCertificate = await hasActiveCertificate();
     if (hasCertificate) {