62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import gi, sys
|
|
|
|
from typing import Callable, Optional, Any, Dict
|
|
|
|
gi.require_version("Astal", "3.0")
|
|
gi.require_version("AstalIO", "0.1")
|
|
from gi.repository import Astal, AstalIO, Gio
|
|
|
|
type Config = Dict[str, Any]
|
|
|
|
class AstalPy(Astal.Application):
|
|
request_handler: Optional[Callable[[str, Callable[[Any], None]], None]] = None
|
|
|
|
def do_astal_application_request(self, msg: str, conn: Gio.SocketConnection):
|
|
if callable(self.request_handler):
|
|
def respond(response: Any):
|
|
AstalIO.write_sock(conn, str(response), None, None)
|
|
self.request_handler(msg, respond)
|
|
else:
|
|
AstalIO.Application.do_request(self, msg, conn)
|
|
|
|
def quit(self, code: int = 0):
|
|
super().quit()
|
|
sys.exit(code)
|
|
|
|
def apply_css(self, css: str, reset: bool = False):
|
|
super().apply_css(css, reset)
|
|
|
|
def start(self, **config: Any):
|
|
config.setdefault("client", lambda *_: (print(f'Astal instance "{self.get_instance_name()}" is already running'), sys.exit(1)))
|
|
config.setdefault("hold", True)
|
|
|
|
self.request_handler = config.get("request_handler")
|
|
|
|
if "css" in config:
|
|
self.apply_css(config["css"])
|
|
if "icons" in config:
|
|
self.add_icons(config["icons"])
|
|
|
|
for key in ["instance_name", "gtk_theme", "icon_theme", "cursor_theme"]:
|
|
if key in config:
|
|
self.set_property(key, config[key])
|
|
|
|
def on_activate(_):
|
|
if callable(config.get("main")):
|
|
config["main"]()
|
|
if config["hold"]:
|
|
self.hold()
|
|
|
|
self.connect("activate", on_activate)
|
|
|
|
try:
|
|
self.acquire_socket()
|
|
except Exception:
|
|
return config["client"](lambda msg: AstalIO.send_message(self.get_instance_name(), msg), *sys.argv[1:])
|
|
|
|
self.run()
|
|
|
|
return self
|
|
|
|
App = AstalPy()
|