#!/usr/bin/env python3
"""
获取微信ID并发送测试消息
"""

import json
import subprocess
import select


def call_mcp_tool(tool_name: str, params: dict):
    """调用 MCP 工具"""
    mcp_request = {
        "jsonrpc": "2.0",
        "method": "tools/call",
        "params": {
            "name": tool_name,
            "arguments": params
        },
        "id": 1
    }
    
    proc = subprocess.Popen(
        ["bunx", "mcp-wechat-server"],
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        text=True
    )
    
    request_json = json.dumps(mcp_request) + "\n"
    proc.stdin.write(request_json)
    proc.stdin.flush()
    proc.stdin.close()
    
    ready, _, _ = select.select([proc.stdout], [], [], 30)
    
    if ready:
        response_line = proc.stdout.readline()
        proc.terminate()
        
        try:
            response = json.loads(response_line)
            if "result" in response:
                result = response["result"]
                if "content" in result and len(result["content"]) > 0:
                    return result["content"][0].get("text", "")
        except:
            pass
    else:
        proc.terminate()
    
    return None


def main():
    print("="*60)
    print("🔍 获取微信联系人列表")
    print("="*60)
    
    # 尝试获取联系人
    result = call_mcp_tool("get_contacts", {})
    
    if result:
        print(f"\n✅ 获取成功:")
        print(result)
    else:
        print("\n❌ 无法获取联系人")
        print("\n尝试直接发送消息到指定微信号...")
        
        # 直接尝试发送消息
        wechat_id = "w402468w"
        message = "🧪 这是来自 bbav20311 的测试消息！如果收到，说明微信集成成功！"
        
        print(f"\n💬 发送消息到 {wechat_id}...")
        result = call_mcp_tool("send_text_message", {
            "user_id": wechat_id,
            "message": message
        })
        
        if result:
            print(f"✅ 发送结果: {result}")
        else:
            print("❌ 发送失败")


if __name__ == "__main__":
    main()