import requests as req

from requests.models import Response
from requests.packages.urllib3.exceptions import InsecureRequestWarning

from prometheus_client import start_http_server, Gauge
import time, os
import re

def mkt_restapi(router_ip,router_username,router_password, path):
    url = 'https://' + router_ip + '/rest/' + path
    req.packages.urllib3.disable_warnings(InsecureRequestWarning)
    try:
        response = req.get(url, auth=(router_username,router_password), verify=False, timeout=5)
    except urllib3.exceptions.ReadTimeoutError:
            print("Connect timeout")
    
    if response.status_code != 200:
        quit()
    
    return response.json()

def init_prometheus_interface_metrics(data,router_ip):

    prom = dict()
    metrics = ['fp_rx_byte','fp_tx_byte','tx_byte','rx_byte']
    # declare all metrics
    for key in metrics:
        name = 'routeros_interface_' + key
        prom[name] = Gauge(name,"RouterOS interface " + key, ['routerip','interface'])

    return prom

def prom_request(data,prom,routerip):
    metrics = {'fp_rx_byte' : 'fp-rx-byte','fp_tx_byte' : 'fp-tx-byte','tx_byte' : 'tx-byte','rx_byte' : 'rx-byte'}

    for key in data:
        for m in metrics:
            name = 'routeros_interface_' +  m      
            interface = key['name']
            value = key[metrics[m]]
            prom[name].labels(routerip=routerip,interface=interface).set(value)


def init_prometheus_cpu_metrics(data,router_ip):

    prom = dict()

    name = 'routeros_load'
    prom[name] = Gauge(name,"RouterOS CPU load " + name, ['routerip','cpu'])
    name = 'routeros_irq'
    prom[name] = Gauge(name,"RouterOS CPU IRQ " + name, ['routerip','cpu'])

    return prom

def prom_cpu_request(data,prom,routerip):

    for key in data:
        name = 'routeros_load'
        cpu = key['cpu']
        value = key['load']
        prom[name].labels(routerip=routerip,cpu=cpu).set(value)

        name = 'routeros_irq'
        cpu = key['cpu']
        value = key['irq']
        prom[name].labels(routerip=routerip,cpu=cpu).set(value)

def init_prometheus_resource(data,router_ip):

    prom = dict()

    name = 'routeros_resource_version'
    prom[name] = Gauge(name,"RouterOS version ", ['routerip','version'])
    name = 'routeros_resource_uptime'
    prom[name] = Gauge(name,"RouterOS Uptime" , ['routerip'])

    return prom

def uptime_extractor(uptime):
    # extract hour
    try:
        re_h = re.compile("(\d)h")
        hour = re_h.match(uptime).group(1)
        hour = int(hour)
    except:
        hour = 0

    # extract minute
    try:
        re_m = re.compile("(\d)m")
        minute = re_m.match(uptime).group(1)
        minute = int(minute)
    except:
        minute = 0

    # extract second
    try:
        re_s = re.compile("(\d)s")
        second = re_s.match(uptime).group(1)
        second = int(second)
    except:
        second = 0

    # remove all value extracted
    uptime = 3600 * hour - minute * 60 - second

    return uptime


def prom_resource_request(data,prom,routerip):
    print(data)
    name = 'routeros_resource_version'
    version = data['version']
    value = 1
    prom[name].labels(routerip=routerip,version=version).set(value)

    name = 'routeros_resource_uptime'
    uptime = uptime_extractor(data['uptime'])
    prom[name].labels(routerip=routerip).set(uptime)



if __name__ == '__main__':


    try:
        router_ip = os.environ['ROUTER_IP']
        router_username = os.environ['ROUTER_USERNAME']
        router_password = os.environ['ROUTER_PASSWORD']
    except KeyError as e:
        print("Fatal Error missing: " + str(e) + " variable, exiting")
        os._exit(-1)

    # PORT
    try:
        exporter_port = os.environ['PORT']
    except KeyError:
        exporter_port = 8000

    # Interval
    try:
        interval = os.environ['INTERVAL']
    except KeyError:
        interval = 30

    # Start up the server to expose the metrics.
    start_http_server(exporter_port)
    print("Running RouterOS exporter on port " + str(exporter_port) + " fetch interval is " + str(interval) + "s")

    # interface metrics
    data_interface = mkt_restapi(router_ip,router_username,router_password,'interface')
    prom_interface = init_prometheus_interface_metrics(data_interface,router_ip)

    # system/resource/cpu metrics
    data_cpu = mkt_restapi(router_ip,router_username,router_password,'system/resource/cpu')
    prom_cpu = init_prometheus_cpu_metrics(data_cpu,router_ip)

    # system/resource
    print("resource")
    data_resource = mkt_restapi(router_ip,router_username,router_password,'system/resource')
    prom_resource = init_prometheus_resource(data_resource,router_ip)

    # Generate requests.
    while True:
        # interface
        data_interface = mkt_restapi(router_ip,router_username,router_password,'interface')
        prom_request(data_interface,prom_interface,router_ip)
        # system/resource/cpu
        data_cpu = mkt_restapi(router_ip,router_username,router_password,'system/resource/cpu')
        prom_cpu_request(data_cpu,prom_cpu,router_ip)

        prom_resource_request(data_resource,prom_resource,router_ip)

        time.sleep(interval)