#!/usr/bin/env python3
#
# Copyright 2022, Mischa Peters <mischa AT alkira DOT net>, Alkira.
# get.py
# Version 0.1 - 20220617 - initial release
# Version 0.2 - 20220621 - simplified structure, added -i for IDs only, added -u for ugly JSON
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#
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="Get JSON config to AlkiraAPI")
parser.add_argument("-t", "--tenant", type=str, default='alkira.cnf', help="location of alikira.cnf (default: alkira.cnf)")
parser.add_argument("-i", "--id", help="", action="store_true")
parser.add_argument("-u", "--ugly", help="make the JSON ugly!", action="store_true")
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
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

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

# Set logging.INFO to logging.DEBUG for debug information
logging.basicConfig(level=loglevel)
logging = logging.getLogger('AlkiraAPI')

# 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'

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

# 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
logging.info('=== Authenticating')
s = alkira_login()
logging.debug(s)

# Get TenantID
logging.info('=== Fetching Tenant Info')
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}')

# Collecting data with base url: /
collect = [
	'tenantenabledfeatures',
	'credentials'
	]

if not args.id:
	for item in collect:
		logging.info(f'=== Collecting {item}')
		r = alkira_get(s, f'/{item}')
		data = r.json()
		print(f'# {item}')
		if args.ugly:
			print(json.dumps(data))
		else:
			print(json.dumps(data, indent=4))

# Collecting data with base url: /tenantnetworks/{tenantNetworkId}
collect = [
	'connectors',
	'services',
	'groups',
	'segments',
	'alkira-remote-access-connectors',
	'global-cidr-lists'
	]

for item in collect:
	logging.info(f'=== Collecting {item}')
	r = alkira_get(s, f'/tenantnetworks/{tenantNetworkId}/{item}')
	data = r.json()
	print(f'# {item}')
	if args.id:
		for i in data:
			str = f"{i['id']:10}"
			if 'type' in i:
				str = str + (f"\t{i['type']:20}")

			if 'name' in i:
				str = str + (f"\t{i['name']:20}")
			print(str)
	else:
		if args.ugly:
			print(json.dumps(data))
		else:
			print(json.dumps(data, indent=4))