From 65e25caa6e836644c6ede0ee5316d09f2cd5dba3 Mon Sep 17 00:00:00 2001
From: palaviv <palaviv@gmail.com>
Date: Sun, 30 Oct 2016 22:48:45 +0200
Subject: [PATCH] Supporting seek

---
 README.md |  5 ++--
 caster.py | 72 +++++++++++++++++++++++++++++++++++++++++--------------
 setup.py  |  1 -
 3 files changed, 57 insertions(+), 21 deletions(-)

diff --git a/README.md b/README.md
index b67ed0b..8d59393 100644
--- a/README.md
+++ b/README.md
@@ -18,9 +18,11 @@ caster /file/to/play.mp4 --subtitles /path/to/subtitles.vtt
 ### Player control
 
 ```
-space -> toggle between space and pause.
+space -> toggle between play and pause.
 up -> volume up.
 down -> volume down.
+right -> jump 30 seconds forward.
+left -> jump 30 seconds backwards.
 s -> stop.
 ```
 
@@ -28,7 +30,6 @@ Dependencies
 ------------
 
 * pychromecast
-* six
 * readchar
 
 Install
diff --git a/caster.py b/caster.py
index d9f5912..0fde97c 100755
--- a/caster.py
+++ b/caster.py
@@ -6,8 +6,8 @@
 import threading
 import socket
 import os
+from socketserver import ThreadingMixIn
 
-from six import b
 import pychromecast
 import readchar
 
@@ -20,37 +20,66 @@ def get_internal_ip(dst_ip):
 
 class RequestHandler(BaseHTTPRequestHandler):
     content_type = "video/mp4"
+    chunk_size = 1024
 
     """ Handle HTTP requests for files which do not need transcoding """
 
+    def parse_range(self, file_size):
+        start = self.headers["range"].split("=")[1].split("-")[0]
+        end = file_size - 1
+        return int(start), int(end)
+
     def do_GET(self):
         self.protocol_version = "HTTP/1.1"
-        self.send_response(200)
-        self.send_header("Content-type", self.content_type)
-        self.send_header('Access-Control-Allow-Origin', '*')
-        self.send_header("Transfer-Encoding", "chunked")
-        self.end_headers()
 
         with open(self.path, "rb") as f:
-            while True:
-                data = f.read(1024)
-                if len(data) == 0:
-                    break
+            f.seek(0, 2)
+            file_size = f.tell()
+
+            if "range" in self.headers:
+                self.send_response(206)
+                start_range, end_range = self.parse_range(file_size)
+                self.send_header('Content-Range', 'bytes {START}-{END}/{TOTAL}'.format(
+                    START=start_range, END=end_range, TOTAL=file_size))
+                self.send_header('Accept-Ranges', 'bytes')
+            else:
+                self.send_response(200)
+                start_range, end_range = 0, file_size
+            self.send_header('Content-Length', end_range - start_range + 1)
+            self.send_header("Content-type", self.content_type)
+            self.send_header('Access-Control-Allow-Origin', '*')
+            self.end_headers()
+
+            f.seek(start_range)
 
-                chunk_size = "%0.2X" % len(data)
-                self.wfile.write(b(chunk_size))
-                self.wfile.write(b("\r\n"))
+            curr = start_range
+
+            while True:
+                if curr + self.chunk_size > end_range:
+                    read_size = end_range - curr
+                else:
+                    read_size = self.chunk_size
+                curr += read_size
+                data = f.read(read_size)
                 self.wfile.write(data)
-                self.wfile.write(b("\r\n"))
+                if curr == end_range:
+                    break
 
-        self.wfile.write(b("0"))
-        self.wfile.write(b("\r\n\r\n"))
+    def handle_one_request(self):
+        try:
+            return BaseHTTPRequestHandler.handle_one_request(self)
+        except socket.error:
+            pass
 
 
 class SubRequestHandler(RequestHandler):
     content_type = "text/vtt"
 
 
+class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
+    pass
+
+
 def handle_input(server_thread, dev, mc):
     while server_thread.is_alive():
         key = readchar.readkey()
@@ -66,6 +95,12 @@ def handle_input(server_thread, dev, mc):
             dev.volume_up()
         elif key == readchar.key.DOWN:
             dev.volume_down()
+        elif key == readchar.key.RIGHT:
+            mc.update_status(blocking=True)
+            mc.seek(mc.status.current_time + 30)
+        elif key == readchar.key.LEFT:
+            mc.update_status(blocking=True)
+            mc.seek(mc.status.current_time - 30)
 
 
 def get_args():
@@ -96,8 +131,8 @@ def main():
 
     server_ip = get_internal_ip(dev.host)
 
-    server = HTTPServer((server_ip, 0), RequestHandler)
-    server_thread = threading.Thread(target=server.handle_request)
+    server = ThreadedHTTPServer((server_ip, 0), RequestHandler)
+    server_thread = threading.Thread(target=server.serve_forever)
     server_thread.start()
 
     if subs:
@@ -117,6 +152,7 @@ def main():
 
     handle_input(server_thread, dev, mc)
 
+    server.shutdown()
     server_thread.join()
     server.server_close()
 
diff --git a/setup.py b/setup.py
index d2e83b0..7660fef 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,6 @@
         'console_scripts': ['caster=caster:main'],
     },
     install_requires=[
-        'six',
         'pychromecast',
         'readchar'
     ],