# -*- 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 .
"""
from select import select
import socket
from threading import Thread
import time
from time import sleep
from traceback import print_exc
import re
from pycurl import FORM_FILE
from module.plugins.Hook import Hook
from module.network.RequestFactory import getURL
from module.utils import formatSize
from module.Api import PackageDoesNotExists, FileDoesNotExists
class IRCInterface(Thread, Hook):
__name__ = "IRCInterface"
__version__ = "0.11"
__type__ = "hook"
__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),
("captcha", "bool", "Send captcha requests", True)]
__description__ = """Connect to irc and let owner perform different tasks"""
__author_name__ = "Jeix"
__author_mail__ = "Jeix@hasnomail.com"
def __init__(self, core, manager):
Thread.__init__(self)
Hook.__init__(self, core, manager)
self.setDaemon(True)
# self.sm = core.server_methods
self.api = core.api # todo, only use api
def coreReady(self):
self.abort = False
self.more = []
self.new_package = {}
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 newCaptchaTask(self, task):
if self.getConfig("captcha") and task.isTextual():
task.handler.append(self)
task.setWaiting(60)
page = getURL("http://www.freeimagehosting.net/upload.php",
post={"attached": (FORM_FILE, task.captchaFile)}, multipart=True)
url = re.search(r"\[img\]([^\[]+)\[/img\]\[/url\]", page).group(1)
self.response(_("New Captcha Request: %s") % url)
self.response(_("Answer with 'c %s text on the captcha'") % task.id)
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.logInfo("pyLoad IRC: Connected to %s!" % host)
self.logInfo("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.logDebug("Sending CTCP VERSION.")
self.sock.send("NOTICE %s :%s\r\n" % (msg['origin'], "pyLoad! IRC Interface"))
return
elif msg['text'] == "\x01TIME\x01":
self.logDebug("Sending CTCP TIME.")
self.sock.send("NOTICE %s :%d\r\n" % (msg['origin'], time.time()))
return
elif msg['text'] == "\x01LAG\x01":
self.logDebug("Received CTCP LAG.") # don't know how to answer
return
trigger = "pass"
args = None
try:
temp = msg['text'].split()
trigger = temp[0]
if len(temp) > 1:
args = temp[1:]
except:
pass
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.logError("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.api.statusDownloads()
if not downloads:
return ["INFO: There are no active downloads currently."]
temp_progress = ""
lines = ["ID - Name - Status - Speed - ETA - Progress"]
for data in downloads:
if data.status == 5:
temp_progress = data.format_wait
else:
temp_progress = "%d%% (%s)" % (data.percent, data.format_size)
lines.append("#%d - %s - %s - %s - %s - %s" %
(
data.fid,
data.name,
data.statusmsg,
"%s/s" % formatSize(data.speed),
"%s" % data.format_eta,
temp_progress
))
return lines
def event_queue(self, args):
ps = self.api.getQueueData()
if not ps:
return ["INFO: There are no packages in queue."]
lines = []
for pack in ps:
lines.append('PACKAGE #%s: "%s" with %d links.' % (pack.pid, pack.name, len(pack.links)))
return lines
def event_collector(self, args):
ps = self.api.getCollectorData()
if not ps:
return ["INFO: No packages in collector!"]
lines = []
for pack in ps:
lines.append('PACKAGE #%s: "%s" with %d links.' % (pack.pid, pack.name, len(pack.links)))
return lines
def event_info(self, args):
if not args:
return ["ERROR: Use info like this: info "]
info = None
try:
info = self.api.getFileData(int(args[0]))
except FileDoesNotExists:
return ["ERROR: Link doesn't exists."]
return ['LINK #%s: %s (%s) [%s][%s]' % (info.fid, info.name, info.format_size, info.statusmsg, info.plugin)]
def event_packinfo(self, args):
if not args:
return ["ERROR: Use packinfo like this: packinfo "]
lines = []
pack = None
try:
pack = self.api.getPackageData(int(args[0]))
except PackageDoesNotExists:
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 pyfile in pack.links:
self.more.append('LINK #%s: %s (%s) [%s][%s]' % (pyfile.fid, 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.api.unpauseServer()
return ["INFO: Starting downloads."]
def event_stop(self, args):
self.api.pauseServer()
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.api.getPackageData(id)
if not pack:
return ["ERROR: Package doesn't exists."]
#TODO add links
return ["INFO: Added %d links to Package %s [#%d]" % (len(links), pack['name'], id)]
except:
# create new package
id = self.api.addPackage(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.api.deletePackages(map(int, args[1:]))
return ["INFO: Deleted %d packages!" % len(args[1:])]
elif args[0] == "-l":
ret = self.api.delLinks(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_push(self, args):
if not args:
return ["ERROR: Push package to queue like this: push "]
id = int(args[0])
try:
info = self.api.getPackageInfo(id)
except PackageDoesNotExists:
return ["ERROR: Package #%d does not exist." % id]
self.api.pushToQueue(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 ."]
id = int(args[0])
if not self.api.getPackageData(id):
return ["ERROR: Package #%d does not exist." % id]
self.api.pullFromQueue(id)
return ["INFO: Pulled package #%d from queue to collector." % id]
def event_c(self, args):
""" captcha answer """
if not args:
return ["ERROR: Captcha ID missing."]
task = self.core.captchaManager.getTaskByID(args[0])
if not task:
return ["ERROR: Captcha Task with ID %s does not exists." % args[0]]
task.setResult(" ".join(args[1:]))
return ["INFO: Result %s saved." % " ".join(args[1:])]
def event_help(self, args):
lines = ["The following commands are available:",
"add [...] Adds link to package. (creates new package if it does not exist)",
"queue Shows all packages in the queue",
"collector Shows all packages in collector",
"del -p|-l [...] Deletes all packages|links with the ids specified",
"info Shows info of the link with id ",
"packinfo Shows info of the package with id ",
"more Shows more info when the result was truncated",
"start Starts all downloads",
"stop Stops the download (but not abort active downloads)",
"push Push package to queue",
"pull Pull package from queue",
"status Show general download status",
"help Shows this help message"]
return lines
class IRCError(Exception):
def __init__(self, value):
self.value = value
def __str__(self):
return repr(self.value)