mcp-ntp 0.1.0__tar.gz

mcp_ntp-0.1.0/.editorconfig

@@ -0,0 +1,12 @@
+# EditorConfig is awesome: http://EditorConfig.org
+
+# top-most EditorConfig file
+root = true
+
+[*]
+charset = utf-8
+indent_style = space
+indent_size = 4
+end_of_line = lf
+insert_final_newline = true
+trim_trailing_whitespace = true

mcp_ntp-0.1.0/.gitignore

@@ -0,0 +1,79 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# uv/uvx specific
+.uv/
+.uvx/
+.venv.uv/
+
+# IDEs and editors
+.idea/
+.vscode/
+*.swp
+*.swo
+*~
+
+# OS specific
+.DS_Store
+Thumbs.db
+
+# Debug and local development
+debug/
+local/
+
+# Credentials
+.pypirc
+
+# Pytest
+.pytest_cache/
+.coverage/
+.coverage
+
+# Pre-commit
+.pre-commit-config.yaml.cache
+
+# debug
+playground/
+
+# cursor
+.cursor*
+
+# Claude
+.claude/
+# repomix
+.repomixignore
+repomix.config.json

mcp_ntp-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Guo
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

mcp_ntp-0.1.0/PKG-INFO

@@ -0,0 +1,149 @@
+Metadata-Version: 2.4
+Name: mcp-ntp
+Version: 0.1.0
+Summary: MCP-based NTP Time Service
+Author-email: Scofield Guo <guozhi5658@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2025 Guo
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+License-File: LICENSE
+Requires-Python: >=3.10
+Requires-Dist: click
+Requires-Dist: fastapi
+Requires-Dist: fastmcp
+Requires-Dist: mcp
+Requires-Dist: ntplib
+Requires-Dist: python-dotenv
+Requires-Dist: requests
+Requires-Dist: uvicorn
+Description-Content-Type: text/markdown
+
+# MCP-NTP 时间服务
+
+这是一个基于MCP（Model Context Protocol）的NTP时间服务，为大模型提供准确的网络时间获取功能。
+
+## 功能特性
+
+- 从NTP服务器获取准确的网络时间
+- 支持stdio和sse两种传输模式
+- 提供`get_current_time`工具，方便大模型获取当前时间
+- 支持环境变量和请求头两种配置NTP服务器的方式
+- 返回详细的时间信息和NTP协议数据
+
+## 安装
+
+```bash
+# 使用 uv 安装依赖
+uv sync
+```
+
+## 配置
+
+### 环境变量配置（stdio模式）
+
+设置`NTP_DOMAIN_URL`环境变量：
+
+```bash
+export NTP_DOMAIN_URL="time.windows.com"
+```
+
+### 请求头配置（sse模式）
+
+在请求头中包含`ntp_url`字段：
+
+```
+ntp_url: time.windows.com
+```
+
+## 使用方法
+
+### stdio模式
+
+```bash
+# 设置环境变量
+export NTP_DOMAIN_URL="time.windows.com"
+
+# 启动服务
+python -m app
+```
+
+### sse模式
+
+```bash
+# 启动服务
+python -m app --transport sse --port 18002
+
+# 使用curl测试
+curl -H "ntp_url: time.windows.com" http://localhost:18002/sse
+```
+
+## 工具说明
+
+### get_current_time
+
+获取当前NTP服务器时间。
+
+**参数：**
+- `ntp_server` (可选): NTP服务器地址，如果不提供则使用环境变量或请求头中的配置
+
+**返回数据：**
+```json
+{
+  "ntp_server": "time.windows.com",
+  "ntp_timestamp": 1703123456.789,
+  "ntp_time": "2023-12-21T10:30:56.789",
+  "local_system_time": "2023-12-21T10:30:56.789",
+  "time_offset": 0.123,
+  "response_delay": 0.045
+}
+```
+
+## 调试
+
+可以使用提供的测试NTP服务器进行调试：
+
+```bash
+export NTP_DOMAIN_URL="time.windows.com"
+python -m app -v
+```
+
+## 开发
+
+### 项目结构
+
+```
+mcp-ntp/
+├── app/
+│   ├── __init__.py      # 主入口
+│   ├── server.py        # MCP服务器实现
+│   ├── client.py        # NTP客户端
+│   └── config.py        # 配置管理
+├── pyproject.toml       # 项目配置
+└── README.md           # 说明文档
+```
+
+### 添加新的NTP服务器
+
+在`client.py`中的`get_current_time`函数可以轻松扩展支持更多NTP服务器配置。
+
+## 许可证
+
+MIT License

mcp_ntp-0.1.0/README.md

@@ -0,0 +1,111 @@
+# MCP-NTP 时间服务
+
+这是一个基于MCP（Model Context Protocol）的NTP时间服务，为大模型提供准确的网络时间获取功能。
+
+## 功能特性
+
+- 从NTP服务器获取准确的网络时间
+- 支持stdio和sse两种传输模式
+- 提供`get_current_time`工具，方便大模型获取当前时间
+- 支持环境变量和请求头两种配置NTP服务器的方式
+- 返回详细的时间信息和NTP协议数据
+
+## 安装
+
+```bash
+# 使用 uv 安装依赖
+uv sync
+```
+
+## 配置
+
+### 环境变量配置（stdio模式）
+
+设置`NTP_DOMAIN_URL`环境变量：
+
+```bash
+export NTP_DOMAIN_URL="time.windows.com"
+```
+
+### 请求头配置（sse模式）
+
+在请求头中包含`ntp_url`字段：
+
+```
+ntp_url: time.windows.com
+```
+
+## 使用方法
+
+### stdio模式
+
+```bash
+# 设置环境变量
+export NTP_DOMAIN_URL="time.windows.com"
+
+# 启动服务
+python -m app
+```
+
+### sse模式
+
+```bash
+# 启动服务
+python -m app --transport sse --port 18002
+
+# 使用curl测试
+curl -H "ntp_url: time.windows.com" http://localhost:18002/sse
+```
+
+## 工具说明
+
+### get_current_time
+
+获取当前NTP服务器时间。
+
+**参数：**
+- `ntp_server` (可选): NTP服务器地址，如果不提供则使用环境变量或请求头中的配置
+
+**返回数据：**
+```json
+{
+  "ntp_server": "time.windows.com",
+  "ntp_timestamp": 1703123456.789,
+  "ntp_time": "2023-12-21T10:30:56.789",
+  "local_system_time": "2023-12-21T10:30:56.789",
+  "time_offset": 0.123,
+  "response_delay": 0.045
+}
+```
+
+## 调试
+
+可以使用提供的测试NTP服务器进行调试：
+
+```bash
+export NTP_DOMAIN_URL="time.windows.com"
+python -m app -v
+```
+
+## 开发
+
+### 项目结构
+
+```
+mcp-ntp/
+├── app/
+│   ├── __init__.py      # 主入口
+│   ├── server.py        # MCP服务器实现
+│   ├── client.py        # NTP客户端
+│   └── config.py        # 配置管理
+├── pyproject.toml       # 项目配置
+└── README.md           # 说明文档
+```
+
+### 添加新的NTP服务器
+
+在`client.py`中的`get_current_time`函数可以轻松扩展支持更多NTP服务器配置。
+
+## 许可证
+
+MIT License

mcp_ntp-0.1.0/app/__init__.py

@@ -0,0 +1,169 @@
+import asyncio
+import logging
+import os
+import sys
+from importlib.metadata import PackageNotFoundError, version
+
+import click
+from dotenv import load_dotenv
+
+try:
+    __version__ = version("mcp-ntp")
+except PackageNotFoundError:
+    __version__ = "0.1.0"
+
+
+def setup_logging(level: int = logging.INFO) -> logging.Logger:
+    """设置日志配置"""
+    logger = logging.getLogger("mcp-ntp")
+    logger.setLevel(level)
+
+    if not logger.handlers:
+        handler = logging.StreamHandler(sys.stderr)
+        formatter = logging.Formatter(
+            '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+        )
+        handler.setFormatter(formatter)
+        logger.addHandler(handler)
+
+    return logger
+
+
+@click.version_option(__version__, prog_name="mcp-ntp")
+@click.command()
+@click.option(
+    "-v",
+    "--verbose",
+    count=True,
+    help="增加详细程度 (可多次使用)",
+)
+@click.option(
+    "--env-file",
+    type=click.Path(exists=True, dir_okay=False),
+    help=".env 文件路径"
+)
+@click.option(
+    "--transport",
+    type=click.Choice(["stdio", "sse"]),
+    default="stdio",
+    help="传输方式 (stdio 或 sse)",
+)
+@click.option(
+    "--port",
+    default=18002,
+    help="SSE 传输方式的监听端口",
+)
+@click.option(
+    "--host",
+    default="0.0.0.0",
+    help="SSE 传输方式的监听地址",
+)
+def main(
+    verbose: int,
+    env_file: str | None,
+    transport: str,
+    port: int,
+    host: str,
+) -> None:
+    """MCP-NTP 服务器
+
+    支持 stdio 和 sse 两种传输方式。
+    """
+    # 设置日志级别
+    if verbose == 1:
+        current_logging_level = logging.INFO
+    elif verbose >= 2:
+        current_logging_level = logging.DEBUG
+    else:
+        current_logging_level = logging.WARNING
+
+    logger = setup_logging(current_logging_level)
+    logger.debug(f"日志级别设置为: {logging.getLevelName(current_logging_level)}")
+
+    # 加载环境变量
+    if env_file:
+        logger.debug(f"从文件加载环境变量: {env_file}")
+        load_dotenv(env_file, override=True)
+    else:
+        logger.debug("尝试从默认 .env 文件加载环境变量")
+        load_dotenv(override=True)
+
+    def was_option_provided(ctx: click.Context, param_name: str) -> bool:
+        return (
+            ctx.get_parameter_source(param_name)
+            != click.core.ParameterSource.DEFAULT_MAP
+            and ctx.get_parameter_source(param_name)
+            != click.core.ParameterSource.DEFAULT
+        )
+
+    # 确定最终的传输方式
+    final_transport = os.getenv("TRANSPORT", transport).lower()
+    if final_transport not in ["stdio", "sse"]:
+        logger.warning(f"无效的传输方式 '{final_transport}'，使用 'stdio'")
+        final_transport = "stdio"
+
+    logger.debug(f"最终传输方式: {final_transport}")
+
+    # 确定端口和主机
+    final_port = int(os.getenv("PORT", port))
+    final_host = os.getenv("HOST", host)
+
+    if final_transport == "stdio":
+        logger.info("使用 STDIO 传输方式启动服务器")
+        asyncio.run(run_stdio_server())
+    else:
+        logger.info(f"使用 SSE 传输方式启动服务器，地址: http://{final_host}:{final_port}/sse")
+        asyncio.run(run_sse_server(final_host, final_port))
+
+
+async def run_stdio_server():
+    """运行 STDIO 传输模式的 MCP 服务器"""
+    from mcp.server.stdio import stdio_server
+    from .server import create_mcp_server
+
+    import os
+    ntp_url = os.getenv("NTP_DOMAIN_URL")
+    app = create_mcp_server(ntp_url=ntp_url)
+
+    async with stdio_server() as (read_stream, write_stream):
+        await app.run(
+            read_stream, write_stream, app.create_initialization_options()
+        )
+
+
+async def run_sse_server(host: str, port: int):
+    """运行 SSE 传输模式的 MCP 服务器"""
+    from mcp.server.sse import SseServerTransport
+    from starlette.applications import Starlette
+    from starlette.requests import Request
+    from starlette.routing import Mount, Route
+    from .server import create_mcp_server
+    import uvicorn
+
+    sse = SseServerTransport("/messages/")
+
+    async def handle_sse(request: Request):
+        ntp_url = request.headers.get("ntp_url")
+        app = create_mcp_server(ntp_url=ntp_url)
+        async with sse.connect_sse(
+            request.scope, request.receive, request._send
+        ) as streams:
+            await app.run(
+                streams[0], streams[1], app.create_initialization_options()
+            )
+
+    starlette_app = Starlette(
+        debug=True,
+        routes=[
+            Route("/sse", endpoint=handle_sse),
+            Mount("/messages/", app=sse.handle_post_message),
+        ],
+    )
+
+    config = uvicorn.Config(starlette_app, host=host, port=port)
+    server = uvicorn.Server(config)
+    await server.serve()
+
+
+if __name__ == "__main__":
+    main()

mcp_ntp-0.1.0/app/__main__.py

@@ -0,0 +1,4 @@
+from . import main
+
+if __name__ == "__main__":
+    main()

mcp_ntp-0.1.0/app/client.py

@@ -0,0 +1,46 @@
+import ntplib
+import time
+from datetime import datetime
+from .config import NTP_DOMAIN_URL
+
+def get_current_time(ntp_server=None):
+    """从NTP服务器获取当前时间"""
+    if ntp_server is None or ntp_server.strip() == "":
+        ntp_server = NTP_DOMAIN_URL
+
+    # 处理空串和None的情况
+    if not ntp_server:
+        raise ValueError("NTP服务器地址未配置，请设置NTP_DOMAIN_URL环境变量或在请求头中提供ntp_url")
+
+    # 去除首尾空格
+    ntp_server = ntp_server.strip()
+
+    try:
+        # 创建NTP客户端
+        ntp_client = ntplib.NTPClient()
+
+        # 从NTP服务器获取时间
+        response = ntp_client.request(ntp_server, version=3, timeout=5)
+
+        # 获取NTP时间戳
+        ntp_time = response.tx_time
+
+        # 转换为本地时间
+        local_time = datetime.fromtimestamp(ntp_time)
+
+        # 获取本地时间作为对比
+        local_system_time = datetime.now()
+
+        return {
+            "ntp_server": ntp_server,
+            "ntp_timestamp": ntp_time,
+            "ntp_time": local_time.isoformat(),
+            "local_system_time": local_system_time.isoformat(),
+            "time_offset": ntp_time - time.time(),
+            "response_delay": response.delay
+        }
+
+    except ntplib.NTPException as e:
+        raise Exception(f"NTP请求失败: {str(e)}")
+    except Exception as e:
+        raise Exception(f"获取NTP时间时发生错误: {str(e)}")

mcp_ntp-0.1.0/app/config.py

@@ -0,0 +1,10 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+def get_env(key: str, default=None):
+    return os.getenv(key, default)
+
+# NTP服务器配置
+NTP_DOMAIN_URL = get_env("NTP_DOMAIN_URL", "pool.ntp.org")