per-device-eq.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
per-device-eq -- per-output-device parametric EQ for PipeWire 1.6+ (entry point).

Thin launcher: locate the `perdeviceeq` package (next to this script when run
from a checkout, or /usr/share/per-device-eq when installed), dispatch the CLI,
and launch the GTK/libadwaita GUI. Implementation lives in the package:
config, eq, profiles, pipewire, integration, cli, gui.

  --list                 list sinks (default marked with *)
  --list-profiles        list known profiles
  --inspect <node.name>  dump node params (pw-dump info.params)
  --apply                apply each bound profile to its sink now
  --install-hook         (re)install the WirePlumber hook + config
  --install-desktop      install the .desktop entry + icon into ~/.local/share
  --uninstall-desktop    remove the .desktop entry + icon from ~/.local/share
  (no args)              launch the GTK4 GUI
"""

import argparse, os, sys

# ---- locate the perdeviceeq package (repo checkout first, then system) -------
_HERE = os.path.dirname(os.path.abspath(__file__))
for _cand in (_HERE, "/usr/share/per-device-eq"):
    if os.path.isdir(os.path.join(_cand, "perdeviceeq")):
        if _cand not in sys.path:
            sys.path.insert(0, _cand)
        break

from perdeviceeq.config import SYS_DESKTOP_FILE, USER_DESKTOP_FILE, USER_ICON_FILE
from perdeviceeq.pipewire import missing_tools, missing_tools_message
from perdeviceeq.integration import (install_hook, restart_wireplumber,
                                      install_desktop_integration,
                                      uninstall_desktop_integration)
from perdeviceeq.cli import cmd_list, cmd_list_profiles, cmd_inspect, cmd_apply


def main():
    try:                      # `kill -USR1 <pid>` dumps a live stack to stderr
        import faulthandler, signal
        faulthandler.register(signal.SIGUSR1)
    except Exception:
        pass
    ap = argparse.ArgumentParser(description="Per-device PipeWire EQ (profiles)")
    g = ap.add_mutually_exclusive_group()
    g.add_argument("--list", action="store_true", help="list sinks")
    g.add_argument("--list-profiles", action="store_true", help="list profiles")
    g.add_argument("--inspect", metavar="NODE_NAME", help="dump node params")
    g.add_argument("--apply", action="store_true",
                   help="push bound profiles into the per-device-eq metadata now")
    g.add_argument("--install-hook", action="store_true",
                   help="install/refresh the WirePlumber hook + metadata config")
    g.add_argument("--install-desktop", action="store_true",
                   help="install the .desktop entry + icon into ~/.local/share")
    g.add_argument("--uninstall-desktop", action="store_true",
                   help="remove the .desktop entry + icon from ~/.local/share")
    args = ap.parse_args()

    if (args.list or args.inspect or args.apply):
        miss = missing_tools()
        if miss:
            print(missing_tools_message(miss), file=sys.stderr)
            return 2

    if args.list:
        return cmd_list()
    if args.list_profiles:
        return cmd_list_profiles()
    if args.inspect:
        return cmd_inspect(args.inspect)
    if args.apply:
        return cmd_apply()
    if args.install_hook:
        try:
            changed = install_hook()
        except FileNotFoundError as e:
            print(str(e), file=sys.stderr)
            return 2
        if changed:
            print("hook + config installed/updated; restarting WirePlumber once...")
            restart_wireplumber()
        else:
            print("hook already up to date (no restart needed)")
        return 0
    if args.install_desktop:
        if os.path.exists(SYS_DESKTOP_FILE):
            print("a system .desktop entry is already installed (%s); nothing to do"
                  % SYS_DESKTOP_FILE)
            return 0
        try:
            install_desktop_integration()
        except FileNotFoundError as e:
            print(str(e), file=sys.stderr)
            return 2
        print("installed: %s" % USER_DESKTOP_FILE)
        print("installed: %s" % USER_ICON_FILE)
        print("(reopen the app to see the icon; run --uninstall-desktop to remove)")
        return 0
    if args.uninstall_desktop:
        if os.path.exists(SYS_DESKTOP_FILE):
            print("the .desktop entry is provided by a system package (%s); remove it "
                  "with your package manager, not this command" % SYS_DESKTOP_FILE)
            return 0
        removed = uninstall_desktop_integration()
        print("desktop entry + icon %s"
              % ("removed from ~/.local/share" if removed else "were not installed"))
        return 0

    # no CLI flag -> launch the GUI (import here so CLI paths never need GTK)
    from perdeviceeq.gui import launch_gui
    return launch_gui()


if __name__ == "__main__":
    sys.exit(main() or 0)