172 lines
6.1 KiB
Python
172 lines
6.1 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
自定义静态文件服务器，支持自定义404页面和SPA路由
|
||
用于部署Next.js静态导出的文件
|
||
"""
|
||
|
||
import http.server
|
||
import socketserver
|
||
import os
|
||
import sys
|
||
import urllib.parse
|
||
from pathlib import Path
|
||
|
||
class CustomHTTPRequestHandler(http.server.SimpleHTTPRequestHandler):
|
||
def __init__(self, *args, **kwargs):
|
||
# 设置静态文件目录
|
||
super().__init__(*args, directory="out", **kwargs)
|
||
|
||
def end_headers(self):
|
||
# 添加安全头和缓存控制
|
||
self.send_header('X-Content-Type-Options', 'nosniff')
|
||
self.send_header('X-Frame-Options', 'DENY')
|
||
self.send_header('X-XSS-Protection', '1; mode=block')
|
||
|
||
# 为HTML文件禁用缓存，为静态资源启用缓存
|
||
if self.path.endswith('.html'):
|
||
self.send_header('Cache-Control', 'no-cache, no-store, must-revalidate')
|
||
self.send_header('Pragma', 'no-cache')
|
||
self.send_header('Expires', '0')
|
||
elif any(self.path.endswith(ext) for ext in ['.css', '.js', '.png', '.jpg', '.jpeg', '.gif', '.ico', '.svg', '.woff', '.woff2']):
|
||
self.send_header('Cache-Control', 'public, max-age=31536000')
|
||
|
||
super().end_headers()
|
||
|
||
def do_GET(self):
|
||
# URL解码
|
||
parsed_path = urllib.parse.unquote(self.path)
|
||
|
||
# 移除查询参数
|
||
if '?' in parsed_path:
|
||
parsed_path = parsed_path.split('?')[0]
|
||
|
||
# 移除fragment
|
||
if '#' in parsed_path:
|
||
parsed_path = parsed_path.split('#')[0]
|
||
|
||
# 构建文件路径
|
||
if parsed_path == '/':
|
||
# 根路径重定向到index.html
|
||
file_path = Path("out/index.html")
|
||
else:
|
||
# 移除开头的斜杠
|
||
clean_path = parsed_path.lstrip('/')
|
||
file_path = Path("out") / clean_path
|
||
|
||
# 如果是目录，尝试查找index.html
|
||
if file_path.is_dir():
|
||
file_path = file_path / "index.html"
|
||
|
||
# 如果没有扩展名，尝试添加.html
|
||
elif not file_path.suffix and not file_path.exists():
|
||
file_path = file_path.with_suffix('.html')
|
||
|
||
print(f"请求路径: {self.path}")
|
||
print(f"解析路径: {parsed_path}")
|
||
print(f"文件路径: {file_path}")
|
||
print(f"文件存在: {file_path.exists()}")
|
||
|
||
# 检查文件是否存在
|
||
if file_path.exists() and file_path.is_file():
|
||
# 文件存在，正常处理
|
||
self.path = '/' + str(file_path.relative_to("out"))
|
||
super().do_GET()
|
||
else:
|
||
# 文件不存在，返回404页面
|
||
self.send_custom_404()
|
||
|
||
def send_custom_404(self):
|
||
"""发送自定义404页面"""
|
||
try:
|
||
# 检查自定义404页面是否存在
|
||
error_page_path = Path("out/404.html")
|
||
if not error_page_path.exists():
|
||
# 如果404.html不存在，使用public/404.html
|
||
error_page_path = Path("public/404.html")
|
||
|
||
if error_page_path.exists():
|
||
with open(error_page_path, 'rb') as f:
|
||
content = f.read()
|
||
|
||
self.send_response(404)
|
||
self.send_header('Content-type', 'text/html; charset=utf-8')
|
||
self.send_header('Content-Length', str(len(content)))
|
||
self.end_headers()
|
||
self.wfile.write(content)
|
||
print(f"返回自定义404页面: {error_page_path}")
|
||
else:
|
||
# 如果自定义404页面也不存在，返回简单的404
|
||
self.send_simple_404()
|
||
except Exception as e:
|
||
print(f"发送自定义404页面时出错: {e}")
|
||
self.send_simple_404()
|
||
|
||
def send_simple_404(self):
|
||
"""发送简单的404响应"""
|
||
message = """
|
||
<!DOCTYPE html>
|
||
<html>
|
||
<head>
|
||
<title>404 - 页面未找到</title>
|
||
<meta charset="utf-8">
|
||
<style>
|
||
body { font-family: Arial, sans-serif; text-align: center; padding: 50px; }
|
||
h1 { color: #333; }
|
||
p { color: #666; }
|
||
a { color: #0066cc; text-decoration: none; }
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<h1>404 - 页面未找到</h1>
|
||
<p>抱歉，您访问的页面不存在。</p>
|
||
<p><a href="/">返回首页</a></p>
|
||
</body>
|
||
</html>
|
||
"""
|
||
|
||
content = message.encode('utf-8')
|
||
self.send_response(404)
|
||
self.send_header('Content-type', 'text/html; charset=utf-8')
|
||
self.send_header('Content-Length', str(len(content)))
|
||
self.end_headers()
|
||
self.wfile.write(content)
|
||
|
||
def log_message(self, format, *args):
|
||
"""自定义日志格式"""
|
||
print(f"[{self.log_date_time_string()}] {format % args}")
|
||
|
||
def main():
|
||
# 检查out目录是否存在
|
||
if not os.path.exists("out"):
|
||
print("错误: 'out' 目录不存在。请先运行构建命令生成静态文件。")
|
||
sys.exit(1)
|
||
|
||
# 设置端口
|
||
PORT = 8080
|
||
if len(sys.argv) > 1:
|
||
try:
|
||
PORT = int(sys.argv[1])
|
||
except ValueError:
|
||
print("警告: 无效的端口号，使用默认端口 8080")
|
||
|
||
# 启动服务器
|
||
try:
|
||
with socketserver.TCPServer(("", PORT), CustomHTTPRequestHandler) as httpd:
|
||
print(f"静态文件服务器已启动")
|
||
print(f"访问地址: http://localhost:{PORT}")
|
||
print(f"静态文件目录: {os.path.abspath('out')}")
|
||
print("按 Ctrl+C 停止服务器")
|
||
print("-" * 50)
|
||
httpd.serve_forever()
|
||
except KeyboardInterrupt:
|
||
print("\n服务器已停止")
|
||
except OSError as e:
|
||
if e.errno == 48: # Address already in use
|
||
print(f"端口 {PORT} 已被占用，请尝试其他端口")
|
||
print(f"使用方法: python {sys.argv[0]} [端口号]")
|
||
else:
|
||
print(f"启动服务器时出错: {e}")
|
||
sys.exit(1)
|
||
|
||
if __name__ == "__main__":
|
||
main() |