by-ae
17858e1963
A letter plus a combining accent was rejected, and when it was already in the file it did not match the composed name.
129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
import logging
|
|
import os
|
|
from urllib.parse import quote
|
|
|
|
from aiohttp import web
|
|
|
|
from . import store
|
|
from .nodes.ideogram4_nodes import bg_temp_filename, bg_temp_path, sanitize_bg_guid
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Maximum JSON payload size: 10 MB (generous for store entries)
|
|
MAX_PAYLOAD_SIZE = 10 * 1024 * 1024
|
|
|
|
|
|
async def get_stores(_request: web.Request) -> web.Response:
|
|
try:
|
|
return web.json_response(store.load_all())
|
|
except Exception:
|
|
logger.exception("ByAE Ideogram: failed to load stores")
|
|
return web.json_response({"error": "Failed to load stores"}, status=500)
|
|
|
|
|
|
async def put_store_entry(request: web.Request) -> web.Response:
|
|
bucket = request.match_info.get("bucket", "")
|
|
name = request.match_info.get("name", "")
|
|
if bucket not in ("characters", "objects", "scenes"):
|
|
return web.json_response({"error": "Invalid bucket"}, status=400)
|
|
try:
|
|
name = store._validate_name(name)
|
|
except ValueError as e:
|
|
return web.json_response({"error": str(e)}, status=400)
|
|
|
|
# Check Content-Length to prevent oversized payloads
|
|
content_length = request.content_length
|
|
if content_length is not None and content_length > MAX_PAYLOAD_SIZE:
|
|
return web.json_response(
|
|
{"error": f"Payload too large (max {MAX_PAYLOAD_SIZE} bytes)"},
|
|
status=413
|
|
)
|
|
|
|
try:
|
|
# Read with size limit for safety
|
|
body = await request.read()
|
|
if len(body) > MAX_PAYLOAD_SIZE:
|
|
return web.json_response(
|
|
{"error": f"Payload too large (max {MAX_PAYLOAD_SIZE} bytes)"},
|
|
status=413
|
|
)
|
|
data = await request.json()
|
|
except Exception:
|
|
return web.json_response({"error": "Invalid JSON body"}, status=400)
|
|
if not isinstance(data, dict):
|
|
return web.json_response({"error": "Request body must be a JSON object"}, status=400)
|
|
try:
|
|
store.save_entry(bucket, name, data)
|
|
return web.json_response({"ok": True, "bucket": bucket, "name": name})
|
|
except ValueError as e:
|
|
return web.json_response({"error": str(e)}, status=400)
|
|
except Exception:
|
|
logger.exception("ByAE Ideogram: failed to save store entry")
|
|
return web.json_response({"error": "Failed to save"}, status=500)
|
|
|
|
|
|
async def delete_store_entry(request: web.Request) -> web.Response:
|
|
bucket = request.match_info.get("bucket", "")
|
|
name = request.match_info.get("name", "")
|
|
if bucket not in ("characters", "objects", "scenes"):
|
|
return web.json_response({"error": "Invalid bucket"}, status=400)
|
|
try:
|
|
name = store._validate_name(name)
|
|
except ValueError as e:
|
|
return web.json_response({"error": str(e)}, status=400)
|
|
|
|
try:
|
|
deleted = store.delete_entry(bucket, name)
|
|
if not deleted:
|
|
return web.json_response({"error": "Entry not found"}, status=404)
|
|
return web.json_response({"ok": True, "bucket": bucket, "name": name})
|
|
except ValueError as e:
|
|
return web.json_response({"error": str(e)}, status=400)
|
|
except Exception:
|
|
logger.exception("ByAE Ideogram: failed to delete store entry")
|
|
return web.json_response({"error": "Failed to delete"}, status=500)
|
|
|
|
|
|
async def get_bg_stat(request: web.Request) -> web.Response:
|
|
guid = request.match_info.get("guid", "")
|
|
try:
|
|
sanitize_bg_guid(guid)
|
|
path = bg_temp_path(guid)
|
|
fname = bg_temp_filename(guid)
|
|
except ValueError:
|
|
return web.json_response({"exists": False, "mtime": 0, "error": "invalid guid"}, status=400)
|
|
temp_dir = os.path.dirname(path)
|
|
if not os.path.isfile(path):
|
|
return web.json_response({
|
|
"exists": False,
|
|
"mtime": 0,
|
|
"path": path,
|
|
"filename": fname,
|
|
"temp_dir": temp_dir,
|
|
})
|
|
try:
|
|
mtime = os.path.getmtime(path)
|
|
except (OSError, IOError) as e:
|
|
logger.warning("ByAE Ideogram: failed to stat background file %s: %s", path, e)
|
|
return web.json_response({"exists": False, "mtime": 0, "error": "stat failed"}, status=500)
|
|
url = f"/view?filename={quote(fname)}&type=temp&subfolder="
|
|
return web.json_response({
|
|
"exists": True,
|
|
"mtime": mtime,
|
|
"url": url,
|
|
"path": path,
|
|
"filename": fname,
|
|
"temp_dir": temp_dir,
|
|
})
|
|
|
|
|
|
def register_routes(app) -> None:
|
|
if app.router.frozen:
|
|
return
|
|
app.add_routes([
|
|
web.get("/byae/ideogram/stores", get_stores),
|
|
web.put("/byae/ideogram/stores/{bucket}/{name}", put_store_entry),
|
|
web.delete("/byae/ideogram/stores/{bucket}/{name}", delete_store_entry),
|
|
web.get("/byae/ideogram/bg/{guid}/stat", get_bg_stat),
|
|
])
|