#!/usr/bin/env python
"""Remove Node from Cluster"""

### usage: ./backupNow.py -v mycluster -u admin -j 'VM Backup'

### import pyhesity wrapper module
from pyhesity import *

### command line arguments
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-v', '--vip', 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('-m', '--maxFull', type=int, default=70)

args = parser.parse_args()

vip = args.vip
username = args.username
domain = args.domain
nodeId = args.nodeId
maxFull = args.maxFull

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

### get cluster stats
cluster = api('get', 'cluster?fetchStats=true')

### check minimum node count
if cluster['nodeCount'] == cluster['supportedConfig']['minNodesAllowed']:
    print("Can not remove a node. Current node count: %i (Minimum required: %i)" % (cluster['nodeCount'], cluster['supportedConfig']['minNodesAllowed']))
    exit()

### check free space
newCapacity = cluster['stats']['usagePerfStats']['physicalCapacityBytes'] / cluster['nodeCount'] * (cluster['nodeCount'] - 1) * maxFull / 100

if cluster['stats']['usagePerfStats']['systemUsageBytes'] > newCapacity:
    print("Can not remove a node. Resulting capacity would be below the maxFull threshold")
    exit()

### check cluster health
clusterStat = api('get', '/nexus/cluster/status')
if clusterStat['healingStatus'] != 'NORMAL':
    print("Can not remove a node. Cluster Health Status is abnormal")
    exit()

### get node for removal
selectedNodeId = 0
nodeList = sorted(clusterStat['clusterConfig']['proto']['nodeVec'], key=lambda kv: kv['clusterNodeIndex'], reverse=True)

### if node is not specified, select last node of cluster
if nodeId == 0:
    selectedNodeId = nodeList[0]['id']
else:
    ### otherwise find specified node
    for node in nodeList:
        if nodeId == node['id']:
            selectedNodeId = nodeId
            break

### if we couldn't find the specified node
if selectedNodeId == 0:
    print("Couldn't find a node to remove")
else:
    ### remove the selected node
    print("Removing node %i" % selectedNodeId)
    result = api('post', "/nodes/%i" % selectedNodeId)