257 lines
11 KiB
Python
257 lines
11 KiB
Python
|
# ============================================================
|
|||
|
|
# 项目 / 会话服务端仓（moduleId: state-projects, AG-08 首切片，可重生 ✅）
|
|||
|
|
# 规则：项目/会话/消息持久化到 JSON；删除项目不触碰世界状态
|
|||
|
|
# ============================================================
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import tempfile
|
|||
|
|
import threading
|
|||
|
|
import uuid
|
|||
|
|
from datetime import datetime
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
from server.timeutil import fmt_dt
|
|||
|
|
|
|||
|
|
PERSONAL_PROJECT_ID = "__personal__"
|
|||
|
|
World = dict[str, Any]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _now() -> str:
|
|||
|
|
return fmt_dt(datetime.now())
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _uid(prefix: str) -> str:
|
|||
|
|
return f"{prefix}_{uuid.uuid4().hex[:8]}"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _seed() -> dict[str, Any]:
|
|||
|
|
sess = {
|
|||
|
|
"id": _uid("sess"),
|
|||
|
|
"projectId": PERSONAL_PROJECT_ID,
|
|||
|
|
"title": "新话题",
|
|||
|
|
"status": "running",
|
|||
|
|
"updatedAt": _now(),
|
|||
|
|
}
|
|||
|
|
return {
|
|||
|
|
"projects": [],
|
|||
|
|
"sessions": [sess],
|
|||
|
|
"files": [],
|
|||
|
|
"messages": {},
|
|||
|
|
"activeProjectId": PERSONAL_PROJECT_ID,
|
|||
|
|
"activeSessionId": sess["id"],
|
|||
|
|
"worldKey": "default", # 与当前全局 WorldStore 绑定（单世界）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ProjectStore:
|
|||
|
|
"""项目制工作区：项目/会话/文件/消息（单文件 JSON + 原子写）。"""
|
|||
|
|
|
|||
|
|
def __init__(self, path: str | None = None) -> None:
|
|||
|
|
self.path = path or os.environ.get("APS_PROJECTS_PATH", "server/data/projects.json")
|
|||
|
|
self._lock = threading.Lock()
|
|||
|
|
self.data: dict[str, Any] = self._load()
|
|||
|
|
|
|||
|
|
def _load(self) -> dict[str, Any]:
|
|||
|
|
try:
|
|||
|
|
with open(self.path, "r", encoding="utf-8") as f:
|
|||
|
|
raw = json.load(f)
|
|||
|
|
if not isinstance(raw, dict) or not raw.get("sessions"):
|
|||
|
|
data = _seed()
|
|||
|
|
self._write(data)
|
|||
|
|
return data
|
|||
|
|
raw.setdefault("projects", [])
|
|||
|
|
raw.setdefault("files", [])
|
|||
|
|
raw.setdefault("messages", {})
|
|||
|
|
raw.setdefault("worldKey", "default")
|
|||
|
|
raw.setdefault("activeProjectId", PERSONAL_PROJECT_ID)
|
|||
|
|
raw.setdefault("activeSessionId", raw["sessions"][0]["id"])
|
|||
|
|
return raw
|
|||
|
|
except (FileNotFoundError, json.JSONDecodeError, TypeError):
|
|||
|
|
data = _seed()
|
|||
|
|
self._write(data)
|
|||
|
|
return data
|
|||
|
|
|
|||
|
|
def _write(self, data: dict[str, Any]) -> None:
|
|||
|
|
os.makedirs(os.path.dirname(self.path) or ".", exist_ok=True)
|
|||
|
|
fd, tmp = tempfile.mkstemp(dir=os.path.dirname(self.path) or ".", suffix=".tmp")
|
|||
|
|
try:
|
|||
|
|
with os.fdopen(fd, "w", encoding="utf-8") as f:
|
|||
|
|
json.dump(data, f, ensure_ascii=False, indent=1)
|
|||
|
|
os.replace(tmp, self.path)
|
|||
|
|
except BaseException:
|
|||
|
|
if os.path.exists(tmp):
|
|||
|
|
os.unlink(tmp)
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
def save(self) -> None:
|
|||
|
|
with self._lock:
|
|||
|
|
self._write(self.data)
|
|||
|
|
|
|||
|
|
def snapshot(self, *, include_messages: bool = True) -> dict[str, Any]:
|
|||
|
|
"""只读投影（P0）。"""
|
|||
|
|
with self._lock:
|
|||
|
|
return self._snapshot_unlocked(include_messages=include_messages)
|
|||
|
|
|
|||
|
|
def _snapshot_unlocked(self, *, include_messages: bool = True) -> dict[str, Any]:
|
|||
|
|
out = {
|
|||
|
|
"projects": list(self.data.get("projects") or []),
|
|||
|
|
"sessions": list(self.data.get("sessions") or []),
|
|||
|
|
"files": list(self.data.get("files") or []),
|
|||
|
|
"activeProjectId": self.data.get("activeProjectId"),
|
|||
|
|
"activeSessionId": self.data.get("activeSessionId"),
|
|||
|
|
"worldKey": self.data.get("worldKey") or "default",
|
|||
|
|
}
|
|||
|
|
if include_messages:
|
|||
|
|
out["messages"] = dict(self.data.get("messages") or {})
|
|||
|
|
return out
|
|||
|
|
|
|||
|
|
def replace_workspace(self, payload: dict[str, Any]) -> dict[str, Any]:
|
|||
|
|
"""整包替换工作区元数据（可含 messages）；不碰 WorldStore。"""
|
|||
|
|
with self._lock:
|
|||
|
|
projects = list(payload.get("projects") or [])
|
|||
|
|
sessions = list(payload.get("sessions") or [])
|
|||
|
|
files = list(payload.get("files") or [])
|
|||
|
|
if not sessions:
|
|||
|
|
seeded = _seed()
|
|||
|
|
sessions = seeded["sessions"]
|
|||
|
|
payload = {**payload, "activeSessionId": sessions[0]["id"],
|
|||
|
|
"activeProjectId": PERSONAL_PROJECT_ID}
|
|||
|
|
messages = payload.get("messages")
|
|||
|
|
if messages is None:
|
|||
|
|
messages = self.data.get("messages") or {}
|
|||
|
|
# 清掉已删除会话的消息，防泄漏
|
|||
|
|
keep = {s["id"] for s in sessions if isinstance(s, dict) and s.get("id")}
|
|||
|
|
messages = {k: v for k, v in dict(messages).items() if k in keep}
|
|||
|
|
self.data = {
|
|||
|
|
"projects": projects,
|
|||
|
|
"sessions": sessions,
|
|||
|
|
"files": files,
|
|||
|
|
"messages": messages,
|
|||
|
|
"activeProjectId": payload.get("activeProjectId") or PERSONAL_PROJECT_ID,
|
|||
|
|
"activeSessionId": payload.get("activeSessionId") or sessions[0]["id"],
|
|||
|
|
"worldKey": payload.get("worldKey") or self.data.get("worldKey") or "default",
|
|||
|
|
}
|
|||
|
|
self._write(self.data)
|
|||
|
|
return self._snapshot_unlocked(include_messages=True)
|
|||
|
|
|
|||
|
|
def create_project(self, name: str, *, scope_label: str = "未设定作用域") -> dict[str, Any]:
|
|||
|
|
with self._lock:
|
|||
|
|
pid = _uid("proj")
|
|||
|
|
sid = _uid("sess")
|
|||
|
|
project = {
|
|||
|
|
"id": pid,
|
|||
|
|
"name": (name or "").strip() or "未命名项目",
|
|||
|
|
"scopeLabel": scope_label or "未设定作用域",
|
|||
|
|
"createdAt": _now(),
|
|||
|
|
"worldKey": self.data.get("worldKey") or "default",
|
|||
|
|
}
|
|||
|
|
session = {
|
|||
|
|
"id": sid, "projectId": pid, "title": "新话题",
|
|||
|
|
"status": "running", "updatedAt": _now(),
|
|||
|
|
}
|
|||
|
|
self.data["projects"] = [project, *(self.data.get("projects") or [])]
|
|||
|
|
self.data["sessions"] = [session, *(self.data.get("sessions") or [])]
|
|||
|
|
self.data["activeProjectId"] = pid
|
|||
|
|
self.data["activeSessionId"] = sid
|
|||
|
|
self._write(self.data)
|
|||
|
|
return {"project": project, "session": session}
|
|||
|
|
|
|||
|
|
def delete_project(self, project_id: str) -> dict[str, Any]:
|
|||
|
|
"""删除项目元数据与其会话/文件/消息；绝不修改世界状态。"""
|
|||
|
|
if project_id == PERSONAL_PROJECT_ID:
|
|||
|
|
raise ValueError("个人话题作用域不可删除")
|
|||
|
|
with self._lock:
|
|||
|
|
before = next((p for p in self.data.get("projects") or [] if p["id"] == project_id), None)
|
|||
|
|
if before is None:
|
|||
|
|
raise ValueError(f"项目不存在：{project_id}")
|
|||
|
|
drop_sessions = {s["id"] for s in self.data.get("sessions") or []
|
|||
|
|
if s.get("projectId") == project_id}
|
|||
|
|
self.data["projects"] = [p for p in self.data["projects"] if p["id"] != project_id]
|
|||
|
|
self.data["sessions"] = [s for s in self.data["sessions"] if s["id"] not in drop_sessions]
|
|||
|
|
self.data["files"] = [f for f in self.data.get("files") or []
|
|||
|
|
if f.get("projectId") != project_id]
|
|||
|
|
msgs = dict(self.data.get("messages") or {})
|
|||
|
|
for sid in drop_sessions:
|
|||
|
|
msgs.pop(sid, None)
|
|||
|
|
self.data["messages"] = msgs
|
|||
|
|
if not self.data["sessions"]:
|
|||
|
|
fresh = {
|
|||
|
|
"id": _uid("sess"), "projectId": PERSONAL_PROJECT_ID,
|
|||
|
|
"title": "新话题", "status": "running", "updatedAt": _now(),
|
|||
|
|
}
|
|||
|
|
self.data["sessions"] = [fresh]
|
|||
|
|
if self.data.get("activeProjectId") == project_id:
|
|||
|
|
self.data["activeProjectId"] = (
|
|||
|
|
self.data["projects"][0]["id"] if self.data["projects"] else PERSONAL_PROJECT_ID)
|
|||
|
|
if self.data.get("activeSessionId") in drop_sessions or not self.data.get("activeSessionId"):
|
|||
|
|
scope = self.data["activeProjectId"]
|
|||
|
|
sib = next((s for s in self.data["sessions"] if s["projectId"] == scope), None)
|
|||
|
|
self.data["activeSessionId"] = (sib or self.data["sessions"][0])["id"]
|
|||
|
|
self._write(self.data)
|
|||
|
|
return {
|
|||
|
|
"deletedProjectId": project_id,
|
|||
|
|
"deletedSessionIds": sorted(drop_sessions),
|
|||
|
|
"worldUntouched": True,
|
|||
|
|
"worldKey": self.data.get("worldKey") or "default",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def create_session(self, project_id: str, title: str = "新话题") -> dict[str, Any]:
|
|||
|
|
with self._lock:
|
|||
|
|
if project_id != PERSONAL_PROJECT_ID:
|
|||
|
|
if not any(p["id"] == project_id for p in self.data.get("projects") or []):
|
|||
|
|
raise ValueError(f"项目不存在：{project_id}")
|
|||
|
|
session = {
|
|||
|
|
"id": _uid("sess"),
|
|||
|
|
"projectId": project_id,
|
|||
|
|
"title": (title or "").strip() or "新话题",
|
|||
|
|
"status": "running",
|
|||
|
|
"updatedAt": _now(),
|
|||
|
|
}
|
|||
|
|
self.data["sessions"] = [session, *(self.data.get("sessions") or [])]
|
|||
|
|
self.data["activeProjectId"] = project_id
|
|||
|
|
self.data["activeSessionId"] = session["id"]
|
|||
|
|
self._write(self.data)
|
|||
|
|
return {"session": session}
|
|||
|
|
|
|||
|
|
def replace_messages(self, session_id: str, messages: list[Any]) -> dict[str, Any]:
|
|||
|
|
with self._lock:
|
|||
|
|
if not any(s["id"] == session_id for s in self.data.get("sessions") or []):
|
|||
|
|
raise ValueError(f"会话不存在：{session_id}")
|
|||
|
|
self.data.setdefault("messages", {})[session_id] = list(messages or [])
|
|||
|
|
for s in self.data["sessions"]:
|
|||
|
|
if s["id"] == session_id:
|
|||
|
|
s["updatedAt"] = _now()
|
|||
|
|
break
|
|||
|
|
self._write(self.data)
|
|||
|
|
return {"sessionId": session_id, "count": len(messages or [])}
|
|||
|
|
|
|||
|
|
def append_message(self, session_id: str, message: dict[str, Any]) -> dict[str, Any]:
|
|||
|
|
with self._lock:
|
|||
|
|
if not any(s["id"] == session_id for s in self.data.get("sessions") or []):
|
|||
|
|
raise ValueError(f"会话不存在：{session_id}")
|
|||
|
|
bucket = list((self.data.get("messages") or {}).get(session_id) or [])
|
|||
|
|
bucket.append(message)
|
|||
|
|
self.data.setdefault("messages", {})[session_id] = bucket
|
|||
|
|
for s in self.data["sessions"]:
|
|||
|
|
if s["id"] == session_id:
|
|||
|
|
s["updatedAt"] = _now()
|
|||
|
|
break
|
|||
|
|
self._write(self.data)
|
|||
|
|
return {"sessionId": session_id, "count": len(bucket)}
|
|||
|
|
|
|||
|
|
def get_messages(self, session_id: str) -> list[Any]:
|
|||
|
|
with self._lock:
|
|||
|
|
return list((self.data.get("messages") or {}).get(session_id) or [])
|
|||
|
|
|
|||
|
|
|
|||
|
|
_store: ProjectStore | None = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_project_store() -> ProjectStore:
|
|||
|
|
global _store
|
|||
|
|
if _store is None:
|
|||
|
|
_store = ProjectStore()
|
|||
|
|
return _store
|