Update aws-ssm-ec2-proxy-command.ps1

Author: qoomon

aws-ssm-ec2-proxy-command.ps1

@@ -1,4 +1,4 @@
-#!/usr/bin/env sh
+#!/usr/bin/env pwsh
 ######## Source ################################################################
 #
 # https://github.com/qoomon/aws-ssm-ec2-proxy-command
@@ -13,15 +13,15 @@
 # Add following SSH Config Entry to ~/.ssh/config
 # host i-* mi-*
 # IdentityFile ~/.ssh/id_rsa
-# ProxyCommand powershell .exe ~/.ssh/aws-ssm-ec2-proxy-command.ps1 %h %r %p ~/.ssh/id_rsa.pub
+# ProxyCommand powershell.exe ~/.ssh/aws-ssm-ec2-proxy-command.ps1 %h %r %p ~/.ssh/id_rsa.pub
 # StrictHostKeyChecking no
 #
 # Ensure SSM Permissions for Target Instance Profile
 # https://docs.aws.amazon.com/systems-manager/latest/userguide/setup-instance-profile.html
 #
 # Open SSH Connection
 # ssh <INSTANCE_USER>@<INSTANCE_ID>
-# 
+#
 # Ensure AWS CLI environment variables are set properly
 # e.g. AWS_PROFILE='default' ssh ec2-user@i-xxxxxxxxxxxxxxxx
 #
@@ -49,31 +49,30 @@ if ($splitted_instance.Length -gt 1)
  $env:AWS_DEFAULT_REGION = $splitted_instance[1]
 }
 
-$authorized_key = "$ssh_public_key ssm-session"
-$script = @"
-\"
-mkdir -p ~$ssh_user/.ssh && cd ~$ssh_user/.ssh || exit 1
+Write-Output "Add public key $ssh_public_key_path for $ssh_user at instance $ec2_instance_id for $ssh_public_key_timeout seconds"
+$addPublicKeyScript = @"
+ \"
+ mkdir -p ~$ssh_user/.ssh && cd ~$ssh_user/.ssh || exit 1
 
-echo '$authorized_key' >> authorized_keys
+ authorized_key='$ssh_public_key ssm-session'
+ echo \\\"`$authorized_key\\\" >> authorized_keys
 
-sleep $ssh_public_key_timeout
+ sleep $ssh_public_key_timeout
 
-grep -v -F '$authorized_key' authorized_keys > .authorized_keys
-mv .authorized_keys authorized_keys
-\"
+ grep -v -F \\\"`$authorized_key\\\" authorized_keys > .authorized_keys
+ mv .authorized_keys authorized_keys
+ \"
 "@
-
-Write-Output "Add public key $ssh_public_key_path for $ssh_user at instance $ec2_instance_id for $ssh_public_key_timeout seconds"
 aws ssm send-command `
  --instance-ids "$ec2_instance_id" `
  --document-name 'AWS-RunShellScript' `
  --comment "Add an SSH public key to authorized_keys for $ssh_public_key_timeout seconds" `
- --parameters commands="$script"
+ --parameters commands="$addPublicKeyScript"
 if($LASTEXITCODE -ne 0) { Write-Error "Failed to add public key with error $output" }
 
 Write-Output "Start ssm session to instance $ec2_instance_id"
 aws ssm start-session `
  --target "$ec2_instance_id" `
  --document-name 'AWS-StartSSHSession' `
  --parameters "portNumber=$ssh_port"
-if($LASTEXITCODE -ne 0) { Write-Error "Failed to start ssm session to instance $output" }
+if($LASTEXITCODE -ne 0) { Write-Error "Failed to start ssm session to instance $output" }