import public.Comment as comments
import public.Scan as scans
import argparse
import time

def pmain():
    print('''
    --
    |       MIMAZ.ORG  CodeFileSCAN  V1.0
            2020 - CMS - FAMILY            |
                                          --
    ''')

def uinput():
    parser = argparse.ArgumentParser(description="CodeFileSCAN  V1.0")
    parser.add_argument("-f", type=str, required=True, help="must need route like: test , do not need \.")
    return parser.parse_args()

def printmsg(msg):
    ptime = time.strftime("%H:%M:%S", time.localtime())
    print('[ {} ] : {}'.format(ptime, msg))

def main():
    printmsg('正在扫描，请耐心等待。。。')
    route = args.f
    X1 = comments.comment('testone')
    X2 = scans.MimazScan('testone')
    #文件原始数组
    FileDic = X1.GetFileAndRoute(route)
    #文件一维数组
    FinArray = []
    for k,v in FileDic.items():
        FinArray.extend(X2.FileScan(k,v))
    #检测数组
    checkArray = ['password=','password:','pass=','pass:','key=','key:','user=','user:']
    printmsg('当前检测规则：{}'.format(checkArray))
    #结果数组
    ResArray = []
    for eFinArray in FinArray:
        #单次扫面结果集
        eresarray = X2.HeartScan(eFinArray, checkArray)
        for eeresarray in eresarray:
            if len(eeresarray[0]) > 0:
                ResArray.append(eeresarray)
    if len(ResArray) == 0:
        printmsg('没有找到任何敏感信息，请检查项目路径，或自定义检测规则。')
    else:
        printmsg('扫描完成！结果集：')
    for eResArray in ResArray:
        printmsg('文件位置：{} 代码行数：{} 触发规则：{}'.format(eResArray[0], eResArray[1], eResArray[2]))

if __name__ == "__main__":
    pmain()
    args = uinput()
    main()