2012-11-06 03:27:09 +00:00
|
|
|
# coding=utf-8
|
|
|
|
|
2012-11-07 15:37:19 +00:00
|
|
|
import http.client
|
|
|
|
import json
|
|
|
|
from urllib.parse import urlparse
|
|
|
|
from urllib.request import urlopen
|
2012-11-06 03:27:09 +00:00
|
|
|
|
|
|
|
nemubotversion = 3.3
|
|
|
|
|
|
|
|
def load(context):
|
|
|
|
from hooks import Hook
|
|
|
|
add_hook("cmd_hook", Hook(cmd_traceurl, "traceurl"))
|
|
|
|
add_hook("cmd_hook", Hook(cmd_isup, "isup"))
|
|
|
|
|
|
|
|
|
|
|
|
def help_tiny ():
|
|
|
|
"""Line inserted in the response to the command !help"""
|
|
|
|
return "The networking module"
|
|
|
|
|
|
|
|
def help_full ():
|
|
|
|
return "!traceurl /url/: Follow redirections from /url/."
|
|
|
|
|
|
|
|
def cmd_traceurl(msg):
|
|
|
|
if 1 < len(msg.cmds) < 6:
|
|
|
|
res = list()
|
|
|
|
for url in msg.cmds[1:]:
|
2012-11-07 15:37:19 +00:00
|
|
|
trace = traceURL(url)
|
|
|
|
res.append(Response(msg.sender, trace, channel=msg.channel, title="TraceURL"))
|
2012-11-06 03:27:09 +00:00
|
|
|
return res
|
|
|
|
else:
|
|
|
|
return Response(msg.sender, "Indiquer une URL a tracer !", channel=msg.channel)
|
|
|
|
|
|
|
|
def cmd_isup(msg):
|
|
|
|
if 1 < len(msg.cmds) < 6:
|
|
|
|
res = list()
|
|
|
|
for url in msg.cmds[1:]:
|
2012-11-07 15:37:19 +00:00
|
|
|
o = urlparse(url, "http")
|
|
|
|
if o.netloc == "":
|
|
|
|
o = urlparse("http://" + url)
|
|
|
|
if o.netloc != "":
|
|
|
|
raw = urlopen("http://isitup.org/" + o.netloc + ".json", timeout=10)
|
|
|
|
isup = json.loads(raw.read().decode())
|
2012-11-06 03:27:09 +00:00
|
|
|
if "status_code" in isup and isup["status_code"] == 1:
|
|
|
|
res.append(Response(msg.sender, "%s est accessible (temps de reponse : %ss)" % (isup["domain"], isup["response_time"]), channel=msg.channel))
|
|
|
|
else:
|
|
|
|
res.append(Response(msg.sender, "%s n'est pas accessible :(" % (isup["domain"]), channel=msg.channel))
|
|
|
|
else:
|
|
|
|
res.append(Response(msg.sender, "%s n'est pas une URL valide" % url, channel=msg.channel))
|
|
|
|
return res
|
|
|
|
else:
|
2012-12-07 20:18:21 +00:00
|
|
|
return Response(msg.sender, "Indiquer une URL à vérifier !", channel=msg.channel)
|
2012-11-07 15:37:19 +00:00
|
|
|
|
|
|
|
def traceURL(url, timeout=5, stack=None):
|
|
|
|
"""Follow redirections and return the redirections stack"""
|
|
|
|
if stack is None:
|
|
|
|
stack = list()
|
|
|
|
stack.append(url)
|
|
|
|
|
2012-11-07 16:19:08 +00:00
|
|
|
if len(stack) > 15:
|
|
|
|
stack.append('stack overflow :(')
|
|
|
|
return stack
|
|
|
|
|
2012-11-07 15:37:19 +00:00
|
|
|
o = urlparse(url, "http")
|
|
|
|
if o.netloc == "":
|
|
|
|
return stack
|
|
|
|
if o.scheme == "http":
|
|
|
|
conn = http.client.HTTPConnection(o.netloc, port=o.port, timeout=timeout)
|
|
|
|
else:
|
|
|
|
conn = http.client.HTTPSConnection(o.netloc, port=o.port, timeout=timeout)
|
|
|
|
try:
|
|
|
|
conn.request("HEAD", o.path, None, {"User-agent": "Nemubot v3"})
|
|
|
|
except socket.timeout:
|
|
|
|
stack.append("Timeout")
|
|
|
|
return stack
|
|
|
|
except socket.gaierror:
|
|
|
|
print ("<tools.web> Unable to receive page %s from %s on %d."
|
|
|
|
% (o.path, o.netloc, o.port))
|
2012-11-07 16:19:08 +00:00
|
|
|
return stack
|
2012-11-07 15:37:19 +00:00
|
|
|
|
|
|
|
try:
|
|
|
|
res = conn.getresponse()
|
|
|
|
except http.client.BadStatusLine:
|
2012-11-07 16:19:08 +00:00
|
|
|
return stack
|
2012-11-07 15:37:19 +00:00
|
|
|
finally:
|
|
|
|
conn.close()
|
|
|
|
|
|
|
|
if res.status == http.client.OK:
|
|
|
|
return stack
|
|
|
|
elif res.status == http.client.FOUND or res.status == http.client.MOVED_PERMANENTLY or res.status == http.client.SEE_OTHER:
|
|
|
|
url = res.getheader("Location")
|
|
|
|
if url in stack:
|
2012-11-07 16:19:08 +00:00
|
|
|
stack.append("loop on " + url)
|
2012-11-07 15:37:19 +00:00
|
|
|
return stack
|
|
|
|
else:
|
|
|
|
return traceURL(url, timeout, stack)
|
|
|
|
else:
|
2012-11-07 16:19:08 +00:00
|
|
|
return stack
|