137 lines
5.7 KiB
Python
137 lines
5.7 KiB
Python
# PYTHON STUFFS #######################################################
|
|
|
|
import datetime
|
|
import urllib
|
|
|
|
from nemubot.exception import IMException
|
|
from nemubot.tools.web import getJSON
|
|
|
|
from nemubot.module.more import Response
|
|
|
|
URL_AVAIL = "https://www.whoisxmlapi.com/whoisserver/WhoisService?cmd=GET_DN_AVAILABILITY&domainName=%%s&outputFormat=json&username=%s&password=%s"
|
|
URL_WHOIS = "https://www.whoisxmlapi.com/whoisserver/WhoisService?da=2&domainName=%%s&outputFormat=json&userName=%s&password=%s"
|
|
|
|
|
|
# LOADING #############################################################
|
|
|
|
def load(CONF, add_hook):
|
|
global URL_AVAIL, URL_WHOIS
|
|
|
|
if not CONF or not CONF.hasNode("whoisxmlapi") or "username" not in CONF.getNode("whoisxmlapi") or "password" not in CONF.getNode("whoisxmlapi"):
|
|
raise ImportError("You need a WhoisXML API account in order to use "
|
|
"the !netwhois feature. Add it to the module "
|
|
"configuration file:\n<whoisxmlapi username=\"XX\" "
|
|
"password=\"XXX\" />\nRegister at "
|
|
"https://www.whoisxmlapi.com/newaccount.php")
|
|
|
|
URL_AVAIL = URL_AVAIL % (urllib.parse.quote(CONF.getNode("whoisxmlapi")["username"]), urllib.parse.quote(CONF.getNode("whoisxmlapi")["password"]))
|
|
URL_WHOIS = URL_WHOIS % (urllib.parse.quote(CONF.getNode("whoisxmlapi")["username"]), urllib.parse.quote(CONF.getNode("whoisxmlapi")["password"]))
|
|
|
|
import nemubot.hooks
|
|
add_hook(nemubot.hooks.Command(cmd_whois, "netwhois",
|
|
help="Get whois information about given domains",
|
|
help_usage={"DOMAIN": "Return whois information on the given DOMAIN"}),
|
|
"in","Command")
|
|
add_hook(nemubot.hooks.Command(cmd_avail, "domain_available",
|
|
help="Domain availability check using whoisxmlapi.com",
|
|
help_usage={"DOMAIN": "Check if the given DOMAIN is available or not"}),
|
|
"in","Command")
|
|
|
|
|
|
# MODULE CORE #########################################################
|
|
|
|
def whois_entityformat(entity):
|
|
ret = ""
|
|
if "organization" in entity:
|
|
ret += entity["organization"]
|
|
if "organization" in entity and "name" in entity:
|
|
ret += " "
|
|
if "name" in entity:
|
|
ret += entity["name"]
|
|
|
|
if "country" in entity or "city" in entity or "telephone" in entity or "email" in entity:
|
|
ret += " (from "
|
|
if "street1" in entity:
|
|
ret += entity["street1"] + " "
|
|
if "city" in entity:
|
|
ret += entity["city"] + " "
|
|
if "state" in entity:
|
|
ret += entity["state"] + " "
|
|
if "country" in entity:
|
|
ret += entity["country"] + " "
|
|
if "telephone" in entity:
|
|
ret += entity["telephone"] + " "
|
|
if "email" in entity:
|
|
ret += entity["email"] + " "
|
|
ret = ret.rstrip() + ")"
|
|
|
|
return ret.lstrip()
|
|
|
|
def available(dom):
|
|
js = getJSON(URL_AVAIL % urllib.parse.quote(dom))
|
|
|
|
if "ErrorMessage" in js:
|
|
raise IMException(js["ErrorMessage"]["msg"])
|
|
|
|
return js["DomainInfo"]["domainAvailability"] == "AVAILABLE"
|
|
|
|
|
|
# MODULE INTERFACE ####################################################
|
|
|
|
def cmd_avail(msg):
|
|
if not len(msg.args):
|
|
raise IMException("Indicate a domain name for having its availability status!")
|
|
|
|
return Response(["%s: %s" % (dom, "available" if available(dom) else "unavailable") for dom in msg.args],
|
|
channel=msg.channel)
|
|
|
|
|
|
def cmd_whois(msg):
|
|
if not len(msg.args):
|
|
raise IMException("Indiquer un domaine ou une IP à whois !")
|
|
|
|
dom = msg.args[0]
|
|
|
|
js = getJSON(URL_WHOIS % urllib.parse.quote(dom))
|
|
|
|
if "ErrorMessage" in js:
|
|
raise IMException(js["ErrorMessage"]["msg"])
|
|
|
|
whois = js["WhoisRecord"]
|
|
|
|
res = []
|
|
|
|
if "registrarName" in whois:
|
|
res.append("\x03\x02registered by\x03\x02 " + whois["registrarName"])
|
|
|
|
if "domainAvailability" in whois:
|
|
res.append(whois["domainAvailability"])
|
|
|
|
if "contactEmail" in whois:
|
|
res.append("\x03\x02contact email\x03\x02 " + whois["contactEmail"])
|
|
|
|
if "audit" in whois:
|
|
if "createdDate" in whois["audit"] and "$" in whois["audit"]["createdDate"]:
|
|
res.append("\x03\x02created on\x03\x02 " + whois["audit"]["createdDate"]["$"])
|
|
if "updatedDate" in whois["audit"] and "$" in whois["audit"]["updatedDate"]:
|
|
res.append("\x03\x02updated on\x03\x02 " + whois["audit"]["updatedDate"]["$"])
|
|
|
|
if "registryData" in whois:
|
|
if "expiresDateNormalized" in whois["registryData"]:
|
|
res.append("\x03\x02expire on\x03\x02 " + whois["registryData"]["expiresDateNormalized"])
|
|
if "registrant" in whois["registryData"]:
|
|
res.append("\x03\x02registrant:\x03\x02 " + whois_entityformat(whois["registryData"]["registrant"]))
|
|
if "zoneContact" in whois["registryData"]:
|
|
res.append("\x03\x02zone contact:\x03\x02 " + whois_entityformat(whois["registryData"]["zoneContact"]))
|
|
if "technicalContact" in whois["registryData"]:
|
|
res.append("\x03\x02technical contact:\x03\x02 " + whois_entityformat(whois["registryData"]["technicalContact"]))
|
|
if "administrativeContact" in whois["registryData"]:
|
|
res.append("\x03\x02administrative contact:\x03\x02 " + whois_entityformat(whois["registryData"]["administrativeContact"]))
|
|
if "billingContact" in whois["registryData"]:
|
|
res.append("\x03\x02billing contact:\x03\x02 " + whois_entityformat(whois["registryData"]["billingContact"]))
|
|
|
|
return Response(res,
|
|
title=whois["domainName"],
|
|
channel=msg.channel,
|
|
nomore="No more whois information")
|