#!/usr/bin/env python
"""Add Node to Cluster"""

### usage:
### ./nodeAdd.py -v 10.1.1.211 -u admin # will choose a free node that is in the same chassis
### ./nodeAdd.py -v 10.1.1.211 -u admin -n nodeID # will try to use free node with specified ID
### ./nodeAdd.py -v 10.1.1.211 -u admin -i 10.0.0.12 -p 10.0.0.13 # provide IP and IPMI addresses
# or will use existing addresses

### import pyhesity wrapper module
from pyhesity import *

### command line arguments
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-s', '--server', type=str, required=True)
parser.add_argument('-u', '--username', type=str, required=True)
parser.add_argument('-d', '--domain', type=str, default='local')
parser.add_argument('-n', '--nodeId', type=int, default=0)
parser.add_argument('-i', '--ipAddress', type=str, default='')
parser.add_argument('-p', '--ipmiAddress', type=str, default='')
parser.add_argument('-v', '--newVip', type=str, default='')

args = parser.parse_args()

server = args.server
username = args.username
domain = args.domain
nodeId = args.nodeId
ipAddress = args.ipAddress
ipmiAddress = args.ipmiAddress
newVip = args.newVip


### ip validate functions
def ipToBinary(ip):
    return ''.join(['{0:08b}'.format(int(octet)) for octet in ip.split('.')])


def ipValid(newIp, oldIp, netMask):
    cidr = ipToBinary(netMask).index('0')
    if('1' not in ipToBinary(newIp)[cidr:]):
        return False
    if('0' not in ipToBinary(newIp)[cidr:]):
        return False
    return ipToBinary(newIp)[0:cidr] == ipToBinary(oldIp)[0:cidr]


### authenticate
apiauth(server, username, domain)

### get cluster config
clusterStat = api('get', '/nexus/cluster/status')
chassisList = []
for chassis in clusterStat['clusterConfig']['proto']['chassisVec']:
    chassisList.append(chassis['name'])

### get free nodes
freeNodes = api('get', '/nexus/avahi/discover_nodes')

selectedNode = 0
foundNode = 0

for node in freeNodes['freeNodes']:

    if foundNode == 0:
        foundNode = node

    if nodeId == 0:
        ### see if node is in one of our chassis
        if(node['chassisSerial'] in chassisList):
            selectedNode = node
            break
    else:
        ### see if this is the node specified from the commandLine
        if(node['nodeId'] == nodeId):
            print("using %i" % node['nodeId'])
            selectedNode = node
            break

# if no node was specified and no node was in the same chassis, use the first available node
if nodeId == 0 and selectedNode == 0 and foundNode != 0:
    selectedNode = foundNode

# if node was specified but not found, fail
if selectedNode == 0:
    print("Couldn't find node with ID: %i" % nodeId)
    exit()

else:

    ### gather cluster subnet details
    clusterSubnet = clusterStat['clusterConfig']['proto']['clusterSubnet']['ip']
    clusterNetMask = clusterStat['clusterConfig']['proto']['clusterSubnet']['netmaskIp4']
    ipmiSubnet = clusterStat['clusterConfig']['proto']['ipmiConfig']['subnet']['ip']
    ipmiNetMask = clusterStat['clusterConfig']['proto']['ipmiConfig']['subnet']['netmaskIp4']

    ### validate VIP

    partitions = api('get', 'clusterPartitions')

    if(newVip == ''):
        createVip = False
        if(len(partitions[0]['nodeIds']) >= len(partitions[0]['vips'])):
            print("please provide a new VIP")
            exit()
    else:
        createVip = True
        if(not ipValid(newVip, clusterSubnet, clusterNetMask)):
            print("VIP address %s is not in the range of the cluster subnet %s(%s)" % (newVip, clusterSubnet, clusterNetMask))
            exit()

        if(newVip in partitions[0]['vips']):
            print("VIP %s is already in use on the cluster. Please select an unused address" % newVip)
            exit()

    ### validate Node IP
    if ipAddress == '':
        if 'ipAddresses' in selectedNode:
            ipAddress = selectedNode['ipAddresses'][0]
        else:
            print("please provide node ip Address")
            exit()

    if(not ipValid(ipAddress, clusterSubnet, clusterNetMask)):
        print("Node IPAddress %s is not in range of the cluster subnet %s(%s)" % (ipAddress, clusterSubnet, clusterNetMask))
        exit()

    ### validate ipmi IP
    if ipmiAddress == '':
        if 'ipmiIp' in selectedNode and selectedNode['ipmiIp'] != '':
            ipmiAddress = selectedNode['ipmiIp']
        else:
            print("please provide ipmi ip address")
            exit()

    if(not ipValid(ipmiAddress, ipmiSubnet, ipmiNetMask)):
        print("Node ipmi address %s is not in range of the ipmi subnet %s(%s)" % (ipmiAddress, ipmiSubnet, ipmiNetMask))
        exit()

    ### new node parameters
    newNodeParams = {
        "nodes": [
            {
                "id": selectedNode['nodeId'],
                "ip": ipAddress,
                "ipmiIp": ipmiAddress
            }
        ],
        "clusterPartitionId": clusterStat['clusterConfig']['proto']['clusterPartitionVec'][0]['id'],
        "autoUpdate": True,
        "ignoreSwIncompatibility": True
    }

    ### add the vip
    if(createVip):
        print("adding VIP %s" % newVip)
        partitions[0]['vips'].append(newVip)
        vipresult = api('put', '/clusterPartitions/%i' % partitions[0]['id'], partitions[0])

    ### execute the node add
    print("Adding node %i to cluster" % selectedNode['nodeId'])

    result = api('post', '/nexus/cluster/expand', newNodeParams)
    if 'message' in result:
        print(result['message'])
    else:
        display(result)