#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
直接使用 HTTP 请求调用小红书 MCP
"""

import json
import requests

MCP_URL = "http://localhost:18060/mcp"

# 小红书 cookies
COOKIES = "acw_tc=0a0bb41a17726804792802066ef22266fb6b0216da2a3e9f089735f8353b77; abRequestId=b5346cbb-6db3-5645-accc-df5d27fd9362; webBuild=5.13.1; xsecappid=xhs-pc-web; a1=19cbbfd8f46tok3grdu3mmi72tpiihd7co02rf9oa30000122754; webId=68ac71cf3f14eb4a280b442b71aad7e5; gid=yjSDDifj0fSfyjSDDifYilij4K9lTqkFf7q68l063WJ9UJq833xfWF888yJJW248dDqfjJ0Y; web_session=040069b8dcb7aa9bcf6957bd9d3b4b7b5c866c; id_token=VjEAAPwuVKkxM5M3tgTrefheWsAsAIisJtFuRYQM3EFnhkneE3Zag62PzVeUMmRAgCOznnJXrYICwToncTBIL4u7bKSd7M8QMVA0TgyLKo+Oknjg00IY1MQziJXeiutd2NTcUd+B; unread={%22ub%22:%2269a8d5d3000000002801e4e2%22%2C%22ue%22:%2269a85762000000002603e4ef%22%2C%22uc%22:30}; loadts=1772680707856"

def call_mcp(method, params):
    """调用 MCP 方法"""
    
    # MCP 请求格式
    payload = {
        "jsonrpc": "2.0",
        "id": 1,
        "method": "tools/call",
        "params": {
            "name": method,
            "arguments": params
        }
    }
    
    headers = {
        "Content-Type": "application/json",
        "Cookie": COOKIES
    }
    
    try:
        response = requests.post(MCP_URL, json=payload, headers=headers, timeout=60)
        return response.json()
    except Exception as e:
        return {"error": str(e)}


if __name__ == "__main__":
    print("调用小红书 MCP 搜索：北京看展 2026")
    print("=" * 50)
    
    result = call_mcp("search_feeds", {"keyword": "北京看展 2026"})
    
    print(json.dumps(result, ensure_ascii=False, indent=2))
    
    # 保存结果
    with open("data/xiaohongshu_result.json", "w", encoding="utf-8") as f:
        json.dump(result, f, ensure_ascii=False, indent=2)
    
    print("\n结果已保存到：data/xiaohongshu_result.json")