#!/usr/bin/env python2

#
# Copyright(c) 2012-2018 Intel Corporation
# SPDX-License-Identifier: BSD-3-Clause-Clear
#

import commands
import sys
import os

args = ' '.join(sys.argv[1:])
script_path = os.path.dirname(os.path.realpath(__file__))
framework_script_path = script_path + os.sep + "../ut-framework/add_new_test_file.py"
framework_script_path = os.path.normpath(framework_script_path)
status, output = commands.getstatusoutput(framework_script_path + " " + args)

print output

if status == 0:
    path = output.split(" ", 1)[0]
    with open(script_path + os.sep + "header.c", "r") as header_file:
        with open(path, "r+") as source_file:
            source = source_file.readlines()

            source_file.seek(0, os.SEEK_SET)
            source_file.truncate()

            source_file.writelines(header_file.readlines())
            source_file.writelines(source)