def connect(file_name, environment_name):
config = load_config(file_name)
environment_websocket = None
for environment in config['environments']:
if environment['name'] == environment_name:
master = Master(
environment_name,
C2Manager(
Factory(PROVIDER_AWS_WEBSOCKETS,
{'url': environment['url']}),
environment['master-password'], environment_name))
master.show_c2_menu()
if environment_websocket is None:
print(
'[-] "{}" environment not found in the "{}" configuration file. Do you want to sync "{}" configuration file from AWS? Y/n'
.format(environment_name, file_name, file_name))
user_input = input()
if user_input.lower() == 'y' or user_input == '':
sync_environments(file_name)
config = load_config(file_name)
for environment in config['environments']:
if environment['name'] == environment_name:
master = Master(
environment_name,
C2Manager(
Factory(PROVIDER_AWS_WEBSOCKETS,
{'url': environment['url']}),
environment['master-password'], environment_name))
master.show_c2_menu()
print(
'[-] {} environment not found in the {} configuration file. Add it manually to config file or create a new environment with -d (--deploy) option'
.format(environment_name, file_name))
