#!/usr/bin/env python
"""
Input: This script takes a csv file containing the campaign name and uuid as attributes as input as following:

campaigns.csv

name              | uuid
--------------------------------------------------------
NIRA_Denmark_2023 | c5c454c1-6470-407c-9b22-b9af4560d65c
NIRA_France_2023  | 9d4af2c8-ca3f-4c4e-99c4-64081581d7c4
.
.
.
--------------------------------------------------------

Output: This script generates a terraform file for a services' request attribute, which is based on a rendered template.
Attention: This script was once used to genrate 50 Endpoints for the Service Chain Connected Friction and caused several
CPU spikes resulting in bad performance!

The input .csv file shall be stored within the default input directory.

Usage: python3 main.py -n Connected_Friction
"""
import argparse
import configparser
import csv
import jinja2
import logging
import pathlib


FORMAT = '%(asctime)s %(message)s'
logging.basicConfig(format=FORMAT)
logger = logging.getLogger('main')
logger.setLevel(logging.INFO)

config = configparser.ConfigParser()
config.read("config.ini")

DEFAULT_INPUT_PATH = pathlib.Path(config['PATHS']['Input']).absolute()
DEFAULT_OUTPUT_PATH = pathlib.Path(config['PATHS']['Output']).absolute()
DEFAULT_TEMPLATES_PATH = pathlib.Path(config['PATHS']['Templates']).absolute()

parser = argparse.ArgumentParser(description="Name of the Services' Chain, e.g. Connected_Friction")
parser.add_argument("--name", "-n", type=str, metavar='', required=True, help="Example: Connected_Friction")
args = parser.parse_args()

env = jinja2.Environment(loader=jinja2.FileSystemLoader(DEFAULT_TEMPLATES_PATH))
template = env.get_template("service.request_attribute.j2")

def checkDirectories(DEFAULT_INPUT_PATH, DEFAULT_OUTPUT_PATH, DEFAULT_TEMPLATES_PATH):
    for dir in [DEFAULT_INPUT_PATH, DEFAULT_OUTPUT_PATH, DEFAULT_TEMPLATES_PATH]:
        
        try:
            pathlib.Path.mkdir(dir)
            logger.info("Directory created: %s", dir)
        except FileExistsError:
            logger.info("Directory already exists: %s", dir)    

def renderFile(args, file, DEFAULT_OUTPUT_PATH):
    logger.info("Generating files for: %s", str(pathlib.PurePath(file).name))
    with open(file, newline='') as csvfile:
        data = csv.DictReader(csvfile, delimiter=';')

        campaignUuids = []
        for row in data:
            campaignUuids.append(row["uuid"])                 
        
        content = template.render(ServiceName = str(args.name), CampaignUuids = campaignUuids)
        filename = pathlib.PurePath.joinpath(DEFAULT_OUTPUT_PATH, str(args.name) + ".request_attribute.tf")

        with open(filename, mode='w+', encoding="utf-8") as output:
            output.write(content)
            logger.info("Generated: %s", pathlib.PurePath(filename).name)


if __name__ == '__main__':
    checkDirectories(DEFAULT_INPUT_PATH, DEFAULT_OUTPUT_PATH, DEFAULT_TEMPLATES_PATH)
    logger.info("Using default input directory: %s", DEFAULT_INPUT_PATH)
    files = sorted(pathlib.Path(DEFAULT_INPUT_PATH).glob("*.csv"))

    for f in files:
        logger.info("File found: %s", str(f))            
        renderFile(args, f, DEFAULT_OUTPUT_PATH)