from InquirerPy import inquirer
from . import Startup_Checks
from . import Decrypt
from . import CDRM_Keys_API

def main_menu():
    local_cdms, remote_cdms = Startup_Checks.check_for_wvds()
    choices = []
    if local_cdms:
        choices.append('Local CDM Decryption')
        choices.append('CDRM-Keys API Mode')
    if remote_cdms:
        choices.append('Remote CDM Decryption')
    choice = inquirer.select(
        message='CDRM-Keys:',
        choices=choices,
    ).execute()
    if choice == 'Local CDM Decryption':
        Decrypt.decrypt_local_cdm()
    if choice == 'Remote CDM Decryption':
        Decrypt.decrypt_remote_cdm()
    if choice == 'CDRM-Keys API Mode':
        CDRM_Keys_API.start_cdrm_keys_api()