1 files changed, 62 insertions, 9 deletions

diff --git a/module/HookManager.py b/module/HookManager.py
index 3a1a3d533..5dbf41936 100644
--- a/module/HookManager.py
+++ b/module/HookManager.py
@@ -17,28 +17,46 @@
     @author: mkaay
     @interface-version: 0.1
 """
+from glob import glob
 import logging
+from os import listdir
+from os.path import basename
+from os.path import join
+import subprocess
+from threading import Lock
 
 from module.XMLConfigParser import XMLConfigParser
 
-from os.path import basename, join
-from glob import glob
-
-from threading import Lock
-
 class HookManager():
     def __init__(self, core):
         self.core = core
-        self.configParser = XMLConfigParser(join("module","config","plugin.xml"))
+        self.configParser = XMLConfigParser(join("module", "config", "plugin.xml"))
         self.configParser.loadData()
         self.config = self.configParser.getConfig()        
         self.logger = logging.getLogger("log")
         self.plugins = []
+        self.scripts = {}
         self.lock = Lock()
         self.createIndex()
     
     def createIndex(self):
         self.lock.acquire()
+
+        f = lambda x: False if x.startswith("#") or x.endswith("~") else True
+	self.scripts = {}
+
+        folder = join(self.core.path, "scripts")
+
+	self.scripts['download_preparing'] = filter(f, listdir(join(folder, 'download_preparing')))
+	self.scripts['download_finished'] = filter(f, listdir(join(folder, 'download_finished')))
+	self.scripts['package_finished'] = filter(f, listdir(join(folder, 'package_finished')))
+	self.scripts['before_reconnect'] = filter(f, listdir(join(folder, 'before_reconnect')))
+        self.scripts['after_reconnect'] = filter(f, listdir(join(folder, 'after_reconnect')))
+
+	self.core.logger.info("Installed Scripts: %s" % str(self.scripts))
+
+        self.folder = folder
+
         pluginFiles = glob(join(self.core.plugin_folder, "hooks", "*.py"))
         plugins = []
         for pluginFile in pluginFiles:
@@ -50,8 +68,8 @@ class HookManager():
                     self.logger.info("Deactivated %s" % pluginName)
                     continue
             else:
-                self.configParser.set(pluginName, {"option": "activated", "type": "bool", "name": "Activated"}, True)

-            module = __import__("module.plugins.hooks."+pluginName, globals(), locals(), [pluginName], -1)
+                self.configParser.set(pluginName, {"option": "activated", "type": "bool", "name": "Activated"}, True)
+            module = __import__("module.plugins.hooks." + pluginName, globals(), locals(), [pluginName], -1)
             pluginClass = getattr(module, pluginName)
             plugin = pluginClass(self.core)
             plugin.readConfig()
@@ -63,24 +81,59 @@ class HookManager():
     
     def downloadStarts(self, pyfile):
         self.lock.acquire()
+
+    	for script in self.scripts['download_preparing']:
+            try:
+                out = subprocess.Popen([join(self.folder, 'download_preparing', script), pyfile.plugin.props['name'], pyfile.url], stdout=subprocess.PIPE)
+                out.wait()
+            except:
+                pass
+
         for plugin in self.plugins:
             plugin.downloadStarts(pyfile)
         self.lock.release()
     
     def downloadFinished(self, pyfile):
         self.lock.acquire()
+
+        for script in self.scripts['download_finished']:
+            try:
+                out = subprocess.Popen([join(self.folder, 'download_finished', script), pyfile.plugin.props['name'], pyfile.url, pyfile.status.name, \
+                join(self.core.path,self.core.config['general']['download_folder'], pyfile.folder, pyfile.status.name)], stdout=subprocess.PIPE)
+            except:
+                pass
+
+        
         for plugin in self.plugins:
             plugin.downloadFinished(pyfile)
         self.lock.release()
-    
+
+    def packageFinished(self, pyfile, package):
+        raise NotImplementedError
+
     def beforeReconnecting(self, ip):
         self.lock.acquire()
+
+        for script in self.scripts['before_reconnect']:
+            try:
+                out = subprocess.Popen([join(self.folder, 'before_reconnect', script), ip], stdout=subprocess.PIPE)
+                out.wait()
+            except:
+                pass
+
         for plugin in self.plugins:
             plugin.beforeReconnecting(ip)
         self.lock.release()
     
     def afterReconnecting(self, ip):
         self.lock.acquire()
+
+        for script in self.scripts['after_reconnect']:
+            try:
+                out = subprocess.Popen([join(self.folder, 'download_preparing', script), ip], stdout=subprocess.PIPE)
+            except:
+                pass
+        
         for plugin in self.plugins:
             plugin.afterReconnecting(ip)
         self.lock.release()