mirror of
https://github.com/caelestia-dots/cli.git
synced 2026-06-06 07:19:29 -05:00
2 * r + 2 * t
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
from argparse import Namespace
|
|
|
|
from caelestia.utils import hypr
|
|
|
|
|
|
class Command:
|
|
args: Namespace
|
|
clients: list[dict[str, any]] = None
|
|
app2unit: str = None
|
|
|
|
def __init__(self, args: Namespace) -> None:
|
|
self.args = args
|
|
|
|
def run(self) -> None:
|
|
getattr(self, self.args.workspace)()
|
|
|
|
def get_clients(self) -> list[dict[str, any]]:
|
|
if self.clients is None:
|
|
self.clients = hypr.message("clients")
|
|
|
|
return self.clients
|
|
|
|
def get_app2unit(self) -> str:
|
|
if self.app2unit is None:
|
|
import shutil
|
|
|
|
self.app2unit = shutil.which("app2unit")
|
|
|
|
return self.app2unit
|
|
|
|
def move_client(self, selector: callable, workspace: str) -> None:
|
|
for client in self.get_clients():
|
|
if selector(client):
|
|
hypr.dispatch("movetoworkspacesilent", f"special:{workspace},address:{client['address']}")
|
|
|
|
def spawn_client(self, selector: callable, spawn: list[str]) -> bool:
|
|
exists = any(selector(client) for client in self.get_clients())
|
|
|
|
if not exists:
|
|
import subprocess
|
|
|
|
subprocess.Popen([self.get_app2unit(), "--", *spawn], start_new_session=True)
|
|
|
|
return not exists
|
|
|
|
def spawn_or_move(self, selector: callable, spawn: list[str], workspace: str) -> None:
|
|
if not self.spawn_client(selector, spawn):
|
|
self.move_client(selector, workspace)
|
|
|
|
def communication(self) -> None:
|
|
self.spawn_or_move(lambda c: c["class"] == "discord", ["discord"], "communication")
|
|
self.move_client(lambda c: c["class"] == "whatsapp", "communication")
|
|
hypr.dispatch("togglespecialworkspace", "communication")
|
|
|
|
def music(self) -> None:
|
|
self.spawn_or_move(
|
|
lambda c: c["class"] == "Spotify" or c["initialTitle"] == "Spotify" or c["initialTitle"] == "Spotify Free",
|
|
["spicetify", "watch", "-s"],
|
|
"music",
|
|
)
|
|
self.move_client(lambda c: c["class"] == "feishin", "music")
|
|
hypr.dispatch("togglespecialworkspace", "music")
|
|
|
|
def sysmon(self) -> None:
|
|
self.spawn_client(
|
|
lambda c: c["class"] == "btop" and c["title"] == "btop" and c["workspace"]["name"] == "special:sysmon",
|
|
["foot", "-a", "btop", "-T", "btop", "--", "btop"],
|
|
)
|
|
hypr.dispatch("togglespecialworkspace", "sysmon")
|
|
|
|
def todo(self) -> None:
|
|
self.spawn_or_move(lambda c: c["class"] == "Todoist", ["todoist"], "todo")
|
|
hypr.dispatch("togglespecialworkspace", "todo")
|
|
|
|
def specialws(self) -> None:
|
|
workspaces = hypr.message("workspaces")
|
|
on_special_ws = any(ws["name"] == "special:special" for ws in workspaces)
|
|
toggle_ws = "special"
|
|
|
|
if not on_special_ws:
|
|
active_ws = hypr.message("activewindow")["workspace"]["name"]
|
|
if active_ws.startswith("special:"):
|
|
toggle_ws = active_ws[8:]
|
|
|
|
hypr.dispatch("togglespecialworkspace", toggle_ws)
|