1#!/bin/bash
2# 通用 MCP 调用脚本（支持 Streamable HTTP + Session ID）
3 
4set -e
5 
6TOOL_NAME="$1"
7TOOL_ARGS="$2"
8MCP_URL="${MCP_URL:-http://localhost:18060/mcp}"
9export no_proxy="${no_proxy:+$no_proxy,}localhost,127.0.0.1"
10 
11# 检查 jq 依赖
12if ! command -v jq &> /dev/null; then
13    echo "错误: 需要安装 jq (apt-get install jq / brew install jq)"
14    exit 1
15fi
16 
17if [ -z "$TOOL_NAME" ]; then
18    echo "用法: $0 <tool_name> [json_args]"
19    echo ""
20    echo "可用工具:"
21    echo "  check_login_status    - 检查登录状态"
22    echo "  search_feeds          - 搜索内容 {\"keyword\": \"...\", \"filters\": {\"sort_by\": \"最新\"}}"
23    echo "  list_feeds            - 获取首页推荐"
24    echo "  get_feed_detail       - 获取帖子详情 {\"feed_id\": \"...\", \"xsec_token\": \"...\"}"
25    echo "  post_comment_to_feed  - 发表评论 {\"feed_id\": \"...\", \"xsec_token\": \"...\", \"content\": \"...\"}"
26    echo "  reply_comment_in_feed - 回复评论 {\"feed_id\": \"...\", \"xsec_token\": \"...\", \"content\": \"...\", \"comment_id\": \"...\", \"user_id\": \"...\"}"
27    echo "  user_profile          - 获取用户主页 {\"user_id\": \"...\", \"xsec_token\": \"...\"}"
28    echo "  like_feed             - 点赞 {\"feed_id\": \"...\", \"xsec_token\": \"...\"}  取消: {\"unlike\": true}"
29    echo "  favorite_feed         - 收藏 {\"feed_id\": \"...\", \"xsec_token\": \"...\"}  取消: {\"unfavorite\": true}"
30    echo "  get_login_qrcode      - 获取登录二维码"
31    echo "  delete_cookies        - 删除 cookies，重置登录状态"
32    echo "  publish_content       - 发布图文"
33    echo "  publish_with_video    - 发布视频"
34    exit 1
35fi
36 
37# 校验 tool name，只允许字母数字和下划线
38if [[ ! "$TOOL_NAME" =~ ^[a-zA-Z_][a-zA-Z0-9_]*$ ]]; then
39    echo "错误: 无效的工具名: $TOOL_NAME"
40    exit 1
41fi
42 
43[ -z "$TOOL_ARGS" ] && TOOL_ARGS="{}"
44 
45# 校验 TOOL_ARGS 是合法 JSON
46if ! echo "$TOOL_ARGS" | jq empty 2>/dev/null; then
47    echo "错误: 参数不是合法的 JSON: $TOOL_ARGS"
48    exit 1
49fi
50 
51# 1. Initialize 并获取 Session ID
52INIT_RESPONSE=$(curl --noproxy '*' -s -i -X POST "$MCP_URL" \
53  -H "Content-Type: application/json" \
54  -d '{"jsonrpc":"2.0","id":1,"method":"initialize","params":{"protocolVersion":"2024-11-05","capabilities":{},"clientInfo":{"name":"openclaw","version":"1.0"}}}')
55 
56SESSION_ID=$(echo "$INIT_RESPONSE" | grep -i "Mcp-Session-Id" | awk '{print $2}' | tr -d '\r\n')
57 
58if [ -z "$SESSION_ID" ]; then
59    echo "错误: 无法获取 MCP Session ID"
60    echo "请确保 MCP 服务正在运行: ./start-mcp.sh"
61    exit 1
62fi
63 
64# 2. Initialized notification
65curl --noproxy '*' -s -X POST "$MCP_URL" \
66  -H "Content-Type: application/json" \
67  -H "Mcp-Session-Id: $SESSION_ID" \
68  -d '{"jsonrpc":"2.0","method":"notifications/initialized"}' > /dev/null
69 
70# 3. Call tool — 使用 jq 安全构建 JSON，避免 shell 注入
71CALL_PAYLOAD=$(jq -n \
72  --arg name "$TOOL_NAME" \
73  --argjson args "$TOOL_ARGS" \
74  '{"jsonrpc":"2.0","id":2,"method":"tools/call","params":{"name":$name,"arguments":$args}}')
75 
76RESULT=$(curl --noproxy '*' -s --max-time 120 -X POST "$MCP_URL" \
77  -H "Content-Type: application/json" \
78  -H "Mcp-Session-Id: $SESSION_ID" \
79  -d "$CALL_PAYLOAD")
80 
81# 输出结果
82echo "$RESULT" | jq .
83