|
|
|
@@ -1,496 +0,0 @@
|
|
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
from __future__ import annotations
|
|
|
|
|
|
|
|
|
|
import html
|
|
|
|
|
import json
|
|
|
|
|
import os
|
|
|
|
|
import subprocess
|
|
|
|
|
import time
|
|
|
|
|
import urllib.parse
|
|
|
|
|
import urllib.request
|
|
|
|
|
from datetime import datetime, timedelta, timezone
|
|
|
|
|
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
|
|
SERVICE = "manual-media-import.service"
|
|
|
|
|
TIMER = "manual-media-import.timer"
|
|
|
|
|
LOG = Path("/var/log/manual-media-import.log")
|
|
|
|
|
IMPORTER_STATUS = Path("/run/manual-media-import/status.json")
|
|
|
|
|
MANUAL_BATCHES = Path("/var/lib/importarr/manual-batches.json")
|
|
|
|
|
QUEUE_ROOTS = {
|
|
|
|
|
"manual": Path(os.getenv("IMPORTARR_DOWNLOAD_ROOT", "/data/downloads/manual")),
|
|
|
|
|
"legacy": Path(os.getenv("IMPORTARR_LEGACY_DOWNLOAD_ROOT", "/data/downloads/legacy")),
|
|
|
|
|
}
|
|
|
|
|
VIDEO_EXT = {".mkv", ".mp4", ".m4v", ".avi", ".mov", ".wmv", ".mpg", ".mpeg", ".ts", ".m2ts", ".webm"}
|
|
|
|
|
SAB_CONFIG = Path(os.getenv("IMPORTARR_SABNZBD_CONFIG", "/config/sabnzbd/sabnzbd.ini"))
|
|
|
|
|
SAB_API = os.getenv("IMPORTARR_SABNZBD_URL", "http://sabnzbd:8080/api")
|
|
|
|
|
LONG_RUNTIME_SECONDS = 25 * 60
|
|
|
|
|
HIGH_MEMORY_BYTES = 8 * 1024**3
|
|
|
|
|
STALE_QUEUE_SECONDS = 6 * 60 * 60
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def run(args: list[str]) -> str:
|
|
|
|
|
return subprocess.run(args, check=False, text=True, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL).stdout
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def systemctl_show(unit: str) -> dict[str, str]:
|
|
|
|
|
data = {}
|
|
|
|
|
for line in run(["systemctl", "show", unit, "--no-pager"]).splitlines():
|
|
|
|
|
if "=" in line:
|
|
|
|
|
key, value = line.split("=", 1)
|
|
|
|
|
data[key] = value
|
|
|
|
|
return data
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def int_value(value: str | None) -> int | None:
|
|
|
|
|
try:
|
|
|
|
|
return int(value or "")
|
|
|
|
|
except ValueError:
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def timestamp_to_iso(usec: str | None) -> str | None:
|
|
|
|
|
value = int_value(usec)
|
|
|
|
|
if not value or value <= 0:
|
|
|
|
|
return None
|
|
|
|
|
return datetime.fromtimestamp(value / 1_000_000, tz=timezone.utc).isoformat()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def monotonic_runtime_seconds(service: dict[str, str]) -> int | None:
|
|
|
|
|
started = int_value(service.get("ActiveEnterTimestampMonotonic"))
|
|
|
|
|
if not started:
|
|
|
|
|
main_pid = int_value(service.get("MainPID"))
|
|
|
|
|
if not main_pid:
|
|
|
|
|
return None
|
|
|
|
|
etimes = run(["ps", "-o", "etimes=", "-p", str(main_pid)]).strip()
|
|
|
|
|
return int_value(etimes)
|
|
|
|
|
boot_ns = time.clock_gettime_ns(time.CLOCK_BOOTTIME)
|
|
|
|
|
runtime = int((boot_ns / 1000 - started) / 1_000_000)
|
|
|
|
|
return max(runtime, 0)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def scan_queue(root: Path) -> dict[str, object]:
|
|
|
|
|
files = dirs = bytes_total = 0
|
|
|
|
|
processing_files = processing_dirs = processing_bytes = 0
|
|
|
|
|
oldest = newest = None
|
|
|
|
|
top_level: list[dict[str, object]] = []
|
|
|
|
|
if not root.exists():
|
|
|
|
|
return {"path": str(root), "exists": False, "files": 0, "dirs": 0, "bytes": 0, "oldest": None, "newest": None, "topLevel": [], "items": [], "processingFiles": 0, "processingDirs": 0, "processingBytes": 0, "processing": [], "processingItems": []}
|
|
|
|
|
top_level_map: dict[Path, dict[str, object]] = {}
|
|
|
|
|
processing_map: dict[Path, dict[str, object]] = {}
|
|
|
|
|
ready_items: list[dict[str, object]] = []
|
|
|
|
|
processing_items: list[dict[str, object]] = []
|
|
|
|
|
for dirpath, dirnames, filenames in os.walk(root):
|
|
|
|
|
for filename in filenames:
|
|
|
|
|
path = Path(dirpath) / filename
|
|
|
|
|
if path.suffix.lower() not in VIDEO_EXT or "sample" in filename.lower() or "sample" in str(path.parent).lower():
|
|
|
|
|
continue
|
|
|
|
|
try:
|
|
|
|
|
st = path.stat()
|
|
|
|
|
except FileNotFoundError:
|
|
|
|
|
continue
|
|
|
|
|
oldest = st.st_mtime if oldest is None else min(oldest, st.st_mtime)
|
|
|
|
|
newest = st.st_mtime if newest is None else max(newest, st.st_mtime)
|
|
|
|
|
try:
|
|
|
|
|
rel = path.relative_to(root)
|
|
|
|
|
except ValueError:
|
|
|
|
|
rel = path
|
|
|
|
|
top = root / rel.parts[0] if rel.parts else path
|
|
|
|
|
sab = sab_state_for(top.name)
|
|
|
|
|
transient = top.name.startswith(('_UNPACK_', '__UNPACK__', '_FAILED_', '_ADMIN_'))
|
|
|
|
|
is_processing = transient or (sab and sab.get("ready") is False and sab.get("status") != "manual")
|
|
|
|
|
relative_dir = str(Path(*rel.parts[:-1])) if len(rel.parts) > 1 else ""
|
|
|
|
|
item_label = path.name if not relative_dir else f"{relative_dir} / {path.name}"
|
|
|
|
|
state = str(sab.get("state") if sab else ("unpacking" if transient else "ready"))
|
|
|
|
|
file_item = {"name": path.name, "label": item_label, "release": top.name, "relativeDir": relative_dir, "path": str(path), "bytes": st.st_size, "mtime": datetime.fromtimestamp(st.st_mtime, tz=timezone.utc).isoformat(), "state": state}
|
|
|
|
|
if is_processing:
|
|
|
|
|
processing_files += 1
|
|
|
|
|
processing_bytes += st.st_size
|
|
|
|
|
processing_items.append(file_item)
|
|
|
|
|
item = processing_map.setdefault(top, {"name": top.name, "type": "dir" if top.is_dir() else "file", "files": 0, "bytes": 0, "mtime": None})
|
|
|
|
|
item["files"] = int(item["files"]) + 1
|
|
|
|
|
item["bytes"] = int(item["bytes"]) + st.st_size
|
|
|
|
|
item["mtime"] = datetime.fromtimestamp(st.st_mtime, tz=timezone.utc).isoformat()
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
files += 1
|
|
|
|
|
bytes_total += st.st_size
|
|
|
|
|
ready_items.append(file_item)
|
|
|
|
|
item = top_level_map.setdefault(top, {"name": top.name, "type": "dir" if top.is_dir() else "file", "files": 0, "bytes": 0, "mtime": None})
|
|
|
|
|
item["files"] = int(item["files"]) + 1
|
|
|
|
|
item["bytes"] = int(item["bytes"]) + st.st_size
|
|
|
|
|
item["mtime"] = datetime.fromtimestamp(st.st_mtime, tz=timezone.utc).isoformat()
|
|
|
|
|
dirs = sum(1 for item in top_level_map.values() if item["type"] == "dir")
|
|
|
|
|
processing_dirs = sum(1 for item in processing_map.values() if item["type"] == "dir")
|
|
|
|
|
top_level = sorted(top_level_map.values(), key=lambda item: str(item["name"]).lower())
|
|
|
|
|
processing = sorted(processing_map.values(), key=lambda item: str(item["name"]).lower())
|
|
|
|
|
return {
|
|
|
|
|
"path": str(root),
|
|
|
|
|
"exists": True,
|
|
|
|
|
"files": files,
|
|
|
|
|
"dirs": dirs,
|
|
|
|
|
"bytes": bytes_total,
|
|
|
|
|
"oldest": datetime.fromtimestamp(oldest, tz=timezone.utc).isoformat() if oldest else None,
|
|
|
|
|
"newest": datetime.fromtimestamp(newest, tz=timezone.utc).isoformat() if newest else None,
|
|
|
|
|
"topLevel": top_level[:100],
|
|
|
|
|
"items": ready_items[:500],
|
|
|
|
|
"processingFiles": processing_files,
|
|
|
|
|
"processingDirs": processing_dirs,
|
|
|
|
|
"processingBytes": processing_bytes,
|
|
|
|
|
"processing": processing[:100],
|
|
|
|
|
"processingItems": processing_items[:500],
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def read_logs(limit: int = 100) -> list[dict[str, object]]:
|
|
|
|
|
if not LOG.exists():
|
|
|
|
|
return []
|
|
|
|
|
lines = LOG.read_text(errors="replace").splitlines()[-max(1, min(limit, 1000)):]
|
|
|
|
|
records = []
|
|
|
|
|
for line in lines:
|
|
|
|
|
try:
|
|
|
|
|
records.append(json.loads(line))
|
|
|
|
|
except json.JSONDecodeError:
|
|
|
|
|
records.append({"level": "RAW", "msg": line})
|
|
|
|
|
return records
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def read_summaries() -> list[dict[str, object]]:
|
|
|
|
|
if not LOG.exists():
|
|
|
|
|
return []
|
|
|
|
|
summaries = []
|
|
|
|
|
for line in LOG.read_text(errors="replace").splitlines():
|
|
|
|
|
try:
|
|
|
|
|
record = json.loads(line)
|
|
|
|
|
except json.JSONDecodeError:
|
|
|
|
|
continue
|
|
|
|
|
if record.get("msg") == "summary":
|
|
|
|
|
summaries.append(record)
|
|
|
|
|
return summaries
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def read_importer_status() -> dict[str, object] | None:
|
|
|
|
|
if not IMPORTER_STATUS.exists():
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def infer_current_from_logs(logs: list[dict[str, object]]) -> dict[str, object] | None:
|
|
|
|
|
for record in reversed(logs):
|
|
|
|
|
if record.get("level") != "MOVE" or record.get("msg") not in {"moving", "moving sidecar"}:
|
|
|
|
|
continue
|
|
|
|
|
src = record.get("src")
|
|
|
|
|
dest = record.get("dest")
|
|
|
|
|
if not src or not dest:
|
|
|
|
|
continue
|
|
|
|
|
src_path = Path(str(src))
|
|
|
|
|
partial = Path(str(dest) + ".partial")
|
|
|
|
|
total = None
|
|
|
|
|
copied = None
|
|
|
|
|
try:
|
|
|
|
|
total = src_path.stat().st_size
|
|
|
|
|
except FileNotFoundError:
|
|
|
|
|
pass
|
|
|
|
|
try:
|
|
|
|
|
copied = partial.stat().st_size
|
|
|
|
|
except FileNotFoundError:
|
|
|
|
|
copied = None
|
|
|
|
|
percent = round((copied / total * 100), 2) if copied is not None and total else None
|
|
|
|
|
return {"phase": "copying", "src": str(src), "dest": str(dest), "partial": str(partial), "bytes_copied": copied, "bytes_total": total, "percent": percent, "media_type": record.get("media_type"), "source_tag": record.get("source_tag"), "kind": "inferred"}
|
|
|
|
|
return None
|
|
|
|
|
try:
|
|
|
|
|
return json.loads(IMPORTER_STATUS.read_text(encoding="utf-8"))
|
|
|
|
|
except (OSError, json.JSONDecodeError):
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def queue_status() -> dict[str, object]:
|
|
|
|
|
global _SAB_HISTORY_CACHE
|
|
|
|
|
_SAB_HISTORY_CACHE = None
|
|
|
|
|
roots = {name: scan_queue(path) for name, path in QUEUE_ROOTS.items()}
|
|
|
|
|
return {
|
|
|
|
|
"roots": roots,
|
|
|
|
|
"files": sum(int(r["files"]) for r in roots.values()),
|
|
|
|
|
"dirs": sum(int(r["dirs"]) for r in roots.values()),
|
|
|
|
|
"bytes": sum(int(r["bytes"]) for r in roots.values()),
|
|
|
|
|
"processingFiles": sum(int(r["processingFiles"]) for r in roots.values()),
|
|
|
|
|
"processingDirs": sum(int(r["processingDirs"]) for r in roots.values()),
|
|
|
|
|
"processingBytes": sum(int(r["processingBytes"]) for r in roots.values()),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def sab_api_key() -> str | None:
|
|
|
|
|
try:
|
|
|
|
|
import re
|
|
|
|
|
m = re.search(r"^api_key\s*=\s*(\S+)", SAB_CONFIG.read_text(errors="replace"), re.M)
|
|
|
|
|
return m.group(1) if m else None
|
|
|
|
|
except OSError:
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def sab_history() -> list[dict[str, object]]:
|
|
|
|
|
key = sab_api_key()
|
|
|
|
|
if not key:
|
|
|
|
|
return []
|
|
|
|
|
try:
|
|
|
|
|
q = urllib.parse.urlencode({"mode": "history", "output": "json", "limit": 200, "apikey": key})
|
|
|
|
|
data = json.load(urllib.request.urlopen(f"{SAB_API}?{q}", timeout=10))
|
|
|
|
|
return data.get("history", {}).get("slots", [])
|
|
|
|
|
except Exception:
|
|
|
|
|
return []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_SAB_HISTORY_CACHE: list[dict[str, object]] | None = None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def sab_state_for(folder_name: str) -> dict[str, object] | None:
|
|
|
|
|
global _SAB_HISTORY_CACHE
|
|
|
|
|
if _SAB_HISTORY_CACHE is None:
|
|
|
|
|
_SAB_HISTORY_CACHE = sab_history()
|
|
|
|
|
normalized = folder_name.removeprefix("_UNPACK_").removeprefix("__UNPACK__")
|
|
|
|
|
for item in _SAB_HISTORY_CACHE:
|
|
|
|
|
name = str(item.get("name") or "")
|
|
|
|
|
if name != normalized and name != folder_name:
|
|
|
|
|
continue
|
|
|
|
|
status = str(item.get("status") or "")
|
|
|
|
|
storage = str(item.get("storage") or "")
|
|
|
|
|
action = str(item.get("action_line") or "")
|
|
|
|
|
category = str(item.get("category") or item.get("cat") or "")
|
|
|
|
|
owned = category == "manual"
|
|
|
|
|
ready = owned and status == "Completed" and bool(storage) and "_UNPACK_" not in storage
|
|
|
|
|
state = "ready" if ready else ("ignored category " + category if not owned else (status.lower() if status else "sab pending"))
|
|
|
|
|
if action:
|
|
|
|
|
state = action
|
|
|
|
|
return {"ready": ready, "owned": owned, "category": category, "status": status, "storage": storage, "state": state}
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def read_manual_batches() -> list[str]:
|
|
|
|
|
try:
|
|
|
|
|
raw = json.loads(MANUAL_BATCHES.read_text(encoding="utf-8"))
|
|
|
|
|
except (OSError, json.JSONDecodeError):
|
|
|
|
|
return []
|
|
|
|
|
return [str(x) for x in raw] if isinstance(raw, list) else []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def write_manual_batches(items: list[str]) -> None:
|
|
|
|
|
MANUAL_BATCHES.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
|
MANUAL_BATCHES.write_text(json.dumps(sorted(set(items)), indent=2), encoding="utf-8")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def add_manual_batch(value: str) -> tuple[bool, str]:
|
|
|
|
|
value = value.strip().strip("/")
|
|
|
|
|
if not value:
|
|
|
|
|
return False, "missing folder"
|
|
|
|
|
base = QUEUE_ROOTS["manual"].resolve()
|
|
|
|
|
path = (base / value).resolve() if not value.startswith("/srv/") else Path(value).resolve()
|
|
|
|
|
if not (path == base or path.is_relative_to(base)):
|
|
|
|
|
return False, "folder must be under manual downloads"
|
|
|
|
|
if not path.is_dir():
|
|
|
|
|
return False, "folder does not exist"
|
|
|
|
|
items = read_manual_batches()
|
|
|
|
|
items.append(str(path))
|
|
|
|
|
write_manual_batches(items)
|
|
|
|
|
return True, str(path)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def status() -> dict[str, object]:
|
|
|
|
|
service = systemctl_show(SERVICE)
|
|
|
|
|
timer = systemctl_show(TIMER)
|
|
|
|
|
logs = read_logs(300)
|
|
|
|
|
all_logs = read_logs(1000)
|
|
|
|
|
summaries = read_summaries()
|
|
|
|
|
last_summary = summaries[-1] if summaries else None
|
|
|
|
|
cutoff_1h = datetime.now() - timedelta(hours=1)
|
|
|
|
|
cutoff_24h = datetime.now() - timedelta(hours=24)
|
|
|
|
|
processed_1h = 0
|
|
|
|
|
processed_24h = 0
|
|
|
|
|
processed_total = 0
|
|
|
|
|
runs_1h = 0
|
|
|
|
|
runs_24h = 0
|
|
|
|
|
runs_total = 0
|
|
|
|
|
for summary in summaries:
|
|
|
|
|
moved = int(summary.get("moved") or 0)
|
|
|
|
|
processed_total += moved
|
|
|
|
|
runs_total += 1
|
|
|
|
|
try:
|
|
|
|
|
ts = datetime.fromisoformat(str(summary.get("ts")))
|
|
|
|
|
except ValueError:
|
|
|
|
|
ts = None
|
|
|
|
|
if ts and ts >= cutoff_1h:
|
|
|
|
|
processed_1h += moved
|
|
|
|
|
runs_1h += 1
|
|
|
|
|
if ts and ts >= cutoff_24h:
|
|
|
|
|
processed_24h += moved
|
|
|
|
|
runs_24h += 1
|
|
|
|
|
queue = queue_status()
|
|
|
|
|
running = service.get("ActiveState") == "activating" or service.get("SubState") in {"start", "running"}
|
|
|
|
|
current = read_importer_status()
|
|
|
|
|
if running and (not current or current.get("phase") == "done"):
|
|
|
|
|
current = infer_current_from_logs(all_logs)
|
|
|
|
|
runtime = monotonic_runtime_seconds(service) if running else None
|
|
|
|
|
memory_current = int_value(service.get("MemoryCurrent"))
|
|
|
|
|
memory_peak = int_value(service.get("MemoryPeak"))
|
|
|
|
|
warnings = []
|
|
|
|
|
if runtime and runtime > LONG_RUNTIME_SECONDS:
|
|
|
|
|
warnings.append("manual-media-import.service has been running longer than 25 minutes")
|
|
|
|
|
if memory_peak and memory_peak > HIGH_MEMORY_BYTES:
|
|
|
|
|
warnings.append("manual-media-import.service peak memory is over 8 GiB")
|
|
|
|
|
if last_summary and int(last_summary.get("errors") or 0) > 0:
|
|
|
|
|
warnings.append("last importer summary reported errors")
|
|
|
|
|
now = time.time()
|
|
|
|
|
for name, root in queue["roots"].items():
|
|
|
|
|
oldest = root.get("oldest")
|
|
|
|
|
if oldest:
|
|
|
|
|
try:
|
|
|
|
|
age = now - datetime.fromisoformat(str(oldest)).timestamp()
|
|
|
|
|
if age > STALE_QUEUE_SECONDS:
|
|
|
|
|
warnings.append(f"{name} queue contains files older than 6 hours")
|
|
|
|
|
except ValueError:
|
|
|
|
|
pass
|
|
|
|
|
health_state = "warning" if warnings else ("running" if running else service.get("Result", "unknown"))
|
|
|
|
|
return {
|
|
|
|
|
"name": "Importarr",
|
|
|
|
|
"service": {
|
|
|
|
|
"unit": SERVICE,
|
|
|
|
|
"activeState": service.get("ActiveState"),
|
|
|
|
|
"subState": service.get("SubState"),
|
|
|
|
|
"result": service.get("Result"),
|
|
|
|
|
"running": running,
|
|
|
|
|
"mainPid": int_value(service.get("MainPID")),
|
|
|
|
|
"startedAt": timestamp_to_iso(service.get("ActiveEnterTimestampUSec")),
|
|
|
|
|
"runtimeSeconds": runtime,
|
|
|
|
|
"memoryCurrentBytes": memory_current,
|
|
|
|
|
"memoryPeakBytes": memory_peak,
|
|
|
|
|
},
|
|
|
|
|
"timer": {
|
|
|
|
|
"unit": TIMER,
|
|
|
|
|
"activeState": timer.get("ActiveState"),
|
|
|
|
|
"subState": timer.get("SubState"),
|
|
|
|
|
"lastTrigger": timestamp_to_iso(timer.get("LastTriggerUSec")),
|
|
|
|
|
"nextElapse": timestamp_to_iso(timer.get("NextElapseUSecRealtime")),
|
|
|
|
|
},
|
|
|
|
|
"queue": queue,
|
|
|
|
|
"lastSummary": last_summary,
|
|
|
|
|
"processed": {"last1h": processed_1h, "last24h": processed_24h, "total": processed_total, "runsLast1h": runs_1h, "runsLast24h": runs_24h, "runsTotal": runs_total},
|
|
|
|
|
"current": current,
|
|
|
|
|
"manualBatches": read_manual_batches(),
|
|
|
|
|
"health": {"state": health_state, "warnings": warnings},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def fast_health() -> dict[str, object]:
|
|
|
|
|
service = systemctl_show(SERVICE)
|
|
|
|
|
timer = systemctl_show(TIMER)
|
|
|
|
|
running = service.get("ActiveState") == "activating" or service.get("SubState") in {"start", "running"}
|
|
|
|
|
runtime = monotonic_runtime_seconds(service) if running else None
|
|
|
|
|
warnings = []
|
|
|
|
|
if runtime and runtime > LONG_RUNTIME_SECONDS:
|
|
|
|
|
warnings.append("manual-media-import.service has been running longer than 25 minutes")
|
|
|
|
|
return {"state": "warning" if warnings else ("running" if running else service.get("Result", "unknown")), "warnings": warnings, "service": service.get("ActiveState"), "timer": timer.get("ActiveState")}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def page() -> bytes:
|
|
|
|
|
s = status()
|
|
|
|
|
logs = read_logs(80)
|
|
|
|
|
def esc(value: object) -> str:
|
|
|
|
|
return html.escape("—" if value is None else str(value))
|
|
|
|
|
def gib(value: object) -> str:
|
|
|
|
|
return "—" if value is None else f"{int(value) / 1024**3:.2f} GiB"
|
|
|
|
|
def secs(value: object) -> str:
|
|
|
|
|
if value is None:
|
|
|
|
|
return "—"
|
|
|
|
|
seconds = int(value)
|
|
|
|
|
return f"{seconds // 60}m {seconds % 60}s"
|
|
|
|
|
def job_name(path: object) -> str:
|
|
|
|
|
if not path:
|
|
|
|
|
return ""
|
|
|
|
|
name = Path(str(path)).name
|
|
|
|
|
return name[:37] + "..." if len(name) > 40 else name
|
|
|
|
|
title_icon = "📥"
|
|
|
|
|
title_runtime = secs(s["service"]["runtimeSeconds"])
|
|
|
|
|
current = s.get("current") or {}
|
|
|
|
|
progress = current.get("percent")
|
|
|
|
|
full_current_name = Path(str(current.get("src", ""))).name if current.get("src") else ""
|
|
|
|
|
current_src_path = Path(str(current.get("src", ""))) if current.get("src") else None
|
|
|
|
|
current_top_name = ""
|
|
|
|
|
current_relative_dir = ""
|
|
|
|
|
if current_src_path:
|
|
|
|
|
for root in QUEUE_ROOTS.values():
|
|
|
|
|
try:
|
|
|
|
|
rel = current_src_path.relative_to(root)
|
|
|
|
|
current_top_name = rel.parts[0] if rel.parts else current_src_path.name
|
|
|
|
|
current_relative_dir = str(Path(*rel.parts[:-1])) if len(rel.parts) > 1 else ""
|
|
|
|
|
break
|
|
|
|
|
except ValueError:
|
|
|
|
|
continue
|
|
|
|
|
current_name = job_name(current.get("src"))
|
|
|
|
|
title = f"{title_icon} {progress}% {title_runtime} - {current_name}" if progress is not None and current_name else f"{title_icon} - idle"
|
|
|
|
|
warnings = s["health"]["warnings"]
|
|
|
|
|
warning_html = "".join(f"<p class='warn'>{esc(w)}</p>" for w in warnings) or "<p class='ok'>None</p>"
|
|
|
|
|
log_text = "\n".join(esc(f"[{r.get('ts','')}] {r.get('level','')} {r.get('msg','')} {json.dumps(r, ensure_ascii=False)}") for r in logs)
|
|
|
|
|
source_target_html = f"<p class='tiny'><span>Source:</span> {esc(current.get('src'))}</p><p class='tiny'><span>Target:</span> {esc(current.get('dest'))}</p>" if current else ""
|
|
|
|
|
progress_html = f"<p class='filename'><strong>{esc(full_current_name)}</strong><br><span class='muted'>{esc(current_relative_dir)}</span></p><p>{esc(progress)}% · runtime {title_runtime}</p><progress max='100' value='{esc(progress or 0)}'></progress><p>{esc(current.get('phase'))} · {gib(current.get('bytes_copied'))} / {gib(current.get('bytes_total'))}</p>{source_target_html}" if current else "<p>—</p>"
|
|
|
|
|
current_row = f"<tr class='active'><td>▶</td><td class='filename'><strong>{esc(full_current_name)}</strong><br><span class='muted'>{esc(current_relative_dir)}</span></td><td>{esc(progress)}%</td><td><progress max='100' value='{esc(progress or 0)}'></progress></td><td>{title_runtime}</td></tr>" if current else ""
|
|
|
|
|
current_path = str(current_src_path) if current_src_path else ""
|
|
|
|
|
processing_items = [item for item in s["queue"]["roots"]["manual"]["processingItems"] if item["path"] != current_path]
|
|
|
|
|
ready_items = [item for item in s["queue"]["roots"]["manual"]["items"] if item["path"] != current_path]
|
|
|
|
|
processing_rows = "".join(f"<tr class='processing'><td>⏳</td><td class='filename'><strong>{esc(item['name'])}</strong><br><span class='muted'>{esc(item.get('relativeDir') or item['release'])}</span></td><td>unpacking</td><td>{gib(item.get('bytes'))}</td><td>waiting</td></tr>" for item in processing_items[:30])
|
|
|
|
|
ready_rows = "".join(f"<tr><td>◷</td><td class='filename'><strong>{esc(item['name'])}</strong><br><span class='muted'>{esc(item.get('relativeDir') or item['release'])}</span></td><td>ready</td><td>{gib(item.get('bytes'))}</td><td>queued</td></tr>" for item in ready_items[:30])
|
|
|
|
|
queue_rows = processing_rows + ready_rows or "<tr><td>✓</td><td colspan='4'>No video files waiting</td></tr>"
|
|
|
|
|
history = [r for r in reversed(logs) if r.get("level") == "MOVE" and r.get("msg") == "moving"][:12]
|
|
|
|
|
history_rows = "".join(f"<tr><td>✓</td><td class='filename'>{esc(Path(str(r.get('dest',''))).name)}</td><td>{esc(r.get('media_type',''))}</td><td colspan='2'>{esc(r.get('ts',''))}</td></tr>" for r in history) or "<tr><td>—</td><td colspan='4'>No recent imports</td></tr>"
|
|
|
|
|
body = f"""<!doctype html><html><head><meta charset='utf-8'><meta name='viewport' content='width=device-width, initial-scale=1'><meta http-equiv='refresh' content='10'><title>{esc(title)}</title><style>body{{font-family:system-ui,sans-serif;margin:0;background:#111;color:#eee}}header{{display:flex;align-items:center;gap:1rem;background:#3b3b3b;padding:.7rem 1.2rem;border-bottom:1px solid #111;flex-wrap:wrap}}header h1{{margin:0;font-size:1.5rem}}.pill{{background:#222;border:1px solid #555;padding:.35rem .7rem}}main{{padding:1rem}}.grid{{display:grid;grid-template-columns:repeat(auto-fit,minmax(260px,1fr));gap:1rem}}.card{{background:#1d1d1d;border:1px solid #333;border-radius:12px;padding:1rem;margin-bottom:1rem;min-width:0;overflow:hidden}}input,button{{padding:.55rem;border:1px solid #555;background:#2b2b2b;color:#eee}}button{{cursor:pointer}}.filename{{overflow-wrap:anywhere;word-break:break-word}}.tiny{{font-size:.78rem;color:#999;line-height:1.25;overflow-wrap:anywhere}}.tiny span{{color:#bbb}}.ok{{color:#60d394}}.warn{{color:#ffd166}}progress{{width:100%;height:1.2rem;accent-color:#7fd37f}}table{{width:100%;border-collapse:collapse;background:#2b2b2b}}th,td{{padding:.65rem;border-bottom:1px solid #111;text-align:left;vertical-align:middle}}th{{background:#444;color:#ddd}}tr:nth-child(even){{background:#333}}tr.active{{background:#3f4a3f}}tr.processing{{background:#4a4232}}pre{{white-space:pre-wrap;overflow-wrap:anywhere;max-height:28rem;overflow:auto}}.muted{{color:#aaa}}a{{color:#8ecae6}}</style></head><body><header><h1>📥 Importarr</h1><span class='pill'>{esc(s['queue']['files'])} videos ready</span><span class='pill'>{esc(s['queue']['processingFiles'])} unpacking</span><span class='pill'>ready {gib(s['queue']['bytes'])}</span><span class='pill'>1h {esc(s['processed']['last1h'])} · 24h {esc(s['processed']['last24h'])} · total {esc(s['processed']['total'])}</span></header><main><div class='grid'><section class='card'><h2>Status</h2><p class='{('warn' if warnings else 'ok')}'>{esc(s['health']['state'])}</p><p>Service: {esc(s['service']['activeState'])}/{esc(s['service']['subState'])}</p><p>Runtime: {secs(s['service']['runtimeSeconds'])}</p><p>Memory current: {gib(s['service']['memoryCurrentBytes'])}</p><p>Memory peak: {gib(s['service']['memoryPeakBytes'])}</p><p class='muted'>Scheduled automatically every 15 minutes. `_UNPACK_` folders are shown as unpacking, not ready.</p></section><section class='card'><h2>Current file</h2>{progress_html}</section><section class='card'><h2>Processed</h2><p>Last 1h: {esc(s['processed']['last1h'])} imported</p><p>Last 24h: {esc(s['processed']['last24h'])} imported</p><p>Total: {esc(s['processed']['total'])} imported</p></section><section class='card'><h2>Warnings</h2>{warning_html}</section></div><section class='card'><h2>Add manual batch</h2><form onsubmit="event.preventDefault();fetch('/api/manual-batches',{{method:'POST',headers:{{'content-type':'application/json'}},body:JSON.stringify({{path:this.path.value}})}}).then(()=>location.reload())"><input name='path' placeholder='folder under manual downloads' size='60'><button>Add folder once</button></form></section><section class='card'><h2>Jobs</h2><table><thead><tr><th></th><th>Name</th><th>State</th><th>Progress / Size</th><th>Runtime</th></tr></thead><tbody>{current_row}{queue_rows}</tbody></table></section><section class='card'><h2>History</h2><table><thead><tr><th></th><th>Name</th><th>Type</th><th colspan='2'>Time</th></tr></thead><tbody>{history_rows}</tbody></table></section><section class='card'><h2>Recent log</h2><pre>{log_text}</pre></section><p><a href='/api/status'>/api/status</a> · <a href='/api/queue'>/api/queue</a> · <a href='/api/logs?limit=100'>/api/logs</a></p></main></body></html>"""
|
|
|
|
|
return body.encode()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Handler(BaseHTTPRequestHandler):
|
|
|
|
|
def send(self, code: int, content_type: str, data: bytes) -> None:
|
|
|
|
|
self.send_response(code)
|
|
|
|
|
self.send_header("Content-Type", content_type)
|
|
|
|
|
self.send_header("Cache-Control", "no-store")
|
|
|
|
|
self.end_headers()
|
|
|
|
|
self.wfile.write(data)
|
|
|
|
|
|
|
|
|
|
def do_GET(self) -> None:
|
|
|
|
|
parsed = urllib.parse.urlparse(self.path)
|
|
|
|
|
if parsed.path == "/":
|
|
|
|
|
self.send(200, "text/html; charset=utf-8", page())
|
|
|
|
|
elif parsed.path == "/api/status":
|
|
|
|
|
self.send(200, "application/json", json.dumps(status()).encode())
|
|
|
|
|
elif parsed.path == "/api/queue":
|
|
|
|
|
self.send(200, "application/json", json.dumps(queue_status()).encode())
|
|
|
|
|
elif parsed.path == "/api/logs":
|
|
|
|
|
params = urllib.parse.parse_qs(parsed.query)
|
|
|
|
|
limit = int(params.get("limit", ["100"])[0])
|
|
|
|
|
self.send(200, "application/json", json.dumps(read_logs(limit)).encode())
|
|
|
|
|
elif parsed.path == "/health":
|
|
|
|
|
self.send(200, "application/json", json.dumps(fast_health()).encode())
|
|
|
|
|
else:
|
|
|
|
|
self.send(404, "text/plain", b"not found")
|
|
|
|
|
|
|
|
|
|
def do_POST(self) -> None:
|
|
|
|
|
parsed = urllib.parse.urlparse(self.path)
|
|
|
|
|
if parsed.path != "/api/manual-batches":
|
|
|
|
|
self.send(404, "text/plain", b"not found")
|
|
|
|
|
return
|
|
|
|
|
length = int(self.headers.get("content-length") or 0)
|
|
|
|
|
try:
|
|
|
|
|
payload = json.loads(self.rfile.read(length) or b"{}")
|
|
|
|
|
except json.JSONDecodeError:
|
|
|
|
|
self.send(400, "application/json", json.dumps({"ok": False, "error": "invalid json"}).encode())
|
|
|
|
|
return
|
|
|
|
|
ok, message = add_manual_batch(str(payload.get("path") or ""))
|
|
|
|
|
self.send(200 if ok else 400, "application/json", json.dumps({"ok": ok, "result": message}).encode())
|
|
|
|
|
|
|
|
|
|
def log_message(self, fmt: str, *args: object) -> None:
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main() -> None:
|
|
|
|
|
host = os.getenv("IMPORTARR_BIND_HOST", "0.0.0.0")
|
|
|
|
|
port = int(os.getenv("IMPORTARR_BIND_PORT", "8095"))
|
|
|
|
|
ThreadingHTTPServer((host, port), Handler).serve_forever()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
main()
|
