103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
from fastapi import FastAPI, Request, HTTPException
|
|
from fastapi.responses import HTMLResponse, FileResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from fastapi.staticfiles import StaticFiles
|
|
from jinja2 import Environment, FileSystemLoader, select_autoescape, TemplateNotFound
|
|
import os
|
|
import markdown
|
|
import re
|
|
|
|
# Get BASE_DIR from environment variables with a default fallback
|
|
BASE_DIR = os.environ.get('BASE_DIR', os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
# Check if BASE_DIR exists
|
|
if not os.path.exists(BASE_DIR):
|
|
raise RuntimeError(f"The BASE_DIR '{BASE_DIR}' does not exist.")
|
|
|
|
sources_dir = os.path.expanduser(f"{BASE_DIR}/poc")
|
|
if not os.path.exists(sources_dir):
|
|
raise RuntimeError(f"The source directory '{sources_dir}' does not exist.")
|
|
static_dir = f"{sources_dir}/static"
|
|
content_dir = f"{sources_dir}/content"
|
|
|
|
def get_content(name: str) -> str:
|
|
"""Get content by name from either HTML or markdown files in content directory"""
|
|
# Remove any leading/trailing slashes
|
|
name = name.strip('/')
|
|
|
|
# Check for file with .html extension
|
|
html_path = os.path.join(content_dir, f"{name}.html")
|
|
if os.path.exists(html_path):
|
|
with open(html_path, 'r') as f:
|
|
return f.read()
|
|
|
|
# Check for file with .md extension
|
|
md_path = os.path.join(content_dir, f"{name}.md")
|
|
if os.path.exists(md_path):
|
|
with open(md_path, 'r') as f:
|
|
content = f.read()
|
|
return markdown.markdown(content)
|
|
|
|
return f"[[{name} not found]]"
|
|
|
|
def process_content(content: str) -> str:
|
|
"""Process content and replace [[name]] with corresponding HTML content"""
|
|
def replace_content(match):
|
|
name = match.group(1)
|
|
return get_content(name)
|
|
|
|
# Replace all [[name]] patterns
|
|
return re.sub(r'\[\[(.*?)\]\]', replace_content, content)
|
|
|
|
app = FastAPI()
|
|
|
|
if not os.path.exists(static_dir):
|
|
raise RuntimeError(f"The directory '{static_dir}' does not exist.")
|
|
|
|
if not os.path.exists(sources_dir):
|
|
raise RuntimeError(f"The templates dir '{sources_dir}' does not exist.")
|
|
|
|
# Mount the static files directory
|
|
app.mount("/static", StaticFiles(directory=static_dir), name="static")
|
|
|
|
env = Environment(
|
|
loader=FileSystemLoader(sources_dir),
|
|
autoescape=select_autoescape(['html', 'xml'])
|
|
)
|
|
|
|
# Initialize Jinja2 templates
|
|
templates = Jinja2Templates(directory=sources_dir)
|
|
|
|
@app.get("/favicon.ico")
|
|
async def favicon():
|
|
# First try to serve from static directory
|
|
favicon_path = os.path.join(static_dir, "favicon.ico")
|
|
if os.path.exists(favicon_path):
|
|
return FileResponse(favicon_path)
|
|
# If not found, return 404
|
|
raise HTTPException(status_code=404, detail="Favicon not found")
|
|
|
|
@app.get("/", response_class=HTMLResponse)
|
|
async def read_index(request: Request):
|
|
template = env.get_template("index.html")
|
|
content = template.render(request=request)
|
|
return process_content(content)
|
|
|
|
@app.get("/{path:path}", response_class=HTMLResponse)
|
|
async def read_template(request: Request, path: str):
|
|
# Add .html extension if not present
|
|
if not path.endswith('.html'):
|
|
path = f"{path}.html"
|
|
|
|
try:
|
|
# Try to load and render the template (this will work for both direct files and templates)
|
|
template = env.get_template(path)
|
|
content = template.render(request=request)
|
|
return process_content(content)
|
|
except TemplateNotFound:
|
|
raise HTTPException(status_code=404, detail=f"Template {path} not found")
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run("server:app", host="127.0.0.1", port=8001, reload=True)
|