1 files changed, 78 insertions, 0 deletions
diff --git a/gjs/src/application.ts b/gjs/src/application.ts
new file mode 100644
index 0000000..19cdaae
--- /dev/null
+++ b/gjs/src/application.ts
@@ -0,0 +1,78 @@
+import Astal from "gi://Astal"
+import GObject from "gi://GObject"
+import Gio from "gi://Gio"
+
+type RequestHandler = {
+    (request: string, res: (response: string) => void): void
+}
+
+type Config = Partial<{
+    instanceName: string
+    gtkTheme: string
+    iconTheme: string
+    cursorTheme: string
+    css: string
+    requestHandler: RequestHandler
+    hold: boolean
+}>
+
+// @ts-expect-error missing types
+// https://github.com/gjsify/ts-for-gir/issues/164
+import { setConsoleLogDomain } from "console"
+import { exit, programArgs } from "system"
+
+class AstalJS extends Astal.Application {
+    static { GObject.registerClass(this) }
+
+    runJS(body: string): Promise<any> {
+        return new Promise((res, rej) => {
+            try {
+                const fn = Function(`return (async function() {
+                    ${body.includes(";") ? body : `return ${body};`}
+                })`)
+                fn()()
+                    .then(res)
+                    .catch(rej)
+            } catch (error) {
+                rej(error)
+            }
+        })
+    }
+
+    requestHandler?: RequestHandler
+
+    vfunc_response(msg: string, conn: Gio.SocketConnection): void {
+        if (typeof this.requestHandler === "function") {
+            this.requestHandler(msg, response => {
+                Astal.write_sock(conn, response, (_, res) =>
+                    Astal.write_sock_finish(res),
+                )
+            })
+        } else {
+            super.vfunc_response(msg, conn)
+        }
+    }
+
+    start({ requestHandler, css, hold, ...cfg }: Config = {}, callback?: (args: string[]) => void) {
+        Object.assign(this, cfg)
+        setConsoleLogDomain(this.instanceName)
+
+        this.requestHandler = requestHandler
+        this.connect("activate", () => callback?.(programArgs))
+        if (!this.acquire_socket()) {
+            print(`Astal instance "${this.instanceName}" already running`)
+            exit(1)
+        }
+
+        if (css)
+            this.apply_css(css, false)
+
+        hold ??= true
+        if (hold)
+            this.hold()
+
+        this.runAsync([])
+    }
+}
+
+export const App = new AstalJS