Initial commit: sprite tool project
This commit is contained in:
82
tool.py
Normal file
82
tool.py
Normal file
@@ -0,0 +1,82 @@
|
||||
import uvicorn
|
||||
import re
|
||||
from fastapi import FastAPI, Request
|
||||
from fastapi.responses import HTMLResponse, Response
|
||||
from starlette.middleware.base import BaseHTTPMiddleware
|
||||
|
||||
# 匯入原本的三個檔案
|
||||
from sprite_tool_fullstack import app as grid_app
|
||||
from sprite_webtool import app as picker_app
|
||||
from shiny_maker import app as shiny_app
|
||||
|
||||
app = FastAPI(title="Game Dev Suite")
|
||||
|
||||
# --- 定義導覽列 HTML ---
|
||||
NAVBAR_HTML = """
|
||||
<nav class="bg-slate-800 border-b border-slate-700 px-6 py-3 flex items-center gap-6 sticky top-0 z-50 shadow-lg font-sans">
|
||||
<span class="font-bold text-blue-400 mr-4">DEV SUITE</span>
|
||||
<a href="/" class="text-sm text-slate-300 hover:text-blue-400 transition">🏠 Home</a>
|
||||
<a href="/grid/" class="text-sm text-slate-300 hover:text-blue-400 transition">📏 Grid Tool</a>
|
||||
<a href="/picker/" class="text-sm text-slate-300 hover:text-blue-400 transition">🎯 Picker Tool</a>
|
||||
<a href="/shiny/" class="text-sm text-slate-300 hover:text-blue-400 transition">✨ Shiny Maker</a>
|
||||
</nav>
|
||||
"""
|
||||
|
||||
class NavbarMiddleware(BaseHTTPMiddleware):
|
||||
async def dispatch(self, request: Request, call_next):
|
||||
response = await call_next(request)
|
||||
|
||||
# 排除首頁,避免重複注入
|
||||
if request.url.path == "/":
|
||||
return response
|
||||
|
||||
# 針對所有子工具的 HTML 回傳進行注入
|
||||
if "text/html" in response.headers.get("content-type", ""):
|
||||
body = b""
|
||||
async for chunk in response.body_iterator:
|
||||
body += chunk
|
||||
|
||||
html_content = body.decode("utf-8")
|
||||
|
||||
# 使用正則表達式匹配 <body> 標籤,不論它有沒有帶 class 或其他屬性
|
||||
# 這會匹配 <body ...> 並在其後方插入 Navbar
|
||||
new_content = re.sub(r'(<body[^>]*>)', r'\1' + NAVBAR_HTML, html_content, flags=re.IGNORECASE)
|
||||
|
||||
return HTMLResponse(content=new_content, status_code=response.status_code)
|
||||
|
||||
return response
|
||||
|
||||
app.add_middleware(NavbarMiddleware)
|
||||
|
||||
# --- 掛載子應用程式 ---
|
||||
app.mount("/grid", grid_app)
|
||||
app.mount("/picker", picker_app)
|
||||
app.mount("/shiny", shiny_app)
|
||||
|
||||
# 首頁入口
|
||||
@app.get("/", response_class=HTMLResponse)
|
||||
async def index():
|
||||
return f"""
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<meta charset="utf-8">
|
||||
<script src="https://cdn.tailwindcss.com"></script>
|
||||
</head>
|
||||
<body class="bg-slate-900 text-white">
|
||||
{NAVBAR_HTML}
|
||||
<div class="flex flex-col items-center justify-center min-h-[80vh] space-y-6">
|
||||
<h1 class="text-4xl font-bold text-blue-400">Game Developer Tool Suite</h1>
|
||||
<p class="text-slate-400">請從上方導覽列選擇要使用的工具</p>
|
||||
<div class="grid grid-cols-3 gap-6 pt-10">
|
||||
<div class="p-6 bg-slate-800 rounded-2xl border border-slate-700 text-center">📏 Grid Tool</div>
|
||||
<div class="p-6 bg-slate-800 rounded-2xl border border-slate-700 text-center">🎯 Picker Tool</div>
|
||||
<div class="p-6 bg-slate-800 rounded-2xl border border-slate-700 text-center">✨ Shiny Maker</div>
|
||||
</div>
|
||||
</div>
|
||||
</body>
|
||||
</html>
|
||||
"""
|
||||
|
||||
if __name__ == "__main__":
|
||||
uvicorn.run(app, host="0.0.0.0", port=8000)
|
||||
Reference in New Issue
Block a user