import sys
import os
import subprocess
import json
import digitalocean
from django.conf import settings

sys.path.append('/srv/newsblur')

os.environ['DJANGO_SETTINGS_MODULE'] = 'newsblur_web.settings'

if __name__ == '__main__':
    # Check and clean second argument (ex: sshdo task 2)
    second_arg = sys.argv[2] if len(sys.argv) > 2 else "1"
    droplet_index = int(second_arg) if second_arg.isnumeric() else 1
    droplet_name = sys.argv[1]

    # Use correct Digital Ocean team based on "old"
    if second_arg == "old":
        doapi = digitalocean.Manager(token=settings.DO_TOKEN_FABRIC)
        # Cycle through droplets finding match
        droplets = doapi.get_all_droplets()
        for droplet in droplets:
            if droplet.name.startswith(droplet_name):
                if droplet_index > 1:
                    droplet_index -= 1
                    continue
                print(droplet.ip_address)
                break
    else:
        hosts = subprocess.check_output(['ansible-inventory', '--list'])
        if not hosts:
            print(" ***> Could not load ansible-inventory!")

        hosts = json.loads(hosts)
        for host, ip_host in hosts['_meta']['hostvars'].items():
            if host.startswith(droplet_name):
                print(ip_host['ansible_host'])
                break