108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
"""
|
|||
|
|
滴答清单 MCP 服务入口点
|
|||
|
|
允许AI模型通过MCP协议访问和操作滴答清单待办事项
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import argparse
|
|||
|
|
import json
|
|||
|
|
from pathlib import Path
|
|||
|
|
import dotenv
|
|||
|
|
from mcp_server import create_server
|
|||
|
|
from tools.official_api import init_api
|
|||
|
|
|
|||
|
|
# 加载环境变量（支持 .env）
|
|||
|
|
dotenv.load_dotenv()
|
|||
|
|
|
|||
|
|
def parse_args():
|
|||
|
|
"""解析命令行参数"""
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description="滴答清单 MCP 服务"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--install",
|
|||
|
|
action="store_true",
|
|||
|
|
help="安装到Claude Desktop或其他MCP客户端"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--token",
|
|||
|
|
help="滴答清单访问令牌"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--email",
|
|||
|
|
help="滴答清单账户邮箱"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--phone",
|
|||
|
|
help="滴答清单账户手机号"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--password",
|
|||
|
|
help="滴答清单账户密码"
|
|||
|
|
)
|
|||
|
|
# 统一 .env-only，不再支持 config 文件路径
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--port",
|
|||
|
|
type=int,
|
|||
|
|
default=3000,
|
|||
|
|
help="服务器端口号（用于SSE传输方式）"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--host",
|
|||
|
|
default="127.0.0.1",
|
|||
|
|
help="服务器主机（用于SSE传输方式）"
|
|||
|
|
)
|
|||
|
|
parser.add_argument(
|
|||
|
|
"--sse",
|
|||
|
|
action="store_true",
|
|||
|
|
help="使用SSE传输方式而不是stdio"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return parser.parse_args()
|
|||
|
|
|
|||
|
|
def ensure_oauth_ready() -> bool:
|
|||
|
|
"""
|
|||
|
|
检查OAuth是否就绪（仅用于独立运行模式）
|
|||
|
|
在集成模式下，token由外部传入，所以这个检查是可选的
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
if os.environ.get("DIDA_ACCESS_TOKEN"):
|
|||
|
|
init_api()
|
|||
|
|
# ✅ 检测到环境变量中的 OAuth token
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
# 未检测到 DIDA_ACCESS_TOKEN 环境变量
|
|||
|
|
# 在集成模式下这是正常的，token将由调用方传递
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
# OAuth初始化警告（在集成模式下正常）
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""主函数"""
|
|||
|
|
args = parse_args()
|
|||
|
|
# OAuth检查不再是强制性的
|
|||
|
|
ensure_oauth_ready()
|
|||
|
|
|
|||
|
|
# 创建MCP服务器
|
|||
|
|
server = create_server({})
|
|||
|
|
|
|||
|
|
# 启动服务器
|
|||
|
|
if args.install:
|
|||
|
|
# 安装到Claude Desktop
|
|||
|
|
# 正在安装到MCP客户端...
|
|||
|
|
os.system("fastmcp install")
|
|||
|
|
else:
|
|||
|
|
# 直接运行
|
|||
|
|
# 启动滴答清单MCP服务器
|
|||
|
|
if args.sse:
|
|||
|
|
# 使用SSE传输方式
|
|||
|
|
server.run(transport="sse", host=args.host, port=args.port)
|
|||
|
|
else:
|
|||
|
|
# 使用默认stdio传输方式
|
|||
|
|
server.run()
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|