from http.server import HTTPServer, BaseHTTPRequestHandler
import os
import socket
import logging
import json

class httpHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        if self.path == '/a':
            self.wfile.write('Helloa! '.encode())
        elif self.path == '/statsd':
            send_statsd_metrics()
            self.send_response(200)
            self.send_header('Content-type', 'application/json')
            self.end_headers()
            self.wfile.write(json.dumps({'traceId': 'dummy'}).encode())
        elif self.path == '/':
            self.send_response(200)
            self.end_headers()
        else:
            self.wfile.write('Hello! '.encode())

def send_statsd_metrics():
    INSTANCE_ID = os.getenv('INSTANCE_ID')
    RECEIVER_ADDRESS = os.getenv('COLLECTOR_UDP_ADDRESS')
    logging.warning(INSTANCE_ID)
    logging.warning(RECEIVER_ADDRESS)
    if INSTANCE_ID is not None and RECEIVER_ADDRESS is not None:
        statsD_metric = bytes('statsdTestMetric1g_%s:1|g|#mykey1:myvalue1,mykey2:myvalue2\nstatsdTestMetric1c_%s:1|c|#mykey3:myvalue3\nstatsdTestMetric1ms_%s:1|ms|#mykey4:myvalue4\nstatsdTestMetric1h_%s:1|h|#mykey5:myvalue5' % (INSTANCE_ID, INSTANCE_ID,  INSTANCE_ID,  INSTANCE_ID),'utf-8')
        opened_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        opened_socket.sendto(statsD_metric, (RECEIVER_ADDRESS.split(':')[0], int(RECEIVER_ADDRESS.split(':')[1])))
    return

def main():
    ADDRESS = os.getenv('LISTEN_ADDRESS')
    if ADDRESS is not None:
        HOST = ADDRESS.split(':')[0]
        PORT = int(ADDRESS.split(':')[1])
    else:
        HOST = '0.0.0.0'
        PORT = 4321
    server = HTTPServer((HOST, PORT), httpHandler)
    server.serve_forever()

if __name__ == '__main__':
    main()