#!/usr/bin/python
#
from spf_core import menu
#
# menu.py
# To add a custom module to the menu, import the module here,
# then add it to the menu function below
#

from client_side import client_side
from evil_ussd_code import ussd
from direct_download import direct_download

def remote_menu():

    menu_list = ['Direct Download Agent',
                 'Client Side Shell',
                 'USSD Webpage Attack (Safe)',
                 'USSD Webpage Attack (Malicious)'
                ]

    choice = menu(menu_list, '[*] Choose an Attack to Launch:\n')

    if choice is 0:
        return_code = 0

    elif choice is 1:
        return_code = direct_download()

    elif choice is 2:
        return_code = client_side()

    elif choice is 3:
        return_code = ussd()

    elif choice is 4:
        return_code = ussd()

#    elif choice is my_number:
#        return_code = custom_module()
# etc...

    else:
        return_code = 1

    return return_code

