@quybt93/pcgf 0.0.1

package/plugins/phaser-casual-game-factory/skills/game-art/client/remote_proto.py

@@ -0,0 +1,135 @@
+"""Giao thức chung giữa MÁY CHỦ (remote_host.py) và MÁY KHÁCH (remote_backend.py).
+
+Chỉ ba thứ cần dùng chung mới đi qua mạng — và chúng là ba thứ KHÔNG nhân bản được:
+  gemini / codex   phiên đăng nhập tài khoản (chỉ máy chủ mới có cookie + `codex login`)
+  ben2             trọng số + GPU
+  esrgan           trọng số + GPU
+Mọi thứ còn lại (dò màu key, unmix, chấm điểm, cắt bbox, dựng sheet, ghi file) là numpy
+thuần trên file LOCAL nên chạy ngay tại máy khách — nhanh hơn, và không cần tin máy chủ.
+
+QUY TẮC SỐNG CÒN: hàm ở đây KHÔNG BAO GIỜ nhận đường dẫn file. Ảnh đi qua mạng dưới dạng
+PNG base64 nằm trong payload. Đó là toàn bộ lý do server này expose được, còn
+`gemini_image_mcp.py` thì không: 11/12 tool của nó nhận đường dẫn đọc/ghi tuỳ ý, mở ra LAN
+là cấp quyền đọc/ghi ổ đĩa cho mọi máy trong tầm (xem cảnh báo đầu file đó).
+"""
+from __future__ import annotations
+
+import base64
+import io
+import os
+
+from PIL import Image
+
+# Header mang token. Dùng tên riêng thay vì `Authorization` để không đụng lớp auth OAuth
+# mà FastMCP có thể tự xử lý.
+TOKEN_HEADER = "x-gameart-token"
+
+ENV_URL = "GAMEART_REMOTE_URL"      # 1 node, vd http://192.168.1.50:8765/mcp
+ENV_NODES = "GAMEART_REMOTE_NODES"  # NHIỀU node, ngăn bằng dấu phẩy; mỗi mục "url" hoặc "url|token"
+ENV_TOKEN = "GAMEART_REMOTE_TOKEN"  # BẮT BUỘC; là token mặc định cho mọi node không khai riêng
+ENV_TIMEOUT = "GAMEART_REMOTE_TIMEOUT"
+ENV_WAIT = "GAMEART_REMOTE_WAIT"    # mọi node bận thì chờ tối đa bao lâu (giây)
+
+# Chờ tối đa 15 phút rồi báo lại — quá đó thì im lặng chờ tiếp gây hại hơn là dừng: người
+# dùng không biết hệ thống còn sống hay đã treo.
+DEFAULT_WAIT = float(os.environ.get(ENV_WAIT, "900"))
+
+# Trần kích thước một ảnh qua mạng. Ảnh 4096² RGBA PNG ~ vài chục MB; base64 phồng thêm 33%.
+# Chặn ở đây để một lời gọi lỡ tay không làm nghẽn cả máy chủ.
+MAX_IMAGE_BYTES = int(os.environ.get("GAMEART_REMOTE_MAX_BYTES", str(48 * 1024 * 1024)))
+MAX_IMAGES = 8
+
+DEFAULT_TIMEOUT = float(os.environ.get(ENV_TIMEOUT, "600"))
+
+
+class RemoteError(RuntimeError):
+    """Lỗi phía máy chủ hoặc đường truyền — caller bắt để rớt về đường local."""
+
+
+class NodeBusy(RemoteError):
+    """Node đã đầy chỗ. KHÁC HẲN lỗi thường: node vẫn khoẻ, chỉ là bận.
+
+    Tách riêng vì cách xử lý ngược nhau — lỗi thường thì bỏ node đó đi, còn bận thì cứ
+    quay lại sau. Gộp chung là hoặc bỏ oan một node tốt, hoặc quay lại mãi một node hỏng.
+    """
+
+    def __init__(self, msg: str, retry_after: float = 5.0):
+        super().__init__(msg)
+        self.retry_after = retry_after
+
+
+class NodeExhausted(RemoteError):
+    """Node hết quota. Còn sống nhưng vô dụng cho tới khi quota hồi."""
+
+    def __init__(self, msg: str, retry_after: float = 300.0):
+        super().__init__(msg)
+        self.retry_after = retry_after
+
+
+def token() -> str:
+    return (os.environ.get(ENV_TOKEN) or "").strip()
+
+
+def url() -> str:
+    return (os.environ.get(ENV_URL) or "").strip()
+
+
+def nodes() -> list[dict]:
+    """Danh sách node của máy khách, theo THỨ TỰ ƯU TIÊN đã khai.
+
+    Đọc `GAMEART_REMOTE_NODES` (nhiều node) rồi tới `GAMEART_REMOTE_URL` (một node, giữ
+    tương thích bản cũ). Mỗi mục là "url" hoặc "url|token" — token riêng cho phép mỗi máy
+    một mật khẩu thay vì dùng chung một chuỗi cho cả LAN.
+
+    Bỏ qua mục thiếu token: gọi mà chắc chắn bị 401 chỉ làm nhiễu bảng trạng thái.
+    """
+    raw = (os.environ.get(ENV_NODES) or "").strip()
+    items = [x.strip() for x in raw.split(",") if x.strip()] if raw else []
+    if not items and url():
+        items = [url()]
+    default_tok = token()
+    out: list[dict] = []
+    seen: set[str] = set()
+    for it in items:
+        u, _, t = it.partition("|")
+        u, t = u.strip(), (t.strip() or default_tok)
+        if not u or not t or u in seen:
+            continue
+        seen.add(u)
+        out.append({"url": u, "token": t})
+    return out
+
+
+def img_to_b64(img: Image.Image) -> str:
+    """PIL Image -> PNG base64. Giữ nguyên alpha, không nén mất mát."""
+    buf = io.BytesIO()
+    img.save(buf, "PNG", optimize=False)   # optimize=False: nhanh hơn nhiều, mạng LAN không thiếu băng thông
+    raw = buf.getvalue()
+    if len(raw) > MAX_IMAGE_BYTES:
+        raise RemoteError(f"ảnh {len(raw)} byte vượt trần {MAX_IMAGE_BYTES}")
+    return base64.b64encode(raw).decode("ascii")
+
+
+def bytes_to_b64(raw: bytes) -> str:
+    if len(raw) > MAX_IMAGE_BYTES:
+        raise RemoteError(f"ảnh {len(raw)} byte vượt trần {MAX_IMAGE_BYTES}")
+    return base64.b64encode(raw).decode("ascii")
+
+
+def b64_to_bytes(s: str) -> bytes:
+    raw = base64.b64decode(s, validate=True)
+    if len(raw) > MAX_IMAGE_BYTES:
+        raise RemoteError(f"ảnh {len(raw)} byte vượt trần {MAX_IMAGE_BYTES}")
+    return raw
+
+
+def b64_to_img(s: str) -> Image.Image:
+    img = Image.open(io.BytesIO(b64_to_bytes(s)))
+    img.load()
+    return img
+
+
+def path_to_b64(path: str) -> str:
+    """Đọc file LOCAL của máy khách rồi mã hoá. Đọc ở phía KHÁCH, không phải phía chủ."""
+    with open(path, "rb") as f:
+        return bytes_to_b64(f.read())

package/plugins/phaser-casual-game-factory/skills/game-art/client/requirements-client.txt

@@ -0,0 +1,22 @@
+# MÁY KHÁCH (chế độ mượn node trong LAN) — cài bản này, KHÔNG cài requirements.txt.
+#
+#     pip install -r client/requirements-client.txt
+#
+# KHÔNG có torch / ben2 / opencv (~2GB): máy khách không chạy model. BEN2 và Real-ESRGAN
+# chạy trên NODE, gọi qua mạng. Mọi thứ còn lại — dò màu nền, unmix, chấm điểm chọn cách
+# tách nền, chấm điểm chọn cách phóng to, cắt bbox, bám size gốc, dựng sheet — là numpy
+# thuần và vẫn chạy TẠI ĐÂY.
+#
+# Máy khách chỉ cần thư mục `client/` (chứa gemini_image_mcp.py + imglib.py + lớp mượn-node).
+# KHÔNG cần `server/` (code node) hay `models/` (trọng số) — hai thứ đó của node.
+
+mcp==1.12.*
+pillow>=10.0
+# numpy <2 để đồng nhất với node; máy khách không có torch nên bản 2.x cũng chạy được,
+# nhưng giữ chung một mốc thì đỡ phải giải thích hai bộ quy tắc.
+numpy>=1.24,<2
+# gemini_image_mcp.py import curl_cffi ở cấp module (lớp gọi Gemini), nên vẫn phải có
+# dù máy khách không tự gọi Gemini.
+curl_cffi==0.14.*
+# Gọi sang node.
+httpx>=0.27