|
|
from fastapi import FastAPI, HTTPException
|
|
|
from fastapi.staticfiles import StaticFiles
|
|
|
from fastapi.responses import HTMLResponse
|
|
|
from pydantic import BaseModel
|
|
|
import markdown
|
|
|
import aiofiles
|
|
|
import os
|
|
|
import uuid
|
|
|
from datetime import datetime
|
|
|
import json
|
|
|
import re
|
|
|
from typing import List, Optional
|
|
|
|
|
|
app = FastAPI(title="HTML/Markdown Preview API", version="1.0.0")
|
|
|
|
|
|
|
|
|
app.mount("/static", StaticFiles(directory="static"), name="static")
|
|
|
|
|
|
class HTMLRequest(BaseModel):
|
|
|
html_content: str
|
|
|
|
|
|
class MarkdownRequest(BaseModel):
|
|
|
markdown_content: str
|
|
|
|
|
|
class PreviewResponse(BaseModel):
|
|
|
url: str
|
|
|
message: str
|
|
|
|
|
|
class FileInfo(BaseModel):
|
|
|
filename: str
|
|
|
title: Optional[str] = None
|
|
|
url: str
|
|
|
created_time: str
|
|
|
|
|
|
class FileListResponse(BaseModel):
|
|
|
files: List[FileInfo]
|
|
|
total: int
|
|
|
|
|
|
def generate_filename(extension: str = ".html") -> str:
|
|
|
"""生成唯一的文件名"""
|
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
|
unique_id = str(uuid.uuid4())[:8]
|
|
|
return f"{timestamp}_{unique_id}{extension}"
|
|
|
|
|
|
async def save_html_file(content: str, filename: str) -> str:
|
|
|
"""保存HTML内容到文件"""
|
|
|
file_path = os.path.join("static", filename)
|
|
|
async with aiofiles.open(file_path, 'w', encoding='utf-8') as f:
|
|
|
await f.write(content)
|
|
|
return filename
|
|
|
|
|
|
def extract_title_from_html(html_content: str) -> Optional[str]:
|
|
|
"""从HTML内容中提取标题"""
|
|
|
|
|
|
title_match = re.search(r'<title[^>]*>(.*?)</title>', html_content, re.IGNORECASE | re.DOTALL)
|
|
|
if title_match:
|
|
|
title = title_match.group(1).strip()
|
|
|
if title:
|
|
|
return title
|
|
|
|
|
|
|
|
|
h1_match = re.search(r'<h1[^>]*>(.*?)</h1>', html_content, re.IGNORECASE | re.DOTALL)
|
|
|
if h1_match:
|
|
|
title = h1_match.group(1).strip()
|
|
|
if title:
|
|
|
return title
|
|
|
|
|
|
return None
|
|
|
|
|
|
async def get_html_files_info() -> List[FileInfo]:
|
|
|
"""获取static目录中所有HTML文件的信息"""
|
|
|
files_info = []
|
|
|
static_dir = "static"
|
|
|
|
|
|
if not os.path.exists(static_dir):
|
|
|
return files_info
|
|
|
|
|
|
for filename in os.listdir(static_dir):
|
|
|
if filename.endswith('.html'):
|
|
|
file_path = os.path.join(static_dir, filename)
|
|
|
|
|
|
|
|
|
stat = os.stat(file_path)
|
|
|
created_time = datetime.fromtimestamp(stat.st_ctime).strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
|
|
|
|
|
try:
|
|
|
async with aiofiles.open(file_path, 'r', encoding='utf-8') as f:
|
|
|
content = await f.read()
|
|
|
title = extract_title_from_html(content)
|
|
|
except:
|
|
|
title = None
|
|
|
|
|
|
files_info.append(FileInfo(
|
|
|
filename=filename,
|
|
|
title=title,
|
|
|
url=f"/static/{filename}",
|
|
|
created_time=created_time
|
|
|
))
|
|
|
|
|
|
|
|
|
files_info.sort(key=lambda x: x.created_time, reverse=True)
|
|
|
return files_info
|
|
|
|
|
|
def markdown_to_html(markdown_content: str) -> str:
|
|
|
"""将Markdown转换为HTML"""
|
|
|
|
|
|
extensions = [
|
|
|
'markdown.extensions.extra',
|
|
|
'markdown.extensions.codehilite',
|
|
|
'markdown.extensions.toc',
|
|
|
'markdown.extensions.tables',
|
|
|
'markdown.extensions.fenced_code'
|
|
|
]
|
|
|
|
|
|
|
|
|
html_content = markdown.markdown(markdown_content, extensions=extensions)
|
|
|
|
|
|
|
|
|
full_html = f"""
|
|
|
<!DOCTYPE html>
|
|
|
<html lang="zh-CN">
|
|
|
<head>
|
|
|
<meta charset="UTF-8">
|
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
|
<title>Markdown Preview</title>
|
|
|
<style>
|
|
|
body {{
|
|
|
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', sans-serif;
|
|
|
line-height: 1.6;
|
|
|
max-width: 800px;
|
|
|
margin: 0 auto;
|
|
|
padding: 20px;
|
|
|
color: #333;
|
|
|
}}
|
|
|
h1, h2, h3, h4, h5, h6 {{
|
|
|
margin-top: 1.5em;
|
|
|
margin-bottom: 0.5em;
|
|
|
}}
|
|
|
code {{
|
|
|
background-color: #f4f4f4;
|
|
|
padding: 2px 4px;
|
|
|
border-radius: 3px;
|
|
|
font-family: 'Consolas', 'Monaco', monospace;
|
|
|
}}
|
|
|
pre {{
|
|
|
background-color: #f4f4f4;
|
|
|
padding: 10px;
|
|
|
border-radius: 5px;
|
|
|
overflow-x: auto;
|
|
|
}}
|
|
|
pre code {{
|
|
|
background-color: transparent;
|
|
|
padding: 0;
|
|
|
}}
|
|
|
blockquote {{
|
|
|
border-left: 4px solid #ddd;
|
|
|
margin: 0;
|
|
|
padding-left: 20px;
|
|
|
color: #666;
|
|
|
}}
|
|
|
table {{
|
|
|
border-collapse: collapse;
|
|
|
width: 100%;
|
|
|
margin: 1em 0;
|
|
|
}}
|
|
|
th, td {{
|
|
|
border: 1px solid #ddd;
|
|
|
padding: 8px;
|
|
|
text-align: left;
|
|
|
}}
|
|
|
th {{
|
|
|
background-color: #f2f2f2;
|
|
|
}}
|
|
|
img {{
|
|
|
max-width: 100%;
|
|
|
height: auto;
|
|
|
}}
|
|
|
</style>
|
|
|
</head>
|
|
|
<body>
|
|
|
{html_content}
|
|
|
</body>
|
|
|
</html>
|
|
|
"""
|
|
|
return full_html
|
|
|
|
|
|
@app.post("/api/html/preview", response_model=PreviewResponse)
|
|
|
async def preview_html(request: HTMLRequest):
|
|
|
"""
|
|
|
接收HTML代码，返回在线访问链接
|
|
|
"""
|
|
|
try:
|
|
|
|
|
|
filename = generate_filename(".html")
|
|
|
|
|
|
|
|
|
await save_html_file(request.html_content, filename)
|
|
|
|
|
|
|
|
|
url = f"/static/{filename}"
|
|
|
|
|
|
return PreviewResponse(
|
|
|
url=url,
|
|
|
message=f"HTML预览已创建，可通过链接访问"
|
|
|
)
|
|
|
|
|
|
except Exception as e:
|
|
|
raise HTTPException(status_code=500, detail=f"创建HTML预览失败: {str(e)}")
|
|
|
|
|
|
@app.post("/api/markdown/preview", response_model=PreviewResponse)
|
|
|
async def preview_markdown(request: MarkdownRequest):
|
|
|
"""
|
|
|
接收Markdown代码，返回渲染后的HTML在线访问链接
|
|
|
"""
|
|
|
try:
|
|
|
|
|
|
html_content = markdown_to_html(request.markdown_content)
|
|
|
|
|
|
|
|
|
filename = generate_filename(".html")
|
|
|
|
|
|
|
|
|
await save_html_file(html_content, filename)
|
|
|
|
|
|
|
|
|
url = f"/static/{filename}"
|
|
|
|
|
|
return PreviewResponse(
|
|
|
url=url,
|
|
|
message=f"Markdown预览已创建，可通过链接访问"
|
|
|
)
|
|
|
|
|
|
except Exception as e:
|
|
|
raise HTTPException(status_code=500, detail=f"创建Markdown预览失败: {str(e)}")
|
|
|
|
|
|
@app.get("/api/files", response_model=FileListResponse)
|
|
|
async def get_files():
|
|
|
"""
|
|
|
获取所有HTML文件列表
|
|
|
"""
|
|
|
try:
|
|
|
files = await get_html_files_info()
|
|
|
return FileListResponse(
|
|
|
files=files,
|
|
|
total=len(files)
|
|
|
)
|
|
|
except Exception as e:
|
|
|
raise HTTPException(status_code=500, detail=f"获取文件列表失败: {str(e)}")
|
|
|
|
|
|
@app.delete("/api/files/{filename}")
|
|
|
async def delete_file(filename: str):
|
|
|
"""
|
|
|
删除指定的HTML文件
|
|
|
"""
|
|
|
try:
|
|
|
|
|
|
if '..' in filename or '/' in filename or '\\' in filename:
|
|
|
raise HTTPException(status_code=400, detail="无效的文件名")
|
|
|
|
|
|
file_path = os.path.join("static", filename)
|
|
|
|
|
|
if not os.path.exists(file_path):
|
|
|
raise HTTPException(status_code=404, detail="文件不存在")
|
|
|
|
|
|
os.remove(file_path)
|
|
|
|
|
|
return {"message": f"文件 {filename} 已删除"}
|
|
|
|
|
|
except HTTPException:
|
|
|
raise
|
|
|
except Exception as e:
|
|
|
raise HTTPException(status_code=500, detail=f"删除文件失败: {str(e)}")
|
|
|
|
|
|
@app.get("/", response_class=HTMLResponse)
|
|
|
async def root():
|
|
|
"""
|
|
|
返回前端管理页面
|
|
|
"""
|
|
|
try:
|
|
|
async with aiofiles.open("templates/index.html", 'r', encoding='utf-8') as f:
|
|
|
content = await f.read()
|
|
|
return HTMLResponse(content=content)
|
|
|
except Exception as e:
|
|
|
return HTMLResponse(content=f"<h1>错误</h1><p>无法加载页面: {str(e)}</p>")
|
|
|
|
|
|
@app.get("/api/docs")
|
|
|
async def api_docs():
|
|
|
"""
|
|
|
API文档，返回使用说明
|
|
|
"""
|
|
|
return {
|
|
|
"message": "HTML/Markdown Preview API",
|
|
|
"version": "1.0.0",
|
|
|
"endpoints": {
|
|
|
"html_preview": {
|
|
|
"url": "/api/html/preview",
|
|
|
"method": "POST",
|
|
|
"description": "接收HTML代码，返回在线访问链接",
|
|
|
"request_body": {
|
|
|
"html_content": "string - HTML代码内容"
|
|
|
}
|
|
|
},
|
|
|
"markdown_preview": {
|
|
|
"url": "/api/markdown/preview",
|
|
|
"method": "POST",
|
|
|
"description": "接收Markdown代码，返回渲染后的HTML在线访问链接",
|
|
|
"request_body": {
|
|
|
"markdown_content": "string - Markdown代码内容"
|
|
|
}
|
|
|
},
|
|
|
"files_list": {
|
|
|
"url": "/api/files",
|
|
|
"method": "GET",
|
|
|
"description": "获取所有HTML文件列表"
|
|
|
},
|
|
|
"delete_file": {
|
|
|
"url": "/api/files/{filename}",
|
|
|
"method": "DELETE",
|
|
|
"description": "删除指定的HTML文件"
|
|
|
}
|
|
|
},
|
|
|
"example_usage": {
|
|
|
"html": "curl -X POST http://localhost:8000/api/html/preview -H 'Content-Type: application/json' -d '{\"html_content\": \"<h1>Hello World</h1>\"}'",
|
|
|
"markdown": "curl -X POST http://localhost:8000/api/markdown/preview -H 'Content-Type: application/json' -d '{\"markdown_content\": \"# Hello World\"}'"
|
|
|
}
|
|
|
}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
import uvicorn
|
|
|
uvicorn.run(app, host="0.0.0.0", port=7860) |
