85 lines
2.0 KiB
Python
Executable File
85 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Music streaming.
|
|
"""
|
|
import os
|
|
import json
|
|
import subprocess
|
|
from urllib import parse
|
|
|
|
from flask import Flask, Response, render_template
|
|
from werkzeug.utils import secure_filename
|
|
|
|
MUSIC_DIR = "/mnt/music/Music"
|
|
FFMPEG_CMD = [
|
|
'ffmpeg', '-y',
|
|
'-loglevel', 'panic',
|
|
'-i', '',
|
|
'-codec:a', 'libopus',
|
|
'-b:a', '64k',
|
|
'-f', 'opus',
|
|
'-'
|
|
]
|
|
|
|
app = Flask(__name__)
|
|
|
|
@app.route('/')
|
|
def index():
|
|
"""Main index page."""
|
|
nav_items = os.listdir(MUSIC_DIR)
|
|
nav_items.sort()
|
|
nav_items = [item + '/' for item in nav_items]
|
|
initial_track = "/Accept/Accept - Balls to the Wall (1983) [FLAC] {74321 93214 2}/01 - Balls to the Wall.flac"
|
|
cd = "/"
|
|
return render_template('index.html', **locals())
|
|
|
|
|
|
@app.route('/stream/<path:track>')
|
|
def stream(track):
|
|
"""View for the raw audio file."""
|
|
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
|
|
|
|
def generate():
|
|
with subprocess.Popen(FFMPEG_CMD, stdout=subprocess.PIPE) as proc:
|
|
data = proc.stdout.read(1024)
|
|
while data:
|
|
yield data
|
|
data = proc.stdout.read(1024)
|
|
return Response(generate(), mimetype="audio/ogg")
|
|
|
|
|
|
@app.route('/get_dir/<path:directory>/')
|
|
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)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host='0.0.0.0', port=5150)
|