from c_comment import generateAllComments as c_generateAllComments
from py_comment import generateAllComments as py_generateAllComments
import sys

def fileExists(filename: str) -> bool:
    try:
        with open(filename, 'r'):
            return True
    except FileNotFoundError:
        return False

if __name__ == '__main__':
    if sys.argv[1] == '-h' or sys.argv[1] == "-H" or sys.argv[1] == '--help' or sys.argv[1] == 'help':
        print('ComGen: A comment generator for C and Python files')
        print('Usage: python3 comgen.py [path/to/file1] [path/to/file2] ...')
        sys.exit()
    elif sys.argv[1] == '-v' or sys.argv[1] == "-V" or sys.argv[1] == '--version' or sys.argv[1] == 'version':
        print('ComGen version 1.0.0')
        sys.exit()
    filenames = sys.argv[1:]
    for filename in filenames:
        if not fileExists(filename):
            print('File not found')
            sys.exit()
        extension = filename.split('.')[-1]
        if extension == 'c' or extension == 'h':
            c_generateAllComments(filename)
        elif extension == 'py':
            py_generateAllComments(filename)
        else:
            print('Language not supported')
            sys.exit()