from datetime import datetime
import glob
import os
import time
import xml.etree.ElementTree as ET


def generate():
    BODY = {}

    ts = time.time()
    formatted_ts = datetime.fromtimestamp(ts).strftime("%Y-%m-%dT%H:%M:%S.%fZ")
    BODY["@timestamp"] = formatted_ts

    # Plots are obtained from csv files ( in archives directory in $WORKSPACE).

    csv_files = glob.glob("archives/*.csv")
    BODY["project"] = "opendaylight"
    BODY["subject"] = "test"

    # If there are no csv files, then it is a functional test.
    # Parse csv files and fill perfomance parameter values

    if len(csv_files) == 0:
        BODY["test-type"] = "functional"
    else:
        BODY["test-type"] = "performance"
        BODY["plots"] = {}
        for f in csv_files:
            key = (f.split("/")[-1])[:-4]
            BODY["plots"][key] = {}
            with open(f) as file:
                lines = file.readlines()
            props = lines[0].strip().split(",")
            vals = lines[1].strip().split(",")
            for i in range(len(props)):
                BODY["plots"][key][props[i]] = float(vals[i])

    # Fill the required parameters whose values are obtained from environment.

    BODY["jenkins-silo"] = os.environ["SILO"]
    BODY["test-name"] = os.environ["JOB_NAME"]
    BODY["test-run"] = int(os.environ["BUILD_NUMBER"])

    # Parsing robot log for stats on start-time, pass/fail tests and duration.

    robot_log = os.environ["WORKSPACE"] + "/output.xml"
    tree = ET.parse(robot_log)
    BODY["id"] = "{}-{}".format(os.environ["JOB_NAME"], os.environ["BUILD_NUMBER"])
    BODY["start-time"] = tree.getroot().attrib["generated"]
    BODY["pass-tests"] = int(tree.getroot().find("statistics")[0][1].get("pass"))
    BODY["fail-tests"] = int(tree.getroot().find("statistics")[0][1].get("fail"))
    endtime = tree.getroot().find("suite").find("status").get("endtime")
    starttime = tree.getroot().find("suite").find("status").get("starttime")
    elap_time = datetime.strptime(endtime, "%Y%m%d %H:%M:%S.%f") - datetime.strptime(
        starttime, "%Y%m%d %H:%M:%S.%f"
    )
    BODY["duration"] = str(elap_time)

    return BODY