ashttpc

#!/bin/env python

import sys
from argparse import ArgumentParser, REMAINDER

from twisted.internet import reactor
from ashttp.client import TunnelClientService
from ashttp.runner import run_with_twistd 
from ashttp.logging import logger

if __name__ == '__main__':
	arg_parser = ArgumentParser(description='Clone an http tunnel client')
	arg_parser.add_argument('listen_port', type=int)
	arg_parser.add_argument('server', metavar='ip:port', help='IP and port of tunnel server')
	arg_parser.add_argument('-t', '--twistd', nargs=REMAINDER, help='Run by twistd. Next options will redirected to twistd')
	parsed_args = arg_parser.parse_args()

	try:
		server_ip, server_port = parsed_args.server.split(':')
		server_port = int(server_port)
	except ValueError:
		arg_parser.error("Invalid IP and Port for tunnel server")

	logger.useTwistdLogger() if parsed_args.twistd is not None else logger.usePythonStdLogger()

	TunnelClientService.remote_host = server_ip
	TunnelClientService.remote_port = server_port
	tunnel_client = TunnelClientService(reactor)
	reactor.listenTCP(parsed_args.listen_port, tunnel_client.getServerFactory())

	if parsed_args.twistd is not None:
		run_with_twistd(parsed_args.twistd, 'HttpTunnelClient')
	else:
		reactor.run()