feat(redirect80): new HTTP→HTTPS redirect service on port 80

Pure Python stdlib http.server + ThreadingMixIn, zero web framework deps. - GET/HEAD → 301, POST/PUT/DELETE/PATCH → 308 - Reads Host header, strips port, redirects to https://{host}{path} - Config via confs (retry_on_refused, hard dependency) + env var override - Logs via log_client → db Logs table (fail-safe async batch) - Port 80 privilege via setcap cap_net_bind_service in _run.sh
2026-08-02 12:50:23 +08:00
1 parent c7364286b7
commit f9c02cd6ad
@@ -0,0 +1,181 @@
+"""
+redirect80 — HTTP 80 端口重定向服务。
+
+监听 80 端口，所有 HTTP 请求返回 301/308 重定向到 HTTPS (443)，
+保留原始 Host 和路径。为搜索引擎和浏览器提供 HTTP→HTTPS 无缝升级。
+
+技术选型：Python 内置 http.server + 多线程混合模式，无外部 Web 框架依赖。
+          日志通过 log_client 异步推送到 db 的 Logs 表（fail-safe）。
+
+配置（两层加载）：
+  1. confs 强依赖拉取 redirect80.ini（retry_on_refused 阻塞等待）
+  2. 环境变量覆盖（Docker/k8s 兼容）
+confs 不可用时直接退出，不做本地回退。
+"""
+
+import configparser
+import os
+import sys as _sys
+from http.server import HTTPServer, BaseHTTPRequestHandler
+from socketserver import ThreadingMixIn
+
+# ── 路径设置 ──────────────────────────────────────────────────────────
+_here = os.path.dirname(os.path.abspath(__file__))
+_sys.path.insert(0, os.path.join(_here, ".."))
+from log_client import logs  # noqa: E402
+
+# ── 配置加载（confs 强依赖 → 环境变量覆盖）────────────────────────
+_config = {
+    "host": "0.0.0.0",
+    "port": 80,
+    "redirect_port": 443,
+    "redirect_code": 301,
+}
+
+# 1. confs 强依赖拉取配置（阻塞等待 confs 就绪，失败则退出）
+_sys.path.insert(0, os.path.join(_here, "..", "pymsg"))
+from pymsg import RpcClient  # noqa: E402
+
+_confs_host = os.environ.get("CONFS_HOST", "127.0.0.1")
+_confs_port = int(os.environ.get("CONFS_PORT", "8010"))
+_confs_client = RpcClient(
+    _confs_host, _confs_port, "redirect80",
+    retry_on_refused=True, timeout=5,
+)
+_ini_text = _confs_client.call("get", {"name": "redirect80.ini"})
+_ini = configparser.ConfigParser(interpolation=None)
+_ini.read_string(_ini_text)
+
+_merge_map = {
+    ("server", "host"): "host",
+    ("server", "port"): "port",
+    ("server", "redirect_port"): "redirect_port",
+    ("server", "redirect_code"): "redirect_code",
+}
+for (sec, key), var in _merge_map.items():
+    if _ini.has_option(sec, key):
+        val = _ini.get(sec, key)
+        _config[var] = type(_config[var])(val)
+logs.info("redirect80", "配置已从 confs 加载")
+
+# 2. 环境变量最高优先级覆盖
+_env_map = {
+    "REDIRECT80_HOST": "host",
+    "REDIRECT80_PORT": ("port", int),
+    "REDIRECT80_REDIRECT_PORT": ("redirect_port", int),
+    "REDIRECT80_REDIRECT_CODE": ("redirect_code", int),
+}
+for env_var, mapping in _env_map.items():
+    val = os.environ.get(env_var)
+    if val is not None:
+        if isinstance(mapping, tuple):
+            _config[mapping[0]] = mapping[1](val)
+        else:
+            _config[mapping] = val
+
+# ── 请求处理器 ──────────────────────────────────────────────────────────
+
+
+class RedirectHandler(BaseHTTPRequestHandler):
+    """HTTP 请求处理器：将所有请求重定向到 HTTPS。
+
+    GET/HEAD → 301 (Moved Permanently)，搜索引擎友好的永久重定向。
+    POST/PUT/DELETE/PATCH → 308 (Permanent Redirect)，保留请求方法和 body。
+    """
+
+    server_version = "UniServe/redirect80"
+    sys_version = ""
+
+    def _redirect(self, code: int):
+        """构造 HTTPS 重定向 URL 并返回。"""
+        host = self.headers.get("Host", "")
+        # 剥离端口号（如果客户端带了 :80）
+        hostname = host.split(":")[0] if host else _config["host"]
+        target_port = _config["redirect_port"]
+        # 标准 443 端口不显式写 :443
+        if target_port == 443:
+            location = f"https://{hostname}{self.path}"
+        else:
+            location = f"https://{hostname}:{target_port}{self.path}"
+
+        self.send_response(code)
+        self.send_header("Location", location)
+        self.send_header("Connection", "close")
+        self.send_header("Cache-Control", "max-age=31536000")
+        self.end_headers()
+
+        # 日志：仅记录摘要（不刷屏）
+        logs.info("redirect80",
+                   f"{code} {self.command} {host} → {location}")
+
+    def do_GET(self):
+        self._redirect(301)
+
+    def do_HEAD(self):
+        self._redirect(301)
+
+    def do_POST(self):
+        self._redirect(308)
+
+    def do_PUT(self):
+        self._redirect(308)
+
+    def do_DELETE(self):
+        self._redirect(308)
+
+    def do_PATCH(self):
+        self._redirect(308)
+
+    def do_OPTIONS(self):
+        self.send_response(204)
+        self.send_header("Allow", "GET, HEAD, POST, PUT, DELETE, PATCH, OPTIONS")
+        self.send_header("Connection", "close")
+        self.end_headers()
+
+    def log_message(self, format, *args):
+        """抑制默认的 stderr 日志（已通过 log_client 记录）。"""
+        pass
+
+
+# ── 多线程 HTTP 服务器 ──────────────────────────────────────────────────
+
+
+class ThreadingRedirectServer(ThreadingMixIn, HTTPServer):
+    """多线程混合 HTTP 服务器，处理并发连接。
+
+    ThreadingMixIn 为每个请求创建新线程，适合 redirect80
+    这种请求处理极快（无 I/O 阻塞）的场景。
+    """
+    daemon_threads = True
+    allow_reuse_address = True
+
+    def handle_error(self, request, client_address):
+        """覆盖默认错误处理，异常写入 log_client 而非 stderr。"""
+        logs.error("redirect80",
+                    f"处理 {client_address} 请求时异常",
+                    extra={"client": str(client_address)})
+
+
+def main():
+    host = _config["host"]
+    port = int(_config["port"])
+
+    server = ThreadingRedirectServer((host, port), RedirectHandler)
+
+    logs.info("redirect80",
+              f"HTTP→HTTPS 重定向服务启动 "
+              f"http://{host}:{port} → https://*:{_config['redirect_port']}")
+
+    print(f"[redirect80] 监听 http://{host}:{port} "
+          f"→ 重定向到 :{_config['redirect_port']}", file=_sys.stderr)
+
+    try:
+        server.serve_forever()
+    except KeyboardInterrupt:
+        logs.info("redirect80", "收到 SIGINT，正在关闭...")
+        server.shutdown()
+        logs.info("redirect80", "服务已停止")
+
+
+if __name__ == "__main__":
+    main()