1 files changed, 55 insertions, 9 deletions

diff --git a/radio/radio.py b/radio/radio.py
index 2c04056..d562627 100755
--- a/radio/radio.py
+++ b/radio/radio.py
@@ -5,8 +5,11 @@ from travnik import Torrent
 import inotify.adapters
 from threading import Thread
 from random import randrange
-from os import stat
+from os import stat, path
 from time import sleep
+from pydub import AudioSegment
+from pydub.playback import play
+from socket import socket, AF_UNIX, AF_INET, SOCK_STREAM
 
 import libtorrent
 
@@ -26,25 +29,31 @@ def random_file(torrent, save_path, endung=".flac"):
 		if (torrent.torrent_file().files().file_path(i).endswith(endung)):
 			endungs += 1
 	if endungs == 0:
+		print("no more endungs")
 		return False
 	chosen = randrange(endungs)
 	tries_left = endungs
+	print(f"there are {endungs} {endung} files")
 	while True:
 		if tries_left == 0:
+			print("no more tries")
 			return False
 		tries_left -= 1
 		try:
-			stat(save_path + "/" + torrent.torrent_file().files().file_path(get_index_from_endung(chosen)))
+			p = save_path + "/" + torrent.torrent_file().files().file_path(get_index_from_endung(chosen))
+			print(f"trying path {p} with stat")
+			stat(p)
 		except FileNotFoundError:
-			chosen = (chosen + 1) % endungs;
 			break
 		else:
+			chosen = (chosen + 1) % endungs;
 			continue
 	prios[get_index_from_endung(chosen)] = 4
 	torrent.prioritize_files(prios)
 	print(f"random file {torrent.torrent_file().files().file_path(get_index_from_endung(chosen))} chosen to download")
+	return True
 
-def watcher(session, torrents, d="/tmp/", s=".", active_torrents=10):
+def watcher(session, torrents, s, d="/tmp/", active_torrents=10):
 	i = inotify.adapters.Inotify()
 	i.add_watch(d)
 	for event in i.event_gen(yield_nones=False):
@@ -63,33 +72,70 @@ def watcher(session, torrents, d="/tmp/", s=".", active_torrents=10):
 						del torrents[0]
 					break
 
-def alerts(session):
+def alerts(session, playstack, save_path, torrents):
 	while True:
 		session.wait_for_alert(86400*1000*21)
 		for a in session.pop_alerts():
 			if a.category() & libtorrent.alert.category_t.error_notification:
 				print(f"error notification alert from libtorrent: {a}")
 			if type(a) == libtorrent.file_completed_alert:
-				print(f"file completed alert: {a}")
-			if type(a) in [libtorrent.external_ip_alert, libtorrent.listen_succeeded_alert, libtorrent.dht_get_peers_alert, libtorrent.dht_outgoing_get_peers_alert, libtorrent.dht_reply_alert, libtorrent.torrent_added_alert, libtorrent.state_changed_alert, libtorrent.torrent_resumed_alert, libtorrent.stats_alert, libtorrent.torrent_checked_alert, libtorrent.peer_connect_alert, libtorrent.alert, libtorrent.peer_disconnected_alert, libtorrent.incoming_connection_alert, libtorrent.block_downloading_alert, libtorrent.dht_bootstrap_alert, libtorrent.dht_announce_alert, libtorrent.block_finished_alert, libtorrent.piece_finished_alert, libtorrent.peer_unsnubbed_alert, libtorrent.peer_snubbed_alert, libtorrent.portmap_error_alert]:
+				if a.handle.get_file_priorities()[a.index] != 4:
+					continue
+				playstack.append(a.handle.torrent_file().files().file_path(a.index))
+				if not random_file(a.handle, save_path):
+					session.remove_torrent(a.handle)
+					torrents.remove(a.handle)
+					continue
+				else:
+					a.handle.resume()
+			if type(a) in [libtorrent.external_ip_alert, libtorrent.listen_succeeded_alert, libtorrent.dht_get_peers_alert, libtorrent.dht_outgoing_get_peers_alert, libtorrent.dht_reply_alert, libtorrent.torrent_added_alert, libtorrent.state_changed_alert, libtorrent.torrent_resumed_alert, libtorrent.stats_alert, libtorrent.torrent_checked_alert, libtorrent.peer_connect_alert, libtorrent.alert, libtorrent.peer_disconnected_alert, libtorrent.incoming_connection_alert, libtorrent.block_downloading_alert, libtorrent.dht_bootstrap_alert, libtorrent.dht_announce_alert, libtorrent.block_finished_alert, libtorrent.piece_finished_alert, libtorrent.peer_unsnubbed_alert, libtorrent.peer_snubbed_alert, libtorrent.portmap_error_alert, libtorrent.block_timeout_alert, libtorrent.block_uploaded_alert, libtorrent.performance_alert, libtorrent.cache_flushed_alert]:
 				continue
 			print(f"{type(a)} | {a}")
 
+def stream(playstack):
+	if False:
+		while True:
+			# sock = socket(AF_INET, SOCK_STREAM)
+			# sock.connect("tranzistor", 1234)
+			sock = socket(AF_UNIX, SOCK_STREAM)
+			sock.connect("/tmp/ls")
+			sock.sendall(b"request.alive\r\n")
+			sleep(1)
+			data = sock.recv(256)
+			a = data.split(b"\r")[0]
+			if a == b"":
+				sock.sendall(b"request.push " + bytes(playstack.pop(), encoding="utf-8"))
+				sleep(1)
+	else:
+		while True:
+			if len(playstack) == 0:
+				sleep(1)
+				continue
+			song = AudioSegment.from_file(path.abspath(playstack.pop()), "flac")
+			play(song)
+			sleep(1)
+
 def progress(session, torrents):
 	while True:
 		for torr in torrents:
 			s = torr.status()
 			print(torr.torrent_file().name() + '\t%.2f%% complete (down: %.1f kB/s up: %.1f kB/s peers: %d) %s' % (s.progress * 100, s.download_rate / 1000, s.upload_rate / 1000, s.num_peers, s.state))
 		sleep(1)
+
 if __name__ == "__main__":
 	session = libtorrent.session({"alert_mask": libtorrent.alert.category_t.all_categories});
 	torrents = []
-	w = Thread(target=watcher, args=(session, torrents))
-	a = Thread(target=alerts, args=(session,))
+	playstack = []
+	save_path = "."
+	w = Thread(target=watcher, args=(session, torrents, save_path))
+	a = Thread(target=alerts, args=(session, playstack, save_path, torrents))
 	p = Thread(target=progress, args=(session, torrents))
+	s = Thread(target=stream, args=(playstack,))
 	w.start()
 	a.start()
 	p.start()
+	s.start()
 	w.join()
 	a.join()
 	p.join()
+	s.join()