#!/usr/bin/env python3

import os
import sys
import re
import json
import time
import logging
import requests
import configparser
import argparse

# Parse all arguments
parser = argparse.ArgumentParser(description="Push JSON connfig to AlkirAPI")
parser.add_argument("-t", "--tenant", type=str, default='alkira.cnf', help="location of alikira.cnf (default: alkira.cnf")
parser.add_argument("-f", "--folder", type=str, default='config', help="location of the JSON connector files (default: config")
parser.add_argument("-c", "--config", type=str, default='connectors.cnf', help="location of the connector config (default: config/connectors.cnf")
parser.add_argument("-v", "--verbose", type=int, default=0, help="Verbose level 0 or 1 (default: 0")

try:
	args = parser.parse_args()
	ALKIRA_CONFIG = args.tenant    
	connector_folder = args.folder      
	CONNECTOR_CONFIG = f'{args.folder}/{args.config}'

except argparse.ArgumentError as e:
	print(str(e))
	sys.exit()

try:
	loglevel = {
		0: logging.INFO,
		1: logging.DEBUG
	}[args.verbose]
except KeyError:
	loglevel = logging.INFO

###############################################

# Tenant config
if not os.path.isfile(ALKIRA_CONFIG):
	logging.error(f"The config file {ALKIRA_CONFIG} doesn't exist")
	sys.exit(1)
alkira = configparser.RawConfigParser()
alkira.read(ALKIRA_CONFIG)
ALKIRA_TENANT = alkira.get('alkira', 'ALKIRA_TENANT')
ALKIRA_USERNAME = alkira.get('alkira', 'ALKIRA_USERNAME')
ALKIRA_PASSWORD = alkira.get('alkira', 'ALKIRA_PASSWORD')
ALKIRA_BASE_URI = f'https://{ALKIRA_TENANT}/api'

# Connector config
if not os.path.isfile(CONNECTOR_CONFIG):
	logging.error(f"The config file {CONNECTOR_CONFIG} doesn't exist")
	sys.exit(1)
config = configparser.ConfigParser()
config.read(CONNECTOR_CONFIG)

# Set loglevel (logging.INFO, logging.DEBUG)
logging.basicConfig(level=loglevel)
logging = logging.getLogger('AlkiraAPI')

###############################################

# Set default headers
headers = {'Content-Type': "application/json"}

# Naming exceptions
service_exceptions = {
	"saas": "internet",
	"pan": "panfw",
	"ftntfw": "ftnt-fw-",
	"chkpfw": "chkp-fw-"
	}

def alkira_login():
	body = {'userName': ALKIRA_USERNAME,
		'password': ALKIRA_PASSWORD}
	session = requests.session()
	response = alkira_post(session, '/login', body)
	return session

def alkira_post(session, uri, body):
	url = f'{ALKIRA_BASE_URI}{uri}'
	try:
		response = session.post(url, data=json.dumps(body), headers=headers)
		response.raise_for_status()
	except Exception as e:
		logging.error(f'Error: {str(e)}')
		sys.exit(1)
	return response

def alkira_get(session, uri):
	url = f'{ALKIRA_BASE_URI}{uri}'
	try:
		response = session.get(url, headers=headers)
		response.raise_for_status()
	except Exception as e:
		logging.error(f'Error: {str(e)}')
		sys.exit(1)
	return response

def alkira_delete(session, uri):
	url = f'{ALKIRA_BASE_URI}{uri}'
	try:
		response = session.delete(url, headers=headers)
		response.raise_for_status()
	except Exception as e:
		logging.error(f'Error: {str(e)}')
		sys.exit(1)
	return response

# Authenticate
s = alkira_login()
logging.debug(s)

# Get TenantID
r = alkira_get(s, '/tenantnetworks')
data = r.json()
tenantNetworkId = data[0]['id']
tenantName = data[0]['name']
logging.info(f'Tenant Name: {tenantName}')
logging.info(f'Tenant ID: {tenantNetworkId}')

# Push connectors
logging.info('Push Connectors')

for connector in config.sections():
	section = config[connector]
	region =  section['cxp']
	segments = section['segments']
	logging.debug(f'{section} - {region} - {segments}')
	if 'group' in config:
		group = section['group']
		logging.debug(group)
	if 'billingtags' in config:
		billingtags = section['billingtags']
		logging.debug(billingtags)

	connector_result = re.match(r'(\w+)(\d+)', connector)
	connector_name = connector_result.group(1)
	connector_number = connector_result.group(2)
	logging.debug(f'{connector_folder}/{connector_name}{connector_number}.txt')
	config_path = (f'{connector_folder}/{connector_name}{connector_number}.txt')
	with open (config_path, 'r') as f:
		body = json.load(f)
		body['cxp'] = region
		body['segments'][0] = segments
		if 'group' in body and 'group' in globals():
			body['group'] = group

		if 'billingTags' in body and 'billingtags' in globals():
			body['billingTags'][0] = billingtags

		logging.debug(json.dumps(body))
		logging.info(f'Pushing {connector_name} to {region} (network segment: {segments})')
		r = alkira_post(s, f'/tenantnetworks/{tenantNetworkId}/{connector_name}', body)
		logging.info(r.status_code)
		logging.debug(r.content)