#!/usr/bin/python3
-from flask import Flask, render_template, request, redirect, send_file
+from functools import wraps
+from flask import Flask, render_template, request, redirect, send_file, Response, jsonify, abort
import variables as var
import util
+import os
import os.path
-from os import listdir
-import random
+import shutil
from werkzeug.utils import secure_filename
import errno
import media
+from media.cache import get_cached_wrapper_from_scrap, get_cached_wrapper_by_id, get_cached_wrappers_by_tags
+import logging
+import time
+
class ReverseProxied(object):
- '''Wrap the application in this middleware and configure the
+ """Wrap the application in this middleware and configure the
front-end server to add these headers, to let you quietly bind
this to a URL other than / and to an HTTP scheme that is
different than what is used locally.
}
:param app: the WSGI application
- '''
+ """
def __init__(self, app):
self.app = app
web = Flask(__name__)
+log = logging.getLogger("bot")
+user = 'Remote Control'
def init_proxy():
if var.is_proxified:
web.wsgi_app = ReverseProxied(web.wsgi_app)
+# https://stackoverflow.com/questions/29725217/password-protect-one-webpage-in-flask-app
+
+
+def check_auth(username, password):
+ """This function is called to check if a username /
+ password combination is valid.
+ """
+ return username == var.config.get("webinterface", "user") and password == var.config.get("webinterface", "password")
+
+
+def authenticate():
+ """Sends a 401 response that enables basic auth"""
+ global log
+ return Response('Could not verify your access level for that URL.\n'
+ 'You have to login with proper credentials', 401,
+ {'WWW-Authenticate': 'Basic realm="Login Required"'})
+
+
+def requires_auth(f):
+ @wraps(f)
+ def decorated(*args, **kwargs):
+ global log
+ auth = request.authorization
+ if var.config.getboolean("webinterface", "require_auth") and (not auth or not check_auth(auth.username, auth.password)):
+ if auth:
+ log.info("web: Failed login attempt, user: %s" % auth.username)
+ return authenticate()
+ return f(*args, **kwargs)
+ return decorated
+
+
+def tag_color(tag):
+ num = hash(tag) % 8
+ if num == 0:
+ return "primary"
+ elif num == 1:
+ return "secondary"
+ elif num == 2:
+ return "success"
+ elif num == 3:
+ return "danger"
+ elif num == 4:
+ return "warning"
+ elif num == 5:
+ return "info"
+ elif num == 6:
+ return "light"
+ elif num == 7:
+ return "dark"
+
-@web.route("/", methods=['GET', 'POST'])
+def build_tags_color_lookup():
+ color_lookup = {}
+ for tag in var.music_db.query_all_tags():
+ color_lookup[tag] = tag_color(tag)
+
+ return color_lookup
+
+
+def build_path_tags_lookup():
+ path_tags_lookup = {}
+ ids = list(var.cache.file_id_lookup.values())
+ if len(ids) > 0:
+ id_tags_lookup = var.music_db.query_tags_by_ids(ids)
+
+ for path, id in var.cache.file_id_lookup.items():
+ path_tags_lookup[path] = id_tags_lookup[id]
+
+ return path_tags_lookup
+
+
+def recur_dir(dirobj):
+ for name, dir in dirobj.get_subdirs().items():
+ print(dirobj.fullpath + "/" + name)
+ recur_dir(dir)
+
+
+@web.route("/", methods=['GET'])
+@requires_auth
def index():
- folder_path = var.music_folder
- files = util.get_recursive_filelist_sorted(var.music_folder)
- music_library = util.Dir(folder_path)
- for file in files:
- music_library.add_file(file)
+ while var.cache.dir_lock.locked():
+ time.sleep(0.1)
+
+ tags_color_lookup = build_tags_color_lookup()
+ path_tags_lookup = build_path_tags_lookup()
+
+ return render_template('index.html',
+ all_files=var.cache.files,
+ tags_lookup=path_tags_lookup,
+ tags_color_lookup=tags_color_lookup,
+ music_library=var.cache.dir,
+ os=os,
+ playlist=var.playlist,
+ user=var.user,
+ paused=var.bot.is_pause,
+ )
+
+
+@web.route("/playlist", methods=['GET'])
+@requires_auth
+def playlist():
+ if len(var.playlist) == 0:
+ return jsonify({'items': [render_template('playlist.html',
+ m=False,
+ index=-1
+ )]
+ })
+
+ tags_color_lookup = build_tags_color_lookup()
+ items = []
+
+ for index, item_wrapper in enumerate(var.playlist):
+ items.append(render_template('playlist.html',
+ index=index,
+ tags_color_lookup=tags_color_lookup,
+ m=item_wrapper.item(),
+ playlist=var.playlist
+ )
+ )
+
+ return jsonify({'items': items})
+
+
+def status():
+ if len(var.playlist) > 0:
+ return jsonify({'ver': var.playlist.version,
+ 'empty': False,
+ 'play': not var.bot.is_pause,
+ 'playhead': var.bot.playhead,
+ 'mode': var.playlist.mode})
+ else:
+ return jsonify({'ver': var.playlist.version,
+ 'empty': True,
+ 'play': False,
+ 'playhead': -1,
+ 'mode': var.playlist.mode})
+
+
+@web.route("/post", methods=['POST'])
+@requires_auth
+def post():
+ global log
if request.method == 'POST':
- print(request.form)
- if 'add_file' in request.form and ".." not in request.form['add_file']:
- item = ('file', request.form['add_file'])
- var.playlist.append(item)
- if ('add_folder' in request.form and ".." not in request.form['add_folder']) or ('add_folder_recursively' in request.form and ".." not in request.form['add_folder_recursively']) :
+ if request.form:
+ log.debug("web: Post request from %s: %s" % (request.remote_addr, str(request.form)))
+ if 'add_file_bottom' in request.form and ".." not in request.form['add_file_bottom']:
+ path = var.music_folder + request.form['add_file_bottom']
+ if os.path.isfile(path):
+ music_wrapper = get_cached_wrapper_by_id(var.bot, var.cache.file_id_lookup[request.form['add_file_bottom']], user)
+
+ var.playlist.append(music_wrapper)
+ log.info('web: add to playlist(bottom): ' + music_wrapper.format_debug_string())
+
+ elif 'add_file_next' in request.form and ".." not in request.form['add_file_next']:
+ path = var.music_folder + request.form['add_file_next']
+ if os.path.isfile(path):
+ music_wrapper = get_cached_wrapper_by_id(var.bot, var.cache.file_id_lookup[request.form['add_file_next']], user)
+ var.playlist.insert(var.playlist.current_index + 1, music_wrapper)
+ log.info('web: add to playlist(next): ' + music_wrapper.format_debug_string())
+
+ elif ('add_folder' in request.form and ".." not in request.form['add_folder']) or ('add_folder_recursively' in request.form and ".." not in request.form['add_folder_recursively']):
try:
folder = request.form['add_folder']
except:
if not folder.endswith('/'):
folder += '/'
- print('folder:', folder)
- if 'add_folder_recursively' in request.form:
- files = music_library.get_files_recursively(folder)
- else:
- files = music_library.get_files(folder)
- files = list(map(lambda file: ('file', folder + '/' + file), files))
- print('Adding to playlist: ', files)
- var.playlist.extend(files)
+ if os.path.isdir(var.music_folder + folder):
+ dir = var.cache.dir
+ if 'add_folder_recursively' in request.form:
+ files = dir.get_files_recursively(folder)
+ else:
+ files = dir.get_files(folder)
+
+ music_wrappers = list(map(
+ lambda file:
+ get_cached_wrapper_by_id(var.bot, var.cache.file_id_lookup[folder + file], user), files))
+
+ var.playlist.extend(music_wrappers)
+
+ for music_wrapper in music_wrappers:
+ log.info('web: add to playlist: ' + music_wrapper.format_debug_string())
+
+ elif 'add_url' in request.form:
+ music_wrapper = get_cached_wrapper_from_scrap(var.bot, type='url', url=request.form['add_url'], user=user)
+ var.playlist.append(music_wrapper)
+
+ log.info("web: add to playlist: " + music_wrapper.format_debug_string())
+ if len(var.playlist) == 2:
+ # If I am the second item on the playlist. (I am the next one!)
+ var.bot.async_download_next()
+
+ elif 'add_radio' in request.form:
+ url = request.form['add_radio']
+ music_wrapper = get_cached_wrapper_from_scrap(var.bot, type='radio', url=url, user=user)
+ var.playlist.append(music_wrapper)
+
+ log.info("cmd: add to playlist: " + music_wrapper.format_debug_string())
+
elif 'delete_music' in request.form:
- try:
- var.playlist.remove("file", request.form['delete_music'])
- except ValueError:
- pass
+ music_wrapper = var.playlist[int(request.form['delete_music'])]
+ log.info("web: delete from playlist: " + music_wrapper.format_debug_string())
+
+ if len(var.playlist) >= int(request.form['delete_music']):
+ index = int(request.form['delete_music'])
+
+ if index == var.playlist.current_index:
+ var.playlist.remove(index)
+
+ if index < len(var.playlist):
+ if not var.bot.is_pause:
+ var.bot.interrupt()
+ var.playlist.current_index -= 1
+ # then the bot will move to next item
+
+ else: # if item deleted is the last item of the queue
+ var.playlist.current_index -= 1
+ if not var.bot.is_pause:
+ var.bot.interrupt()
+ else:
+ var.playlist.remove(index)
+
+ elif 'play_music' in request.form:
+ music_wrapper = var.playlist[int(request.form['play_music'])]
+ log.info("web: jump to: " + music_wrapper.format_debug_string())
+
+ if len(var.playlist) >= int(request.form['play_music']):
+ var.playlist.point_to(int(request.form['play_music']) - 1)
+ if not var.bot.is_pause:
+ var.bot.interrupt()
+ else:
+ var.bot.is_pause = False
+ time.sleep(0.1)
+
+ elif 'delete_music_file' in request.form and ".." not in request.form['delete_music_file']:
+ path = var.music_folder + request.form['delete_music_file']
+ if os.path.isfile(path):
+ log.info("web: delete file " + path)
+ os.remove(path)
+
+ elif 'delete_folder' in request.form and ".." not in request.form['delete_folder']:
+ path = var.music_folder + request.form['delete_folder']
+ if os.path.isdir(path):
+ log.info("web: delete folder " + path)
+ shutil.rmtree(path)
+ time.sleep(0.1)
+
+ elif 'add_tag' in request.form:
+ music_wrappers = get_cached_wrappers_by_tags(var.bot, [request.form['add_tag']], user)
+ for music_wrapper in music_wrappers:
+ log.info("cmd: add to playlist: " + music_wrapper.format_debug_string())
+ var.playlist.extend(music_wrappers)
+
elif 'action' in request.form:
action = request.form['action']
if action == "randomize":
- random.shuffle(var.playlist)
- if var.current_music:
- source = var.current_music[0]
- # format for current_music below:
- # (sourcetype, title, url or None)
- if source == "radio":
- current_music = (
- "[radio]",
- media.get_radio_title(var.current_music[1]),
- var.current_music[2]
- )
- elif source == "url":
- current_music = (
- "[url]",
- var.current_music[2],
- var.current_music[1]
- )
- elif source == "file":
- current_music = (
- "[file]",
- var.current_music[2],
- None
- )
- else:
- current_music = (
- "(??)[" + var.current_music[0] + "]",
- var.current_music[1],
- var.current_music[2],
- )
- else:
- current_music = None
+ if var.playlist.mode != "random":
+ var.playlist = media.playlist.get_playlist("random", var.playlist)
+ else:
+ var.playlist.randomize()
+ var.bot.interrupt()
+ var.db.set('playlist', 'playback_mode', "random")
+ log.info("web: playback mode changed to random.")
+ if action == "one-shot":
+ var.playlist = media.playlist.get_playlist("one-shot", var.playlist)
+ var.db.set('playlist', 'playback_mode', "one-shot")
+ log.info("web: playback mode changed to one-shot.")
+ if action == "repeat":
+ var.playlist = media.playlist.get_playlist("repeat", var.playlist)
+ var.db.set('playlist', 'playback_mode', "repeat")
+ log.info("web: playback mode changed to repeat.")
+ if action == "autoplay":
+ var.playlist = media.playlist.get_playlist("autoplay", var.playlist)
+ var.db.set('playlist', 'playback_mode', "autoplay")
+ log.info("web: playback mode changed to autoplay.")
+ if action == "rescan":
+ var.cache.build_dir_cache(var.bot)
+ log.info("web: Local file cache refreshed.")
+ elif action == "stop":
+ var.bot.stop()
+ elif action == "pause":
+ var.bot.pause()
+ elif action == "resume":
+ var.bot.resume()
+ elif action == "clear":
+ var.bot.clear()
+ elif action == "volume_up":
+ if var.bot.volume_set + 0.03 < 1.0:
+ var.bot.volume_set = var.bot.volume_set + 0.03
+ else:
+ var.bot.volume_set = 1.0
+ var.db.set('bot', 'volume', str(var.bot.volume_set))
+ log.info("web: volume up to %d" % (var.bot.volume_set * 100))
+ elif action == "volume_down":
+ if var.bot.volume_set - 0.03 > 0:
+ var.bot.volume_set = var.bot.volume_set - 0.03
+ else:
+ var.bot.volume_set = 0
+ var.db.set('bot', 'volume', str(var.bot.volume_set))
+ log.info("web: volume up to %d" % (var.bot.volume_set * 100))
- return render_template('index.html',
- current_music=current_music,
- user=var.user,
- playlist=var.playlist,
- all_files=files,
- music_library=music_library)
+ return status()
@web.route('/upload', methods=["POST"])
def upload():
- file = request.files['file']
- if not file:
- return redirect("./", code=406)
+ global log
- filename = secure_filename(file.filename).strip()
- if filename == '':
+ files = request.files.getlist("file[]")
+ if not files:
return redirect("./", code=406)
- targetdir = request.form['targetdir'].strip()
- if targetdir == '':
- targetdir = 'uploads/'
- elif '../' in targetdir:
- return redirect("./", code=406)
-
- #print('Uploading file:')
- #print('filename:', filename)
- #print('targetdir:', targetdir)
- #print('mimetype:', file.mimetype)
+ # filename = secure_filename(file.filename).strip()
+ for file in files:
+ filename = file.filename
+ if filename == '':
+ return redirect("./", code=406)
- if "audio" in file.mimetype:
- storagepath = os.path.abspath(os.path.join(var.music_folder, targetdir))
- if not storagepath.startswith(var.music_folder):
+ targetdir = request.form['targetdir'].strip()
+ if targetdir == '':
+ targetdir = 'uploads/'
+ elif '../' in targetdir:
return redirect("./", code=406)
- try:
- os.makedirs(storagepath)
- except OSError as ee:
- if ee.errno != errno.EEXIST:
- return redirect("./", code=500)
+ log.info('web: Uploading file from %s:' % request.remote_addr)
+ log.info('web: - filename: ' + filename)
+ log.info('web: - targetdir: ' + targetdir)
+ log.info('web: - mimetype: ' + file.mimetype)
- filepath = os.path.join(storagepath, filename)
- if os.path.exists(filepath):
- return redirect("./", code=406)
+ if "audio" in file.mimetype or os.path.splitext(filename)[-1] in ('.ogg', '.opus', '.mp3', '.flac', '.wav'):
+ storagepath = os.path.abspath(os.path.join(var.music_folder, targetdir))
+ print('storagepath:', storagepath)
+ if not storagepath.startswith(os.path.abspath(var.music_folder)):
+ return redirect("./", code=406)
+
+ try:
+ os.makedirs(storagepath)
+ except OSError as ee:
+ if ee.errno != errno.EEXIST:
+ return redirect("./", code=500)
+
+ filepath = os.path.join(storagepath, filename)
+ log.info(' - filepath: ' + filepath)
+ if os.path.exists(filepath):
+ continue
+
+ file.save(filepath)
+ else:
+ continue
+
+ var.cache.build_dir_cache(var.bot)
+ log.info("web: Local file cache refreshed.")
+
+ return redirect("./", code=302)
- file.save(filepath)
- return redirect("./", code=302)
- else:
- return redirect("./", code=409)
@web.route('/download', methods=["GET"])
def download():
+ global log
+
if 'file' in request.args:
requested_file = request.args['file']
+ log.info('web: Download of file %s requested from %s:' % (requested_file, request.remote_addr))
if '../' not in requested_file:
folder_path = var.music_folder
- files = util.get_recursive_filelist_sorted(var.music_folder)
+ files = var.cache.files
if requested_file in files:
filepath = os.path.join(folder_path, requested_file)
try:
return send_file(filepath, as_attachment=True)
except Exception as e:
- self.log.exception(e)
- self.Error(400)
+ log.exception(e)
+ abort(404)
elif 'directory' in request.args:
requested_dir = request.args['directory']
folder_path = var.music_folder
try:
return send_file(zipfile, as_attachment=True)
except Exception as e:
- self.log.exception(e)
- self.Error(400)
+ log.exception(e)
+ abort(404)
return redirect("./", code=400)