#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import os.path
from os.path import join
from traceback import print_exc
import shutil
from shutil import copyfileobj

from bottle import route, request, HTTPError, validate

from webinterface import PYLOAD

from utils import login_required, render_to_response, toDict

from module.utils import decode, formatSize


def format_time(seconds):
    seconds = int(seconds)

    hours, seconds = divmod(seconds, 3600)
    minutes, seconds = divmod(seconds, 60)
    return "%.2i:%.2i:%.2i" % (hours, minutes, seconds)


def get_sort_key(item):
    return item["order"]


@route("/json/status")
@route("/json/status", method="POST")
@login_required('see_downloads')
def status():
    try:
        status = toDict(PYLOAD.statusServer())
        status['captcha'] = PYLOAD.isCaptchaWaiting()
        return status
    except:
        return HTTPError()


@route("/json/links")
@route("/json/links", method="POST")
@login_required('see_downloads')
def links():
    try:
        links = [toDict(x) for x in PYLOAD.statusDownloads()]
        ids = []
        for link in links:
            ids.append(link['fid'])

            if link['status'] == 12:
                link['info'] = "%s @ %s/s" % (link['format_eta'], formatSize(link['speed']))
            elif link['status'] == 5:
                link['percent'] = 0
                link['size'] = 0
                link['bleft'] = 0
                link['info'] = _("waiting %s") % link['format_wait']
            else:
                link['info'] = ""

        data = {'links': links, 'ids': ids}
        return data
    except Exception, e:
        print_exc()
        return HTTPError()


@route("/json/queue")
@login_required('see_downloads')
def queue():
    print "/json/queue"
    try:
        return PYLOAD.getQueue()

    except:
        return HTTPError()


@route("/json/pause")
@login_required('status')
def pause():
    try:
        return PYLOAD.pauseServer()

    except:
        return HTTPError()


@route("/json/unpause")
@login_required('status')
def unpause():
    try:
        return PYLOAD.unpauseServer()

    except:
        return HTTPError()


@route("/json/cancel")
@login_required('status')
def cancel():
    try:
        return PYLOAD.stopAllDownloads()
    except:
        return HTTPError()


@route("/json/packages")
@login_required('see_downloads')
def packages():
    print "/json/packages"
    try:
        data = PYLOAD.getQueue()

        for package in data:
            package['links'] = []
            for file in PYLOAD.get_package_files(package['id']):
                package['links'].append(PYLOAD.get_file_info(file))

        return data

    except:
        return HTTPError()


@route("/json/package/:id")
@validate(id=int)
@login_required('see_downloads')
def package(id):
    try:
        data = toDict(PYLOAD.getPackageData(id))
        data["links"] = [toDict(x) for x in data["links"]]

        for pyfile in data["links"]:
            if pyfile["status"] == 0:
                pyfile["icon"] = "status_finished.png"
            elif pyfile["status"] in (2, 3):
                pyfile["icon"] = "status_queue.png"
            elif pyfile["status"] in (9, 1):
                pyfile["icon"] = "status_offline.png"
            elif pyfile["status"] == 5:
                pyfile["icon"] = "status_waiting.png"
            elif pyfile["status"] == 8:
                pyfile["icon"] = "status_failed.png"
            elif pyfile["status"] == 4:
                pyfile["icon"] = "arrow_right.png"
            elif pyfile["status"] in (11, 13):
                pyfile["icon"] = "status_proc.png"
            else:
                pyfile["icon"] = "status_downloading.png"

        tmp = data["links"]
        tmp.sort(key=get_sort_key)
        data["links"] = tmp
        return data

    except:
        print_exc()
        return HTTPError()


@route("/json/package_order/:ids")
@login_required('add')
def package_order(ids):
    try:
        pid, pos = ids.split("|")
        PYLOAD.orderPackage(int(pid), int(pos))
        return {"response": "success"}
    except:
        return HTTPError()


@route("/json/link/:id")
@validate(id=int)
@login_required('see_downloads')
def link(id):
    print "/json/link/%d" % id
    try:
        data = toDict(PYLOAD.getFileData(id))
        return data
    except:
        return HTTPError()


@route("/json/remove_link/:id")
@validate(id=int)
@login_required('delete')
def remove_link(id):
    try:
        PYLOAD.deleteFiles([id])
        return {"response": "success"}
    except Exception, e:
        return HTTPError()


@route("/json/restart_link/:id")
@validate(id=int)
@login_required('add')
def restart_link(id):
    try:
        PYLOAD.restartFile(id)
        return {"response": "success"}
    except Exception:
        return HTTPError()


@route("/json/abort_link/:id")
@validate(id=int)
@login_required('delete')
def abort_link(id):
    try:
        PYLOAD.stopDownloads([id])
        return {"response": "success"}
    except:
        return HTTPError()


@route("/json/link_order/:ids")
@login_required('add')
def link_order(ids):
    try:
        pid, pos = ids.split("|")
        PYLOAD.orderFile(int(pid), int(pos))
        return {"response": "success"}
    except:
        return HTTPError()


@route("/json/add_package")
@route("/json/add_package", method="POST")
@login_required('add')
def add_package():
    name = request.forms.get("add_name", "New Package").strip()
    queue = int(request.forms['add_dest'])
    links = decode(request.forms['add_links'])
    links = links.split("\n")
    pw = request.forms.get("add_password", "").strip("\n\r")

    try:
        f = request.files['add_file']

        if not name or name == "New Package":
            name = f.name

        fpath = join(PYLOAD.getConfigValue("general", "download_folder"), "tmp_" + f.filename)
        destination = open(fpath, 'wb')
        copyfileobj(f.file, destination)
        destination.close()
        links.insert(0, fpath)
    except:
        pass

    name = name.decode("utf8", "ignore")

    links = map(lambda x: x.strip(), links)
    links = filter(lambda x: x != "", links)

    pack = PYLOAD.addPackage(name, links, queue)
    if pw:
        pw = pw.decode("utf8", "ignore")
        data = {"password": pw}
        PYLOAD.setPackageData(pack, data)


@route("/json/remove_package/:id")
@validate(id=int)
@login_required('delete')
def remove_package(id):
    try:
        PYLOAD.deletePackages([id])
        return {"response": "success"}
    except Exception, e:
        return HTTPError()


@route("/json/restart_package/:id")
@validate(id=int)
@login_required('add')
def restart_package(id):
    try:
        PYLOAD.restartPackage(id)
        return {"response": "success"}
    except Exception:
        print_exc()
        return HTTPError()


@route("/json/move_package/:dest/:id")
@validate(dest=int, id=int)
@login_required('add')
def move_package(dest, id):
    try:
        PYLOAD.movePackage(dest, id)
        return {"response": "success"}
    except:
        return HTTPError()


@route("/json/edit_package", method="POST")
@login_required('add')
def edit_package():
    try:
        id = int(request.forms.get("pack_id"))
        data = {"name": request.forms.get("pack_name").decode("utf8", "ignore"),
                "folder": request.forms.get("pack_folder").decode("utf8", "ignore"),
                "priority": request.forms.get("pack_prio"),
                "password": request.forms.get("pack_pws").decode("utf8", "ignore")}

        PYLOAD.setPackageData(id, data)
        return {"response": "success"}

    except:
        return HTTPError()


@route("/json/set_captcha")
@route("/json/set_captcha", method="POST")
@login_required('add')
def set_captcha():
    if request.environ.get('REQUEST_METHOD', "GET") == "POST":
        try:
            PYLOAD.setCaptchaResult(request.forms["cap_id"], request.forms["cap_result"])
        except:
            pass

    task = PYLOAD.getCaptchaTask()

    if task.tid >= 0:
        src = "data:image/%s;base64,%s" % (task.type, task.data)

        return {'captcha': True, 'id': task.tid, 'src': src, 'result_type' : task.resultType}
    else:
        return {'captcha': False}


@route("/json/delete_finished")
@login_required('delete')
def delete_finished():
    return {"del": PYLOAD.deleteFinished()}


@route("/json/restart_failed")
@login_required('delete')
def restart_failed():
    restart = PYLOAD.restartFailed()

    if restart: return restart
    return {"response": "success"}


@route("/json/load_config/:category/:section")
@login_required("settings")
def load_config(category, section):
    conf = None
    if category == "general":
        conf = PYLOAD.getConfigDict()
    elif category == "plugin":
        conf = PYLOAD.getPluginConfigDict()

    for key, option in conf[section].iteritems():
        if key in ("desc","outline"): continue

        if ";" in option["type"]:
            option["list"] = option["type"].split(";")

        option["value"] = decode(option["value"])

    return render_to_response("settings_item.html", {"skey": section, "section": conf[section]})


@route("/json/save_config/:category", method="POST")
@login_required("settings")
def save_config(category):
    for key, value in request.POST.iteritems():
        try:
            section, option = key.split("|")
        except:
            continue

        if category == "general": category = "core"

        PYLOAD.setConfigValue(section, option, decode(value), category)


@route("/json/add_account", method="POST")
@login_required("settings")
def add_account():
    login = request.POST["account_login"]
    password = request.POST["account_password"]
    type = request.POST["account_type"]

    PYLOAD.updateAccount(type, login, password)


@route("/json/update_accounts", method="POST")
@login_required("settings")
def update_accounts():
    deleted = [] #dont update deleted accs or they will be created again

    for name, value in request.POST.iteritems():
        value = value.strip()
        if not value: continue
        
        tmp, user = name.split(";")
        plugin, action = tmp.split("|")

        if (plugin, user) in deleted: continue

        if action == "password":
            PYLOAD.updateAccount(plugin, user, value)
        elif action == "time" and "-" in value:
            PYLOAD.updateAccount(plugin, user, options={"time": [value]})
        elif action == "limitdl" and value.isdigit():
            PYLOAD.updateAccount(plugin, user, options={"limitDL": [value]})
        elif action == "delete":
            deleted.append((plugin,user))
            PYLOAD.removeAccount(plugin, user)

@route("/json/change_password", method="POST")
def change_password():

    user = request.POST["user_login"]
    oldpw = request.POST["login_current_password"]
    newpw = request.POST["login_new_password"]

    if not PYLOAD.changePassword(user, oldpw, newpw):
        print "Wrong password"
        return HTTPError()

#@route("/json/filemanager/rename", method="POST")
#@login_required('filemanager')
def rename_dir():
    try:
        path = decode(request.forms.get("path"))
        old_name = path + "/" + decode(request.forms.get("old_name"))
        new_name = path + "/" + decode(request.forms.get("new_name"))

        try:
            #check if file exists
            os.rename(old_name, new_name)
        except Exception, e:
            return {"response": "fail", "error": str(e) + "\n" + old_name + " => " + new_name}

        return {"response": "success"}

    except:
        return HTTPError()


#@route("/json/filemanager/delete", method="POST")
#@login_required('filemanager')
def delete_dir():
    try:
        try:
            path = decode(request.forms.get("path"))
            name = decode(request.forms.get("name"))
            shutil.rmtree(path + "/" + name)
        except Exception, e:
            return {"response": "fail", "error": str(e) + "\n" + path + "/" + name}

        return {"response": "success"}

    except:
        return HTTPError()


#@route("/json/filemanager/mkdir", method="POST")
#@login_required('filemanager')
def make_dir():
    try:
        path = decode(request.forms.get("path"))
        name = decode(request.forms.get("name"))
        try:
            #i = 1
            #full_name = path + "/" + name
            #while os.path.exists(full_name)
            #    full_name = full_name + i
            #    i = i + 1
            #
            #os.mkdir(full_name)

            os.mkdir(path + "/" + name)
        except Exception, e:
            return {"response": "fail", "error": str(e) + "\nUnable to create directory: " + path + "/" + name}

        return {"response": "success", "path": path, "name": name}

    except:
        return HTTPError()