3a096f59a9
- 核心功能从 store/ 迁移至 oss/core/ 框架层 - 实现 NBPF 包格式：多重签名（Ed25519+RSA-PSS+HMAC）+ 多重加密（AES-256-GCM） - 实现 NIR 编译器：基于 compile()+marshal 的跨平台中间表示 - 新增 nebula nbpf CLI 命令组（pack/unpack/verify/sign/keygen） - 新增 19 个 NBPF 测试用例，覆盖全链路 - 彻底重写 README，大型项目标准框架风格，所有图表使用 SVG - 更新 LICENSE 版权声明 - 清理旧版 store 插件目录（已迁移至 oss/core）
125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
"""HTTP 服务器核心"""
|
|
import threading
|
|
from http.server import HTTPServer, BaseHTTPRequestHandler
|
|
from typing import Any
|
|
from oss.config import get_config
|
|
from oss.logger.logger import Log
|
|
|
|
|
|
class Request:
|
|
"""请求对象"""
|
|
def __init__(self, method, path, headers, body):
|
|
self.method = method
|
|
self.path = path
|
|
self.headers = headers
|
|
self.body = body
|
|
|
|
|
|
class Response:
|
|
"""响应对象"""
|
|
def __init__(self, status=200, headers=None, body=""):
|
|
self.status = status
|
|
self.headers = headers or {}
|
|
self.body = body
|
|
|
|
|
|
class HttpServer:
|
|
"""HTTP 服务器"""
|
|
|
|
def __init__(self, router, middleware, host=None, port=None):
|
|
config = get_config()
|
|
self.host = host or config.get("HOST", "127.0.0.1")
|
|
self.port = port or config.get("HTTP_API_PORT", 8080)
|
|
self.router = router
|
|
self.middleware = middleware
|
|
self._server = None
|
|
self._thread = None
|
|
|
|
def start(self):
|
|
"""启动服务器"""
|
|
handler = self._create_handler()
|
|
self._server = HTTPServer((self.host, self.port), handler)
|
|
self._thread = threading.Thread(target=self._server.serve_forever, daemon=True)
|
|
self._thread.start()
|
|
Log.info("Core", f"HTTP 服务器启动: {self.host}:{self.port}")
|
|
|
|
def stop(self):
|
|
"""停止服务器"""
|
|
if self._server:
|
|
self._server.shutdown()
|
|
Log.info("Core", "HTTP 服务器已停止")
|
|
|
|
def _create_handler(self):
|
|
"""创建请求处理器"""
|
|
router = self.router
|
|
middleware = self.middleware
|
|
|
|
class Handler(BaseHTTPRequestHandler):
|
|
def do_GET(self):
|
|
self._handle("GET")
|
|
|
|
def do_POST(self):
|
|
self._handle("POST")
|
|
|
|
def do_PUT(self):
|
|
self._handle("PUT")
|
|
|
|
def do_DELETE(self):
|
|
self._handle("DELETE")
|
|
|
|
def do_OPTIONS(self):
|
|
"""处理 CORS 预检请求"""
|
|
config = get_config()
|
|
allowed_origins = config.get("CORS_ALLOWED_ORIGINS", ["http://localhost:3000", "http://127.0.0.1:3000"])
|
|
origin = self.headers.get("Origin", "")
|
|
|
|
if origin in allowed_origins or "*" in allowed_origins:
|
|
self.send_response(200)
|
|
self.send_header("Access-Control-Allow-Origin", origin if origin else "*")
|
|
self.send_header("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
|
self.send_header("Access-Control-Allow-Headers", "Content-Type, Authorization")
|
|
self.send_header("Access-Control-Allow-Credentials", "true")
|
|
else:
|
|
self.send_response(204)
|
|
self.end_headers()
|
|
|
|
def _handle(self, method):
|
|
content_length = int(self.headers.get("Content-Length", 0))
|
|
body = self.rfile.read(content_length) if content_length else b""
|
|
|
|
req = Request(
|
|
method=method,
|
|
path=self.path,
|
|
headers=dict(self.headers),
|
|
body=body.decode("utf-8")
|
|
)
|
|
|
|
# 执行中间件
|
|
ctx = {"request": req, "response": None}
|
|
result = middleware.run(ctx)
|
|
if result:
|
|
self._send_response(result)
|
|
return
|
|
|
|
# 路由匹配
|
|
resp = router.handle(req)
|
|
self._send_response(resp)
|
|
|
|
def _send_response(self, resp: Response):
|
|
try:
|
|
self.send_response(resp.status)
|
|
for k, v in resp.headers.items():
|
|
self.send_header(k, v)
|
|
self.end_headers()
|
|
if isinstance(resp.body, str):
|
|
self.wfile.write(resp.body.encode("utf-8"))
|
|
else:
|
|
self.wfile.write(resp.body)
|
|
except (BrokenPipeError, ConnectionAbortedError, ConnectionResetError):
|
|
pass # 忽略客户端断开
|
|
|
|
def log_message(self, format, *args):
|
|
Log.debug("Core", format % args)
|
|
|
|
return Handler
|