90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
Markmap 本地服务器
|
||
|
||
启动一个简单的 HTTP 服务器，用于本地预览 Markmap 编辑器。
|
||
支持 Python 3.x，无需安装任何额外依赖。
|
||
|
||
用法:
|
||
python serve.py # 默认端口 8080
|
||
python serve.py 3000 # 指定端口 3000
|
||
python serve.py 3000 0.0.0.0 # 绑定到所有网络接口
|
||
"""
|
||
|
||
import http.server
|
||
import socketserver
|
||
import sys
|
||
import os
|
||
import webbrowser
|
||
|
||
# 默认配置
|
||
DEFAULT_PORT = 8080
|
||
DEFAULT_HOST = '127.0.0.1'
|
||
|
||
def main():
|
||
port = DEFAULT_PORT
|
||
host = DEFAULT_HOST
|
||
|
||
# 解析命令行参数
|
||
if len(sys.argv) >= 2:
|
||
try:
|
||
port = int(sys.argv[1])
|
||
except ValueError:
|
||
print(f'无效的端口号: {sys.argv[1]}')
|
||
sys.exit(1)
|
||
if len(sys.argv) >= 3:
|
||
host = sys.argv[2]
|
||
|
||
# 切换到脚本所在目录
|
||
script_dir = os.path.dirname(os.path.abspath(__file__))
|
||
os.chdir(script_dir)
|
||
|
||
handler = http.server.SimpleHTTPRequestHandler
|
||
|
||
# 添加 CORS 头和正确的 MIME 类型
|
||
class MarkmapHandler(handler):
|
||
def end_headers(self):
|
||
self.send_header('Access-Control-Allow-Origin', '*')
|
||
self.send_header('Cache-Control', 'no-cache')
|
||
super().end_headers()
|
||
|
||
def guess_type(self, path):
|
||
mimetype = super().guess_type(path)
|
||
# 确保正确的 MIME 类型
|
||
if path.endswith('.mjs') or path.endswith('.js'):
|
||
return 'application/javascript'
|
||
if path.endswith('.css'):
|
||
return 'text/css'
|
||
if path.endswith('.svg'):
|
||
return 'image/svg+xml'
|
||
if path.endswith('.md'):
|
||
return 'text/markdown'
|
||
if path.endswith('.woff2'):
|
||
return 'font/woff2'
|
||
return mimetype
|
||
|
||
try:
|
||
with socketserver.TCPServer((host, port), MarkmapHandler) as httpd:
|
||
url = f'http://{host}:{port}'
|
||
print(f'')
|
||
print(f' Markmap 编辑器已启动')
|
||
print(f' 访问地址: {url}')
|
||
print(f' 按 Ctrl+C 停止服务器')
|
||
print(f'')
|
||
|
||
# 自动打开浏览器
|
||
webbrowser.open(url)
|
||
|
||
httpd.serve_forever()
|
||
except KeyboardInterrupt:
|
||
print(f'\n服务器已停止')
|
||
except OSError as e:
|
||
if e.errno == 48 or 'Address already in use' in str(e):
|
||
print(f'端口 {port} 已被占用，请尝试其他端口')
|
||
else:
|
||
print(f'启动失败: {e}')
|
||
sys.exit(1)
|
||
|
||
if __name__ == '__main__':
|
||
main()
|