#!/usr/bin/python
import requests
import getpass
import json
from pprint import pprint

from requests.packages.urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

host = "0.188.131.20"
base_url = "https://" + host + "/rest/"

fabric_key = ""

def get_fabrics(session,host):
    url = base_url + "resourcegroups/All/fcfabrics"
    #print url
    try:
        response = session.get(url=url)
        #print response
        return json.loads(response.content)
    except:
        print "Unable to retrieve fabrics"
        return None
    return response.content

def login(session):
    url = base_url + "login"
    try:
        response = session.post(url=url)
        #print dir(response)
        #print response
        #print response.headers
    except Exception as e:
        print e
        print "Unable to login"
        return None
    return response.headers['WStoken']

def logout(connection, WStoken):
    url = base_url + 'logout'
    print("URL: " + url)
    connection.request('POST', url,
                       headers={"WStoken": WStoken,
                                "Accept": "application/vnd.brocade.networkadvisor+json;version=v1"})
    print("Logging out of BNA...")
    # Get the response
    response = connection.getresponse()
    # Display the response status
    print("Status=", response.status)
    # If successful (status = 204), display the returned Network Advisor session token
    if response.status == 204:
        print("Logout successful!")
    else:
        print(response.status, response.reason)
    connection.close()

#step1 - dont use
def start_zoning(WStoken, session, fabric_key):
    print("Starting Zoning Transaction...")
    data = {
        "lsanZoning": "false",
        "action": "START"
    }
    headers = {
        "WStoken": WStoken,
        "Accept": "application/vnd.brocade.networkadvisor+json;version=v1",
        "Content-type": "application/vnd.brocade.networkadvisor+json;version=v1"
    }
    url = base_url + 'resourcegroups/All/fcfabrics/{}/controlzonetransaction'
    print("URL: " + url.format(fabric_key))
    r = requests.post(url.format(fabric_key), data=json.dumps(data), headers=headers, verify=False)
    print("Status=", r.status_code)
    # If successful (status = 204), continue with creation of Aliases, Zones and commit transaction
    if r.status_code == 204:
        return True
    else:
        return False

#step2
def create_zoning_objects(WStoken, payload, fabric_key):
    print("Creating Aliases and Zones...")
    url = base_url + 'resourcegroups/All/fcfabrics/{}/createzoningobject'
    data=payload
    headers={
    "WStoken": WStoken,
    "Accept": "application/vnd.brocade.networkadvisor+json;version=v1",
    "Content-type": "application/vnd.brocade.networkadvisor+json;version=v1"
    }
    if confirm("Create the following zoning object:", data):
        r = requests.post(url.format(fabric_key), data=json.dumps(data), headers=headers, verify=False)
        print("Status=", r.status_code)
        return True
    else:
        return False

def make_zoning_payload():
    payload = {"zones": [], "zoneAliases": []}
    while again("Do you want to create an alias?"):
        name = raw_input('Alias name: ')
        wwn = raw_input('Enter all wwns seperated by commas: ')
        payload['zoneAliases'].append({"name":name, "memberNames":wwn.split(',')})
    while again("Do you want to create a zone? "):
        name = raw_input("Zone name")
        aliases = raw_input("Enter all aliases for this zone seperated by commas:")
        payload['zones'].append({"name": name, "aliasNames": aliases.split(',')})
    return payload

def test_zoning_payload():
    return {
    "zones": [{
    "name": "host1_hba0_array_sp0", #make parameter
    "aliasNames": [
    "host_hba0",
    "array_sp0"],
    "type": "STANDARD"
    }],
    "zoneAliases": [{
    "name": "host_hba0",
    "memberNames": [
    "11:22:33:44:55:66:77:88"
    ]},{
    "name": "array_sp0",
    "memberNames": [
    "88:77:66:55:44:33:22:11"
    ]}]}

#step 3
def make_cfg_payload():
    payload = {"action":"ADD", "zoneSets":[]}
    while again("Do you want to add a zone to cfg?"):
        name = raw_input("fabric name:")
        zones = raw_input("zone_names:")
        payload['zoneNames'].append({"name": name, "zoneNames": zones.split(',')})
    return payload

def test_cfg_payload():
    return {
        "action": "ADD",
        "zoneSets": [
            {
                "name": "home_fabric",
                "zoneNames": [
                    "host1_hba0_array_sp0"
                ]
            }
        ]
    }

def add_zoning_to_CFG(WStoken, payload, fabric_key):
    print("Adding Zones to CFG...")
    url = base_url + 'resourcegroups/All/fcfabrics/{}/updatezoningobject'
    data = make_cfg_payload()
    headers = {
        "WStoken": WStoken,
        "Accept": "application/vnd.brocade.networkadvisor+json;version=v1",
        "Content-type": "application/vnd.brocade.networkadvisor+json;version=v1"
    }
    if confirm("Add the following zone to the cfg:", data):
        r = requests.post(url.format(fabric_key), data=json.dumps(data), headers=headers)
        print("Status=", r.status_code)
        return True
    else:
        return False

#step 4
def commit_zoning_transaction(WStoken, fabric_key):
    print("Committing Zoning Transaction...")
    url = base_url + 'resourcegroups/All/fcfabrics/{}/controlzonetransaction'
    data = {
        "lsanZoning": "false",
        "action": "COMMIT"
    }
    headers = {
        "WStoken": WStoken,
        "Accept": "application/vnd.brocade.networkadvisor+json;version=v1",
        "Content-type": "application/vnd.brocade.networkadvisor+json;version=v1"
    }
    if confirm("Are you ready to commit?", None):
        r = requests.post(url.format(fabric_key), data=json.dumps(data), headers=headers)
        print("Status=", r.status_code)
        return True
    else:
        return False

#step 5
def activate_CFG(connection, WStoken, fabric_key):
    print("Activating CFG...")
    url = base_url + 'resourcegroups/All/fcfabrics/{}/zonesets/home_fabric-false/activate'
    data = {
    }
    headers = {
        "WStoken": WStoken,
        "Accept": "application/vnd.brocade.networkadvisor+json;version=v1",
        "Content-type": "application/vnd.brocade.networkadvisor+json;version=v1"
    }
    print("URL: " + url)
    r = requests.post(url.format(fabric_key), data=json.dumps(data), headers=headers)
    print("Status=", r.status_code)
    print()
    url = base_url + 'resourcegroups/All/fcfabrics/' + fabric_key + '/zonedbs'
    print("URL: " + url)
    print("Displaying updated contents of Zoning DB for fabric...")
    connection.connect()
    connection.request('GET', url,
                       headers={"WStoken": WStoken,
                                "Accept": "application/vnd.brocade.networkadvisor+json;version=v1"})
    # Get the response
    response = connection.getresponse()
    # Display the response status
    print("Status=", response.status)
    # If successful (status = 200), display the returned list in JSON format
    if response.status == 200:
        print("Zone DB:")
        json_response_bytes = response.read()
        json_response_string = str(json_response_bytes, encoding='utf8')
        zone_db_dict = json.loads(json_response_string)
        print(json.dumps(zone_db_dict, indent=4))
    else:
        print(response.status, response.reason)
        connection.close()

###GET USER INTERACTION#####
def again(prompt):
    response = raw_input(prompt)
    if response in ['Y', 'y', 'Yes', 'YES', 'yes', 'ye']:
        return True
    return False

def confirm(message, object):
    print(message)
    pprint(object)
    return again("Does this look right? :")




user = getpass.getuser()
password = getpass.getpass("Enter your password: ")
session = requests.Session()
session.verify = False
session.headers.update({"WSUsername":"%s" % user ,"WSPassword":"%s" % password,"Accept":"application/vnd.brocade.networkadvisor+json;version=v1"})
session.auth = (user,password)
token = login(session)
session.headers.update({"WStoken":"%s" % token})

content = get_fabrics(session,host)

fabric_key = raw_input("Fabric Key:")

start_zoning(token, fabric_key)
payload = make_zoning_payload()
create_zoning_objects(token, payload, fabric_key)
#commit_zoning_transaction(token, fabric_key)
#activate_CFG(session, token, fabric_key)


#print json.dumps(content,indent=4)

#wwn = content['fcFabrics'][0]['principalSwitchWwn']
#print content

#TODO: look for collisions