From c797394d45a1ff820b4f4ffc465bc51c00fed7b1 Mon Sep 17 00:00:00 2001
From: Jon Bergli Heier <snakebite@jvnv.net>
Date: Thu, 16 Feb 2012 23:27:11 +0100
Subject: Serve and play raw tracks from /track/<id> using SoundManager 2.

---
 app.py | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 54 insertions(+), 5 deletions(-)

(limited to 'app.py')

diff --git a/app.py b/app.py
index 0358b96..18c48f5 100755
--- a/app.py
+++ b/app.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python2
 
-import db, os, json, mimetypes
+import db, os, json, mimetypes, datetime
 from config import config
 
 class JSONApplication(object):
@@ -30,6 +30,8 @@ class JSONApplication(object):
 			return self.handlers[module](self, environ, start_response, path)
 
 class Application(object):
+	rfc1123_format = '%a, %d %b %Y %H:%M:%S +0000'
+
 	def __init__(self):
 		self.jsonapp = JSONApplication()
 
@@ -37,20 +39,67 @@ class Application(object):
 		path.pop(0)
 		return self.jsonapp(environ, start_response, path)
 
+	def _serve_path(self, environ, start_response, filename):
+		if not os.path.exists(filename) or '..' in filename.split(os.path.sep):
+			start_response('404 Not Found', [])
+			return []
+
+		do_range = 'HTTP_RANGE' in environ
+		if do_range:
+			file_range = environ['HTTP_RANGE'].split('bytes=')[1]
+
+		mime = mimetypes.guess_type(filename, strict = False)[0] or 'application/octet-stream'
+		last_modified = datetime.datetime.fromtimestamp(os.path.getmtime(filename)).strftime(self.rfc1123_format)
+
+		# Range handling
+		if do_range:
+			start, end = [int(x or 0) for x in file_range.split('-')]
+			size = os.path.getsize(filename)
+
+			if end == 0:
+				end = size-1
+
+			write_out = start_response('206 Partial Content', [('Content-Type', mime),
+				('Content-Range', 'bytes {start}-{end}/{size}'.format(start = start, end = end, size = size)),
+				('Content-Length', str(end - start + 1)), ('Last-Modified', last_modified)])
+
+			f = open(filename, 'rb')
+			f.seek(start)
+			remaining = end-start+1
+			s = f.read(min(remaining, 1024))
+			while s:
+				write_out(s)
+				remaining -= len(s)
+				s = f.read(min(remaining, 1024))
+			return []
+
+		start_response('200 OK', [('Content-Type', mime), ('Content-Length', str(os.path.getsize(filename))),
+			('Last-Modified', last_modified)])
+		return open(filename, 'rb')
+
 	def static(self, environ, start_response, path):
 		filename = os.path.join('static', *path[1:])
+		return self._serve_path(environ, start_response, filename)
 
-		if not os.path.exists(filename) or '..' in path:
+	def track(self, environ, start_response, path):
+		track = int(path[1])
+		session = db.Session()
+		try:
+			track = db.Track.get_by_id(session, track)
+			filename = track.get_path()
+		except db.NoResultFound:
 			start_response('404 Not Found', [])
 			return []
+		finally:
+			session.close()
 
-		mime = mimetypes.guess_type(filename, strict = False)[0] or 'application/octet-stream'
-		start_response('200 OK', [('Content-Type', mime)])
-		return open(filename, 'rb')
+		return self._serve_path(environ, start_response, filename)
 
 	handlers = {
 		'json': json,
 		'static': static,
+		'file': file,
+		'track': track,
 	}
 
 	def __call__(self, environ, start_response):
-- 
cgit v1.2.3