#!/bin/bash
source ~/.bashrc
source logger.sh
source utils.sh

if [[ -z $AWS_CLOUD_GAMING_PROFILE ]]; then
	AWS_CLOUD_GAMING_PROFILE=personal
fi

checkAwsProfile() {
	printInfo "Checking for the $AWS_CLOUD_GAMING_PROFILE AWS Profile in AWS config"
  if ! (aws --profile $AWS_CLOUD_GAMING_PROFILE sts get-caller-identity --no-cli-auto-prompt > /dev/null 2>&1); then
  	printError "AWS config is missing the $AWS_CLOUD_GAMING_PROFILE profile. Add it to your ~/.aws/config file and try running this application again"
  	exit 1
  fi
}

getInstanceState() {
  aws --profile $AWS_CLOUD_GAMING_PROFILE ec2 describe-instances --instance-ids "$AWS_TEAM_BUILDING_EC2_INSTANCE_ID" --query 'Reservations[0].Instances[0].State.Name' --no-cli-auto-prompt --output text &
}

showGaugeBoxForAwsCommand() {
  declare increment
  declare i
  increment=$(echo "scale=1; 100 / 30" | bc)
  i="0"

  printInfo "$2"

  {
    while [[ $(getInstanceState) != "$1" ]]; do
      declare percent
      percent=$(printf "%.0f" $i)
      echo -e "XXX\n$percent\n$2... \nXXX"
      i=$(echo "$i + $increment" | bc)
      sleep 5
    done

    if [[ $(getInstanceState) != "$1" ]]; then
    	printError "$4"
      echo -e "XXX\n0\n$4\nXXX"
      return 1
    else
      echo -e "XXX\n100\n$3\nXXX"
      sleep 1
    fi
  } | whiptail --title "$2..." --gauge "$2..." "$GAUGE_BOX_HEIGHT" "$GAUGE_BOX_WIDTH" 0

  printInfo "$3"
}

waitForInstanceToBecomeAvailable() {
	printInfo "Waiting for instance to become available"

	{
		declare increment
		increment=$(echo "scale=1; 100/90" | bc)
		for ((i=0; i<=100; i=$(printf "%.0f" $(echo "scale=1; $i + $increment" | bc)))); do
			if (timeout 10s nc -vz "$AWS_TEAM_BUILDING_EC2_INSTANCE_IP" 8443); then
				break
			fi
			echo "$i"
		done
		echo 100
		sleep 1
	} | whiptail --title "Instance Startup" --gauge "Waiting for instance to become available..." "$GAUGE_BOX_HEIGHT" "$GAUGE_BOX_WIDTH" 0
}

getInstanceIp() {
	printInfo "Fetching instance IP for id: $AWS_TEAM_BUILDING_EC2_INSTANCE_ID"
	aws --profile $AWS_CLOUD_GAMING_PROFILE ec2 describe-instances --instance-ids "$AWS_TEAM_BUILDING_EC2_INSTANCE_ID" --query "Reservations[0].Instances[0].PublicIpAddress" --output text --no-cli-auto-prompt
}

createDcvConnectionProfileFromTemplate() {
	printInfo "Creating DCV connection profile from template"
	PASSWORD="$(aws --profile $AWS_CLOUD_GAMING_PROFILE ec2 get-password-data --instance-id "$AWS_TEAM_BUILDING_EC2_INSTANCE_ID" --priv-launch-key ~/.ssh/insights-team-building-key-pair.pem --query 'PasswordData' --output text --no-cli-auto-prompt)"
  PASSWORD=$(echo -n $PASSWORD)
	sed -i "/^host=/c\host=$AWS_TEAM_BUILDING_EC2_INSTANCE_IP" cloud_gaming_dcv_profile.dcv
	sed -i "/^password=/c\password=$PASSWORD" cloud_gaming_dcv_profile.dcv
}

startInstance() {
  declare state
  declare desiredState=running
  state=$(getInstanceState)
  printInfo "Starting instance"

  if [[ $state == "$desiredState" ]]; then
  	printError "Instance is already running. Doing nothing"
    msgBox "Instance is already running."
  else
  	declare instanceIp
    aws --profile $AWS_CLOUD_GAMING_PROFILE ec2 start-instances --instance-ids "$AWS_TEAM_BUILDING_EC2_INSTANCE_ID" --no-cli-auto-prompt > /dev/null 2>&1 &
    showGaugeBoxForAwsCommand "$desiredState" "Starting Your Instance" "Successfully Started Your Instance!" "Failed to start your instance!"
    printInfo "Checking to see if IP changed"
  	instanceIp=$(getInstanceIp)
  	if [[ $instanceIp != $AWS_TEAM_BUILDING_EC2_INSTANCE_IP ]]; then
  		setConfigValue "AWS_TEAM_BUILDING_EC2_INSTANCE_IP" "$instanceIp"
  		createDcvConnectionProfileFromTemplate
  	fi

    waitForInstanceToBecomeAvailable

  fi
}

stopInstance() {
  declare state
  declare desiredState=stopped
  state=$(getInstanceState)
  printInfo "Stopping instance"

  if [[ $state == "$desiredState" ]]; then
  	printError "Instance is already stopped."
    msgBox "Instance is already stopped."
  else
    aws --profile $AWS_CLOUD_GAMING_PROFILE ec2 stop-instances --instance-ids "$AWS_TEAM_BUILDING_EC2_INSTANCE_ID" --no-cli-auto-prompt > /dev/null 2>&1 &
    showGaugeBoxForAwsCommand "$desiredState" "Stopping Your Instance" "Successfully Stopped Your Instance!" "Failed to stop your instance!"
  fi
}

rebootInstance() {
  declare desiredState=running

  printInfo "Rebooting instance"

  aws --profile $AWS_CLOUD_GAMING_PROFILE ec2 reboot-instances --instance-ids "$AWS_TEAM_BUILDING_EC2_INSTANCE_ID" --no-cli-auto-prompt > /dev/null 2>&1 &

  if ! (showGaugeBoxForAwsCommand "$desiredState" "Restarting Your Instance" "Successfully Restarted Your Instance!"); then
  	printError "Failed to restart instance. Waiting for user to manually start instance before continuing."
    msgBox "$(cat <<-EOF
			Failed to restart your instance! Please make sure your instance is started before continuing!

			Your instance ID is $AWS_TEAM_BUILDING_EC2_INSTANCE_ID

			Hit 'OK' Once your instance is started
		EOF
		)"
  fi
}

getMyIp() {
	curl -s -L -X GET http://checkip.amazonaws.com
}

deployCdk() {
	printInfo "Deploying CDK"

  cd cdk

  declare user
  declare localIp
  declare logFile=/tmp/cdk
  user="$(whoami)"
	localIp="$(getMyIp)"

  {
    echo -e "XXX\n0\nRunning npm install... \nXXX"
    printInfo "Running npm install"
    npm install > /dev/null 2>&1
    echo -e "XXX\n50\nBuilding CDK... \nXXX"
    printInfo "Running npm run build"
    npm run build > /dev/null 2>&1
    echo -e "XXX\n100\nDone! \nXXX"
    sleep 1
  } | whiptail --title "Preparing CDK..." --gauge "Preparing CDK..." "$GAUGE_BOX_HEIGHT" "$GAUGE_BOX_WIDTH" 0

  declare pid
  declare synthLogFile="${logFile}-synth.log"
  printInfo "Running CDK synth and logging to $synthLogFile"
  yes | npx cdk --no-color --require-approval never --profile $AWS_CLOUD_GAMING_PROFILE -c "user=$user" -c "localIp=$localIp" synth "TeamBuildingCloudGaming-$user" > $synthLogFile 2>&1 &
  pid=$!
  showTailBox "Synthesizing CDK" $pid $synthLogFile

  declare bootstrapLogFile="${logFile}-bootstrap.log"
  printInfo "Bootstrapping CDK and logging to $bootstrapLogFile"
  yes | npx cdk --no-color --require-approval never --profile $AWS_CLOUD_GAMING_PROFILE -c "user=$user" -c "localIp=$localIp" bootstrap > $bootstrapLogFile 2>&1 &
  pid=$!
  showTailBox "Bootstrapping CDK" $pid $bootstrapLogFile

  declare deployLogFile="${logFile}-deploy.log"
  printInfo "Deploying CDK and logging to $deployLogFile"
  yes | npx cdk --no-color --require-approval never --profile $AWS_CLOUD_GAMING_PROFILE -c "user=$user" -c "localIp=$localIp" deploy "TeamBuildingCloudGaming-$user" > $deployLogFile 2>&1 &
  pid=$!
  showTailBox "Deploy EC2 Instance" $pid $deployLogFile

  unset AWS_TEAM_BUILDING_EC2_INSTANCE_ID
  unset AWS_TEAM_BUILDING_EC2_INSTANCE_IP

  AWS_TEAM_BUILDING_EC2_INSTANCE_ID=$(cat $deployLogFile | grep InstanceId | awk '{print $NF;}')
  setConfigValue "AWS_TEAM_BUILDING_EC2_INSTANCE_ID" "$AWS_TEAM_BUILDING_EC2_INSTANCE_ID"

  AWS_TEAM_BUILDING_EC2_INSTANCE_IP=$(getInstanceIp)
  setConfigValue "AWS_TEAM_BUILDING_EC2_INSTANCE_IP" "$AWS_TEAM_BUILDING_EC2_INSTANCE_IP"

  cd ..

  waitForInstanceToBecomeAvailable
	rebootInstance
	waitForInstanceToBecomeAvailable
}

connectToInstanceViaDcvViewer() {
	printInfo "Connecting to instance desktop via DCV Viewer"

	if ! (hash dcvviewer 2> /dev/null); then
		printError "dcvviewer is not installed. Cannot connect to personal instance until first time setup is run."
		msgBox "Can't connect to personal instance via DCV Viewer without first time setup. Run the deploy instance task from the instance management menu first!"
	fi

	if (pgrep -f dcvviewer); then
		printError "DCV Viewer is already running."
		msgBox "DCV Viewer is already running"
		return 0
	fi

	declare state
	state=$(getInstanceState)
	if [[ $state != "running" ]]; then
		if (whiptail --fb --title "Start your instance?" --yesno "In order to stream, you instance needs to be started. Would you like to start your personal instance?" "$BOX_BOX_HEIGHT" "$BOX_WIDTH"); then
			startInstance
		else
			printError "Unable to start desktop connection: Instance is not running"
			msgBox "Can't start desktop connection! Instance is not running. You can start the instance from the personal Instance Management menu."
			return 0
		fi
	fi

	dcvviewer cloud_gaming_dcv_profile.dcv --certificate-validation-policy=accept-untrusted > /dev/null 2>&1 &
}