Initial commit

app/services/apt.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+import logging
+from pathlib import Path
+
+from app.models import Mirror
+from app.services.host import get_ubuntu_codename, host_path, run_on_host, validate_codename
+
+logger = logging.getLogger(__name__)
+
+SOURCES_PATH = "etc/apt/sources.list"
+SOURCES_D_PATH = "etc/apt/sources.list.d"
+MIRROR_MARKER = "mirror-manager"
+
+
+def read_current_apt() -> dict:
+    sources = host_path(SOURCES_PATH)
+    return {
+        "sources_list": sources.read_text(encoding="utf-8") if sources.exists() else None,
+        "path": str(sources),
+    }
+
+
+def _build_sources_content(base_url: str, codename: str) -> str:
+    url = base_url.rstrip("/")
+    return f"""# Managed by Mirror Manager - {MIRROR_MARKER}
+deb {url} {codename} main restricted universe multiverse
+deb {url} {codename}-updates main restricted universe multiverse
+deb {url} {codename}-backports main restricted universe multiverse
+deb {url} {codename}-security main restricted universe multiverse
+"""
+
+
+def apply_apt(mirror: Mirror) -> dict:
+    if not mirror.url:
+        return {"success": False, "error": "URL میرور APT تعریف نشده"}
+
+    codename = validate_codename(get_ubuntu_codename())
+    sources_path = host_path(SOURCES_PATH)
+    sources_path.parent.mkdir(parents=True, exist_ok=True)
+
+    content = _build_sources_content(mirror.url, codename)
+    sources_path.write_text(content, encoding="utf-8")
+
+    sources_d = host_path(SOURCES_D_PATH)
+    if sources_d.exists():
+        for f in sources_d.glob("*.list"):
+            if MIRROR_MARKER not in f.read_text(encoding="utf-8", errors="ignore"):
+                backup_name = f.with_suffix(".list.disabled")
+                f.rename(backup_name)
+
+    result = run_on_host(["apt-get", "update", "-qq"], timeout=180)
+    if result.returncode != 0:
+        return {
+            "success": False,
+            "error": (result.stderr or result.stdout or "apt-get update ناموفق")[:500],
+            "codename": codename,
+        }
+
+    return {
+        "success": True,
+        "message": f"APT mirror اعمال شد: {mirror.name}",
+        "codename": codename,
+        "url": mirror.url,
+    }