From 5aefaafd26598cd5d9b40d98eab25623e659ebf8 Mon Sep 17 00:00:00 2001
From: Matt Behrens <matt@zigg.com>
Date: Tue, 26 Dec 2017 16:34:39 -0500
Subject: [PATCH] =?UTF-8?q?IT=20LIVES=20=F0=9F=99=8C=F0=9F=8F=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fediplay.py | 91 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 stream      | 19 +++++++++++
 2 files changed, 110 insertions(+)
 create mode 100644 fediplay.py
 create mode 100755 stream

diff --git a/fediplay.py b/fediplay.py
new file mode 100644
index 0000000..6cb4f88
--- /dev/null
+++ b/fediplay.py
@@ -0,0 +1,91 @@
+from subprocess import run
+from threading import Thread, Lock
+
+from lxml.etree import HTML
+import mastodon
+Mastodon = mastodon.Mastodon
+from youtube_dl import YoutubeDL
+
+class Queue(object):
+    def __init__(self):
+        self.lock = Lock()
+        self.playing = False
+        self.queue = []
+
+    def add(self, url):
+        filename = Getter().get(url)
+        
+        with self.lock:
+            self.queue.append(filename)
+            if not self.playing:
+                self._play(self.queue.pop(0), self._play_finished)
+
+    def _play(self, filename, cb_complete):
+        self.playing = True
+
+        def run_thread(filename, cb_complete):
+            print('==> Playing', filename)
+            run(['afplay', filename])
+            cb_complete()
+
+        thread = Thread(target=run_thread, args=(filename, cb_complete))
+        thread.start()
+
+    def _play_finished(self):
+        with self.lock:
+            self.playing = False
+            if len(self.queue) > 0:
+                self._play(self.queue.pop(0), self._play_finished)
+
+class Getter(object):
+    def _progress_hook(self, progress):
+        if progress['status'] == 'finished':
+            self.filename = progress['filename']
+
+    def get(self, url):
+        options = {
+                'format': 'mp3/mp4',
+                'progress_hooks': [self._progress_hook]
+                }
+        with YoutubeDL(options) as downloader:
+            downloader.download([url])
+
+        return self.filename
+
+class StreamListener(mastodon.StreamListener):
+    def __init__(self):
+        self.queue = Queue()
+
+    def on_update(self, status):
+        tags = extract_tags(status)
+        if 'fediplay' in tags:
+            links = extract_links(status)
+            self.queue.add(links[0])
+
+def register(api_base_url):
+    Mastodon.create_app('fediplay', api_base_url=api_base_url, to_file='clientcred.secret')
+
+def login(api_base_url, email, password):
+    client = Mastodon(client_id='clientcred.secret', api_base_url=api_base_url)
+    client.log_in(email, password, to_file='usercred.secret')
+
+def stream(api_base_url):
+    client = Mastodon(client_id='clientcred.secret', access_token='usercred.secret', api_base_url=api_base_url)
+    listener = StreamListener()
+    client.user_stream(listener)
+
+def extract_tags(toot):
+    return [tag['name'] for tag in toot['tags']]
+
+def has_external_link_class(class_string):
+    classes = class_string.split(' ')
+    if classes:
+        return 'mention' in classes
+
+    return False
+
+def extract_links(toot):
+    html = HTML(toot['content'])
+    all_links = html.cssselect('a')
+    return [link.attrib['href'] for link in all_links if not has_external_link_class(link.attrib.get('class', ''))]
+
diff --git a/stream b/stream
new file mode 100755
index 0000000..e4cbba1
--- /dev/null
+++ b/stream
@@ -0,0 +1,19 @@
+#!/usr/bin/env python
+
+API_BASE_URL = 'https://cybre.space'
+
+import os
+from getpass import getpass
+
+from fediplay import register, login, stream
+
+if not os.path.exists('clientcred.secret'):
+    register(API_BASE_URL)
+
+if not os.path.exists('usercred.secret'):
+    email = input('Email: ')
+    password = getpass('Password: ')
+    login(API_BASE_URL, email, password)
+
+stream('https://cybre.space')
+