import httplib, json 
from HTMLParser import HTMLParser
from sre_compile import isstring

class DevelWtfResponseParser(HTMLParser):
   
    def __init__(self):
        HTMLParser.__init__(self)
        self.hosts = []
    def handle_data(self, data):
        if (len(data.strip()) > 0):
            self.hosts.append(data)
    def result(self):
        return self.hosts

# Default callback for DevelWtfClass
# Just print domains to console out
# You should extend this class to save result in file or
# database or anything else.

class DevelWtfCallback:
   
    def onSuccess(self, data):
        assert isinstance(data, list)
       
        for domain in data:
            print domain

class DevelWtf:
    def __init__(self, host, callback):
        assert isstring(host)
        assert isinstance(callback, DevelWtfCallback)
       
        self.host = host
        self.callback = callback
        assert self.host != None
        assert self.callback != None
       
    def run(self):
        conn = httplib.HTTPSConnection("devel.wtf")
        conn.request("GET", "/reverse/" + self.host + "/js")
        resp = conn.getresponse()
       
        if (resp.status == 200):
            data = resp.read()
            respObj = json.loads(data)
            parser = DevelWtfResponseParser()
            parser.feed(respObj['body']);
            self.callback.onSuccess(parser.result())
        else:
            print "Request failed"