quoting and unquoting url paths

.gitignore

@@ -1,3 +1,4 @@
 __pycache__/
 *.swp
 *.swo
+*.json

README.md

@@ -4,7 +4,7 @@ Stream some music.
 ## Requirements  
 Python 3.6+  
 FFmpeg compiled with `--enable-libopus`  
-Python packages: `flask gunicorn mutagen`  
+Python packages: `flask gunicorn mutagen Flask-RESTful`  
 
 ## Install  
 1. Get on the floor  

musik.py

@@ -3,17 +3,16 @@
 Music streaming.
 """
 import os
-import re
 import json
 import random
 import subprocess
 from urllib import parse
-from collections import defaultdict
 
-from flask import Flask, Response, render_template, send_file
-from werkzeug.utils import secure_filename
+from flask import Flask, Response, render_template, send_file, url_for
+from flask_restful import reqparse, abort, Api, Resource
+import mutagen
 
-MUSIC_DIR = "/mnt/music/Music"
+MUSIC_DIR = "/home/iou1name/music/Music"
 MUSIC_EXT = ['flac', 'mp3', 'wav', 'm4a']
 FFMPEG_CMD = [
 	'ffmpeg', '-y',
@@ -25,46 +24,178 @@ FFMPEG_CMD = [
 	'-'
 ]
 
-def build_tree(root_dir):
-	"""Walks the music directory and builds a tree."""
-	print("Building tree.")
-	tree = defaultdict(dict)
-	for dirName, subDirs, files in os.walk(root_dir):
-		if dirName == root_dir:
-			continue
-		reg = re.search(".*/(.+?) - (.+) \(\d{4}\)", dirName)
-		if not reg:
-			print(dirName)
-			continue
-		artist, album = reg.groups()
-		tracks = [f for f in files if f.rpartition('.')[2] in MUSIC_EXT]
-		tree[artist][album] = tracks
-	return tree
+class Track:
+	def __init__(self, filepath=None, d=None, coverart=""):
+		if d:
+			for attr, value in d.items():
+				setattr(self, attr, value)
+			return
+
+		if filepath.endswith("mp3"):
+			m = mutagen.mp3.EasyMP3(filepath)
+		else:
+			m = mutagen.File(filepath)
+		self.tracknumber = m.get('tracknumber', [''])[0]
+		self.title = m.get('title', [''])[0]
+		self.artist = m.get('artist', [''])[0]
+		self.album = m.get('album', [''])[0]
+		self.date = m.get('date', [''])[0]
+		self.length = str(int(m.info.length) // 60) + ":"
+		self.length += str(int(m.info.length) % 60)
+		self.filepath = filepath
+		self.coverart = coverart
+
+
+def build_library(root_dir):
+	"""Walks the music directory and builds a library of tracks."""
+	print("Building library")
+	tracks = []
+	for dir_name, sub_dirs, files in os.walk(root_dir):
+		for file in files:
+			if not os.path.splitext(file)[1][1:] in MUSIC_EXT:
+				continue
+			filepath = os.path.join(root_dir, dir_name, file)
+			if "folder.jpg" in files:
+				coverart = os.path.join(root_dir, dir_name, "folder.jpg")
+			else:
+				coverart = ""
+			track = Track(filepath, coverart=coverart)
+			tracks.append(track)
+	print("Done")
+	return tracks
+
+
+def init_library():
+	"""Loads the library from file, or builds a new one if one isn't found."""
+	if os.path.isfile("library.json"):
+		with open("library.json", "r") as file:
+			tracks = json.loads(file.read())
+		tracks = [Track(d=d) for d in tracks]
+	else:
+		tracks = build_library(MUSIC_DIR)
+		with open("library.json", "w") as file:
+			file.write(json.dumps([vars(t) for t in tracks]))
+	return tracks
 
 
 app = Flask(__name__)
+api = Api(app)
+tracks = init_library()
 
 @app.route('/')
 def index():
 	"""Main index page."""
-	nav_items = os.listdir(MUSIC_DIR)
-	nav_items.sort()
-	nav_items = [item + '/' for item in nav_items]
-	cd = "/"
+	artists = list(set(t.artist for t in tracks))
+	artists.sort()
 	return render_template('index.html', **locals())
 
 
-@app.route('/stream/<path:track>')
-def stream(track):
+parser = reqparse.RequestParser()
+parser.add_argument('artist')
+parser.add_argument('album')
+parser.add_argument('track')
+
+def validate_select_args(args):
+	"""
+	If a track is specified, both artist and album must also be specified.
+	If an album is specified, the artist must also be specified.
+	"""
+	if args.get('track'):
+		if not args.get('artist') or not args.get('album'):
+			abort(400, message="Artist and album must also be specified.")
+	elif args.get('album'):
+		if not args.get('artist'):
+			abort(400, message="Artist must also be specified.")
+	elif not args.get('artist'):
+		abort(400, message="You must specify at least an artist.")
+
+class Selection(Resource):
+	def get(self):
+		global tracks
+		args = parser.parse_args()
+		validate_select_args(args)
+
+		if args.get('track'):
+			for track in tracks:
+				if (track.artist == args.get('artist') and
+					track.album == args.get('album') and
+					track.title == args.get('track')):
+					break
+			else:
+				abort(404, message="Track does not exist.")
+			found = dict(vars(track))
+			found.pop('filepath')
+			found['streampath'] = url_for(
+				'stream',
+				artist=parse.quote(track.artist, safe=''),
+				album=parse.quote(track.album, safe=''),
+				track=parse.quote(track.title, safe=''))
+			found['coverart'] = url_for(
+				'coverart',
+				artist=parse.quote(track.artist, safe=''),
+				album=parse.quote(track.album, safe=''),
+				track=parse.quote(track.title, safe=''))
+			return found
+
+		elif args.get('album'):
+			found = []
+			for track in tracks:
+				if (track.artist == args.get('artist') and
+					track.album == args.get('album')):
+					found.append(track)
+			if not found:
+				abort(404, message="Album does not exist.")
+			found = [t.tracknumber + " - " + t.title for t in found]
+			found.sort()
+			return found
+
+		elif args.get('artist'):
+			found = []
+			for track in tracks:
+				if track.artist == args.get('artist'):
+					found.append(track)
+			if not found:
+				abort(404, message="Artist does not exist.")
+			found = list(set(t.album for t in found))
+			return sorted(found)
+
+
+class RandomSelection(Resource):
+	def get(self):
+		global tracks
+		track = random.choice(tracks)
+		found = dict(vars(track))
+		found.pop('filepath')
+		found['streampath'] = url_for(
+			'stream',
+			artist=track.artist,
+			album=track.album,
+			track=track.title)
+		found['coverart'] = url_for(
+			'coverart',
+			artist=track.artist,
+			album=track.album,
+			track=track.title)
+		return found
+
+api.add_resource(Selection, '/select')
+api.add_resource(RandomSelection, '/select/random')
+api.init_app(app)
+
+
+@app.route('/stream/<artist>/<album>/<track>')
+def stream(artist, album, track):
 	"""View for the raw audio file."""
-	track = parse.unquote(track)
-	path = os.path.join(MUSIC_DIR, track)
-	path = os.path.abspath(path)
-	if not path.startswith(MUSIC_DIR):
-		return "False"
-	if not os.path.isfile(path):
-		return "False"
-	FFMPEG_CMD[5] = path
+	artist, album, track = map(parse.unquote, (artist, album, track))
+	for t in tracks:
+		if (t.artist == artist and
+			t.album == album and
+			t.title == track):
+			break
+	else:
+		abort(404, message="Track does not exist.")
+
+	FFMPEG_CMD[5] = t.filepath
 
 	def generate():
 		with subprocess.Popen(FFMPEG_CMD, stdout=subprocess.PIPE) as proc:
@@ -75,63 +206,22 @@ def stream(track):
 	return Response(generate(), mimetype="audio/ogg")
 
 
-@app.route('/get_dir/<path:directory>/')
-@app.route('/get_dir/')
-def get_dir(directory=""):
-	"""Returns the contents of the requested directory."""
-	directory = directory.replace("DOTDOT", "..")
-	directory = os.path.join(MUSIC_DIR, directory)
-	directory = os.path.abspath(directory)
-	if not directory.startswith(MUSIC_DIR):
-		return "False"
-	if not os.path.isdir(directory):
-		return "False"
-
-	nav_items = os.listdir(directory)
-	nav_items.sort()
-	if directory != MUSIC_DIR:
-		nav_items = [".."] + nav_items
-
-	nav_items_new = []
-	for item in nav_items:
-		if os.path.isdir(os.path.join(directory, item)):
-			item += '/'
-		nav_items_new.append(item)
-	nav_items_new = [directory.replace(MUSIC_DIR, '') + '/'] + nav_items_new
-
-	return json.dumps(nav_items_new)
-
-
-@app.route('/get_shuffle')
-def shuffle():
-	"""Returns a randomly selected track from the library."""
-	item = random.choice(os.listdir(MUSIC_DIR))
-	path = os.path.join(MUSIC_DIR, item)
-	n = 0
-	while not item.rpartition('.')[2] in MUSIC_EXT:
-		n += 1
-		item = random.choice(os.listdir(path))
-		if os.path.isdir(os.path.join(path, item)):
-			path = os.path.join(path, item)
-		if n == 5:
-			item = random.choice(os.listdir(MUSIC_DIR))
-			path = os.path.join(MUSIC_DIR, item)
-			n = 0
-	path = os.path.join(path, item)
-	return path.replace(MUSIC_DIR, '')
-
-
-@app.route('/album_cover/<path:cover>')
-def album_cover(cover):
+@app.route('/coverart/<artist>/<album>/<track>')
+def coverart(artist, album, track):
 	"""View for the raw audio file."""
-	path = os.path.join(MUSIC_DIR, cover)
-	path = os.path.abspath(path)
-	if not path.startswith(MUSIC_DIR):
-		return "False"
-	if not os.path.isfile(path):
-		return "False"
+	artist, album, track = map(parse.unquote, (artist, album, track))
+	for t in tracks:
+		if (t.artist == artist and
+			t.album == album and
+			t.title == track):
+			break
+	else:
+		abort(404, message="Track does not exist.")
 	
-	return send_file(path)
+	if t.coverart:
+		return send_file(t.coverart)
+	else:
+		return "False"
 
 
 if __name__ == "__main__":

static/musik.css

@@ -21,16 +21,16 @@ img {
 #navigationContainer {
 	flex: auto;
 	overflow: auto;
+	display: flex;
+	flex-direction: row;
+	height: 100%;
 	border-top: 2px solid #ccc;
 	border-bottom: 2px solid #ccc;
 }
 
-#currentDirectory {
-	padding-left: 0.5em;
-}
-
-#navItems {
-	list-style-type: none;
+.list {
+	height: 100%;
+	width: 100%;
 }
 
 #playerContainer {

static/musik.js

@@ -5,74 +5,77 @@ function load() {
 		if (document.getElementById('shuffle').checked) {
 			var httpRequest;
 			httpRequest = new XMLHttpRequest();
-			httpRequest.onreadystatechange = function() {
+			httpRequest.onreadystatechange = function () {
 				if (httpRequest.readyState !== XMLHttpRequest.DONE) { return; }
 				if (httpRequest.status !== 200) { return; }
-				let track = httpRequest.responseText;
+
+				let track = JSON.parse(httpRequest.responseText);
 				change_track(track);
 			};
-			httpRequest.open('GET', '/musik/get_shuffle', true);
+			httpRequest.open('GET', api_uri + '/random', true);
 			httpRequest.send();
 		}
 	});
 }
 
-
-function navigate(item) {
-	if (/\..{3,5}$/.test(item)) {
-		select_track(item);
-	}
-	else {
-		get_dir(item);
-	}
-}
-
-function select_track(item) {
-	let cd = document.getElementById('currentDirectory').innerText;
-	let track = cd + item;
-	change_track(track);
-}
-
-function change_track(track) {
-	let source = document.getElementById('stream');
-	source.src = '/musik/stream' + encodeURIComponent(track);
-	let player = document.getElementById('player');
-	player.load();
-	player.play();
-	document.getElementById('nowPlaying').innerHTML = track;
-
-	let arr = track.split('/');
-	let art = document.getElementById('albumCover');
-	art.firstChild.src = '/musik/album_cover' + arr.slice(0, arr.length-1).join("/") + '/folder.jpg';
-}
-
-function get_dir(item) {
-	let cd = document.getElementById('currentDirectory').innerText;
-	if (item === "../") {
-		if (cd !== "/") {
-			item = cd.slice(0, cd.slice(0, -1).lastIndexOf("/")+1);
-		}
-		else {
-			return;
-		}
-	}
-	else {
-		item = cd + item;
-	}
-
+function select_artist(select) {
 	var httpRequest;
 	httpRequest = new XMLHttpRequest();
 	httpRequest.onreadystatechange = function () {
 		if (httpRequest.readyState !== XMLHttpRequest.DONE) { return; }
 		if (httpRequest.status !== 200) { return; }
 		nav_items = JSON.parse(httpRequest.responseText);
-		document.getElementById('currentDirectory').innerText = nav_items.shift();
 		let html_str = '';
 		for (let i = 0; i < nav_items.length; i++) {
-			html_str += `<li><a href="javascript:void(0);" onclick="navigate(\`${nav_items[i]}\`)">${nav_items[i]}</a></li>`;
+			html_str += '<option value="' + nav_items[i] + '">' + nav_items[i] + '</option>';
 		}
-		document.getElementById('navItems').innerHTML = html_str;
+		document.getElementById('albumList').innerHTML = html_str;
+		document.getElementById('trackList').innerHTML = '';
 	};
-	httpRequest.open('GET', '/musik/get_dir/' + item, true);
+	httpRequest.open('GET', api_uri + '?artist=' + select.value, true);
 	httpRequest.send();
 }
+
+function select_album(select) {
+	var httpRequest;
+	httpRequest = new XMLHttpRequest();
+	httpRequest.onreadystatechange = function () {
+		if (httpRequest.readyState !== XMLHttpRequest.DONE) { return; }
+		if (httpRequest.status !== 200) { return; }
+		nav_items = JSON.parse(httpRequest.responseText);
+		let html_str = '';
+		for (let i = 0; i < nav_items.length; i++) {
+			html_str += '<option value="' + nav_items[i].replace(/\d* - /, '') + '">' + nav_items[i] + '</option>';
+		}
+		document.getElementById('trackList').innerHTML = html_str;
+	};
+	httpRequest.open('GET', api_uri + '?artist=' + document.getElementById('artistList').value + '&album=' + select.value, true);
+	httpRequest.send();
+}
+
+function select_track(select) {
+	var httpRequest;
+	httpRequest = new XMLHttpRequest();
+	httpRequest.onreadystatechange = function () {
+		if (httpRequest.readyState !== XMLHttpRequest.DONE) { return; }
+		if (httpRequest.status !== 200) { return; }
+
+		let track = JSON.parse(httpRequest.responseText);
+		change_track(track);
+	};
+	httpRequest.open('GET', api_uri + '?artist=' + document.getElementById('artistList').value + '&album=' + document.getElementById('albumList').value + '&track=' + select.value, true);
+	httpRequest.send();
+}
+
+function change_track(track) {
+	let source = document.getElementById('stream');
+	source.src = track.streampath;
+	let player = document.getElementById('player');
+	player.load();
+	player.play();
+	document.getElementById('nowPlayingArtist').innerHTML = track.artist;
+	document.getElementById('nowPlayingAlbum').innerHTML = track.album;
+	document.getElementById('nowPlayingTitle').innerHTML = track.title;
+
+	document.getElementById('albumCover').firstChild.src = track.coverart;
+}

templates/index.html

@@ -4,23 +4,36 @@
 	<title>Musik</title>
 	<link rel="stylesheet" type="text/css" href="/static/musik.css">
 	<script type="text/javascript" src="/static/musik.js"></script>
+	<script>
+		const api_uri = "{{ url_for('selection') }}";
+	</script>
 	<script>window.onload = load;</script>
 </head>
 <body>
 	<div id="globalContainer">
 		<div id="titleContainer"><h1>Musik</h1></div>
 		<div id="navigationContainer">
-			<div id="currentDirectory">{{ cd }}</div>
-			<ul id="navItems">
-				{% for item in nav_items %}
-				<li><a href="javascript:void(0);" onclick="navigate('{{ item }}')">{{ item }}</a></li>
-				{% endfor %}
-			</ul>
+			<div id="artistListContainer" class="list">
+				<select id="artistList" size="2" class="list" onchange="select_artist(this)">
+					{% for artist in artists %}
+					<option value="{{ artist }}">{{ artist }}</option>
+					{% endfor %}
+				</select>
+			</div>
+			<div id="albumListContainer" class="list">
+				<select id="albumList" size="2" class="list" onchange="select_album(this)">
+				</select>
+			</div>
+			<div id="trackListContainer" class="list">
+				<select id="trackList" size="2" class="list" onchange="select_track(this)">
+				</select>
+			</div>
 		</div>
 		<div id="playerContainer">
 			<span id="albumCover"><img src=""/></span>
 			<span id="playerControls">
-				<h3 id="nowPlaying"></h3>
+				<h4><span id="nowPlayingArtist"></span> - <span id="nowPlayingAlbum"></span></h4>
+				<h3 id="nowPlayingTitle"></h3>
 				<audio id="player" controls>
 					<source id="stream" src="" type="audio/ogg">
 				</audio>