import os
import subprocess
import logging
import argparse
import json
import time
from prometheus_client import start_http_server, Summary, Gauge, Counter
from prometheus_client.core import GaugeMetricFamily

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                    level=logging.DEBUG)
logger = logging.getLogger(__name__)


parser = argparse.ArgumentParser(description='Script is monitor network status')

parser.add_argument("-i","--destination", dest="hostname", type=str,default="localhost",help="Ping destination")
parser.add_argument("-p","--parametres", dest="parametres", type=str,default="-c",help="Ping parametres")
parser.add_argument("-c","--count", dest="count", type=str,default="1",help="Ping parametres")

args = parser.parse_args()

hostname = args.hostname
parametres = args.parametres
count = args.count

if "EXTERNAL_URL" in os.environ :
    # export EXTERNAL_URL='["google.com","localhost"]'
    logging.debug("Variable exist")
    hostname=json.loads(os.environ["EXTERNAL_URL"])
else:
    logging.debug("Variable not exist")
    hostname=[hostname]

# Decorate function with metric.

def ping_time( parametres, count, hostname):
    for ip in hostname:
        out = subprocess.check_output("ping " + parametres + " " + count + " " + ip + " | grep -oP '.*time=\\K\\d+'", shell=True)
        output = out.decode()
        logging.debug("Host + {i} + Time + {o}".format(i=ip, o=output))
        result = output
    return result

network_jitter = Gauge('ping_time', 'Network jitter to hostname', ["destination"])
print(network_jitter)

if __name__ == '__main__':
    start_http_server(8000)
    while True:
        network_jitter.labels(hostname).set(ping_time(parametres, count, hostname))