#!/usr/bin/python

#
# Copyright (c) 2016 NEC Corporation and others
# All rights reserved.
#
# This program and the accompanying materials are made available under the
# terms of the Eclipse Public License v1.0 which accompanies this
# distribution, and is available at http://www.eclipse.org/legal/epl-v10.html
#

import requests
import json
import sys
import re


# response code to check the rest calls
RESP_GET_SUCCESS = 200
RESP_NOT_FOUND = 404


con_header = {"Accept": "application/json", "content-type": "application/json"}
authentication = ("admin", "admin")


def validate_cluster(ipaddress):
    """method to check the cluster status
    Args:
        ipaddress(str): ip address of the ODL controller
    """
    url = "http://" + ipaddress + ":8181/jolokia/read/akka:type=Cluster"
    try:
        resp = requests.get(url, headers=con_header, auth=authentication)
    except requests.exceptions.RequestException:
        print("controller is unreachable")
        sys.exit(1)

    if resp.status_code == RESP_NOT_FOUND:
        print("jolokia not installed, resp code", resp.status_code)
        print("Problem with accessing jolokia")
        sys.exit(1)

    elif resp.status_code != RESP_GET_SUCCESS:
        print("error in getting response, resp code", resp.status_code)
        sys.exit(1)

    data = json.loads(resp.content)
    cluster_status = data["value"]["ClusterStatus"]
    status = json.loads(cluster_status)
    members = status["members"]
    member_list = []
    entity_owner_list = []

    for member in members:
        # spliting the ip address of the node from json object
        # sample json data
        # "akka.tcp://opendaylight-cluster-data@10.106.138.137:2550"
        ip = re.search("@(.+?):", member["address"]).group(1)
        node_status = ip + "-" + member["status"]
        member_list.append(node_status)
        url1 = (
            "http://" + ip + ":8181/jolokia/read/org.opendaylight.controller:"
            "Category=ShardManager,name=shard-manager-operational,"
            "type=DistributedOperationalDatastore"
        )
        resp1 = requests.get(url1, headers=con_header, auth=authentication)
        if resp1.status_code != RESP_GET_SUCCESS:
            print("error in getting response for the node", ip)
            print("response content", resp1.content)
            continue
        data2 = json.loads(resp1.content)
        member_role = data2["value"]["MemberName"]
        entity_owner_list.append(ip + ":" + member_role)
    leader = data["value"]["Leader"]

    leaderNode = leader[leader.index("@") + 1 : leader.rindex(":")]
    for leader_node in member_list:
        address = leader_node.split("-")
        if address[0] == leaderNode:
            print("=================== Leader Node ======================\n")
            print(leader_node)
            member_list.remove(leader_node)
            print("=================== Follower Node ====================\n")
            print(member_list)
    list_entity_owners(ipaddress, entity_owner_list)


def list_entity_owners(ipaddress, entity_owner_list):
    """method to list the entity owners
    Args:
        ipaddress(str): ip address of the ODL controller
        entity_owner_list(list): list of member role in cluster
    """
    entity = ":8181/restconf/operational/entity-owners:entity-owners"
    url = "http://" + ipaddress + entity
    resp = requests.get(url, headers=con_header, auth=authentication)
    if resp.status_code != RESP_GET_SUCCESS:
        print("controller is down, resp_code", resp.status_code)
        print("response content", resp.content)
        sys.exit(1)
    data = json.loads(resp.content)
    ovsdb = data["entity-owners"]["entity-type"]
    print("\n\n=================== Entity Details ===================\n")
    for e_type in ovsdb:
        entities = e_type["entity"]
        for entity in entities:
            id = entity["id"]
            if len(entity["owner"]) > 0:
                print("NODE ID", str(id[id.rindex("=") + 2 : len(id) - 2]))
                print("OWNER", str(entity["owner"]))
            for owner in entity_owner_list:
                owner_role = owner.split(":")
                if entity["owner"] == owner_role[1]:
                    print("IP Address", str(owner_role[0]))
                    print("\n")


# Main Block
if __name__ == "__main__":
    print("*****Cluster Status******")
    ipaddress = raw_input("Please enter ipaddress to find Leader Node : ")
    validate_cluster(ipaddress)

else:
    print("Cluster checker loaded as Module")