app.py

#!/usr/bin/env python2

import db, codec, os, json, mimetypes, datetime, re, cgi
from config import config

class JSONApplication(object):
	@staticmethod
	def cache_check(track):
		ext = os.path.splitext(track.filename)[-1][1:]
		if ext in config.get('allow_extensions').split():
			return True
		try:
			r = codec.Recoder(track.get_path(), config.get('encoder'), track.file_index, os.path.join(config.get('cache_dir'), str(track.id)))
		except codec.DecoderNotFoundError:
			return False
		return os.path.exists(r.enc_destination)

	@staticmethod
	def format_track(track):
		d = track.dict()
		d['cache'] = JSONApplication.cache_check(track)
		return d

	@staticmethod
	def format_album(album, artist = False):
		d = {
			'id': album.id,
			'name': album.name,
		}
		if artist:
			d['artist'] = {'name': album.artist.name}
		return d

	def list(self, environ, start_response, path):
		root_id = int(path[1]) if len(path) > 1 and len(path[1]) else 0
		session = db.Session()
		try:
			if root_id > 0:
				directory = db.Directory.get_by_id(session, root_id)
			else:
				directory = db.Directory.get(session, config.get('music_root'))
			directories = directory.children
			tracks = directory.tracks
			contents = json.dumps([x.dict() for x in directories] +
				[self.format_track(x) for x in tracks])
		finally:
			session.close()
		start_response('200 OK', [('Content-Type', 'application/json'), ('Content-Length', str(len(contents)))])
		return [contents]

	def hint(self, environ, start_response, path):
		track_id = int(path[1])
		session = db.Session()
		try:
			track = db.Track.get_by_id(session, track_id)
			cached = self.cache_check(track)
		finally:
			session.close()

		if not cached:
			r = codec.Recoder(track.get_path(), config.get('encoder'), track.file_index, os.path.join(config.get('cache_dir'), str(track.id)))
			r.recode()

		start_response('200 OK', [('Content-Type', 'application/json')])
		return [json.dumps({'recoded': not cached})]

	re_search = re.compile(r'(")?((?(1)[^"]|[^ ])+)(?(1)")')

	def search(self, environ, start_response, path):
		args = cgi.FieldStorage(environ = environ)
		query = args.getvalue('q')
		if not (query or '').strip():
			start_response('200 OK', [])
			return json.dumps({'total': 0, 'results': []})

		r = self.re_search.findall(query)
		d = {}
		l = []
		for _, v in r:
			if ':' in v:
				k, v = v.split(':', 1)
				d[k] = v
			else:
				l.append(v)

		total_results = 0
		results = []
		try:
			session = db.Session()
			r = db.Track.search(session, *l, **d)
			total_results = r.count()
			results = [self.format_track(x) for x in r.limit(100)]
		finally:
			session.close()

		start_response('200 OK', [])
		return json.dumps({
			'total': total_results,
			'results': results,
		})

	def albums(self, environ, start_response, path):
		page = int(path[1] if len(path) > 1 else 0)
		page_size = config.getint('album_pagesize', default = 75)
		try:
			session = db.Session()
			albums = session.query(db.Album).offset(page*page_size).limit(page_size)
			results = [self.format_album(a, artist = True) for a in albums]
		finally:
			session.close()

		start_response('200 OK', [])
		return json.dumps(results)

	def album(self, environ, start_response, path):
		album = int(path[1])
		try:
			session = db.Session()
			album = session.query(db.Album).filter(db.Album.id == album).one()
			results = [self.format_track(x) for x in album.tracks]
		finally:
			session.close()

		start_response('200 OK', [])
		return json.dumps(results)

	def artist(self, environ, start_response, path):
		artist = int(path[1])
		try:
			session = db.Session()
			artist = session.query(db.Artist).filter(db.Artist.id == artist).one()
			results = {
				'id': artist.id,
				'name': artist.name,
				'albums': [self.format_album(a) for a in artist.albums],
			}
		finally:
			session.close()

		start_response('200 OK', [])
		return json.dumps(results)

	handlers = {
		'list': list,
		'hint': hint,
		'search': search,
		'albums': albums,
		'album': album,
		'artist': artist,
	}

	def __call__(self, environ, start_response, path):
		module = path[0]
		if module in self.handlers:
			return self.handlers[module](self, environ, start_response, path)
		else:
			start_response('404 Not Found', [])
			return []

class Application(object):
	rfc1123_format = '%a, %d %b %Y %H:%M:%S +0000'

	def __init__(self):
		self.jsonapp = JSONApplication()

	def json(self, environ, start_response, path):
		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)

	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()
			ext = os.path.splitext(filename)[-1][1:]
			if not ext in config.get('allow_extensions').split():
				r = codec.Recoder(filename, config.get('encoder'), track.file_index, os.path.join(config.get('cache_dir'), str(track.id)))
				filename = r.enc_destination
				if not os.path.exists(filename):
					r.recode()
		except db.NoResultFound:
			start_response('404 Not Found', [])
			return []
		finally:
			session.close()

		return self._serve_path(environ, start_response, filename)

	def album_cover(self, environ, start_response, path):
		album = int(path[1].split('.')[0])
		session = db.Session()
		cover = None
		try:
			album = db.Album.get_by_id(session, album)
			import coverart
			cover = coverart.get_coverart(album)
		except db.NoResultFound:
			start_response('404 Not Found', [])
			return []
		finally:
			session.close()

		if not cover:
			start_response('302 Found', [('Location', '/static/nocover.jpg')])
			return []

		return self._serve_path(environ, start_response, cover)

	handlers = {
		'json': json,
		'static': static,
		'file': file,
		'track': track,
		'album-cover': album_cover,
	}

	def __call__(self, environ, start_response):
		path = environ['PATH_INFO'].split('/')[1:]
		module = path[0] or None
		if not module:
			module = 'static'
			path = ['static', 'index.html']

		if module in self.handlers:
			return self.handlers[module](self, environ, start_response, path)

		start_response('404 Not Found', [])
		return []

if __name__ == '__main__':
	import sys
	if len(sys.argv) == 3:
		from flup.server.fcgi import WSGIServer
		WSGIServer(Application(), bindAddress = (sys.argv[1], int(sys.argv[2]))).run()
	else:
		from wsgiref.simple_server import make_server, WSGIServer
		from SocketServer import ThreadingMixIn
		class ThreadingWSGIServer(ThreadingMixIn, WSGIServer): pass
		# enable IPv6
		ThreadingWSGIServer.address_family |= 10
		httpd = make_server('', 8000, Application(), server_class = ThreadingWSGIServer)
		httpd.serve_forever()