From 28c46086b5a5cc27d5ad509b307e1403f5157295 Mon Sep 17 00:00:00 2001
From: Jack Phillips <jack.phillips@datadoghq.com>
Date: Mon, 25 Nov 2024 14:53:05 -0500
Subject: [PATCH] Update deploy to copy private key path to pulumi env

---
 tasks/aws/deploy.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tasks/aws/deploy.py b/tasks/aws/deploy.py
index 0da9749f3..d9195a525 100644
--- a/tasks/aws/deploy.py
+++ b/tasks/aws/deploy.py
@@ -11,6 +11,7 @@
 from tasks.deploy import deploy as common_deploy
 
 default_public_path_key_name = "ddinfra:aws/defaultPublicKeyPath"
+default_private_path_key_name = "ddinfra:aws/defaultPrivateKeyPath"
 
 
 def deploy(
@@ -42,6 +43,10 @@ def deploy(
 
     flags[default_public_path_key_name] = _get_public_path_key_name(cfg, public_key_required)
 
+    privateKeyPath = cfg.get_aws().privateKeyPath
+    if privateKeyPath is not None:
+        flags[default_private_path_key_name] = privateKeyPath
+
     awsKeyPairName = cfg.get_aws().keyPairName
 
     flags["ddinfra:aws/defaultKeyPairName"] = awsKeyPairName