# -*- 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 .
@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 ']
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 ']
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 links". '\
'This will add the link to to the package / the package with 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 [...] (-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> [...] (-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 [...] 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 [...] Deletes all packages|links with the ids specified")
lines.append("info Shows info of the link with id ")
lines.append("packinfo Shows info of the package with 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)