http_

#!/usr/bin/env python3

import http.client
import os
import sys
import time
import urllib.parse


CLIENTS = {
    'http': http.client.HTTPConnection,
    'https': http.client.HTTPSConnection,
}

if len(sys.argv) > 1 and sys.argv[1] == 'config':
    print('graph_title Response time - {}'.format(os.getenv('title')))
    print('graph_category {}'.format(os.getenv('category', 'HTTP')))
    print('graph_args --lower-limit 0')
    print('graph_vlabel Time (milliseconds)')
    print('r.label {}'.format(os.getenv('url')))
    
    for c in ('warning', 'critical'):
        if os.getenv(c):
            print('r.{} {}'.format(c, os.getenv(c)))

else:
    o = urllib.parse.urlparse(os.getenv('url'))
    try:
        t1 = time.time()
        conn = CLIENTS[o.scheme](o.hostname)
        conn.request('GET', o.path)
        resp = conn.getresponse()
        resp.read()
        t2 = time.time()
        if resp.status == int(os.getenv('status', 200)):
            print('r.value {:.2f}'.format((t2 - t1)*1000))
        else:
            print('r.value U')
    except:
        print('r.value U')