Merge pull request #28 from linhtrantuan/main

[Add] - ssm portforwarding

Author: lamhaison

common/peco.sh

@@ -138,6 +138,10 @@ peco_aws_list_db_clusters() {
 	peco_aws_input 'aws rds describe-db-clusters --query "*[].DBClusterIdentifier"' 'true'
 }
 
+peco_aws_list_db_endpoint() {
+	peco_aws_input 'aws rds describe-db-clusters --query "*[].[Endpoint, ReaderEndpoint]"' 'true'
+}
+
 peco_aws_list_db_instances() {
 	peco_aws_input 'aws rds describe-db-instances --query "*[].DBInstanceIdentifier"' 'true'
 }

services/ec2.sh

@@ -6,6 +6,12 @@ local_aws_ec2_instance_id_peco_menu() {
 	echo ${aws_ec2_instance_id}
 }
 
+local_aws_rds_endpoint_peco_menu() {
+	local aws_rds_endpoint=$(peco_create_menu 'peco_aws_list_db_endpoint')
+	aws_rds_endpoint=$(echo "${aws_rds_endpoint}" | awk -F "_" '{print $1}')
+	echo ${aws_rds_endpoint}
+}
+
 # AWS ec2
 # List all ec2 instance(don't care stopped or running instances)
 aws_ec2_list_all() {
@@ -122,6 +128,14 @@ aws_ec2_connect_with_hint() {
 	aws_ssm_connection_ec2 $(local_aws_ec2_instance_id_peco_menu)
 }
 
+aws_ec2_connect_forwarding_for_rds_with_hint() {
+	local ec2_instance_id=$(local_aws_ec2_instance_id_peco_menu)
+	local rds_endpoint=$(local_aws_rds_endpoint_peco_menu)
+	echo "Enter your local port forwarding to the instance"
+	read local_port
+	aws_ssm_port_forwarding_ec2 $ec2_instance_id $rds_endpoint $local_port
+}
+
 aws_ec2_list_eips() {
 	aws_run_commandline 'aws ec2 describe-addresses'
 }

services/ssm.sh

@@ -27,3 +27,21 @@ aws_ssm_connection_ec2() {
         fi
 
 }
+
+aws_ssm_port_forwarding_ec2_for_rds() {
+        aws_ec2_instance_id=$1
+        aws_rds_endpoint=$2
+        local_port=$3
+        aws_commandline_logging "\
+                aws ssm start-session --target ${aws_ec2_instance_id:?'aws_ec2_instance_id is unset or empty'}
+                        --document-name AWS-StartPortForwardingSessionToRemoteHost \
+                        --parameters host="${aws_rds_endpoint:?'aws_rds_endpoint is unset or empty'}",portNumber="3306",localPortNumber="${local_port:?'local_port is unset or empty'}"
+        "
+
+        if [[ -n "${aws_ec2_instance_id}" || -n "${aws_rds_endpoint}"|| -n "${local_port}" ]]; then
+                aws ssm start-session --target $aws_ec2_instance_id \
+                        --document-name AWS-StartPortForwardingSessionToRemoteHost \
+                        --parameters host="$aws_rds_endpoint",portNumber="3306",localPortNumber="$local_port"
+        fi
+
+}