import frida
import sys
import time

# Connect to USB device
device = frida.get_usb_device(timeout=10)
print(f"Connected to: {device.name}")

# Attach to running WeChat
session = device.attach("com.tencent.mm")
print(f"Attached to com.tencent.mm")

# Read JS script
with open("/Users/yibo/Documents/aipcrm/wechat_plain/scripts/frida-daemon/debug_builder_class.js", "r") as f:
    script_source = f.read()

# Load script
script = session.create_script(script_source)

def on_message(message, data):
    if message["type"] == "send":
        print(f"[Script] {message['payload']}")
    elif message["type"] == "error":
        print(f"[Error] {message['stack']}")

script.on("message", on_message)
script.load()
print("Script loaded, waiting for output...")

# Wait for script to complete
time.sleep(8)

# Cleanup
session.detach()
print("Done.")