pkgmate 0.0.1__py3-none-any.whl

pkgtool/apt/__init__.py

@@ -0,0 +1,9 @@
+"""pkgtool.apt — deb/apt 的数据源与写操作。
+
+子模块：
+  version  Debian 版本号比较（dpkg verrev，无 python-debian 依赖）
+  lists    /var/lib/apt/lists 的唯一解析器（仓库索引 + 源标签 + section/priority）
+  logs     dpkg.log / apt history.log 时间线（安装通道判定的证据来源）
+  dpkg     /var/lib/dpkg 状态：status、.list 清单、extended_states、apt-mark
+  actions  写操作：apt-get download / 特权执行
+"""

pkgtool/apt/actions.py

@@ -0,0 +1,193 @@
+"""pkgtool.apt.actions — 需要特权的写操作。
+
+统一提权入口：原实现有三份互不复用的特权执行路径（Web UI 的 sudo -S apt 升级、
+sudo -S flatpak 升级、safety.py 的 pkexec bash -c），这里收敛成一个
+run_privileged()，remove.py 与各后端的升级动作都走它。
+
+安全约束：
+  · 密码只经 stdin 传给 sudo，不进 argv（argv 在 /proc/<pid>/cmdline 全局可见）
+  · 一律用 argv 列表调用，不经过 shell
+  · 包名/版本先过 base.is_safe_name 白名单，拒绝前导 '-' 造成的参数注入
+"""
+import glob
+import os
+import shutil
+import signal
+import subprocess
+import tempfile
+from dataclasses import dataclass, field
+
+from ..base import is_safe_name
+from ..config import CFG
+
+_SUDO_NOPASS = None
+
+
+@dataclass
+class Result:
+    """一次写操作的结果。command 是不含密码的可复现命令（供失败时提示手动执行）。"""
+    ok: bool
+    returncode: int = 0
+    output: str = ""
+    error: str = ""
+    file: str = ""
+    command: list = field(default_factory=list)
+
+    @property
+    def command_text(self):
+        return " ".join(shutil.quote(c) for c in self.command)
+
+
+def is_root():
+    return os.geteuid() == 0
+
+
+def sudo_nopass():
+    """是否有免密 sudo（结果缓存）。没有时特权命令会在终端上交互提示。"""
+    global _SUDO_NOPASS
+    if _SUDO_NOPASS is None:
+        if is_root():
+            _SUDO_NOPASS = True
+            return True
+        try:
+            _SUDO_NOPASS = subprocess.run(
+                ["sudo", "-n", "true"], capture_output=True, timeout=10).returncode == 0
+        except (OSError, subprocess.SubprocessError):
+            _SUDO_NOPASS = False
+    return _SUDO_NOPASS
+
+
+def privileged_argv(argv, password=None):
+    """套上提权前缀。已是 root → 原样；给了密码 → sudo -S；否则裸 sudo 走终端提示。"""
+    if is_root():
+        return list(argv)
+    if password:
+        return ["sudo", "-S", "-p", "", *argv]
+    return ["sudo", *argv]
+
+
+def _kill(p, own_session):
+    """中止子进程。
+    own_session=True 时子进程自成一个会话，可以 killpg 干净地带走整棵树；
+    False 时它和我们同组，killpg 会把自己一起杀掉，只能 terminate 它本身
+    （apt-get 收到 SIGTERM 会自行收拾 dpkg）。"""
+    try:
+        if own_session:
+            os.killpg(os.getpgid(p.pid), signal.SIGTERM)
+        else:
+            p.terminate()
+        p.wait(timeout=5)
+    except (OSError, subprocess.SubprocessError):
+        try:
+            if own_session:
+                os.killpg(os.getpgid(p.pid), signal.SIGKILL)
+            else:
+                p.kill()
+        except OSError:
+            pass
+
+
+def _error_summary(rc, lines):
+    """失败原因：取输出的最后一行非空内容。
+    只报"退出码 1"等于没报——sudo 认证失败、apt 依赖冲突这些真正的原因
+    都在输出里，用户看到的却是一个光秃秃的数字。"""
+    for line in reversed(lines):
+        line = line.strip()
+        if line:
+            return f"{line[:200]}（退出码 {rc}）"
+    return f"退出码 {rc}"
+
+
+def run_privileged(argv, password=None, timeout=None, on_line=None, cfg=CFG):
+    """执行特权命令并逐行回显。Ctrl-C 中止（替代原 Web UI 的 cancel 接口）。
+
+    只有"密码经 stdin 喂进去"这种非交互场景才新建会话（那样能 killpg 整棵树）。
+    交互式必须留在当前会话里：start_new_session 会 setsid() 切断控制终端，
+    sudo 就再也打不开 /dev/tty 读密码，直接报
+    "A terminal is required to authenticate"——密码提示根本出不来。"""
+    cmd = privileged_argv(argv, password)
+    shown = list(argv) if is_root() else ["sudo", *argv]   # 可复现命令，永不含密码
+    own_session = bool(password)
+    lines = []
+    try:
+        p = subprocess.Popen(cmd, stdin=subprocess.PIPE if password else None,
+                             stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
+                             text=True, start_new_session=own_session)
+    except OSError as e:
+        return Result(ok=False, returncode=-1, error=f"{type(e).__name__}: {e}",
+                      command=shown)
+    try:
+        if password:
+            try:
+                p.stdin.write(password + "\n")
+                p.stdin.close()
+            except OSError:
+                pass          # sudo 可能已因缓存凭据直接放行
+        for line in p.stdout:
+            line = line.rstrip("\n")
+            lines.append(line)
+            if on_line:
+                on_line(line)
+        rc = p.wait(timeout=timeout)
+    except KeyboardInterrupt:
+        _kill(p, own_session)
+        return Result(ok=False, returncode=-2, output="\n".join(lines),
+                      error="已中止（Ctrl-C）", command=shown)
+    except subprocess.TimeoutExpired:
+        _kill(p, own_session)
+        return Result(ok=False, returncode=-3, output="\n".join(lines),
+                      error=f"超时（{timeout}s）", command=shown)
+    except OSError as e:
+        _kill(p, own_session)
+        return Result(ok=False, returncode=-1, output="\n".join(lines),
+                      error=f"{type(e).__name__}: {e}", command=shown)
+    return Result(ok=rc == 0, returncode=rc,
+                  output="\n".join(lines)[-cfg.output_tail_chars:],
+                  error="" if rc == 0 else _error_summary(rc, lines),
+                  command=shown)
+
+
+def run_plain(argv, cwd=None, timeout=None, cfg=CFG):
+    """非特权命令（apt-get download / dpkg-deb 一类）。"""
+    try:
+        p = subprocess.run(argv, cwd=cwd, capture_output=True, text=True,
+                           timeout=timeout or cfg.timeout_query)
+    except subprocess.TimeoutExpired:
+        return Result(ok=False, returncode=-3, error=f"超时（{timeout}s）", command=argv)
+    except OSError as e:
+        return Result(ok=False, returncode=-1, error=f"{type(e).__name__}: {e}", command=argv)
+    return Result(ok=p.returncode == 0, returncode=p.returncode,
+                  output=((p.stdout or "") + (p.stderr or "")).strip()[-cfg.output_tail_chars:],
+                  error=(p.stderr or "").strip()[-500:], command=argv)
+
+
+def download(name, version="", dest_dir=None, cfg=CFG):
+    """apt-get download（非 root 可用）→ .deb 落到 dest_dir（默认 ~/Downloads）。"""
+    if not is_safe_name(name) or (version and not is_safe_name(version)):
+        return Result(ok=False, error=f"包名或版本非法：{name}={version}")
+    spec = f"{name}={version}" if version else name
+    dest_dir = dest_dir or cfg.download_dir
+    tmp = tempfile.mkdtemp(prefix="pkgtool_dl_")
+    try:
+        r = run_plain(["apt-get", "download", spec], cwd=tmp,
+                      timeout=cfg.timeout_download, cfg=cfg)
+        debs = glob.glob(os.path.join(tmp, "*.deb"))
+        if not r.ok or not debs:
+            tail = [ln for ln in (r.output or r.error).splitlines() if ln.strip()]
+            return Result(ok=False, error=(tail[-1] if tail else "下载失败")[:300],
+                          command=r.command)
+        os.makedirs(dest_dir, exist_ok=True)
+        dest = os.path.join(dest_dir, os.path.basename(debs[0]))
+        shutil.move(debs[0], dest)
+        return Result(ok=True, file=dest, command=r.command)
+    finally:
+        shutil.rmtree(tmp, ignore_errors=True)   # 原实现把临时目录留在 /tmp 不清
+
+
+def upgrade(name, version="", password=None, on_line=None, cfg=CFG):
+    """apt-get install -y 到指定版本（version 为空则装候选版本）。"""
+    if not is_safe_name(name) or (version and not is_safe_name(version)):
+        return Result(ok=False, error=f"包名或版本非法：{name}={version}")
+    spec = f"{name}={version}" if version else name
+    return run_privileged(["apt-get", "install", "-y", spec], password=password,
+                          timeout=cfg.timeout_upgrade, on_line=on_line, cfg=cfg)

pkgtool/apt/deps.py

@@ -0,0 +1,94 @@
+"""pkgtool.apt.deps — 已安装包的依赖图：独占依赖闭包与体积核算。
+
+「独占依赖」= 只被这一个包硬依赖（Depends / Pre-Depends）、且 apt 标记为
+auto 的包，取传递闭包。回答的问题是"这个软件拖了多少只属于它自己的东西"。
+
+它和 `apt-get autoremove` 的答案不是一回事，两者不要混用：
+  · 独占依赖        这个包**用到**的、没人共用的东西      → 体积核算用
+  · autoremove 结果 删掉这个包后**连带孤立**的全部自动包   → 卸载预览用
+后者含反向依赖级联，数字通常大得多：node-css-loader 的独占依赖是 0，
+但删它会让 npm（依赖它）连带 396 个自动包一起走。卸载预览走 remove.py
+的 apt dry-run，不用这里的数字。
+
+依赖串解析要处理：逗号分隔的多项、竖线分隔的可选项、版本约束括号、
+:arch 后缀。虚拟包（如 default-mta）不在已安装集合里，自然被丢弃。
+"""
+import re
+
+_DEP_NAME_RE = re.compile(r"([A-Za-z0-9][A-Za-z0-9.+\-]*)")
+_KB_PER_MB = 1024.0
+
+
+def dep_names(raw, known):
+    """依赖字段 → 已安装包名集合。raw 可以是多个字段拼接的串。"""
+    out = set()
+    for group in (raw or "").split(","):
+        for alt in group.split("|"):
+            m = _DEP_NAME_RE.match(alt.strip())
+            if not m:
+                continue
+            name = m.group(1)
+            if name in known:
+                out.add(name)
+    return out
+
+
+class DepGraph:
+    """硬依赖图。构造一次，多次查询（exclusive 结果带缓存）。"""
+
+    def __init__(self, entries, manual=frozenset()):
+        self._size = {n: e.installed_size_kb for n, e in entries.items()}
+        self._deps = {n: dep_names(f"{e.depends}, {e.pre_depends}", entries)
+                      for n, e in entries.items()}
+        self._manual = set(manual)
+        refcount = {}
+        for deps in self._deps.values():
+            for d in deps:
+                refcount[d] = refcount.get(d, 0) + 1
+        self._refcount = refcount
+        self._cache = {}
+
+    def __len__(self):
+        return len(self._deps)
+
+    def size_mb(self, name):
+        """包自身已安装体积（来自 status 的 Installed-Size，不遍历文件树）。"""
+        return round(self._size.get(name, 0) / _KB_PER_MB, 1)
+
+    def is_shared(self, name):
+        """被两个以上的包依赖 = 共用，删掉任何一个都不会释放它。"""
+        return self._refcount.get(name, 0) > 1
+
+    def _children(self, name):
+        """name 的依赖中，只被 name 一个包依赖、且 apt 未标记为 manual 的那些。
+        manual 标记的包 apt 不会自动清理，算进来会虚高"能释放多少"。"""
+        out = []
+        for d in self._deps.get(name, ()):
+            if d == name or d in self._manual:
+                continue
+            if self._refcount.get(d, 0) == 1:
+                out.append(d)
+        return out
+
+    def exclusive(self, name):
+        """独占依赖的传递闭包（带环保护：A↔B 互相独占时不会死循环）。"""
+        if name in self._cache:
+            return self._cache[name]
+        seen = set()
+        stack = [name]
+        while stack:
+            cur = stack.pop()
+            for d in self._children(cur):
+                if d == name or d in seen:
+                    continue
+                seen.add(d)
+                stack.append(d)
+        self._cache[name] = seen
+        return seen
+
+    def total_mb(self, name):
+        """→ (自身体积, 独占依赖体积, 合计)。"""
+        deps = self.exclusive(name)
+        own = self.size_mb(name)
+        extra = round(sum(self._size.get(d, 0) for d in deps) / _KB_PER_MB, 1)
+        return own, extra, round(own + extra, 1)

pkgtool/apt/dpkg.py

@@ -0,0 +1,153 @@
+"""pkgtool.apt.dpkg — /var/lib/dpkg 与 apt 标记状态的读取。
+
+extended_states 是 auto/manual 标记的权威存储，兼容两种格式/位置：
+  apt 3.x:  /var/lib/apt/extended_states   stanza 格式（Package: / Auto-Installed:）
+  dpkg 旧版: /var/lib/dpkg/extended_states  行格式 "pkg:arch auto"
+注意新版 dpkg(1.23+) 的 status 第三字段已不再区分 user/installed，不能再用它
+判断 manual/auto，所以要与 apt-mark 交叉核对，不一致时标记 conflict。
+"""
+import os
+import subprocess
+from dataclasses import dataclass
+
+from ..config import CFG
+
+
+@dataclass(slots=True)
+class StatusEntry:
+    """/var/lib/dpkg/status 里一个包的全部所需字段。
+    体积与依赖都在这里一次读出来，避免为算大小再去遍历文件树、
+    为算依赖图再解析一遍 status。"""
+    name: str
+    version: str = ""
+    status: str = ""
+    installed_size_kb: int = 0
+    depends: str = ""
+    pre_depends: str = ""
+    recommends: str = ""
+
+    @property
+    def size_mb(self):
+        return round(self.installed_size_kb / 1024.0, 1)
+
+
+_STATUS_FIELDS = {
+    "Package": "name", "Version": "version", "Status": "status",
+    "Installed-Size": "installed_size_kb", "Depends": "depends",
+    "Pre-Depends": "pre_depends", "Recommends": "recommends",
+}
+
+
+def installed(cfg=CFG):
+    """解析 dpkg status → {包名: StatusEntry}。
+    多架构同名包（gcc:amd64 / gcc:i386）按名字归并，后者覆盖前者。"""
+    pkgs = {}
+    cur = {}
+
+    def flush():
+        name = cur.get("name")
+        if name:
+            size = cur.get("installed_size_kb", "")
+            pkgs[name] = StatusEntry(
+                name=name, version=cur.get("version", ""),
+                status=cur.get("status", ""),
+                installed_size_kb=int(size) if str(size).isdigit() else 0,
+                depends=cur.get("depends", ""),
+                pre_depends=cur.get("pre_depends", ""),
+                recommends=cur.get("recommends", ""))
+        cur.clear()
+
+    try:
+        fh = open(cfg.dpkg_status, errors="replace")
+    except OSError:
+        return pkgs
+    with fh:
+        for line in fh:
+            if line == "\n":
+                flush()
+                continue
+            if line[0] in " \t":          # 续行（Description 正文等）
+                continue
+            key, sep, value = line.partition(": ")
+            if sep:
+                field = _STATUS_FIELDS.get(key)
+                if field and field not in cur:
+                    cur[field] = value.strip()
+    flush()
+    return pkgs
+
+
+def package_files(pkg, cfg=CFG):
+    """从 <pkg>.list 读文件清单（比逐个 dpkg -L 快得多）→ (files, dirs)。"""
+    path = os.path.join(cfg.dpkg_info_dir, pkg + ".list")
+    files, dirs = [], []
+    if not os.path.isfile(path):
+        return files, dirs
+    try:
+        with open(path, errors="replace") as fh:
+            for line in fh:
+                p = line.rstrip("\n")
+                if not p or p == "/":
+                    continue
+                (dirs if p.endswith("/") else files).append(p)
+    except OSError:
+        return files, dirs
+    return files, dirs
+
+
+def apt_marks(cfg=CFG):
+    """→ (manual 集合, auto 集合)。apt-mark 不可用时返回两个空集。"""
+    def run(sub):
+        try:
+            out = subprocess.run(["apt-mark", sub], capture_output=True,
+                                 text=True, timeout=cfg.timeout_query).stdout
+            return set(out.split())
+        except (OSError, subprocess.SubprocessError):
+            return set()
+    return run("showmanual"), run("showauto")
+
+
+def extended_states(cfg=CFG):
+    """解析 extended_states → ({包名: 是否 auto}, 命中的文件路径)。
+    只记录 auto 的包；未出现的包视为 manual。文件都不存在时返回 ({}, None)。"""
+    for path in cfg.ext_states_paths:
+        if not os.path.isfile(path):
+            continue
+        try:
+            with open(path, errors="replace") as fh:
+                lines = [ln.strip() for ln in fh if ln.strip()]
+        except OSError:
+            continue
+        autos = {}
+        if any(ln.startswith("Package: ") for ln in lines):      # stanza 格式
+            name = None
+            for ln in lines:
+                if ln.startswith("Package: "):
+                    name = ln.split(":", 1)[1].strip().split(":")[0]
+                elif ln.startswith("Auto-Installed: ") and name is not None:
+                    autos[name] = ln.split(":", 1)[1].strip() == "1"
+                    name = None
+        else:                                                    # 行格式
+            for ln in lines:
+                parts = ln.split()
+                if len(parts) >= 2 and ":" in parts[0] and parts[1] in ("auto", "manual"):
+                    autos[parts[0].split(":")[0]] = parts[1] == "auto"
+        return autos, path
+    return {}, None
+
+
+def deb_file_info(path, *fields, cfg=CFG):
+    """dpkg-deb -f 读散落 .deb 的元数据 → {字段: 值}；读不出包名返回 None。
+    合并原先 scan_deb_files 与 _deb_file_info 两份同构实现。"""
+    want = fields or ("Package", "Version")
+    try:
+        p = subprocess.run(["dpkg-deb", "-f", path, *want],
+                           capture_output=True, text=True, timeout=cfg.timeout_query)
+    except (OSError, subprocess.SubprocessError):
+        return None
+    out = {}
+    for line in p.stdout.splitlines():
+        key, sep, value = line.partition(": ")
+        if sep:
+            out[key.strip()] = value.strip()
+    return out if out.get("Package") else None