#!/usr/bin/env python3
#####################################################################################
#  The MIT License (MIT)
#
#  Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
#
#  Permission is hereby granted, free of charge, to any person obtaining a copy
#  of this software and associated documentation files (the "Software"), to deal
#  in the Software without restriction, including without limitation the rights
#  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
#  copies of the Software, and to permit persons to whom the Software is
#  furnished to do so, subject to the following conditions:
#
#  The above copyright notice and this permission notice shall be included in
#  all copies or substantial portions of the Software.
#
#  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
#  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
#  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
#  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
#  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
#  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
#  THE SOFTWARE.
#####################################################################################
import subprocess
import sys

debug = False
# The filetypes we want to check for that are stamped
# LICENSE is included here as it SHOULD have a liscence in it otherwise flag it as unstamped
supported_file_types = (".cpp", ".hpp", ".h", ".ipynb", ".py", ".txt", ".sh",
                        ".bsh", "LICENSE", ".cmake")

#add general stuff we shouldn't stamp and any exceptions here
unsupported_file_types = [
    ".onnx", ".pb", ".rst", ".jpg", ".jpeg", ".proto", ".md", ".clang",
    ".weight", ".ini", ".json", ".docker", ".git", ".rules", ".yml"
]

specificIgnores = ("digits.txt", "Dockerfile", "Jenkinsfile", "")


def hasKeySequence(inputfile, key_message):
    result = False
    if key_message in inputfile:
        result = True
    return result


#Simple just open and write stuff to each file with the license stamp
def openAndCheckFile(filename):
    result = False
    #open save old contents and append things here
    if debug is True:
        print("Open", filename, end='')

    try:
        file = open(filename, 'r')
    except OSError as e:
        if debug is True:
            print(str(e) + "....Open Error: Skipping  file ")
        file.close()
        return
    else:
        with file as contents:
            try:
                save = contents.read()
                hasAmdLic = hasKeySequence(
                    save, "Advanced Micro Devices, Inc. All rights reserved")

                #Check if we have a licence stamp already
                if hasAmdLic is True:
                    if debug is True:
                        print("....Already Stamped: Skipping  file ")
                    contents.close()
                    result = True

            except UnicodeDecodeError as eu:
                if debug is True:
                    print(str(eu) + "...Skipping binary file ")
                contents.close()
                result = True

    return result


# Deterine if filename is desired in the fileTuple past in
def check_filename(filename, fileTuple):
    supported = False
    for key in fileTuple:
        if key in filename:
            supported = True
            break
    return supported


def main():
    unsupported_file_types.extend(specificIgnores)

    #Get a list of all the tracked files in our git repo
    proc = subprocess.run("git ls-files --exclude-standard",
                          shell=True,
                          stdout=subprocess.PIPE)
    fileList = proc.stdout.decode().split('\n')

    if debug is True:
        print("Target file list:\n" + str(fileList))

    unsupportedFiles = []
    unstampedFiles = []
    unknownFiles = []

    for file in fileList:
        supported = check_filename(file, supported_file_types)
        if supported is True:
            isStamped = openAndCheckFile(file)
            if isStamped is False:
                unstampedFiles.append(file)

        else:
            unsupported = check_filename(file, unsupported_file_types)
            if unsupported is True:
                unsupportedFiles.append(file)
            else:
                unknownFiles.append(file)

    #Do a bunch of checks based on our file lists
    if len(unstampedFiles) > 0:
        print("Error: The following " + str(len(unstampedFiles)) +
              " files are currently without a license:")
        print(str(unstampedFiles))
        sys.exit(1)

    if len(unknownFiles) > 0:
        print("Error: The following " + str(len(unknownFiles)) +
              " files not handled:")
        print(str(unknownFiles))
        sys.exit(2)

    sys.exit(0)


if __name__ == "__main__":
    main()