virtus
1b1217ace2
Build and Push Reader API Image / docker (push) Successful in 26s
chore(docker): remove MongoDB service and related configurations from local setup feat(migrations): create ChapterMeta table and add search_name, size_bytes, mtime_epoch, lastScannedAt, review_status, and review_payload columns to SourceAsset chore(dependencies): remove motor and pymongo from project dependencies
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import hashlib
|
|
from pathlib import Path
|
|
|
|
from app.config import settings
|
|
|
|
|
|
class NasContentStorage:
|
|
def __init__(self, root_dir: str):
|
|
self.root = Path(root_dir).resolve()
|
|
self.root.mkdir(parents=True, exist_ok=True)
|
|
|
|
def _resolve(self, href: str) -> Path:
|
|
rel = href.strip().lstrip("/")
|
|
target = (self.root / rel).resolve()
|
|
if self.root not in target.parents and target != self.root:
|
|
raise ValueError("Invalid storage href")
|
|
return target
|
|
|
|
def read_text(self, href: str) -> str:
|
|
path = self._resolve(href)
|
|
return path.read_text(encoding="utf-8")
|
|
|
|
def write_text(self, href: str, content: str) -> dict[str, str | int]:
|
|
path = self._resolve(href)
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
path.write_text(content, encoding="utf-8")
|
|
digest = hashlib.sha256(content.encode("utf-8")).hexdigest()
|
|
return {"href": href, "sha256": digest, "size": len(content.encode("utf-8"))}
|
|
|
|
def delete_href(self, href: str) -> bool:
|
|
path = self._resolve(href)
|
|
if not path.exists() or not path.is_file():
|
|
return False
|
|
path.unlink(missing_ok=True)
|
|
parent = path.parent
|
|
while parent != self.root:
|
|
try:
|
|
parent.rmdir()
|
|
except OSError:
|
|
break
|
|
parent = parent.parent
|
|
return True
|
|
|
|
|
|
storage = NasContentStorage(settings.nas_content_root)
|