#! /usr/bin/env python

import sys, requests, time
sys.path.insert(0, '.')
from docker import buildsha
print("Expected sha: ", buildsha.sha)

for i in range(30):

    try:
        print("Pinging...")
        sha = requests.get("https://{}.whtlst.in/api/ping".format(sys.argv[1])).json()["version"]["sha"]
        if sha == buildsha.sha:
            print("Success! {} at latest sha!".format(sys.argv[1]))
            sys.exit(0)
        else:
            raise AssertionError("{} != {}".format(sha, buildsha.sha))
    except Exception as e:
        print("Failure {}: {}".format(i, e))
        time.sleep(10)

print("Did not match sha within 300 seconds. Total Fatality")
sys.exit(2)