import socket

from advanceddispatcher import AdvancedDispatcher
import asyncore_pollchoose as asyncore
from proxy import ProxyError
from socks5 import Socks5Connection, Socks5Resolver
from socks4a import Socks4aConnection, Socks4aResolver


class HttpError(ProxyError):
    pass


class HttpConnection(AdvancedDispatcher):
    def __init__(self, host, path="/"):
        AdvancedDispatcher.__init__(self)
        self.path = path
        self.destination = (host, 80)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.connect(self.destination)
        print "connecting in background to %s:%i" % (self.destination[0], self.destination[1])

    def state_init(self):
        self.append_write_buf(
            "GET %s HTTP/1.1\r\nHost: %s\r\nConnection: close\r\n\r\n" % (
                self.path, self.destination[0]))
        print "Sending %ib" % (len(self.write_buf))
        self.set_state("http_request_sent", 0)
        return False

    def state_http_request_sent(self):
        if len(self.read_buf) > 0:
            print "Received %ib" % (len(self.read_buf))
            self.read_buf = b""
        if not self.connected:
            self.set_state("close", 0)
        return False


class Socks5HttpConnection(Socks5Connection, HttpConnection):
    def __init__(self, host, path="/"):
        self.path = path
        Socks5Connection.__init__(self, address=(host, 80))

    def state_socks_handshake_done(self):
        HttpConnection.state_init(self)
        return False


class Socks4aHttpConnection(Socks4aConnection, HttpConnection):
    def __init__(self, host, path="/"):
        Socks4aConnection.__init__(self, address=(host, 80))
        self.path = path

    def state_socks_handshake_done(self):
        HttpConnection.state_init(self)
        return False


if __name__ == "__main__":
    # initial fill

    for host in ("bootstrap8080.bitmessage.org", "bootstrap8444.bitmessage.org"):
        proxy = Socks5Resolver(host=host)
        while len(asyncore.socket_map) > 0:
            print "loop %s, len %i" % (proxy.state, len(asyncore.socket_map))
            asyncore.loop(timeout=1, count=1)
        proxy.resolved()

        proxy = Socks4aResolver(host=host)
        while len(asyncore.socket_map) > 0:
            print "loop %s, len %i" % (proxy.state, len(asyncore.socket_map))
            asyncore.loop(timeout=1, count=1)
        proxy.resolved()

    for host in ("bitmessage.org",):
        direct = HttpConnection(host)
        while len(asyncore.socket_map) > 0:
            # print "loop, state = %s" % (direct.state)
            asyncore.loop(timeout=1, count=1)

        proxy = Socks5HttpConnection(host)
        while len(asyncore.socket_map) > 0:
            # print "loop, state = %s" % (proxy.state)
            asyncore.loop(timeout=1, count=1)

        proxy = Socks4aHttpConnection(host)
        while len(asyncore.socket_map) > 0:
            # print "loop, state = %s" % (proxy.state)
            asyncore.loop(timeout=1, count=1)