fix: add clean public compatibility patcher

2026-07-15 02:09:36 +08:00
parent 0e21eb7673
commit aed4f69de6
1 changed files with 130 additions and 0 deletions
@@ -0,0 +1,130 @@
+#!/usr/bin/env python3
+from pathlib import Path
+
+path = Path(__file__).resolve().with_name("apply_full_safe_modularization.py")
+text = path.read_text(encoding="utf-8")
+
+
+def replace_once(source, old, new, label):
+    count = source.count(old)
+    if count != 1:
+        raise RuntimeError(f"{label}: expected one match, got {count}")
+    return source.replace(old, new, 1)
+
+
+# Keep monkeypatch compatibility for generate_username after moving mail code.
+text = replace_once(
+    text,
+    '''def _bind_mail_service():
+    _mail_service.bind_runtime(globals())
+''',
+    '''def _bind_mail_service():
+    _mail_service.bind_runtime(globals())
+    _current = globals().get("generate_username")
+    _standard = _MAIL_COMPAT_PROXIES.get("generate_username")
+    if _current is not None and _current is not _standard:
+        _mail_service.generate_username = _current
+    elif _standard is not None:
+        _mail_service.generate_username = _MAIL_ORIGINALS["generate_username"]
+''',
+    "mail binder",
+)
+
+text = replace_once(
+    text,
+    '''for _name in {MAIL_NAMES!r}:
+    globals()[_name] = _make_compat_proxy(_mail_service, _name, _bind_mail_service)
+for _name in {REGISTRATION_NAMES!r}:
+''',
+    '''_MAIL_ORIGINALS = dict((name, getattr(_mail_service, name)) for name in {MAIL_NAMES!r})
+_MAIL_COMPAT_PROXIES = dict()
+for _name in {MAIL_NAMES!r}:
+    _proxy = _make_compat_proxy(_mail_service, _name, _bind_mail_service)
+    _MAIL_COMPAT_PROXIES[_name] = _proxy
+    globals()[_name] = _proxy
+for _name in {REGISTRATION_NAMES!r}:
+''',
+    "mail proxy registry",
+)
+
+# Preserve cf_mail_debug.requests and create_address for tests/external callers.
+text = replace_once(
+    text,
+    '''import argparse
+import time
+
+from mail_service import CloudflareMailClient, extract_verification_code
+
+
+def main():
+''',
+    '''import argparse
+import time
+
+from curl_cffi import requests
+from mail_service import CloudflareMailClient, extract_verification_code
+
+
+def create_address(api_base, auth_mode="none", api_key="", create_path="/api/new_address",
+                   domain="", name="", timeout=20):
+    import mail_service as _mail_service
+    client = CloudflareMailClient(
+        api_base, auth_mode=auth_mode, api_key=api_key,
+        create_path=create_path, timeout=timeout,
+    )
+    original_requests = _mail_service.requests
+    _mail_service.requests = requests
+    try:
+        return client.create_address(domain=domain, name=name)
+    finally:
+        _mail_service.requests = original_requests
+
+
+def main():
+''',
+    "debug public API",
+)
+
+text = replace_once(
+    text,
+    '''        address, credential = client.create_address(domain=args.domain, name=args.name)
+''',
+    '''        address, credential = create_address(
+            args.api_base,
+            auth_mode=args.auth_mode,
+            api_key=args.api_key,
+            create_path=args.create_path,
+            domain=args.domain,
+            name=args.name,
+        )
+''',
+    "debug create call",
+)
+
+# External cpa_xai packages may use relative imports; register before execution.
+text = replace_once(
+    text,
+    "import shutil\nimport time\nfrom pathlib import Path\n",
+    "import shutil\nimport sys\nimport time\nfrom pathlib import Path\n",
+    "cpa sys import",
+)
+text = replace_once(
+    text,
+    '''    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
+    return module.mint_and_export
+''',
+    '''    module = importlib.util.module_from_spec(spec)
+    sys.modules[module_name] = module
+    try:
+        spec.loader.exec_module(module)
+    except Exception:
+        sys.modules.pop(module_name, None)
+        raise
+    return module.mint_and_export
+''',
+    "cpa external package load",
+)
+
+path.write_text(text, encoding="utf-8")
+print("public compatibility patch applied")