feat: add IAP in native bridge plugin for iOS (#1673)

apps/readest-app/src-tauri/plugins/tauri-plugin-native-bridge/ios/Sources/NativeBridgePlugin.swift

@@ -3,6 +3,7 @@ import AuthenticationServices
 import CoreText
 import MediaPlayer
 import ObjectiveC
+import StoreKit
 import SwiftRs
 import Tauri
 import UIKit
@@ -48,6 +49,37 @@ class LockScreenOrientationRequestArgs: Decodable {
   let orientation: String?
 }
 
+struct InitializeRequest: Decodable {
+  let publicKey: String?
+}
+
+struct FetchProductsRequest: Decodable {
+  let productIds: [String]
+}
+
+struct PurchaseProductRequest: Decodable {
+  let productId: String
+}
+
+struct ProductData: Codable {
+  let id: String
+  let title: String
+  let description: String
+  let price: String
+  let priceCurrencyCode: String?
+  let priceAmountMicros: Int64
+  let productType: String
+}
+
+struct PurchaseData: Codable {
+  let productId: String
+  let transactionId: String
+  let originalTransactionId: String
+  let purchaseDate: String
+  let purchaseState: String
+  let platform: String
+}
+
 class VolumeKeyHandler: NSObject {
   private var audioSession: AVAudioSession?
   private var originalVolume: Float = 0.0
@@ -422,6 +454,85 @@ class NativeBridgePlugin: Plugin {
       }
     }
   }
+
+  @objc public func iap_initialize(_ invoke: Invoke) {
+    StoreKitManager.shared.initialize()
+    invoke.resolve(["success": true])
+  }
+
+  @objc public func iap_fetch_products(_ invoke: Invoke) {
+    do {
+      let args = try invoke.parseArgs(FetchProductsRequest.self)
+
+      StoreKitManager.shared.fetchProducts(productIds: args.productIds) { products in
+        let productsData: [ProductData] = products.map { product in
+          return ProductData(
+            id: product.productIdentifier,
+            title: product.localizedTitle,
+            description: product.localizedDescription,
+            price: product.price.stringValue,
+            priceCurrencyCode: product.priceLocale.currencyCode,
+            priceAmountMicros: Int64(product.price.doubleValue * 1_000_000),
+            productType: product.productIdentifier.contains("monthly")
+              || product.productIdentifier.contains("yearly") ? "subscription" : "consumable"
+          )
+        }
+        invoke.resolve(["products": productsData])
+      }
+    } catch {
+      invoke.reject("Failed to parse fetch products arguments: \(error.localizedDescription)")
+    }
+  }
+
+  @objc public func iap_purchase_product(_ invoke: Invoke) {
+    do {
+      let args = try invoke.parseArgs(PurchaseProductRequest.self)
+
+      StoreKitManager.shared.fetchProducts(productIds: [args.productId]) { products in
+        guard let product = products.first else {
+          invoke.reject("Product not found")
+          return
+        }
+
+        StoreKitManager.shared.purchase(product: product) { result in
+          switch result {
+          case .success(let txn):
+            let purchase = PurchaseData(
+              productId: txn.payment.productIdentifier,
+              transactionId: txn.transactionIdentifier ?? "",
+              originalTransactionId: txn.original?.transactionIdentifier ?? txn
+                .transactionIdentifier ?? "",
+              purchaseDate: ISO8601DateFormatter().string(from: txn.transactionDate ?? Date()),
+              purchaseState: "purchased",
+              platform: "ios"
+            )
+            invoke.resolve(["purchase": purchase])
+          case .failure(let error):
+            invoke.reject("Purchase failed: \(error.localizedDescription)")
+          }
+        }
+      }
+    } catch {
+      invoke.reject("Failed to parse purchase arguments: \(error.localizedDescription)")
+    }
+  }
+
+  @objc public func iap_restore_purchases(_ invoke: Invoke) {
+    StoreKitManager.shared.restorePurchases { transactions in
+      let restored = transactions.map { txn -> PurchaseData in
+        return PurchaseData(
+          productId: txn.payment.productIdentifier,
+          transactionId: txn.transactionIdentifier ?? "",
+          originalTransactionId: txn.original?.transactionIdentifier ?? txn.transactionIdentifier
+            ?? "",
+          purchaseDate: ISO8601DateFormatter().string(from: txn.transactionDate ?? Date()),
+          purchaseState: "restored",
+          platform: "ios"
+        )
+      }
+      invoke.resolve(["purchases": restored])
+    }
+  }
 }
 
 @_cdecl("init_plugin_native_bridge")

apps/readest-app/src-tauri/plugins/tauri-plugin-native-bridge/ios/Sources/StoreKitManager.swift

@@ -0,0 +1,121 @@
+import StoreKit
+import os
+
+private let logger = Logger(subsystem: Bundle.main.bundleIdentifier!, category: "StoreKitManager")
+
+class StoreKitManager: NSObject, SKProductsRequestDelegate, SKPaymentTransactionObserver {
+  static let shared = StoreKitManager()
+
+  private var productsRequest: SKProductsRequest?
+  private var productResponseHandler: (([SKProduct]) -> Void)?
+  private var purchaseHandler: ((Result<SKPaymentTransaction, Error>) -> Void)?
+  private var restoreHandler: (([SKPaymentTransaction]) -> Void)?
+
+  private override init() {
+    super.init()
+  }
+
+  func initialize() {
+    SKPaymentQueue.default().add(self)
+  }
+
+  deinit {
+    SKPaymentQueue.default().remove(self)
+  }
+
+  func fetchProducts(productIds: [String], completion: @escaping ([SKProduct]) -> Void) {
+    let ids = Set(productIds)
+    productsRequest = SKProductsRequest(productIdentifiers: ids)
+    productsRequest?.delegate = self
+    productResponseHandler = completion
+    productsRequest?.start()
+  }
+
+  func purchase(
+    product: SKProduct, completion: @escaping (Result<SKPaymentTransaction, Error>) -> Void
+  ) {
+    guard SKPaymentQueue.canMakePayments() else {
+      completion(
+        .failure(
+          NSError(
+            domain: "iap", code: 0, userInfo: [NSLocalizedDescriptionKey: "Purchases disabled."])))
+      return
+    }
+    purchaseHandler = completion
+    let payment = SKPayment(product: product)
+    SKPaymentQueue.default().add(payment)
+  }
+
+  func restorePurchases(completion: @escaping ([SKPaymentTransaction]) -> Void) {
+    restoreHandler = completion
+    SKPaymentQueue.default().restoreCompletedTransactions()
+  }
+
+  // MARK: - SKProductsRequestDelegate
+  func productsRequest(_ request: SKProductsRequest, didReceive response: SKProductsResponse) {
+    DispatchQueue.main.async { [weak self] in
+      self?.productResponseHandler?(response.products)
+      self?.productResponseHandler = nil
+    }
+  }
+
+  func request(_ request: SKRequest, didFailWithError error: Error) {
+    DispatchQueue.main.async { [weak self] in
+      logger.error("Products request failed: \(error.localizedDescription)")
+      self?.productResponseHandler?([])
+      self?.productResponseHandler = nil
+    }
+  }
+
+  // MARK: - SKPaymentTransactionObserver
+  func paymentQueue(
+    _ queue: SKPaymentQueue, updatedTransactions transactions: [SKPaymentTransaction]
+  ) {
+    for transaction in transactions {
+      switch transaction.transactionState {
+      case .purchased:
+        SKPaymentQueue.default().finishTransaction(transaction)
+        DispatchQueue.main.async { [weak self] in
+          self?.purchaseHandler?(.success(transaction))
+          self?.purchaseHandler = nil
+        }
+      case .failed:
+        SKPaymentQueue.default().finishTransaction(transaction)
+        let error =
+          transaction.error
+          ?? NSError(
+            domain: "iap", code: -1, userInfo: [NSLocalizedDescriptionKey: "Unknown purchase error"]
+          )
+        DispatchQueue.main.async { [weak self] in
+          self?.purchaseHandler?(.failure(error))
+          self?.purchaseHandler = nil
+        }
+      case .restored:
+        SKPaymentQueue.default().finishTransaction(transaction)
+      case .deferred, .purchasing:
+        // Handle these states if needed
+        break
+      @unknown default:
+        break
+      }
+    }
+  }
+
+  func paymentQueueRestoreCompletedTransactionsFinished(_ queue: SKPaymentQueue) {
+    let restored = queue.transactions.filter { $0.transactionState == .restored }
+    DispatchQueue.main.async { [weak self] in
+      self?.restoreHandler?(restored)
+      self?.restoreHandler = nil
+    }
+  }
+
+  func paymentQueue(
+    _ queue: SKPaymentQueue, restoreCompletedTransactionsFailedWithError error: Error
+  ) {
+    DispatchQueue.main.async { [weak self] in
+      logger.error("Restore purchases failed: \(error.localizedDescription)")
+      self?.restoreHandler?([])
+      self?.restoreHandler = nil
+    }
+  }
+}