import verticalPermutationCipher as vpc
import substitutionCipher as sc

import os

exit = False


while True:
    os.system("cls" if os.name == "nt" else "clear")
    print(
        "\n==================================:Шифрування/Розшифрування:=================================="
    )
    print(
        "----Використовується \033[1;32;34mпоєднаний шифр\033[0m та \033[1;32;34mшифр вертикальної перестановки\033[0m------------"
    )
    print("\n\nОберіть дію:\n")
    print("1. Шифрування повідомлення;")
    print("2. Розшифрування повідомлення;")
    print("0. Вихід з застосунку.")

    do = input("\n\nОберіть дію: ")

    if do == "1":
        os.system("cls" if os.name == "nt" else "clear")
        while True:
            print("\n=========:ШИФРУВАННЯ:=========")
            print("\n\nЗа яким алгоритмом будемо виконувати шифрування?\n")
            print("1. Шифр вертикальної перестановки;")
            print("2. Поєднаний шифр;")
            print("9. Повернутися назад;")
            print("0. Вихід із застосунку.")

            choice = input("\n\nОбраний алгоритм/дія: ")

            if choice == "1":
                os.system("cls" if os.name == "nt" else "clear")
                print("\n=========:Шифр вертикальної перестановки:=========")
                key = input("\nУвідть ключ: ")
                message = input("Увідть повідомлення: ")

                print("\nВаше зашфироване повідомлення можно скопіювати нижче\n\n")
                print(vpc.encode(message, key))

                input("\n\nНатисніть будь-яку клавішу для продовження...")
                os.system("cls" if os.name == "nt" else "clear")

            elif choice == "2":
                os.system("cls" if os.name == "nt" else "clear")
                print("\n=========:Поєднаний шифр:=========")
                while True:
                    print("\nДля початку оберіть абетку\n")
                    print("1. Українська;")
                    print("2. Англійська;")
                    choice = input("\n\nОберіть абетку: ")

                    if choice == "1":
                        alpabet = "ukr"
                        break
                    elif choice == "2":
                        alpabet = "en"
                        break
                    else:
                        print("\033[31mНе зрозумілий вибір! Оберіть з переліку\033[0m")
                        input("\n\nНатисніть будь-яку клавішу для продовження...")
                        os.system("cls" if os.name == "nt" else "clear")

                key = input("\nУвідть ключ: ")
                message = input("Увідть повідомлення: ")

                print("\nВаше зашфироване повідомлення можно скопіювати нижче\n\n")
                print(sc.encode(message, key, alpabet))

                input("\n\nНатисніть будь-яку клавішу для продовження...")
                os.system("cls" if os.name == "nt" else "clear")

            elif choice == "9":
                os.system("cls" if os.name == "nt" else "clear")
                break
            elif choice == "0":
                os.system("cls" if os.name == "nt" else "clear")
                exit = True
                break
            else:
                print("\n\033[31mНе зрозумілий вибір. Оберіть з переліку\033[0m")
                input("\nНатисніть будь-яку клавішу для продовження...")
                os.system("cls" if os.name == "nt" else "clear")

    elif do == "2":
        os.system("cls" if os.name == "nt" else "clear")
        while True:
            print("\n=========:РОЗШИФРУВАННЯ:=========")
            print("\n\nЗа яким алгоритмом будемо виконувати розшифрування?\n")
            print("1. Шифр вертикальної перестановки;")
            print("2. Поєднаний шифр;")
            print("9. Повернутися назад;")
            print("0. Вихід із застосунку.")

            choice = input("\n\nОбраний алгоритм: ")

            if choice == "1":
                os.system("cls" if os.name == "nt" else "clear")
                print("\n=========:Шифр вертикальної перестановки:=========")
                key = input("\nУвідть ключ: ")
                message = input("Увідть зашифроване повідомлення: ")

                print("\nВаше розшифроване повідомлення можна скопіювати нижче\n\n")
                print(vpc.decode(message, key))

                input("\n\nНатисніть будь-яку клавішу для продовження...")
                os.system("cls" if os.name == "nt" else "clear")

            elif choice == "2":
                os.system("cls" if os.name == "nt" else "clear")
                print("\n=========:Поєднаний шифр:=========")
                while True:
                    print("\nДля початку оберіть абетку\n")
                    print("1. Українська;")
                    print("2. Англійська;")
                    choice = input("\n\nОберіть абетку: ")

                    if choice == "1":
                        alpabet = "ukr"
                        break
                    elif choice == "2":
                        alpabet = "en"
                        break
                    else:
                        print("\033[31mНе зрозумілий вибір! Оберіть з переліку\033[0m")
                        input("\n\nНатисніть будь-яку клавішу для продовження...")
                        os.system("cls" if os.name == "nt" else "clear")

                key = input("\nУвідть ключ: ")
                message = input("Увідть зашифроване повідомлення: ")

                print("\nВаше розшифроване повідомлення можна скопіювати нижче\n\n")
                print(sc.decode(message, key, alpabet))

                input("\n\nНатисніть будь-яку клавішу для продовження...")
                os.system("cls" if os.name == "nt" else "clear")

            elif choice == "9":
                os.system("cls" if os.name == "nt" else "clear")
                break
            elif choice == "0":
                os.system("cls" if os.name == "nt" else "clear")
                exit = True
                break
            else:
                print("\n\033[31mНе зрозумілий вибір. Оберіть з переліку\033[0m")
                input("\nНатисніть будь-яку клавішу для продовження...")
                os.system("cls" if os.name == "nt" else "clear")
    elif do == "0":
        os.system("cls" if os.name == "nt" else "clear")
        print("\nДо зустрічі :)")
        break
    else:
        print("\n\033[31mНе зрозуміла дія. Оберіть з переліку!\033[0m")
        input("\nНатисніть будь-яку клавішу для продовження...")
        os.system("cls" if os.name == "nt" else "clear")

    if exit:
        os.system("cls" if os.name == "nt" else "clear")
        print("\nДо зустрічі :)")
        break