#!/usr/bin/env python

import shlex
import sys
import subprocess


def ec2host(tag):
    output = subprocess.check_output(["ec2hostcache", tag])
    return output.strip()


def start_proxy(host, port):
    cmd = shlex.split("nc %s %s" % (host, port))
    return subprocess.call(cmd)


def main():
    try:
        host, port = sys.argv[1:]
    except ValueError:
        print >> sys.stderr, "Invalid arguments"
        return 1

    tag = host.rsplit(".")[0]
    return start_proxy(ec2host(tag), port)


if __name__ == "__main__":
    sys.exit(main())
