from argparse import RawTextHelpFormatter
from utils.commons import Commons
from utils.banner import generate_banner
from utils.colors import Color
import argparse


def print_colored_message(message, color):
    print(color + message + Color.END)

def main():
    HELPALL = """
             
                        [+] python  main.py -h  
                        [+] python  main.py -x  F12              
                        [+] python  main.py -c  F12
                        [+] python  main.py -all   F12F12
                                     
    """
    parser = argparse.ArgumentParser(description=HELPALL, formatter_class=RawTextHelpFormatter)
    parser.add_argument('-x', action='store_true', help='F12')
    parser.add_argument('-c', action='store_true', help='F12')
    parser.add_argument('-all', action='store_true', help='F12F12')
    args = parser.parse_args()

    if args.x:
       commons.load_wechatEx_configs()
    elif args.c:
        commons.load_wechatEXE_configs()
    elif args.all:
        commons.load_wechatEXE_and_wechatEx()
    else:

        print_colored_message(HELPALL, Color.RED)

if __name__ == "__main__":
    generate_banner()
    commons = Commons()
    main()