#!/usr/bin/env python

import os
import time
import sys
import subprocess
import digitalocean

OLD = False
# Set env var OLD=1 to use existing servers
if os.environ.get('OLD', False):
    OLD = True

if OLD:
    TOKEN_FILE = "/srv/secrets-newsblur/keys/digital_ocean.readprod.token"
else:
    TOKEN_FILE = "/srv/secrets-newsblur/keys/digital_ocean.token"

try:
    api_token = open(TOKEN_FILE, 'r').read().strip()
except IOError:
    print(f" ---> Missing Digital Ocean API token: {TOKEN_FILE}")
    exit()

outfile = f"/srv/newsblur/ansible/inventories/digital_ocean{'.old' if OLD else ''}.ini"

# Install from https://github.com/do-community/do-ansible-inventory/releases
ansible_inventory_cmd = f'do-ansible-inventory -t {api_token} --out {outfile}'
subprocess.call(ansible_inventory_cmd, shell=True)

with open(outfile, 'r') as original: 
    data = original.read()
with open(outfile, 'w') as modified: 
    modified.write("127.0.0.1 ansible_connection=local\n" + data)

exit() # Too many requests if we run the below code

do = digitalocean.Manager(token=api_token)
droplets = do.get_all_droplets()

print("\n ---> Checking droplets: %s\n" % (' '.join([d.name for d in droplets])))


def check_droplets_created():
    i = 0
    droplets = do.get_all_droplets()

    for instance in droplets:
        if instance.status == 'new':
            print(".", end=' ')
            sys.stdout.flush()
            i += 1
            time.sleep(i)
            break
    else:
        print(" ---> All booted!")
        return True

i = 0
while True:
    if check_droplets_created():
        break