hard code ec2 ami ids (#708)

Author: GracefulLemming

terraform/bastion-host.nix

@@ -9,10 +9,18 @@
       "github.com/cloudposse/terraform-aws-ec2-bastion-server?ref=7f8fc52095ef466fedd1c06876e281a1ac6bb75b";
     enabled = true;
     instance_type = "t4g.micro";
-    # ami = "ami-03190fe20ef6b1419";
+    
+    # TODO Make this more flexible
+    ami = if config.setup.stage == "prod" 
+      then "ami-037d882b31eae26a2" 
+      else if config.setup.stage == "uat"
+      then "ami-03190fe20ef6b1419"
+      else "ami-0d1c8113ba7b8b12a";
+
     # ami_filter = {
-    #   name = [ "amzn2-ami-kernel-5.10-hvm-*-arm64-gp2" ]
-    # }
+    #   name = [ "amzn2-ami-*-hvm-*-arm64-gp2" ]
+    # };
+
 
     # ID will be constructed from these namespace, stage, and name for some reason.
     namespace = "dailp";