From 5baea84a6a11c66aa7bcb3f398ba033087e5d305 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Thu, 19 Sep 2019 10:52:11 -0400
Subject: [PATCH] restore and error messages on inapp purchase

---
 src/app/accounts/premium.component.html |  4 ++
 src/app/accounts/premium.component.ts   | 68 +++++++++++++++++++------
 src/locales/en/messages.json            |  6 +++
 3 files changed, 62 insertions(+), 16 deletions(-)

diff --git a/src/app/accounts/premium.component.html b/src/app/accounts/premium.component.html
index 1327efa86e..f8be3bc76b 100644
--- a/src/app/accounts/premium.component.html
+++ b/src/app/accounts/premium.component.html
@@ -55,6 +55,10 @@
                     [disabled]="purchaseBtn.loading" [appApiAction]="purchasePromise">
                     <b>{{'premiumPurchase' | i18n}}</b>
                 </button>
+                <button #restoreBtn type="button" class="primary" appBlurClick (click)="restore()"
+                    *ngIf="canRestorePurchase" [disabled]="restoreBtn.loading" [appApiAction]="restorePromise">
+                    <b>{{'premiumRestore' | i18n}}</b>
+                </button>
                 <button type="button" data-dismiss="modal">{{'close' | i18n}}</button>
                 <div class="right" *ngIf="!isPremium">
                     <button #refreshBtn type="button" appBlurClick (click)="refresh()" [disabled]="refreshBtn.loading"
diff --git a/src/app/accounts/premium.component.ts b/src/app/accounts/premium.component.ts
index c55c22d1e1..7cb9a95669 100644
--- a/src/app/accounts/premium.component.ts
+++ b/src/app/accounts/premium.component.ts
@@ -27,7 +27,9 @@ import { Utils } from 'jslib/misc/utils';
 })
 export class PremiumComponent extends BasePremiumComponent {
     purchasePromise: Promise<any>;
+    restorePromise: Promise<any>;
     canMakeMacAppStorePayments = false;
+    canRestorePurchase = false;
 
     constructor(i18nService: I18nService, platformUtilsService: PlatformUtilsService,
         tokenService: TokenService, apiService: ApiService,
@@ -45,6 +47,7 @@ export class PremiumComponent extends BasePremiumComponent {
         if (!this.canMakeMacAppStorePayments) {
             return;
         }
+        this.setCanRestorePurchase();
         remote.inAppPurchase.on('transactions-updated', (event, transactions) => {
             this.ngZone.run(async () => {
                 if (!Array.isArray(transactions)) {
@@ -64,27 +67,15 @@ export class PremiumComponent extends BasePremiumComponent {
                             if (payment.productIdentifier !== 'premium_annually') {
                                 return;
                             }
-                            const receiptUrl = remote.inAppPurchase.getReceiptURL();
-                            const receiptBuffer = fs.readFileSync(receiptUrl);
-                            const receiptB64 = Utils.fromBufferToB64(receiptBuffer);
-                            const fd = new FormData();
-                            fd.append('paymentMethodType', '6');
-                            fd.append('paymentToken', receiptB64);
-                            fd.append('additionalStorageGb', '0');
-                            try {
-                                this.purchasePromise = this.apiService.postPremium(fd).then((paymentResponse) => {
-                                    if (paymentResponse.success) {
-                                        return this.finalizePremium();
-                                    }
-                                });
-                                await this.purchasePromise;
-                            } catch { }
+                            await this.makePremium(this.purchasePromise);
                             // Finish the transaction.
                             remote.inAppPurchase.finishTransactionByDate(transaction.transactionDate);
                             break;
                         case 'failed':
                             // tslint:disable-next-line
-                            console.log(`Failed to purchase ${payment.productIdentifier}.`);
+                            console.log(`Failed to purchase ${payment.productIdentifier}.` +
+                                `${transaction.errorCode} = ${transaction.errorMessage}`);
+                            this.platformUtilsService.showToast('error', null, transaction.errorMessage);
                             // Finish the transaction.
                             remote.inAppPurchase.finishTransactionByDate(transaction.transactionDate);
                             break;
@@ -122,11 +113,56 @@ export class PremiumComponent extends BasePremiumComponent {
         } catch { }
     }
 
+    async restore() {
+        if (this.isPremium || !this.canMakeMacAppStorePayments) {
+            return;
+        }
+        let makePremium = false;
+        try {
+            const request = new IapCheckRequest();
+            request.paymentMethodType = PaymentMethodType.AppleInApp;
+            this.restorePromise = this.apiService.postIapCheck(request);
+            await this.restorePromise;
+            makePremium = true;
+        } catch { }
+        if (makePremium) {
+            await this.makePremium(this.restorePromise);
+        }
+    }
+
+    private async makePremium(promise: Promise<any>) {
+        const receiptUrl = remote.inAppPurchase.getReceiptURL();
+        const receiptBuffer = fs.readFileSync(receiptUrl);
+        const receiptB64 = Utils.fromBufferToB64(receiptBuffer);
+        const fd = new FormData();
+        fd.append('paymentMethodType', '6');
+        fd.append('paymentToken', receiptB64);
+        fd.append('additionalStorageGb', '0');
+        try {
+            promise = this.apiService.postPremium(fd).then((paymentResponse) => {
+                if (paymentResponse.success) {
+                    return this.finalizePremium();
+                }
+            });
+            await promise;
+        } catch { }
+    }
+
     private async finalizePremium() {
         await this.apiService.refreshIdentityToken();
         await this.syncService.fullSync(true);
         this.platformUtilsService.showToast('success', null, this.i18nService.t('premiumUpdated'));
         this.messagingService.send('purchasedPremium');
         this.isPremium = this.tokenService.getPremium();
+        this.setCanRestorePurchase();
+    }
+
+    private setCanRestorePurchase() {
+        if (!this.isPremium && this.canMakeMacAppStorePayments) {
+            const receiptUrl = remote.inAppPurchase.getReceiptURL();
+            this.canRestorePurchase = receiptUrl != null;
+        } else {
+            this.canRestorePurchase = false;
+        }
     }
 }
diff --git a/src/locales/en/messages.json b/src/locales/en/messages.json
index eea58fb34b..10f10263a3 100644
--- a/src/locales/en/messages.json
+++ b/src/locales/en/messages.json
@@ -983,6 +983,9 @@
   "premiumPurchase": {
     "message": "Purchase Premium"
   },
+  "premiumRestore": {
+    "message": "Restore Premium"
+  },
   "premiumPurchaseAlert": {
     "message": "You can purchase premium membership on the bitwarden.com web vault. Do you want to visit the website now?"
   },
@@ -1248,5 +1251,8 @@
   },
   "selectOneCollection": {
     "message": "You must select at least one collection."
+  },
+  "premiumUpdated": {
+    "message": "You've upgraded to premium."
   }
 }