1// Set this binding to post real-time actions into the trainer.
2@State private var trainerAction: [String: Any]? = nil
3
4kit.createTrainerChatView(
5    user: UserDetails(age: 32, height: 178, weight: 76, gender: .Male, lifestyle: .Active), // or nil
6    style: IStyle(style: "dark"),
7    isLoading: $isLoading,
8    customParams: [
9        "isSubscribed": SubscriptionManager.shared.isActive
10    ],
11    workoutAction: $trainerAction,     // KinesteXAIKit ≥ 1.1.4
12    onMessageReceived: handleMessage
13)

val webView = KinesteXSDK.createTrainerChatView(
    context = this,
    user = UserDetails(age = 32, height = 178, weight = 76, gender = Gender.MALE, lifestyle = Lifestyle.ACTIVE), // or null
    style = IStyle(style = "dark"),
    customParams = mapOf(
        "isSubscribed" to subscriptionManager.isActive
    ),
    isLoading = isLoading,
    onMessageReceived = ::handleWebViewMessage,
    permissionHandler = this
)

// Include isSubscribed in the postData you send after kinestex_loaded.
const postData = {
  userId: "user-123",
  company: "YOUR_COMPANY",
  key: "YOUR_API_KEY",          // or session: "<session_id>" (recommended)
  integration: "AI_TRAINER_CHAT",
  isSubscribed: subscriptionIsActive,   // only an explicit false gates generation
};
iframe.contentWindow.postMessage(JSON.stringify(postData), "https://ai.kinestex.com");

1private func handleMessage(_ message: KinestexMessage) {
2    switch message {
3    case .exit_kinestex:
4        showKinesteX = false
5    case .custom_type(let data):
6        guard let type = data["type"] as? String else { return }
7        if type == "open_subscription_flow" {
8            presentSubscriptionFlow()
9        }
10    default:
11        break
12    }
13}
14
15func presentSubscriptionFlow() {
16    let paywall = PaywallViewController()
17    paywall.onPurchaseCompleted = { [weak self] in
18        self?.trainerAction = ["subscription_result": "purchased"] // generation resumes
19        paywall.dismiss(animated: true)
20    }
21    paywall.onDismissed = { [weak self] in
22        self?.trainerAction = ["subscription_result": "dismissed"] // trainer resumes
23    }
24    present(paywall, animated: true)
25}

private fun handleWebViewMessage(message: WebViewMessage) {
    when (message) {
        is WebViewMessage.ExitKinestex -> closeTrainer()
        is WebViewMessage.CustomType -> when (message.data["type"] as? String) {
            "open_subscription_flow" -> presentSubscriptionFlow()
            else -> Unit
        }
        else -> Unit
    }
}

private fun presentSubscriptionFlow() {
    PaywallSheet(
        onPurchaseCompleted = {
            KinesteXWebViewController.getInstance()
                .sendAction("subscription_result", "purchased") // generation resumes
        },
        onDismissed = {
            KinesteXWebViewController.getInstance()
                .sendAction("subscription_result", "dismissed") // trainer resumes
        }
    ).show(supportFragmentManager, "paywall")
}

window.addEventListener("message", (event) => {
  if (event.origin !== "https://ai.kinestex.com") return;
  const message = JSON.parse(event.data);
  if (message.type === "open_subscription_flow") {
    openPaywall({
      onPurchased: () =>
        iframe.contentWindow.postMessage(
          JSON.stringify({ subscription_result: "purchased" }),
          "https://ai.kinestex.com"
        ),
      onDismissed: () =>
        iframe.contentWindow.postMessage(
          JSON.stringify({ subscription_result: "dismissed" }),
          "https://ai.kinestex.com"
        ),
    });
  }
});