package com.amazonaws.samples;

import java.io.Console;
import java.io.IOException;
import java.net.ConnectException;
import java.net.SocketTimeoutException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.List;
import java.util.Scanner;
import java.util.concurrent.TimeUnit;

import com.jcraft.jsch.*;

import net.schmizz.sshj.SSHClient;
import net.schmizz.sshj.common.IOUtils;
import net.schmizz.sshj.connection.channel.direct.Session;
import net.schmizz.sshj.connection.channel.direct.Session.Command;
import software.amazon.awssdk.services.ec2.Ec2Client;
import software.amazon.awssdk.services.ec2.model.AuthorizeSecurityGroupIngressRequest;
import software.amazon.awssdk.services.ec2.model.AuthorizeSecurityGroupIngressResponse;
import software.amazon.awssdk.services.ec2.model.CreateKeyPairRequest;
import software.amazon.awssdk.services.ec2.model.CreateKeyPairResponse;
import software.amazon.awssdk.services.ec2.model.CreateSecurityGroupRequest;
import software.amazon.awssdk.services.ec2.model.CreateSecurityGroupResponse;
import software.amazon.awssdk.services.ec2.model.DeleteKeyPairRequest;
import software.amazon.awssdk.services.ec2.model.DeleteKeyPairResponse;
import software.amazon.awssdk.services.ec2.model.DeleteSecurityGroupRequest;
import software.amazon.awssdk.services.ec2.model.DeleteSecurityGroupResponse;
import software.amazon.awssdk.services.ec2.model.DescribeInstancesRequest;
import software.amazon.awssdk.services.ec2.model.DescribeInstancesResponse;
import software.amazon.awssdk.services.ec2.model.Instance;
import software.amazon.awssdk.services.ec2.model.InstanceType;
import software.amazon.awssdk.services.ec2.model.IpPermission;
import software.amazon.awssdk.services.ec2.model.IpRange;
import software.amazon.awssdk.services.ec2.model.Reservation;
import software.amazon.awssdk.services.ec2.model.RunInstancesRequest;
import software.amazon.awssdk.services.ec2.model.RunInstancesResponse;
import software.amazon.awssdk.services.ec2.model.StartInstancesRequest;
import software.amazon.awssdk.services.ec2.model.StartInstancesResponse;
import software.amazon.awssdk.services.ec2.model.StopInstancesRequest;
import software.amazon.awssdk.services.ec2.model.TerminateInstancesRequest;
import software.amazon.awssdk.services.ec2.model.TerminateInstancesResponse;



public class StartVM {
    static final String vpc_id ="vpc-bdaf5dd7";

    public static void CreateSecurityGroup(String name, String desc, Ec2Client ec2){
        CreateSecurityGroupResponse createResponse = ec2.createSecurityGroup(CreateSecurityGroupRequest.builder().groupName(name).description(desc).vpcId(vpc_id).build());
    }

    public static void IpRangePermissionAndSecurityGroupIngressAuthorization (Ec2Client ec2, String securityGroupName) {
        IpRange ip_range = IpRange.builder().cidrIp("0.0.0.0/0").build();
        IpPermission ip_perm = IpPermission.builder().ipProtocol("tcp").toPort(80).fromPort(80).ipRanges(ip_range).build();
        IpPermission ip_perm2 = IpPermission.builder().ipProtocol("tcp").toPort(22).fromPort(22).ipRanges(ip_range).build();

        AuthorizeSecurityGroupIngressRequest auth_request = AuthorizeSecurityGroupIngressRequest.builder().groupName(securityGroupName).ipPermissions(ip_perm, ip_perm2).build();
        ec2.authorizeSecurityGroupIngress(auth_request);
    }

    public static String createSshSecurityGroup(Ec2Client ec2, String name) {
        CreateSecurityGroupResponse createResponse = ec2.createSecurityGroup(CreateSecurityGroupRequest.builder().groupName(name).description(name).vpcId(vpc_id).build());

        IpRange anyIpv4 = IpRange.builder().cidrIp("0.0.0.0/0").build();

        IpPermission allowSsh = IpPermission.builder()
            .ipProtocol("tcp")
            .fromPort(22)
            .toPort(22)
            .ipRanges(anyIpv4)
            .build();

        ec2.authorizeSecurityGroupIngress(AuthorizeSecurityGroupIngressRequest.builder().groupId(createResponse.groupId()).ipPermissions(allowSsh).build());

        return createResponse.groupId();
    }

    public static String CreateKeyPair (String kpname, Ec2Client ec2) throws IOException {
        CreateKeyPairResponse response = ec2.createKeyPair(CreateKeyPairRequest.builder().keyName(kpname).build());
        return Files.write(Paths.get(kpname + ".pem"), response.keyMaterial().getBytes()).toString();
    }

    public static void deleteKeyPair (Ec2Client ec2, String kpname) throws IOException {
        DeleteKeyPairResponse response = ec2.deleteKeyPair(DeleteKeyPairRequest.builder().keyName(kpname).build());
        Files.delete(Paths.get(kpname + ".pem"));
    }

    public static RunInstancesResponse RunInstances (Ec2Client ec2, String ami_id, String kpname) {
        RunInstancesRequest request = RunInstancesRequest.builder().imageId(ami_id).keyName(kpname).instanceType(InstanceType.T2_MICRO).maxCount(1).minCount(1).build();
        RunInstancesResponse response = ec2.runInstances(request);
        return response;
    }

    public static void startInstances (Ec2Client ec2, String instance_id) {
        StartInstancesRequest StartRequest = StartInstancesRequest.builder().instanceIds(instance_id).build();
        StartInstancesResponse response = ec2.startInstances(StartRequest);
    }



    public static void deleteSecurityGroup (Ec2Client ec2, String SecurityGroupName) {
        DeleteSecurityGroupRequest delSecGroupRequest = DeleteSecurityGroupRequest.builder().groupName(SecurityGroupName).build();
        DeleteSecurityGroupResponse delSGResponse = ec2.deleteSecurityGroup(delSecGroupRequest);
    }

    public static void terminateInstance (Ec2Client ec2, String instance_id) {
        ec2.terminateInstances(TerminateInstancesRequest.builder().instanceIds(instance_id).build());
    }

    public static String instanceIpv4Address (Ec2Client ec2) {
        String nextToken = null;
        String ipv4;
        DescribeInstancesRequest request = DescribeInstancesRequest.builder().maxResults(6).nextToken(nextToken).build();
        ec2.describeInstances(request);
        for (Reservation r : ec2.describeInstances().reservations()) {
            for (Instance i : r.instances()) {
                return ipv4 = i.publicIpAddress();
            }
        }
        return null;
    }

    private static final Console con = System.console();

    public static void main(String[] args) throws IOException {
        Scanner scanner = new Scanner(System.in);
        String ami_id = "ami-010fae13a16763bb4";
        String kpname = "Key Pair Name_test";
        String name = "name_test";
        String desc = "SecurityGroupDesc";
        String user = "ec2-user";

        Ec2Client ec2 = Ec2Client.create();


        CreateSecurityGroup(name, desc, ec2);
        IpRangePermissionAndSecurityGroupIngressAuthorization(ec2, name);
        String keyFilePath = CreateKeyPair(kpname, ec2);
        RunInstancesResponse riresponse = RunInstances(ec2, ami_id, kpname);
        String instance_id = riresponse.instances().get(0).instanceId();
        startInstances(ec2, instance_id);

        System.out.println("wait 40 seconds till instance has started, then enter a new line");
        scanner.nextLine();

        //ssh
        final SSHClient ssh = new SSHClient();
        ssh.loadKnownHosts();
        ssh.connect(instanceIpv4Address(ec2));
        Session session = null;
        try {
            ssh.authPublickey(user, keyFilePath);
            session = ssh.startSession();
            final Command cmd = session.exec("sudo apt-get install openjdk-8-jre openjfx");
            con.writer().print(IOUtils.readFully(cmd.getInputStream()).toString());
            cmd.join(5, TimeUnit.SECONDS);
            con.writer().print("\n** exit status: " + cmd.getExitStatus());
        } finally {
            try {
                if (session != null) {
                    session.close();
                }
            } catch (IOException e) {
                // Do Nothing
            }

            ssh.disconnect();
            ssh.close();
        }

        System.out.println("waiting for newline to terminate the instance.");
        scanner.nextLine();
        try {}
        finally {
            terminateInstance (ec2, instance_id);
            deleteKeyPair(ec2, kpname);
            deleteSecurityGroup(ec2, name);
        }

        System.out.println("Instance shutdown initiated.");

    }
}