重构路由架构

backend/api/routes/worldbooksRoute.py

@@ -0,0 +1,170 @@
+from fastapi import APIRouter, HTTPException, status
+from pathlib import Path
+from typing import Dict, List
+from backend.core.models.WorldBook import WorldBook
+from backend.core.models.WorldItem import WorldInfoEntry
+
+router = APIRouter(prefix="/worldbooks", tags=["worldbooks"])
+
+
+# ========== 世界书基础路由 ==========
+
+@router.get("", response_model=Dict[str, List[Dict]])
+async def list_worldbooks():
+	"""获取所有世界书列表"""
+	worldbook_dir = Path("data/worldbooks")
+	if not worldbook_dir.exists():
+		return {"worldbooks": []}
+
+	worldbooks = []
+	for wb_file in worldbook_dir.glob("*.json"):
+		try:
+			worldbook = WorldBook.from_sillytavern_json(str(wb_file))
+			worldbooks.append(worldbook.to_summary_dict())
+		except Exception as e:
+			print(f"警告: 跳过文件 {wb_file.name}，加载失败: {e}")
+			continue
+
+	return {"worldbooks": worldbooks}
+
+
+@router.get("/{worldbook_uid}")
+async def get_worldbook(worldbook_uid: str):
+	"""获取指定世界书完整内容"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+
+	try:
+		worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
+		return worldbook.to_dict()
+	except Exception as e:
+		raise HTTPException(status_code=500, detail=f"Failed to load worldbook: {str(e)}")
+
+
+@router.post("", status_code=status.HTTP_201_CREATED)
+async def create_worldbook(worldbook_data: Dict):
+	"""创建新世界书"""
+	worldbook_dir = Path("data/worldbooks")
+	worldbook_dir.mkdir(parents=True, exist_ok=True)
+
+	worldbook = WorldBook.from_dict(worldbook_data)
+	worldbook_path = worldbook_dir / f"{worldbook.uid}.json"
+
+	if worldbook_path.exists():
+		raise HTTPException(status_code=400, detail="WorldBook already exists")
+
+	worldbook.to_sillytavern_json(str(worldbook_path))
+	return {"message": "WorldBook created successfully", "uid": worldbook.uid}
+
+
+@router.put("/{worldbook_uid}")
+async def update_worldbook(worldbook_uid: str, update_data: Dict):
+	"""更新世界书基本信息"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+
+	try:
+		worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
+		for key, value in update_data.items():
+			if hasattr(worldbook, key):
+				setattr(worldbook, key, value)
+		worldbook.to_sillytavern_json(str(worldbook_path))
+		return {"message": "WorldBook updated successfully"}
+	except Exception as e:
+		raise HTTPException(status_code=500, detail=f"Failed to update worldbook: {str(e)}")
+
+
+@router.delete("/{worldbook_uid}")
+async def delete_worldbook(worldbook_uid: str):
+	"""删除世界书"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+	worldbook_path.unlink()
+	return {"message": "WorldBook deleted successfully"}
+
+
+# ========== 世界书条目路由 ==========
+
+@router.get("/{worldbook_uid}/entries")
+async def list_worldbook_entries(worldbook_uid: str):
+	"""获取世界书所有条目列表"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+
+	try:
+		worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
+		return {"entries": [entry.dict() for entry in worldbook.entries]}
+	except Exception as e:
+		raise HTTPException(status_code=500, detail=f"Failed to load entries: {str(e)}")
+
+
+@router.get("/{worldbook_uid}/entries/{entry_uid}")
+async def get_worldbook_entry(worldbook_uid: str, entry_uid: str):
+	"""获取指定条目详情"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+
+	try:
+		worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
+		entry = worldbook.get_entry(entry_uid)
+		if not entry:
+			raise HTTPException(status_code=404, detail="Entry not found")
+		return entry.dict()
+	except Exception as e:
+		raise HTTPException(status_code=500, detail=f"Failed to load entry: {str(e)}")
+
+
+@router.post("/{worldbook_uid}/entries", status_code=status.HTTP_201_CREATED)
+async def add_worldbook_entry(worldbook_uid: str, entry_data: Dict):
+	"""向世界书添加新条目"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+
+	try:
+		worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
+		entry = WorldInfoEntry(**entry_data)
+		worldbook.add_entry(entry)
+		worldbook.to_sillytavern_json(str(worldbook_path))
+		return {"message": "Entry added successfully", "entry_uid": entry.uid}
+	except Exception as e:
+		raise HTTPException(status_code=500, detail=f"Failed to add entry: {str(e)}")
+
+
+@router.put("/{worldbook_uid}/entries/{entry_uid}")
+async def update_worldbook_entry(worldbook_uid: str, entry_uid: str, update_data: Dict):
+	"""更新指定条目"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+
+	try:
+		worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
+		if not worldbook.update_entry(entry_uid, **update_data):
+			raise HTTPException(status_code=404, detail="Entry not found")
+		worldbook.to_sillytavern_json(str(worldbook_path))
+		return {"message": "Entry updated successfully"}
+	except Exception as e:
+		raise HTTPException(status_code=500, detail=f"Failed to update entry: {str(e)}")
+
+
+@router.delete("/{worldbook_uid}/entries/{entry_uid}")
+async def delete_worldbook_entry(worldbook_uid: str, entry_uid: str):
+	"""从世界书删除指定条目"""
+	worldbook_path = Path("data/worldbooks") / f"{worldbook_uid}.json"
+	if not worldbook_path.exists():
+		raise HTTPException(status_code=404, detail="WorldBook not found")
+
+	try:
+		worldbook = WorldBook.from_sillytavern_json(str(worldbook_path))
+		if not worldbook.remove_entry(entry_uid):
+			raise HTTPException(status_code=404, detail="Entry not found")
+		worldbook.to_sillytavern_json(str(worldbook_path))
+		return {"message": "Entry deleted successfully"}
+	except Exception as e:
+		raise HTTPException(status_code=500, detail=f"Failed to delete entry: {str(e)}")