#!/usr/bin/env python3

# Test program for CmConnection

import CmConnection
import argparse
import logging

logging.basicConfig(level=logging.INFO)

parser = argparse.ArgumentParser(description='CM Connectin Test Utility')
parser.add_argument('ip', help='CM IP address or domain name')
parser.add_argument('port', help='CM port', type=int)
parser.add_argument('username', help='Username')
parser.add_argument('password', help='Password')

parser.add_argument('--list-constellations', action='store_true')
parser.add_argument('--show-constellation-by-hub-name', nargs='?', type=str)
parser.add_argument('--create-connection', nargs='?', type=str, help="uuid;ifname;ifname")
parser.add_argument('--modify-connection', nargs='?', type=str, help="href;uuid;ifname;ifname")
parser.add_argument('--show-connection-by-name', nargs='?', type=str)
parser.add_argument('--list-connections', action='store_true')
parser.add_argument('--delete-connection', nargs='?', type=str, help="connection id, e.g. \"/network-connections/4505d5d3-b2f3-40b8-8ec2-4a5b28523c03\"")

args = parser.parse_args()

cm = CmConnection.CmConnection(args.ip, args.port, args.username, args.password, tls_verify=False)
if not cm.Connect():
    exit(-1)

if args.list_constellations:
    constellations = cm.list_constellations()
    for cid, if_list in constellations:
        print("Constellation:", cid)
        for if_name in if_list:
            print(f"    {if_name}")

if args.show_constellation_by_hub_name:
    constellation = cm.get_constellation_by_hub_name(args.show_constellation_by_hub_name)
    if constellation:
        (cid, if_list) = constellation
        print("Constellation:", cid)
        for if_name in if_list:
            print(f"    {if_name}")

if args.create_connection:
    cc_args = args.create_connection.split(";")
    if len(cc_args) != 3:
        print("Invalid create connection arguments. Expecting \"oid;ifname1;ifname2\", where ifname is form \"MODULE|PORT\"")
        exit(-1)
    cm.create_connection_ifnames(*cc_args)

if args.modify_connection:
    mc_args = args.modify_connection.split(";")
    if len(mc_args) == 2:
        cm.modify_connection_ifnames(mc_args[0], mc_args[1], None, None)
    elif len(mc_args) == 4:
        cm.modify_connection_ifnames(*mc_args)
    else:
        print("Invalid modify connection arguments. Expecting \"href;oid\" or \"href;oid;ifname1;ifname2\", where ifname is form \"MODULE|PORT\"")
        exit(-1)

if args.show_connection_by_name:
    connection = cm.get_connection_by_name(args.show_connection_by_name)
    if connection:
        print(str(connection))

if args.list_connections:
    connections = cm.get_connections()
    for c in connections:
        print(str(c))

if args.delete_connection:
    was_deleted = cm.delete_connection(args.delete_connection)
    if was_deleted:
        print(f"Successfully deleted {args.delete_connection}")
    else:
        print(f"Failed to delete {args.delete_connection}")