1 files changed, 60 insertions, 0 deletions
diff --git a/youtube/youtube.py b/youtube/youtube.py
new file mode 100644
index 0000000..7ec75c0
--- /dev/null
+++ b/youtube/youtube.py
@@ -0,0 +1,60 @@
+import mimetypes
+import urllib.parse
+from youtube import watch_later, watch, search, playlist, channel, comments
+YOUTUBE_FILES = (
+    "/shared.css",
+    "/opensearch.xml",
+    '/comments.css',
+)
+
+def youtube(env, start_response):
+    path, method, query_string = env['PATH_INFO'], env['REQUEST_METHOD'], env['QUERY_STRING']
+    if method == "GET":
+        if path in YOUTUBE_FILES:
+            with open("youtube" + path, 'rb') as f:
+                mime_type = mimetypes.guess_type(path)[0] or 'application/octet-stream'
+                start_response('200 OK',  (('Content-type',mime_type),) )
+                return f.read()
+
+        elif path == "/comments":
+            start_response('200 OK',  (('Content-type','text/html'),) )
+            return comments.get_comments_page(query_string).encode()
+
+        elif path == "/watch":
+            start_response('200 OK',  (('Content-type','text/html'),) )
+            return watch.get_watch_page(query_string).encode()
+        
+        elif path == "/search":
+            start_response('200 OK',  (('Content-type','text/html'),) )
+            return search.get_search_page(query_string).encode()
+        
+        elif path == "/playlist":
+            start_response('200 OK',  (('Content-type','text/html'),) )
+            return playlist.get_playlist_page(query_string).encode()
+        
+        elif path.startswith("/channel/"):
+            start_response('200 OK',  (('Content-type','text/html'),) )
+            return channel.get_channel_page(path[9:], query_string=query_string).encode()
+
+        elif path.startswith("/user/"):
+            start_response('200 OK',  (('Content-type','text/html'),) )
+            return channel.get_user_page(path[6:], query_string=query_string).encode()
+
+        else:
+            start_response('404 Not Found',  () )
+            return b'404 Not Found'
+
+    elif method == "POST":
+        if path == "/edit_playlist":
+            fields = urllib.parse.parse_qs(env['wsgi.input'].read().decode())
+            if fields['action'][0] == 'add' and fields['playlist_name'][0] == 'watch_later':
+                watch_later.add_to_watch_later(fields['video_info_list'])
+                
+            start_response('204 No Content', ())
+        else:
+            start_response('404 Not Found', ())
+            return b'404 Not Found' 
+
+    else:
+        start_response('501 Not Implemented', ())
+        return b'501 Not Implemented'
+\ No newline at end of file