#!/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.")