import frida
import sys

# Try USB device first, fallback to TCP
try:
    device = frida.get_usb_device(timeout=5)
    print(f"USB device: {device.name}")
except Exception as e:
    print(f"USB failed: {e}")
    device = frida.get_device_manager().add_remote_device("127.0.0.1:27042")
    print(f"TCP device: {device.name}")

# Try attach by PID directly
pid = 15059  # current WeChat PID
print(f"Attaching to PID {pid}...")
session = device.attach(pid)
print(f"Attached!")

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

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}")

script.on("message", on_message)
script.load()
print("Script loaded. Waiting...")

import time
time.sleep(10)
print("Done.")