#!/usr/bin/python3

from flask import Flask, render_template, request, redirect
import variables as var
import os.path
from os import listdir
import random
from werkzeug.utils import secure_filename


class ReverseProxied(object):
    '''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.

    In nginx:
    location /myprefix {
        proxy_pass http://192.168.0.1:5001;
        proxy_set_header Host $host;
        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
        proxy_set_header X-Scheme $scheme;
        proxy_set_header X-Script-Name /myprefix;
        }

    :param app: the WSGI application
    '''

    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):
        script_name = environ.get('HTTP_X_SCRIPT_NAME', '')
        if script_name:
            environ['SCRIPT_NAME'] = script_name
            path_info = environ['PATH_INFO']
            if path_info.startswith(script_name):
                environ['PATH_INFO'] = path_info[len(script_name):]

        scheme = environ.get('HTTP_X_SCHEME', '')
        if scheme:
            environ['wsgi.url_scheme'] = scheme
        real_ip = environ.get('HTTP_X_REAL_IP', '')
        if real_ip:
            environ['REMOTE_ADDR'] = real_ip
        return self.app(environ, start_response)


web = Flask(__name__)


def init_proxy():
    global web
    if var.is_proxified:
        web.wsgi_app = ReverseProxied(web.wsgi_app)


@web.route("/", methods=['GET', 'POST'])
def index():
    folder_path = var.music_folder
    files = {}
    dirs = [f for f in listdir(folder_path) if os.path.isdir(os.path.join(folder_path, f))]
    for director in dirs:
        files[director] = [f for f in listdir(folder_path + director) if os.path.isfile(os.path.join(folder_path + director, f))]

    if request.method == 'POST':
        if 'add_music' in request.form and ".." not in request.form['add_music']:
            var.playlist.append(['file', request.form['add_music']])

        if 'add_url' in request.form :
            var.playlist.append(['url', request.form['add_url']])

        if 'add_radio' in request.form:
            var.playlist.append(['radio', request.form['add_radio']])

        if 'add_folder' in request.form and ".." not in request.form['add_folder']:
            dir_files = [["file", request.form['add_folder'] + '/' + i] for i in files[request.form['add_folder']]]
            var.playlist.extend(dir_files)
        elif 'delete_music' in request.form:
            try:
                var.playlist.remove(["file", request.form['delete_music']])
            except ValueError:
                pass
        elif 'action' in request.form:
            action = request.form['action']
            if action == "randomize":
                random.shuffle(var.playlist)

    return render_template('index.html',
                           current_music=var.current_music,
                           user=var.user,
                           playlist=var.playlist,
                           all_files=files)


@web.route('/download', methods=["POST"])
def download():
    print(request.form)

    file = request.files['music_file']
    if not file:
        return redirect("./", code=406)
    elif file.filename == '':
        return redirect("./", code=406)
    elif '..' in request.form['directory']:
        return redirect("./", code=406)

    if file.name == "music_file" and "audio" in file.headers.to_list()[1][1]:
        web.config['UPLOAD_FOLDER'] = var.music_folder + request.form['directory']
        filename = secure_filename(file.filename)
        print(filename)
        file.save(os.path.join(web.config['UPLOAD_FOLDER'], filename))
        return redirect("./", code=302)
    else:
        return redirect("./", code=409)


if __name__ == '__main__':
    web.run(port=8181, host="0.0.0.0")