From 2faa38fb87e4066ae72022f5b1926f7767bd4836 Mon Sep 17 00:00:00 2001
From: Aaron Liang <76561968+AaronL725@users.noreply.github.com>
Date: Tue, 14 Jul 2026 20:44:09 +0800
Subject: [PATCH] fix(auth): add GUI CPA hook patch script

---
 tools/apply_gui_cpa_hook.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 tools/apply_gui_cpa_hook.py

diff --git a/tools/apply_gui_cpa_hook.py b/tools/apply_gui_cpa_hook.py
new file mode 100644
index 0000000..04337f9
--- /dev/null
+++ b/tools/apply_gui_cpa_hook.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+import ast
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+APP_PATH = ROOT / "grok_register_ttk.py"
+
+app = APP_PATH.read_text(encoding="utf-8-sig")
+old = '                    add_token_to_grok2api_pools(sso, email=email, log_callback=self.log)\n                    self.success_count += 1'
+new = '''                    add_token_to_grok2api_pools(sso, email=email, log_callback=self.log)
+                    maybe_export_cpa_xai_after_success(
+                        email=email,
+                        password=profile.get("password", ""),
+                        sso=sso,
+                        log_callback=self.log,
+                        cancel_callback=self.should_stop,
+                    )
+                    self.success_count += 1'''
+if 'maybe_export_cpa_xai_after_success(' not in app.split('class CliStopController', 1)[0]:
+    if old not in app:
+        raise RuntimeError("GUI CPA hook anchor not found")
+    app = app.replace(old, new, 1)
+ast.parse(app)
+APP_PATH.write_text(app, encoding="utf-8-sig")
+print("GUI CPA hook applied.")