#!/usr/bin/env python3
import argparse
import math
import sys

from pyEDSM.edsm.exception import ServerError, NotFoundError
from pyEDSM.edsm.models import System, Commander

# ===========================================================================

def getBodyCount(system):
  try:
    bodyCount = System(system).bodyCount
  except ServerError as e:
    print(e)
    sys.exit(1)
  except NotFoundError as e:
    print(e)
    sys.exit(2)
  else:
    print(bodyCount)
    sys.exit(0)

def distanceBetween(system1, system2):
  try:
    coords1 = System(system1).coords
    coords2 = System(system2).coords
  except ServerError as e:
    print(e)
    sys.exit(1)
  except NotFoundError as e:
    print(e)
    syst.exit(2)
  else:
    print(int(round(math.sqrt( (coords1['x']-coords2['x'])**2
        + (coords1['y']-coords2['y'])**2
        + (coords1['z']-coords2['z'])**2 ),0)))
    sys.exit(0)

# ===========================================================================

parser = argparse.ArgumentParser(description="A collection of tools useful for "
    + "exploration.")
subparsers = parser.add_subparsers(title="subcommands", help="sub-command help",
    dest="subCommand")

parser_bodycount = subparsers.add_parser("bodycount",
    help="Returns the number of bodies in a system. Will exit with code 1 on "
    + "server error and code 2 if the system could not be found in EDSM.")
parser_bodycount.add_argument("system", nargs=1)

parser_distance = subparsers.add_parser("distancebetween",
    help="Calculates the distance between two systems. Will exti with code 1 "
    + "on server error and code 2 if (one of) the systems could not be found "
    + "on EDSM.")
parser_distance.add_argument("system", nargs=2)

args = parser.parse_args()

# ===========================================================================

if args.subCommand == "bodycount":
  getBodyCount(args.system[0])
elif args.subCommand == "distancebetween":
  distanceBetween(args.system[0], args.system[1])