# -*- coding: utf-8 -*-

import os
import subprocess

from itertools import chain

from pyload.plugin.Addon import Addon
from pyload.utils import fs_join


class ExternalScripts(Addon):
    __name    = "ExternalScripts"
    __type    = "addon"
    __version = "0.29"

    __config = [("activated", "bool", "Activated"         , True ),
                  ("wait"     , "bool", "Wait script ending", False)]

    __description = """Run external scripts"""
    __license     = "GPLv3"
    __authors     = [("mkaay", "mkaay@mkaay.de"),
                     ("RaNaN", "ranan@pyload.org"),
                     ("spoob", "spoob@pyload.org"),
                     ("Walter Purcaro", "vuolter@gmail.com")]


    event_map = {'archive-extracted'      : "archive_extracted",
                 'package-extracted'      : "package_extracted",
                 'all_archives-extracted' : "all_archives_extracted",
                 'all_archives-processed' : "all_archives_processed",
                 'all_downloads-finished' : "allDownloadsFinished",
                 'all_downloads-processed': "allDownloadsProcessed"}


    def setup(self):
        self.scripts = {}

        folders = ["download_preparing", "download_finished", "all_downloads_finished", "all_downloads_processed",
                   "before_reconnect", "after_reconnect",
                   "package_finished", "package_extracted",
                   "archive_extracted", "all_archives_extracted", "all_archives_processed",
                   # deprecated folders
                   "unrar_finished", "all_dls_finished", "all_dls_processed"]

        for folder in folders:
            self.scripts[folder] = []

            self.initPluginType(folder, os.path.join(pypath, 'scripts', folder))
            self.initPluginType(folder, os.path.join('scripts', folder))

        for script_type, names in self.scripts.iteritems():
            if names:
                self.logInfo(_("Installed scripts for"), script_type, ", ".join(map(os.path.basename, names)))


    def initPluginType(self, folder, path):
        if not os.path.exists(path):
            try:
                os.makedirs(path)

            except Exception:
                self.logDebug("Script folder %s not created" % folder)
                return

        for f in os.listdir(path):
            if f.startswith("#") or f.startswith(".") or f.startswith("_") or f.endswith("~") or f.endswith(".swp"):
                continue

            if not os.access(os.path.join(path, f), os.X_OK):
                self.logWarning(_("Script not executable:") + " %s/%s" % (folder, f))

            self.scripts[folder].append(os.path.join(path, f))


    def callScript(self, script, *args):
        try:
            cmd = [script] + [str(x) if not isinstance(x, basestring) else x for x in args]

            self.logDebug("Executing", os.path.abspath(script), " ".join(cmd))

            p = subprocess.Popen(cmd, bufsize=-1)  #@NOTE: output goes to pyload
            if self.getConfig('wait'):
                p.communicate()

        except Exception, e:
            self.logError(_("Error in %(script)s: %(error)s") % {"script": os.path.basename(script), "error": e})


    def downloadPreparing(self, pyfile):
        for script in self.scripts['download_preparing']:
            self.callScript(script, pyfile.pluginname, pyfile.url, pyfile.id)


    def downloadFinished(self, pyfile):
        download_folder = self.config['general']['download_folder']
        for script in self.scripts['download_finished']:
            filename = fs_join(download_folder, pyfile.package().folder, pyfile.name)
            self.callScript(script, pyfile.pluginname, pyfile.url, pyfile.name, filename, pyfile.id)


    def packageFinished(self, pypack):
        download_folder = self.config['general']['download_folder']
        for script in self.scripts['package_finished']:
            folder = fs_join(download_folder, pypack.folder)
            self.callScript(script, pypack.name, folder, pypack.password, pypack.id)


    def beforeReconnecting(self, ip):
        for script in self.scripts['before_reconnect']:
            self.callScript(script, ip)


    def afterReconnecting(self, ip):
        for script in self.scripts['after_reconnect']:
            self.callScript(script, ip)


    def archive_extracted(self, pyfile, folder, filename, files):
        for script in self.scripts['archive_extracted']:
            self.callScript(script, folder, filename, files)
        for script in self.scripts['unrar_finished']:  #: deprecated
            self.callScript(script, folder, filename)


    def package_extracted(self, pypack):
        download_folder = self.config['general']['download_folder']
        for script in self.scripts['package_extracted']:
            folder = fs_join(download_folder, pypack.folder)
            self.callScript(script, pypack.name, folder, pypack.password, pypack.id)


    def all_archives_extracted(self):
        for script in self.scripts['all_archives_extracted']:
            self.callScript(script)


    def all_archives_processed(self):
        for script in self.scripts['all_archives_processed']:
            self.callScript(script)


    def allDownloadsFinished(self):
        for script in chain(self.scripts['all_downloads_finished'], self.scripts['all_dls_finished']):
            self.callScript(script)


    def allDownloadsProcessed(self):
        for script in chain(self.scripts['all_downloads_processed'], self.scripts['all_dls_processed']):
            self.callScript(script)