from api import app
from flask import Flask,jsonify,request,abort, make_response
from utils.inventory import *
from nornir import InitNornir
from nornir_utils.plugins.functions import print_result
from nornir_napalm.plugins.tasks import napalm_get,napalm_configure, napalm_cli
from nornir_netmiko.tasks import netmiko_send_config,netmiko_send_command, netmiko_save_config, netmiko_commit

def get_inventory():
    hosts = app.config.get('nr').inventory.dict().get('hosts').keys()
    return list(map(lambda item: app.config.get('nr').inventory.dict().get('hosts').get(item), hosts))

def add_device(device):
    device = add_item_to_hosts_yaml(item=device)  
    return device

def get_device_by_name(name):
    host = app.config.get('nr').inventory.hosts.get(name)
    if not host:
        abort(make_response(jsonify(message="device not found"), 404))
    return host.dict()

def delete_device(device):
    try:
        delete_item_from_hosts_yaml(item=device)
    except Exception as e:
        abort(make_response(jsonify(message="Unable to delete this device", error=str(e)), 500))

def get_device_interfaces(device):
    nr = app.config.get('nr')
    result = nr.filter(device_name=device.get('name')).run(task=napalm_get,
    getters=["get_interfaces"])
    print_result(result)
    print(result)
    interfaces = result[device.get('name')][0].result.get('get_interfaces')
    return list(map(lambda item: dict(name=item, **interfaces.get(item)), interfaces))

def get_device_technical_info(device):
    nr = app.config.get('nr')
    result = nr.filter(device_name=device.get('name')).run(
        task=napalm_get,
        getters=["get_facts"]
    )

    print_result(result)

    facts = result[device.get('name')][0].result.get('get_facts')

    return {
        "hostname": facts.get("hostname"),
        "vendor": facts.get("vendor"),
        "model": facts.get("model"),
        "serial_number": facts.get("serial_number"),
        "os_version": facts.get("os_version"),
        "uptime": facts.get("uptime"),
    }


def get_device_interfaces_ip(device):
    nr = app.config.get('nr')
    result = nr.filter(device_name=device.get('name')).run(task=napalm_get,
    getters=["get_interfaces_ip"])
    print_result(result)
    interfaces = result[device.get('name')][0].result.get('get_interfaces_ip')
    def transformer(item):
        ip_address = list(interfaces.get(item).get('ipv4').keys())[0]
        prefix_length = interfaces.get(item).get('ipv4').get(list(interfaces.get(item).get('ipv4').keys())[0]).get('prefix_length')
        return dict(name=item, ip=ip_address, prefix_length=prefix_length)
    return list(map(lambda item: transformer(item), interfaces)) 
    #return interfaces