#!/usr/bin/python3

import socket
import time
import subprocess

UDP_PORT = 2021
MESSAGE = b"""NOTIFY * HTTP/1.1
HOST: 239.255.255.250:1900
Server: https://git.f3l.de/asdil12/bambulab-wireguard-proxy/
Location: 10.73.37.29
NT: urn:bambulab-com:device:3dprinter:1
USN: 01S00C321501449
Cache-Control: max-age=1800
DevModel.bambu.com: C11
DevName.bambu.com: Nerdberg Bambu P1P
DevSignal.bambu.com: -41
DevConnect.bambu.com: lan
DevBind.bambu.com: free
Devseclink.bambu.com: secure

""".replace(b"\n", b"\r\n")

while True:
	for pl in subprocess.check_output(["wg", "show", "wg-firezone", "dump"]).decode().strip().split("\n")[1:]:
		ips = pl.split("\t")[3]
		for ip in ips.split(","):
			if ip.endswith("/32"):
				print(ip[:-3])
				print(f"{ip}:{UDP_PORT} {MESSAGE=}" % MESSAGE)
				sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
				sock.bind(('0.0.0.0', 1900)) # 1900 is SSDP src port
				sock.sendto(MESSAGE, (ip, UDP_PORT))
				sock.close()
	time.sleep(10)