from credentials import aws_access_key_id, aws_secret_access_key, aws_session_token, default_vpc
import boto3

INDEKS = "444018"
key_name = f"{INDEKS}-aws-ssh-key"
security_group_name = f"{INDEKS}-security-group"

user_data = f'''
#!/bin/bash
sudo yum update -y
sudo yum install git -y
git clone https://git.wmi.amu.edu.pl/s444018/DPZC-5.git
cd DPZC-5
sudo yum install docker -y
sudo service docker start
sudo usermod -a -G docker ec2-user
sudo docker build -t invoice .
sudo docker run -d -p 80:8000 -t invoice
'''

if __name__ == '__main__':
    ec2 = boto3.resource(
        'ec2',
        region_name='us-east-1',
        aws_access_key_id=aws_access_key_id,
        aws_secret_access_key=aws_secret_access_key,
        aws_session_token=aws_session_token,
    )

    key_pair = ec2.create_key_pair(
        KeyName=key_name,
        KeyType='ed25519',
        KeyFormat='pem',
    )

    security_group = ec2.create_security_group(
        Description=security_group_name,
        GroupName=security_group_name,
        VpcId=default_vpc,
    )

    inbound_rules = security_group.authorize_ingress(
        GroupId=security_group.group_id,
        CidrIp='0.0.0.0/0',
        IpProtocol='tcp',
        FromPort=80,
        ToPort=80,
    )

    instance = ec2.create_instances(
        ImageId='ami-0b5eea76982371e91',
        MinCount=1,
        MaxCount=1,
        InstanceType='t2.micro',
        KeyName=key_pair.name,
        UserData=user_data,
        SecurityGroups=[security_group.group_name],
    )