[iOS] - Inject VPN subscription info on script message

ios/brave-ios/Sources/Brave/BraveSkus/BraveSkusAccountLink.swift

@@ -12,7 +12,7 @@ import os.log
 
 extension BraveStoreProduct {
   /// The key to use when storing the receipt in WebKit's LocalStorage
-  fileprivate var localStorageKey: String {
+  var localStorageKey: String {
     switch self {
     case .vpnMonthly, .vpnYearly: return "braveVpn.receipt"
     case .leoMonthly, .leoYearly: return "braveLeo.receipt"

ios/brave-ios/Sources/Brave/Frontend/UserContent/UserScripts/Scripts_Dynamic/ScriptHandlers/Paged/BraveSkusScriptHandler.swift

@@ -6,8 +6,10 @@
 import AIChat
 import BraveCore
 import BraveShared
+import BraveStore
 import BraveVPN
 import Foundation
+import Preferences
 import Shared
 import WebKit
 import os.log
@@ -36,8 +38,7 @@ class BraveSkusScriptHandler: TabContentScript {
 
     return WKUserScript(
       source: secureScript(
-        handlerNamesMap: ["$<message_handler>": messageHandlerName]
-          .merging(Method.map, uniquingKeysWith: { $1 }),
+        handlerNamesMap: ["$<message_handler>": messageHandlerName],
         securityToken: scriptId,
         script: script
       ),
@@ -109,6 +110,30 @@ class BraveSkusScriptHandler: TabContentScript {
     case .credentialsSummary:
       let summary = try CredentialSummaryMessage.from(message: message)
       return await skusManager.credentialSummary(for: summary.domain)
+
+    case .setLocalStorageReceipt:
+      let storeMessage = try StoreReceiptMessage.from(message: message)
+      if storeMessage.message == "vpn" {
+        if let vpnSubscriptionProductId = Preferences.VPN.subscriptionProductId.value,
+          let product = BraveStoreProduct(rawValue: vpnSubscriptionProductId)
+        {
+          let storageKey = product.localStorageKey
+          let receipt = try BraveSkusSDK.receipt(for: product)
+          return ["key": storageKey, "data": receipt]
+        }
+      }
+
+      if storeMessage.message == "leo" {
+        if let aiChatSubscriptionProductId = Preferences.AIChat.subscriptionProductId.value,
+          let orderId = Preferences.AIChat.subscriptionOrderId.value,
+          let product = BraveStoreProduct(rawValue: aiChatSubscriptionProductId)
+        {
+          let storageKey = product.localStorageKey
+          let receipt = try BraveSkusSDK.receipt(for: product)
+          return ["key": storageKey, "data": receipt, "braveLeo.orderId": orderId]
+        }
+      }
+      throw SkusWebMessageError.invalidFormat
     }
   }
 }
@@ -119,6 +144,7 @@ extension BraveSkusScriptHandler {
     case fetchOrderCredentials = 2
     case prepareCredentialsPresentation = 3
     case credentialsSummary = 4
+    case setLocalStorageReceipt = 5
 
     static var map: [String: String] {
       var jsDict = [String: String]()
@@ -141,6 +167,10 @@ extension BraveSkusScriptHandler {
   private struct CredentialSummaryMessage: SkusWebMessage {
     let domain: String
   }
+
+  private struct StoreReceiptMessage: SkusWebMessage {
+    let message: String
+  }
 }
 
 private enum SkusWebMessageError: Error {

ios/brave-ios/Sources/Brave/Frontend/UserContent/UserScripts/Scripts_Dynamic/Scripts/DomainSpecific/Paged/BraveSkusScript.js

@@ -17,6 +17,31 @@ window.__firefox__.includeOnce("BraveSkusScript", function($) {
   if (!window.chrome) {
     window.chrome = {};
   }
+
+  // Request VPN Receipt
+  sendMessage($<setLocalStorageReceipt>, { "message": "vpn" })
+  .then(function(jsonData) {
+    if (jsonData) {
+      window.localStorage.setItem(jsonData["key"], jsonData["data"]);
+    } else {
+      console.error("No VPN Subscription Data");
+    }
+  }).catch(function(err) {
+    console.error(`An error occurred setting the local storage receipt: ${err}`);
+  });
+
+  // Request Leo Receipt
+  sendMessage($<setLocalStorageReceipt>, { "message": "leo" })
+  .then(function(jsonData) {
+    if (jsonData) {
+      window.localStorage.setItem("braveLeo.orderId", jsonData["braveLeo.orderId"])
+      window.localStorage.setItem(jsonData["key"], jsonData["data"]);
+    } else {
+      console.error("No Leo Subscription Data");
+    }
+  }).catch(function(err) {
+    console.error(`An error occurred setting the local storage receipt: ${err}`);
+  });
 
   Object.defineProperty(window.chrome, 'braveSkus', {
     enumerable: false,

ios/brave-ios/Sources/BraveVPN/Components/BuyVPN/Paywall/BraveVPNPaywallView.swift

@@ -202,9 +202,9 @@ public struct BraveVPNPaywallView: View {
             .font(.body.weight(.semibold))
             .foregroundColor(Color(.white))
             .padding()
-            .frame(maxWidth: .infinity)
           }
         }
+        .frame(maxWidth: .infinity)
         .background(
           LinearGradient(
             gradient: