From 7c318896be0292b6734429ae2b9c9f44577fd971 Mon Sep 17 00:00:00 2001
From: zuckerberg <5-zuckerberg@users.noreply.git.neet.dev>
Date: Thu, 27 May 2021 09:51:05 -0400
Subject: [PATCH] wip

---
 downloader.py | 75 ++++++++++++++++++++++++++-------------------------
 radio.py      | 21 ++++++++++++---
 2 files changed, 56 insertions(+), 40 deletions(-)

diff --git a/downloader.py b/downloader.py
index c1e0b59..b7fbd72 100644
--- a/downloader.py
+++ b/downloader.py
@@ -9,50 +9,51 @@ import tempfile
 import sys
 import subprocess
 import os
-
-youtube_dl = None
-
-def execute(cmd):
-  popen = subprocess.Popen(cmd, stdout=subprocess.PIPE)
-  # monitor the stdout
-  BUFFER_SIZE = 1024
-  for chunk in iter(lambda: popen.stdout.read(BUFFER_SIZE), b''):
-    yield chunk
-  popen.stdout.close()
-  popen.wait()
+import pip
+import signal
 
 def updateYoutubeDL():
   pip.main(['install', '--target=' + dirpath, '--upgrade', 'youtube_dl'])
 
 def importYoutubeDL():
-  global youtube_dl
-  youtube_dl = __import__('youtube_dl')
+  return __import__('youtube_dl')
 
-def download(url):
+dirpath = tempfile.mkdtemp()
+sys.path.append(dirpath)
+updateYoutubeDL()
+
+def executeYoutubeDL(url, cb):
+  env = dict(os.environ)
+  env["PYTHONPATH"] = dirpath
+  cmd = [
+    sys.executable,
+    dirpath + "/bin/youtube-dl",
+    "-o", "-",
+    "-f", "bestaudio/best",
+    # "--extract-audio",
+    "--audio-format", "wav",
+    "--prefer-ffmpeg",
+    #"--postprocessor-args", "-re",
+    #"--external-downloader", "ffmpeg",
+    #"--external-downloader-args", "-re",
+    url
+  ]
+  popen = subprocess.Popen(cmd, env=env, stdout=subprocess.PIPE, preexec_fn=os.setsid)
+
+  # monitor the stdout and send to callback, if result from callback function is true,
+  # then kill the download process
+  BUFFER_SIZE = 8096
+  for chunk in iter(lambda: popen.stdout.read(BUFFER_SIZE), b''):
+    if cb(chunk):
+      os.killpg(os.getpgid(popen.pid), signal.SIGTERM)
+      break
+  popen.stdout.close()
+  popen.wait()
+
+def download(url, cb):
   # update youtube-dl
   # TODO: do this only every once in a while
-  #updateYoutubeDL()
+  # updateYoutubeDL()
 
   # start downloader so that it's stdout (with fragments) may be captured
-  for s in execute([sys.executable, os.path.realpath(__file__), dirpath,url]):
-    print(s)
-
-def runYoutubeDL(url):
-  ydl_opts = {
-    'logtostderr': True,
-    'outtmpl': '-',
-  }
-
-  with youtube_dl.YoutubeDL(ydl_opts) as ydl:
-    ydl.download([url])
-
-if __name__ == "__main__":
-  dirpath = sys.argv[1]
-  sys.path.append(dirpath)
-  importYoutubeDL()
-  runYoutubeDL(sys.argv[2])
-else:
-  import pip
-  dirpath = tempfile.mkdtemp()
-  sys.path.append(dirpath)
-  updateYoutubeDL()
\ No newline at end of file
+  executeYoutubeDL(url, cb)
diff --git a/radio.py b/radio.py
index d472ee1..00377c1 100644
--- a/radio.py
+++ b/radio.py
@@ -1,9 +1,24 @@
-import pip
 import downloader
+import ffmpeg
+import sys
 
 def run():
-  downloader.download('https://www.youtube.com/watch?v=BaW_jenozKc')
-  downloader.download('https://www.youtube.com/watch?v=kgBcg4uBd9Q')
+  # process = (
+  #   ffmpeg
+  #   .input('pipe:', format='wav')
+  #   .output("icecast://source:hackme@localhost:8000/stream.mp3", format='mp3', content_type="audio/mpeg")
+  #   .global_args("-re") # argument to act as a live stream
+  #   .run_async(pipe_stdin=True)
+  # )
+
+  def cb(chunk):
+    # process.stdin.write(chunk)
+    sys.stdout.write("".join(map(chr, chunk)))
+    return False
+
+  downloader.download('https://www.youtube.com/watch?v=BaW_jenozKc', cb)
+  # downloader.download('https://www.youtube.com/watch?v=kgBcg4uBd9Q', cb)
+  # downloader.download('https://www.youtube.com/watch?v=EbnH3VHzhu8', cb)
 
 if __name__ == "__main__":
   run()
\ No newline at end of file