blob: 30dc5da62bae6f388b8021762dddee25e3d01bd9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#!/usr/bin/env python3
import asyncio
import subprocess
import json
import sys
from urllib.parse import urlencode, quote
EXTENSION = "irth/dmenu"
COMMAND = "index"
HOST = "127.0.0.1"
PORT = 8888
SHUTDOWN_EVENT = asyncio.Event()
elements = [el for el in list(sys.stdin) if not el.isspace()]
async def handle_echo(reader, writer):
if not SHUTDOWN_EVENT.is_set():
data_len = f"{len(elements)}\n".encode("utf-8")
writer.write(data_len)
for el in elements:
data_line = f"{el.strip()}\n".encode("utf-8")
writer.write(data_line)
await writer.drain()
selection = await reader.read()
writer.close()
await writer.wait_closed()
sys.stdout.write(selection.decode())
SHUTDOWN_EVENT.set()
async def main():
server = await asyncio.start_server(handle_echo, HOST, PORT)
addrs = ", ".join(str(sock.getsockname()) for sock in server.sockets)
async with server:
asyncio.create_task(server.serve_forever())
arguments = {
"host": HOST,
"port": str(PORT),
}
query = urlencode({"arguments": json.dumps(arguments)}, quote_via=quote)
url = f"raycast://extensions/{EXTENSION}/{COMMAND}?{query}"
await asyncio.create_subprocess_shell(f"open -g {url}")
await SHUTDOWN_EVENT.wait()
asyncio.run(main())
|
