import argparse
import os
import uuid

def generate_env(username, password):
    env = []
    compose_location = os.path.join(os.getcwd(), 'services/docker-compose.yml')

    env.append(f'ADMIN_USERNAME={username}')
    env.append(f'ADMIN_PASSWORD={password}')
    env.append(f'COMPOSE_LOCATION={compose_location}')

    credentials = [f'PASSWORD_{(i * 1000) + 10000}={uuid.uuid4().hex}' for i in range(20)]
    env.extend(credentials)
    return '\n'.join(env)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('-u', '--username', type=str, default='kekuatanayam')
    parser.add_argument('-p', '--password', type=str, default='adalahkseimbangan')
    parser.add_argument('-c', '--challenges', type=str, default='', help='challenge names')
    args = parser.parse_args()

    cwd = os.getcwd()
    receiver_env = os.path.join(cwd, 'receiver/.env')
    services_env = os.path.join(cwd, 'services/.env')

    # Only generate .env files if they don't already exist
    if not os.path.exists(receiver_env) or not os.path.exists(services_env):
        env = generate_env(args.username, args.password)
        
        with open(receiver_env, 'w') as f:
            f.write(env)
        
        with open(services_env, 'w') as f:
            f.write(env)
        
        print("Generated new .env files")
    else:
        print("Using existing .env files")

    os.chdir(os.path.join(cwd, 'services'))
    # os.system(f'docker compose -f docker-compose.yml up -d {args.challenges}')
    os.system(f'docker compose -f docker-compose.yml up --build -d {args.challenges}')
    
    os.chdir(os.path.join(cwd, 'receiver'))
    os.system('sudo yum install -y gcc python3-devel gmp-devel openssl-devel libffi-devel')
    os.system('python3 -m venv .')
    os.system('./bin/python3 -m pip install -r requirements.txt')
    os.system('sudo ./bin/python3 -m uvicorn main:app --reload --host 0.0.0.0 --port 80')

if __name__ == '__main__':
    main()