warp.pyw

import subprocess
import pystray
from pystray import MenuItem as item
from PIL import Image, ImageDraw


def get_warp_status():
    result = subprocess.run(
        ["warp-cli", "status"], capture_output=True, text=True, shell=True
    )
    return "connected" if "Connected" in result.stdout else "disconnected"


def switch(icon, item):
    status = get_warp_status()
    if status == "connected":
        icon.icon = disconnected_ico
        subprocess.run(["warp-cli", "disconnect"], shell=True)
    else:
        icon.icon = connected_ico
        subprocess.run(["warp-cli", "connect"], shell=True)


def close(icon, item):
    icon.stop()


connected_ico = Image.new("RGBA", (16, 16), (255, 255, 255, 0))
draw = ImageDraw.Draw(connected_ico)
draw.line((2, 8, 6, 12), fill="green", width=2)
draw.line((6, 12, 14, 4), fill="green", width=2)

disconnected_ico = Image.new("RGBA", (16, 16), (255, 255, 255, 0))
draw = ImageDraw.Draw(disconnected_ico)
draw.line((2, 2, 13, 13), fill="red", width=2)
draw.line((3, 13, 14, 2), fill="red", width=2)


pystray.Icon(
    "warp",
    connected_ico if get_warp_status() == "connected" else disconnected_ico,
    "Warp",
    (item("Connect/Disconnect", switch), item("Exit", close)),
).run()