1#!/usr/bin/env python3
2import argparse
3import json
4import sys
5import urllib.request
6 
7DEFAULT_URL = "https://mcp.sendpulse.com/mcp"
8DEFAULT_PROTOCOL_VERSION = "2025-03-26"
9 
10 
11def post_json(url: str, payload: dict, headers: dict):
12    data = json.dumps(payload).encode("utf-8")
13    req = urllib.request.Request(url, data=data, method="POST")
14    req.add_header("Content-Type", "application/json")
15    req.add_header("Accept", "application/json, text/event-stream")
16    for k, v in headers.items():
17        req.add_header(k, v)
18    with urllib.request.urlopen(req, timeout=30) as resp:
19        raw = resp.read().decode("utf-8")
20        return resp.status, dict(resp.headers.items()), raw
21 
22 
23def init_session(url: str, sp_id: str, sp_secret: str, protocol_version: str):
24    common_headers = {
25        "X-SP-ID": sp_id,
26        "X-SP-SECRET": sp_secret,
27    }
28    initialize_payload = {
29        "jsonrpc": "2.0",
30        "id": 1,
31        "method": "initialize",
32        "params": {
33            "protocolVersion": protocol_version,
34            "capabilities": {},
35            "clientInfo": {"name": "sendpulse-mcp-tool-call", "version": "1.0"},
36        },
37    }
38    status, headers, body = post_json(url, initialize_payload, common_headers)
39    init_data = json.loads(body)
40    session_id = headers.get("Mcp-Session-Id") or headers.get("mcp-session-id")
41    if status != 200 or not session_id:
42        raise SystemExit(json.dumps({
43            "ok": False,
44            "step": "initialize",
45            "status": status,
46            "hasSessionId": bool(session_id),
47            "body": init_data,
48        }, indent=2))
49 
50    notify_headers = dict(common_headers)
51    notify_headers["Mcp-Session-Id"] = session_id
52    notify_payload = {
53        "jsonrpc": "2.0",
54        "method": "notifications/initialized",
55        "params": {},
56    }
57    post_json(url, notify_payload, notify_headers)
58    return notify_headers, init_data, session_id
59 
60 
61def main():
62    parser = argparse.ArgumentParser(description="Call a SendPulse MCP tool through raw MCP HTTP.")
63    parser.add_argument("--sp-id", required=True, help="SendPulse MCP/OpenAI ID header value")
64    parser.add_argument("--sp-secret", required=True, help="SendPulse MCP/OpenAI Secret header value")
65    parser.add_argument("--tool", required=True, help="Tool name to call")
66    parser.add_argument("--args-json", default="{}", help="Tool args as JSON object string")
67    parser.add_argument("--url", default=DEFAULT_URL, help="MCP URL")
68    parser.add_argument("--protocol-version", default=DEFAULT_PROTOCOL_VERSION, help="MCP protocol version")
69    parser.add_argument("--pretty", action="store_true", help="Pretty-print JSON")
70    args = parser.parse_args()
71 
72    try:
73        tool_args = json.loads(args.args_json)
74    except json.JSONDecodeError as e:
75        print(f"Invalid --args-json: {e}", file=sys.stderr)
76        sys.exit(2)
77    if not isinstance(tool_args, dict):
78        print("--args-json must decode to a JSON object", file=sys.stderr)
79        sys.exit(2)
80 
81    headers, init_data, session_id = init_session(args.url, args.sp_id, args.sp_secret, args.protocol_version)
82 
83    call_payload = {
84        "jsonrpc": "2.0",
85        "id": 2,
86        "method": "tools/call",
87        "params": {
88            "name": args.tool,
89            "arguments": tool_args,
90        },
91    }
92    status, _, body = post_json(args.url, call_payload, headers)
93    call_data = json.loads(body)
94 
95    result = {
96        "ok": status == 200 and "error" not in call_data,
97        "initialize": {
98            "serverInfo": init_data.get("result", {}).get("serverInfo", {}),
99            "protocolVersion": init_data.get("result", {}).get("protocolVersion"),
100            "sessionId": session_id,
101        },
102        "toolCall": {
103            "status": status,
104            "tool": args.tool,
105            "arguments": tool_args,
106            "response": call_data,
107        },
108    }
109 
110    if args.pretty:
111        print(json.dumps(result, indent=2))
112    else:
113        print(json.dumps(result))
114 
115 
116if __name__ == "__main__":
117    main()
118