#!/usr/bin/env python

import sys
import os
import subprocess
import socket
import urllib


def find_hostname_and_command(parms):
  host = None
  rest = parms
  while True:
    if not rest:
      break
    if rest[0] == "--":
      if host is None:
        _, host, rest = rest[0], rest[1], rest[2:]
      else:
        _, rest = rest[0], rest[1:]
      break
    elif rest[0].startswith("-o") and len(rest[0]) > 2:
      _, rest = rest[0], rest[1:]
    elif rest[0].startswith("-o"):
      _, rest = rest[0:1], rest[2:]
    elif rest[0].startswith("-"):
      _, rest = rest[0], rest[1:]
    else:
      if host is None:
        host, rest = rest[0], rest[1:]
      else:
        break
  host, port = urllib.splitport(host)
  return host, rest


def is_qubes_host(host):
  return host.endswith(".__qubes__")


def get_vmname_and_management_proxy(host):
  host = host[:-len(".__qubes__")]
  if host.endswith("__"):
    host, proxy, _ = host.rsplit("__", 2)
    return host, proxy
  return host, None


parms = sys.argv[1:]
host, rest = find_hostname_and_command(parms)

if os.path.basename(sys.argv[0]) in ("scp", "qscp"):
  if not is_qubes_host(host):
    os.execv("/usr/bin/scp", ["/usr/bin/scp"] + parms)

  path_to_ssh = os.path.join(path_to_this_file, "qssh")
  scmd = ["/usr/bin/scp"] + ["-S", path_to_ssh] + parms
  os.execvp(scmd[0], scmd)

else:
  if not is_qubes_host(host):
    os.execv("/usr/bin/ssh", ["/usr/bin/ssh"] + parms)

  path_to_bombshell = os.path.abspath(os.path.join(os.path.dirname(__file__), "bombshell-client"))
  vmname, proxy = get_vmname_and_management_proxy(host)
  if proxy:
    assert 0, "While connecting to %s (VM name %s): management proxy not supported yet" % (host, vmname)

  cmd = [
        path_to_bombshell,
        vmname,
    ] + ["sh", "-c", " ".join(rest)]
  os.execvp(cmd[0], cmd)