64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
import time
|
||
|
|
import uuid
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import Any, Dict, List, Optional
|
||
|
|
|
||
|
|
|
||
|
|
def repl_queue_dir(root: Path) -> Path:
|
||
|
|
return Path(root) / "Log" / "repl_queue"
|
||
|
|
|
||
|
|
|
||
|
|
def enqueue_repl_command(
|
||
|
|
root: Path,
|
||
|
|
command: str,
|
||
|
|
*,
|
||
|
|
source: str = "external",
|
||
|
|
metadata: Optional[Dict[str, Any]] = None,
|
||
|
|
) -> Path:
|
||
|
|
queue_dir = repl_queue_dir(root)
|
||
|
|
queue_dir.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
payload: Dict[str, Any] = {
|
||
|
|
"id": uuid.uuid4().hex,
|
||
|
|
"command": str(command or "").strip(),
|
||
|
|
"source": str(source or "external").strip() or "external",
|
||
|
|
"created_at": time.time(),
|
||
|
|
}
|
||
|
|
if isinstance(metadata, dict) and metadata:
|
||
|
|
payload["metadata"] = metadata
|
||
|
|
|
||
|
|
stamp = int(time.time() * 1000)
|
||
|
|
token = payload["id"][:8]
|
||
|
|
final_path = queue_dir / f"{stamp:013d}-{token}.json"
|
||
|
|
temp_path = final_path.with_suffix(".tmp")
|
||
|
|
temp_path.write_text(json.dumps(payload, ensure_ascii=False), encoding="utf-8")
|
||
|
|
temp_path.replace(final_path)
|
||
|
|
return final_path
|
||
|
|
|
||
|
|
|
||
|
|
def pop_repl_commands(root: Path, *, limit: int = 8) -> List[Dict[str, Any]]:
|
||
|
|
queue_dir = repl_queue_dir(root)
|
||
|
|
if not queue_dir.exists():
|
||
|
|
return []
|
||
|
|
|
||
|
|
items: List[Dict[str, Any]] = []
|
||
|
|
for entry in sorted(queue_dir.glob("*.json"))[: max(1, int(limit or 1))]:
|
||
|
|
try:
|
||
|
|
payload = json.loads(entry.read_text(encoding="utf-8"))
|
||
|
|
except Exception:
|
||
|
|
payload = {
|
||
|
|
"id": entry.stem,
|
||
|
|
"command": "",
|
||
|
|
"source": "invalid",
|
||
|
|
"created_at": entry.stat().st_mtime,
|
||
|
|
}
|
||
|
|
try:
|
||
|
|
entry.unlink()
|
||
|
|
except Exception:
|
||
|
|
continue
|
||
|
|
if isinstance(payload, dict):
|
||
|
|
items.append(payload)
|
||
|
|
return items
|