# -*- 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: %(name)s @ %(plugin)s ") % { "name" : pyfile.name, "plugin": 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("PONG %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."] temp_progress = "" lines = [] lines.append("ID - Name - Status - Speed - ETA - Progress") for data in downloads: if data['statusmsg'] == 'waiting': temp_progress = data['format_wait'] else: temp_progress = "%d%% (%s)" % (data['percent'], data['format_size']) lines.append("#%d - %s - %s - %s - %s - %s" % ( data['id'], data['name'], data['statusmsg'], "%.2f kb/s" % data['speed'], "%s" % data['format_eta'], temp_progress ) ) 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_push(self, args): if not args: return ["ERROR: Push package to queue like this: push <package id>"] id = int(args[0]) if not self.sm.get_package_data(id): return ["ERROR: Package #%d does not exist." % id] self.sm.push_package_to_queue(id) return ["INFO: Pushed package #%d to queue." % id] def event_pull(self, args): if not args: return ["ERROR: Pull package from queue like this: pull <package id>"] id = int(args[0]) if not self.sm.get_package_data(id): return ["ERROR: Package #%d does not exist." % id] self.sm.pull_out_package(id) return ["INFO: Pulled package #%d from queue to collector." % id] 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("push <id> Push package to queue") lines.append("pull <id> Pull package from queue") 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)