refactor, fix uri encoding

README.md

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

musik.py

@@ -3,10 +3,12 @@
 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
@@ -23,6 +25,23 @@ 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
+
+
 app = Flask(__name__)
 
 @app.route('/')
@@ -40,6 +59,7 @@ def index():
 @app.route('/stream/<path:track>')
 def stream(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):
@@ -58,7 +78,8 @@ def stream(track):
 
 
 @app.route('/get_dir/<path:directory>/')
-def get_dir(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)

static/musik.css

@@ -1,5 +1,6 @@
 body {
 	margin: 0;
+	font-family: Helvetica, sans-serif;
 }
 
 img {
@@ -33,6 +34,7 @@ img {
 }
 
 #playerContainer {
+	height: 30vh;
 	flex: 0 0 auto;
 	display: flex;
 	flex-direction: row;
@@ -42,6 +44,8 @@ img {
 #albumCover {
 	padding-right: 1em;
 	padding-left: 1em;
+	padding-bottom: 3px;
+	box-sizing: border-box;
 }
 
 #playerControls {

static/musik.js

@@ -1,23 +1,26 @@
-var httpRequest
-
 function load() {
 	// track has ended
 	document.getElementById('player').addEventListener('ended', function() {
 		// shuffle
 		if (document.getElementById('shuffle').checked) {
-			httpRequest = new XMLHttpRequest();
-			httpRequest.onreadystatechange = function() {
+			ajax('/musik/get_shuffle', function() {
 				if (httpRequest.readyState !== XMLHttpRequest.DONE) { return; }
 				if (httpRequest.status !== 200) { return; }
 				let track = httpRequest.responseText;
 				change_track(track);
-			}
-			httpRequest.open('GET', '/musik/get_shuffle', true);
-			httpRequest.send();
+			});
 		}
 	});
 }
 
+var httpRequest;
+function ajax(url, func) {
+	httpRequest = new XMLHttpRequest();
+	httpRequest.onreadystatechange = func;
+	httpRequest.open('GET', url, true);
+	httpRequest.send();
+}
+
 function navigate(item) {
 	if (/\..{3,5}$/.test(item)) {
 		select_track(item);
@@ -35,7 +38,7 @@ function select_track(item) {
 
 function change_track(track) {
 	let source = document.getElementById('stream');
-	source.src = '/musik/stream' + track;
+	source.src = '/musik/stream' + encodeURIComponent(track);
 	let player = document.getElementById('player');
 	player.load();
 	player.play();
@@ -48,12 +51,18 @@ function change_track(track) {
 
 function get_dir(item) {
 	let cd = document.getElementById('currentDirectory').innerText;
-	item = cd + item;
-	item = item.replace(/\.\./g, 'DOTDOT');
-	httpRequest = new XMLHttpRequest();
-	httpRequest.onreadystatechange = update_nav_items;
-	httpRequest.open('GET', '/musik/get_dir' + item, true);
-	httpRequest.send();
+	if (item === "../") {
+		if (cd !== "/") {
+			item = cd.slice(0, cd.slice(0, -1).lastIndexOf("/")+1);
+		}
+		else {
+			return;
+		}
+	}
+	else {
+		item = cd + item;
+	}
+	ajax('/musik/get_dir/' + item, update_nav_items);
 }
 
 function update_nav_items() {

templates/index.html

@@ -18,8 +18,8 @@
 			</ul>
 		</div>
 		<div id="playerContainer">
-			<div id="albumCover"><img src="/musik/album_cover{{ initial_cover }}"/></div>
-			<div id="playerControls">
+			<span id="albumCover"><img src="/musik/album_cover{{ initial_cover }}"/></span>
+			<span id="playerControls">
 				<h3 id="nowPlaying">{{ initial_track }}</h3>
 				<audio id="player" controls>
 					<source id="stream" src="{{ url_for('stream', track=initial_track) }}" type="audio/ogg">
@@ -28,7 +28,7 @@
 					<input type="checkbox" name="shuffle" id="shuffle" checked="true">
 					<label for="shuffle">Shuffle</label>
 				</div>
-			</div>
+			</span>
 		</div>
 	</div>
 </body>