new dirs

1 files changed, 364 insertions, 0 deletions

diff --git a/module/plugins/hooks/IRCInterface.py b/module/plugins/hooks/IRCInterface.py
new file mode 100644
index 000000000..e7e1e6797
--- /dev/null
+++ b/module/plugins/hooks/IRCInterface.py
@@ -0,0 +1,364 @@
+# -*- coding: utf-8 -*-
+
+"""
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 3 of the License,
+    or (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+    See the GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, see <http://www.gnu.org/licenses/>.
+    
+    @author: RaNaN
+    @author: jeix
+    @interface-version: 0.2
+"""
+
+from select import select
+import socket
+import sys
+from threading import Thread
+import time
+from time import sleep
+from traceback import print_exc
+
+from module.plugins.Hook import Hook
+
+class IRCInterface(Thread, Hook):
+    __name__ = "IRCInterface"
+    __version__ = "0.1"
+    __description__ = """connect to irc and let owner perform different tasks"""
+    __config__ = [("activated", "bool", "Activated", "False"),
+        ("host", "str", "IRC-Server Address", "Enter your server here!"),
+        ("port", "int", "IRC-Server Port", "6667"),
+        ("ident", "str", "Clients ident", "pyload-irc"),
+        ("realname", "str", "Realname", "pyload-irc"),
+        ("nick", "str", "Nickname the Client will take", "pyLoad-IRC"),
+        ("owner", "str", "Nickname the Client will accept commands from", "Enter your nick here!"),
+        ("info_file", "bool", "Inform about every file finished", "False"),
+        ("info_pack", "bool", "Inform about every package finished", "True")]
+    __author_name__ = ("Jeix")
+    __author_mail__ = ("Jeix@hasnomail.com")
+    
+    def __init__(self, core):
+        Thread.__init__(self)
+        Hook.__init__(self, core)
+        self.setDaemon(True)
+        self.sm = core.server_methods
+        
+    def coreReady(self):
+        self.new_package = {}
+        
+        self.abort = False
+        
+        self.links_added = 0
+        self.more = []
+
+        self.start()
+        
+        
+    def packageFinished(self, pypack):
+        try:
+            if self.getConfig("info_pack"):
+                self.response(_("Package finished: %s") % pypack.name)
+        except:
+            pass
+        
+    def downloadFinished(self, pyfile):
+        try:
+            if self.getConfig("info_file"):
+                self.response(_("Download finished: %s @ %s") % (pyfile.name, pyfile.pluginname) )
+        except:
+            pass
+             
+    def run(self):
+        # connect to IRC etc.
+        self.sock = socket.socket()
+        host = self.getConfig("host")
+        self.sock.connect((host, self.getConfig("port")))
+        nick = self.getConfig("nick")
+        self.sock.send("NICK %s\r\n" % nick)
+        self.sock.send("USER %s %s bla :%s\r\n" % (nick, host, nick))
+        for t in self.getConfig("owner").split():
+            if t.strip().startswith("#"):
+                self.sock.send("JOIN %s\r\n" % t.strip())
+        self.log.info("pyLoad IRC: Connected to %s!" % host)
+        self.log.info("pyLoad IRC: Switching to listening mode!")
+        try:        
+            self.main_loop()
+            
+        except IRCError, ex:
+            self.sock.send("QUIT :byebye\r\n")
+            print_exc()
+            self.sock.close()
+
+            
+    def main_loop(self):
+        readbuffer = ""
+        while True:
+            sleep(1)
+            fdset = select([self.sock], [], [], 0)
+            if self.sock not in fdset[0]:
+                continue
+            
+            if self.abort:
+                raise IRCError("quit")
+            
+            readbuffer += self.sock.recv(1024)
+            temp = readbuffer.split("\n")
+            readbuffer = temp.pop()
+
+            for line in temp:
+                line  = line.rstrip()
+                first = line.split()
+
+                if(first[0] == "PING"):
+                    self.sock.send("PING %s\r\n" % first[1])
+                    
+                if first[0] == "ERROR":
+                    raise IRCError(line)
+                    
+                msg = line.split(None, 3)
+                if len(msg) < 4:
+                    continue
+                    
+                msg = {
+                    "origin":msg[0][1:],
+                    "action":msg[1],
+                    "target":msg[2],
+                    "text":msg[3][1:]
+                }
+                
+                self.handle_events(msg)
+        
+        
+    def handle_events(self, msg):
+        if not msg["origin"].split("!", 1)[0] in self.getConfig("owner").split():
+            return
+            
+        if msg["target"].split("!", 1)[0] != self.getConfig("nick"):
+            return
+            
+        if msg["action"] != "PRIVMSG":
+            return
+            
+        # HANDLE CTCP ANTI FLOOD/BOT PROTECTION
+        if msg["text"] == "\x01VERSION\x01":
+            self.log.debug("Sending CTCP VERSION.")
+            self.sock.send("NOTICE %s :%s\r\n" % (msg['origin'], "pyLoad! IRC Interface"))
+            return
+        elif msg["text"] == "\x01TIME\x01":
+            self.log.debug("Sending CTCP TIME.")
+            self.sock.send("NOTICE %s :%d\r\n" % (msg['origin'], time.time()))
+            return
+        elif msg["text"] == "\x01LAG\x01":
+            self.log.debug("Received CTCP LAG.") # don't know how to answer
+            return
+         
+        trigger = "pass"
+        args = None
+        
+        temp = msg["text"].split()
+        trigger = temp[0]
+        if len(temp) > 1:
+            args = temp[1:]
+        
+        handler = getattr(self, "event_%s" % trigger, self.event_pass)
+        try:
+            res = handler(args)
+            for line in res:
+                self.response(line, msg["origin"])
+        except Exception, e:
+            self.log.error("pyLoad IRC: "+ repr(e))
+        
+        
+    def response(self, msg, origin=""):
+        if origin == "":
+            for t in self.getConfig("owner").split():
+                self.sock.send("PRIVMSG %s :%s\r\n" % (t.strip(), msg))
+        else:
+            self.sock.send("PRIVMSG %s :%s\r\n" % (origin.split("!", 1)[0], msg))
+        
+        
+#### Events
+    def event_pass(self, args):
+        return []
+        
+    def event_status(self, args):
+        downloads = self.sm.status_downloads()
+        if not downloads:
+            return ["INFO: There are no active downloads currently."]
+            
+        lines = []
+        lines.append("ID - Name - Status - Speed - ETA - Progress")
+        for data in downloads:
+            lines.append("#%d - %s - %s - %s - %s - %s" %
+                     (
+                     data['id'],
+                     data['name'],
+                     data['statusmsg'],
+                     "%.2f kb/s" % data['speed'],
+                     "%s" % data['format_eta'],
+                     "%d%% (%s)" % (data['percent'], data['format_size'])
+                     )
+                     )
+        return lines
+            
+    def event_queue(self, args):
+        ps = self.sm.get_queue()
+        
+        if not ps:
+            return ["INFO: There are no packages in queue."]
+        
+        lines = []
+        for id, pack in ps.iteritems():
+            lines.append('PACKAGE #%s: "%s" with %d links.' % (id, pack['name'], len(pack['links']) ))
+                
+        return lines
+        
+    def event_collector(self, args):
+        ps = self.sm.get_collector()
+        if not ps:
+            return ["INFO: No packages in collector!"]
+        
+        lines = []
+        for id, pack in ps.iteritems():
+            lines.append('PACKAGE #%s: "%s" with %d links.' % (id, pack['name'], len(pack['links']) ))
+                
+        return lines
+            
+    def event_info(self, args):
+        if not args:
+            return ['ERROR: Use info like this: info <id>']
+            
+        info = self.sm.get_file_data(int(args[0]))
+        
+        if not info:
+            return ["ERROR: Link doesn't exists."]
+        
+        id = info.keys()[0]
+        data = info[id]
+        
+        return ['LINK #%s: %s (%s) [%s][%s]' % (id, data['name'], data['format_size'], data['statusmsg'], data['plugin'])]
+        
+    def event_packinfo(self, args):
+        if not args:
+            return ['ERROR: Use packinfo like this: packinfo <id>']
+            
+        lines = []
+        pack = self.sm.get_package_data(int(args[0]))
+        
+        if not pack:
+            return ["ERROR: Package doesn't exists."]
+        
+        id = args[0]
+
+        self.more = []
+        
+        lines.append('PACKAGE #%s: "%s" with %d links' % (id, pack['name'], len(pack["links"])) )
+        for id, pyfile in pack["links"].iteritems():
+            self.more.append('LINK #%s: %s (%s) [%s][%s]' % (id, pyfile["name"], pyfile["format_size"], pyfile["statusmsg"], pyfile["plugin"]))
+
+        if len(self.more) < 6:
+            lines.extend(self.more)
+            self.more = []
+        else:
+            lines.extend(self.more[:6])
+            self.more = self.more[6:]
+            lines.append("%d more links do display." % len(self.more))
+            
+            
+        return lines
+    
+    def event_more(self, args):
+        if not self.more:
+            return ["No more information to display."]
+        
+        lines = self.more[:6]
+        self.more = self.more[6:]
+        lines.append("%d more links do display." % len(self.more))
+        
+        return lines
+    
+    def event_start(self, args):
+        
+        self.sm.unpause_server()
+        return ["INFO: Starting downloads."]
+        
+    def event_stop(self, args):
+    
+        self.sm.pause_server()
+        return ["INFO: No new downloads will be started."]
+    
+    
+    def event_add(self, args):
+        if len(args) < 2:
+            return ['ERROR: Add links like this: "add <packagename|id> links". '\
+                     'This will add the link <link> to to the package <package> / the package with id <id>!']
+            
+
+            
+        pack = args[0].strip()
+        links = [x.strip() for x in args[1:]]
+        
+        count_added = 0
+        count_failed = 0
+        try:
+            id = int(pack) 
+            pack = self.sm.get_package_data(id)
+            if not pack:
+                return ["ERROR: Package doesn't exists."]
+            
+            #add links
+            
+            
+            return ["INFO: Added %d links to Package %s [#%d]" % (len(links), pack["name"], id)]
+            
+        except:
+            # create new package
+            id = self.sm.add_package(pack, links, 1)
+            return ["INFO: Created new Package %s [#%d] with %d links." % (pack, id, len(links))]
+             
+        
+    def event_del(self, args):
+        if len(args) < 2:
+            return ["ERROR: Use del command like this: del -p|-l <id> [...] (-p indicates that the ids are from packages, -l indicates that the ids are from links)"]
+            
+        if args[0] == "-p":
+            ret = self.sm.del_packages(map(int, args[1:]))
+            return ["INFO: Deleted %d packages!" % len(args[1:])]
+            
+        elif args[0] == "-l":
+            ret = self.sm.del_links(map(int, args[1:]))
+            return ["INFO: Deleted %d links!" % len(args[1:])]
+
+        else:
+            return ["ERROR: Use del command like this: del <-p|-l> <id> [...] (-p indicates that the ids are from packages, -l indicates that the ids are from links)"]
+            
+    def event_help(self, args):
+        lines = []
+        lines.append("The following commands are available:")
+        lines.append("add <package|packid> <links> [...] Adds link to package. (creates new package if it does not exist)")
+        lines.append("queue                       Shows all packages in the queue")
+        lines.append("collector                   Shows all packages in collector")
+        lines.append("del -p|-l <id> [...]        Deletes all packages|links with the ids specified")
+        lines.append("info <id>                   Shows info of the link with id <id>")
+        lines.append("packinfo <id>               Shows info of the package with id <id>")
+        lines.append("more                        Shows more info when the result was truncated")
+        lines.append("start                       Starts all downloads")
+        lines.append("stop                        Stops the download (but not abort active downloads)")
+        lines.append("status                      Show general download status")
+        lines.append("help                        Shows this help message")
+        return lines
+        
+        
+class IRCError(Exception):
+    def __init__(self, value):
+        self.value = value
+    def __str__(self):
+        return repr(self.value)
\ No newline at end of file