Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

smoother up (fixes #57) #59

Merged
merged 11 commits into from
Aug 3, 2024
Merged

smoother up (fixes #57) #59

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
8f14681
feat: Add checkKeyName function to prevent the script from asking to …
hiroTochigi Jul 25, 2024
a9d1637
fix variable name
hiroTochigi Jul 25, 2024
83b4730
camel case
hiroTochigi Jul 25, 2024
fdf58db
variable name
hiroTochigi Jul 25, 2024
7b31854
Enhance main.sh to check existing AWS resources before creation
hiroTochigi Aug 1, 2024
18bcd11
fix syntax error in up function by adding missing closing fi
hiroTochigi Aug 1, 2024
4fc81d6
fix the function name
hiroTochigi Aug 1, 2024
8ca0b8a
Enhance up function to handle terminated EC2 instances
hiroTochigi Aug 1, 2024
c48491d
feat: ignore the terminated states currently
hiroTochigi Aug 1, 2024
306d7f0
Update up.sh
dogi Aug 1, 2024
9f5adff
Update up.sh
dogi Aug 3, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
294 changes: 154 additions & 140 deletions src/aws/up.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,175 +10,189 @@ groupName=luftballons-sg
instanceName=luftballon
checkSSH=~/.ssh/$publickey

checkSshKey() {
aws ec2 describe-key-pairs --key-names $keyname &> /dev/null
return $?
}

checkSecurityGroup() {
aws ec2 describe-security-groups --group-names $groupName &> /dev/null
return $?
}

checkInstance() {
aws ec2 describe-instances --filters "Name=tag:Name,Values=$instanceName" --query "Reservations[*].Instances[*].InstanceId" --output text
}

checkInstanceState() {
ID=$1
aws ec2 describe-instances --instance-ids $ID --query "Reservations[*].Instances[*].State.Name" --output text
}

function importSshKey()
{
if [[ -f ~/.ssh/$publickey ]]
then
aws ec2 import-key-pair --key-name "$keyname" --public-key-material fileb://~/.ssh/$publickey
else
echo 'ssh key pair (~/.ssh/$publickey) do not exist ~/.ssh/$publickey'
echo 'Please generate the ssh key by the commad "ssh-keygen -t rsa"'
exit 1
fi
if [[ -f ~/.ssh/$publickey ]]; then
aws ec2 import-key-pair --key-name "$keyname" --public-key-material fileb://~/.ssh/$publickey
else
echo 'ssh key pair (~/.ssh/$publickey) do not exist ~/.ssh/$publickey'
echo 'Please generate the ssh key by the commad "ssh-keygen -t rsa"'
exit 1
fi
}

function addPort(){
aws ec2 authorize-security-group-ingress \
--group-name $groupName \
--protocol tcp \
--port $1 \
--cidr 0.0.0.0/0
aws ec2 authorize-security-group-ingress \
--group-name $groupName \
--protocol tcp \
--port $1 \
--cidr 0.0.0.0/0
}

function addUDPPort() {
aws ec2 authorize-security-group-ingress \
--group-name $groupName \
--protocol udp \
--port $1 \
--cidr 0.0.0.0/0
aws ec2 authorize-security-group-ingress \
--group-name $groupName \
--protocol udp \
--port $1 \
--cidr 0.0.0.0/0
}


function createSecurityGroups(){
aws ec2 create-security-group \
--group-name $groupName \
--description "luftballons security group"

if [ -z "$portConfigArray" ]
then
portConfigArray="8080:80,8443:443,2022:22"
fi

portArray=($(makePortArray "$portConfigArray"))

for i in "${portArray[@]}"
do
addPort $i
echo $i
done

if [ -z "$udpPortConfigArray" ]
then
udpPortConfigArray="1194"
fi

portArray=($udpPortConfigArray)

for i in "${portArray[@]}"
do
addUDPPort $i
echo $i
done
aws ec2 create-security-group \
--group-name $groupName \
--description "luftballons security group"
if [ -z "$portConfigArray" ]; then
portConfigArray="8080:80,8443:443,2022:22"
fi
portArray=($(makePortArray "$portConfigArray"))
for i in "${portArray[@]}"; do
addPort $i
echo $i
done
if [ -z "$udpPortConfigArray" ]; then
udpPortConfigArray="1194"
fi
portArray=($udpPortConfigArray)
for i in "${portArray[@]}"; do
addUDPPort $i
echo $i
done
}

function createEc2(){
image="ami-0750fb43a63427eff"
#image="ami-01e5ff16fd6e8c542"
aws ec2 run-instances \
--count 1 \
--image-id $image \
--instance-type t2.micro \
--key-name $keyname \
--security-groups $groupName
image="ami-0750fb43a63427eff"
#image="ami-01e5ff16fd6e8c542"
aws ec2 run-instances \
--count 1 \
--image-id $image \
--instance-type t2.micro \
--key-name $keyname \
--security-groups $groupName
}

function findData(){
keyWord=$1
grep $keyWord | awk -F':' '{ print $2 }' | sed 's/ //g; s/"//g; s/,//g'
keyWord=$1
grep $keyWord | awk -F':' '{ print $2 }' | sed 's/ //g; s/"//g; s/,//g'
}

function deleteKeyword(){
keyWord=$1
sed "s/$keyWord//g; s/ //g"
keyWord=$1
sed "s/$keyWord//g; s/ //g"
}

function getValueByKeyword(){
keyWord=$1
findData $keyWord | deleteKeyword $keyWord
keyWord=$1
findData $keyWord | deleteKeyword $keyWord
}

function usage {
echo "script usage: $(basename \$0 aws up) [-n ssh key name] [-p] [-a change key name, instance name, and group name]" >&2
echo 'Start Luftballon.'
echo ' -n Change SSH key name on AWS'
echo ' -a Change SSH key name, instance name, and group name'
echo ' -p Use stored port Numbers instead of the default port number.'
exit 1
echo "script usage: $(basename \$0 aws up) [-n ssh key name] [-p] [-a change key name, instance name, and group name]" >&2
echo 'Start Luftballon.'
echo ' -n Change SSH key name on AWS'
echo ' -a Change SSH key name, instance name, and group name'
echo ' -p Use stored port Numbers instead of the default port number.'
exit 1
}

function up {
while getopts 'n:pN:a:' OPTION; do
case "$OPTION" in
n)
keyname=$OPTARG
;;
p)
portConfigArray=$(getArrayValueAsStringByKey $instanceName tcpPortArray)
udpPortConfigArray=$(getArrayValueAsStringByKey $instanceName udpPortArray)
if [ -z "$portConfigArray" ]
then
echo "There is no stored port numbers. The default port numbers are used"
fi
if [ -z "$udpPortConfigArray" ]
then
echo "There is no stored udp port numbers. The default port numbers are used"
fi
;;
a)
groupName=$OPTARG-sg
instanceName=$OPTARG
keyname=$OPTARG
;;
?)
usage
;;
esac
done
shift "$(($OPTIND -1))"

aws --version || ( echo "Run './installAwsCli.sh' first. AWS CLI is not installed." && exit 1 )

if test ! -f "$checkSSH"; then
echo "Run 'ssh-keygen' first, with an empty passphrase for no passphrase. Missing ssh key." && exit 1
fi

if [ -z $keyname ]
then
keyname=luftballon
fi


keyName=$(importSshKey | getValueByKeyword KeyName )

if [ -z $keyName ]
then
exit 1
fi

echo "Success to add ssh key: $keyName"

createSecurityGroups
echo "Add security group"

instanceId=$(createEc2 | getValueByKeyword InstanceId )
echo "Create EC2 Instance"
echo "Instance id is $instanceId"


aws ec2 create-tags --resources $instanceId --tags Key=Name,Value=$instanceName
aws ec2 create-tags --resources $instanceId --tags Key=Class,Value=treehouses


publicIp=$(waitForOutput "getLatestIpAddress $instanceId")
echo "Public IP Address is $publicIp"

echo "Will open ssh tunnel soon"
isOpen=$(waitForOutput "ssh-keyscan -H $publicIp | grep ecdsa-sha2-nistp256")
echo "Opened ssh tunnel"

openSSHTunnel $instanceName $publicIp $portConfigArray

storeConfigIntoTreehousesConfigAsStringfiedJson $instanceName $keyName $instanceId $publicIp $groupName
while getopts 'n:pN:a:' OPTION; do
case "$OPTION" in
n)
keyname=$OPTARG
;;
p)
portConfigArray=$(getArrayValueAsStringByKey $instanceName tcpPortArray)
udpPortConfigArray=$(getArrayValueAsStringByKey $instanceName udpPortArray)
if [ -z "$portConfigArray" ]; then
echo "There is no stored port numbers. The default port numbers are used"
fi
if [ -z "$udpPortConfigArray" ]; then
echo "There is no stored udp port numbers. The default port numbers are used"
fi
;;
a)
groupName=$OPTARG-sg
instanceName=$OPTARG
keyname=$OPTARG
;;
?)
usage
;;
esac
done
shift "$(($OPTIND -1))"

aws --version || ( echo "Run './installAwsCli.sh' first. AWS CLI is not installed." && exit 1 )

if test ! -f "$checkSSH"; then
echo "Run 'ssh-keygen' first, with an empty passphrase for no passphrase. Missing ssh key." && exit 1
fi

if [ -z $keyname ]; then
keyname=luftballon
fi

if ! checkSshKey ; then
importedKeyName=$(importSshKey | getValueByKeyword KeyName )
if [ -z $importedKeyName ]; then
exit 1
fi
echo "Success to add ssh key: $importedKeyName"
else
echo "The key pair $keyname already exists. Please use another key name."
fi

if ! checkSecurityGroup; then
createSecurityGroups
echo "Add security group"
# Add rules to Security Group as needed
else
echo "Security Group already exists."
fi

instanceId=$(checkInstance)
if [ -z "$instanceId" ]; then
instanceState=""
instanceId=$(createEc2 | getValueByKeyword InstanceId)
echo "Creating and running EC2 instance..."
echo "Instance id is $instanceId"
aws ec2 create-tags --resources $instanceId --tags Key=Name,Value=$instanceName
aws ec2 create-tags --resources $instanceId --tags Key=Class,Value=treehouses
publicIp=$(waitForOutput "getLatestIpAddress $instanceId")
echo "Public IP Address is $publicIp"
echo "Will open ssh tunnel soon"
isOpen=$(waitForOutput "ssh-keyscan -H $publicIp | grep ecdsa-sha2-nistp256")
echo "Opened ssh tunnel"
openSSHTunnel $instanceName $publicIp $portConfigArray
storeConfigIntoTreehousesConfigAsStringfiedJson $instanceName $importedKeyName $instanceId $publicIp $groupNameaws ec2 create-tags --resources $instanceId --tags Key=Class,Value=treehouses
else
instanceState=$(checkInstanceState $instanceId)
if [ "$instanceState" = "running" ]; then
echo "EC2 instance is already running."
elif [ "$instanceState" = "stopped" ]; then
echo "Starting stopped EC2 instance..."
start $instanceName
else
echo "EC2 instance is in state: $instanceState."
fi
fi
}
Loading