feat(terraform-templates): implementing aws-key-pair route

Author: mohammadll

app/media/terraform.tfvars

@@ -1,4 +1,2 @@
-create_zone = true
-create_record = true
-create_delegation_set = false
-create_resolver_rule_association = false
+create = true
+create_private_key = false

app/models/terraform_models.py

@@ -141,3 +141,8 @@ class IaCTemplateGenerationRoute53(BaseModel):
     record:bool = True
     delegation_set:bool = False
     resolver_rule_association:bool = False
+
+class IaCTemplateGenerationKeyPair(BaseModel):
+    
+    key_pair:bool = True
+    private_key:bool = False

app/routes/terraform.py

@@ -20,7 +20,8 @@
         IaCTemplateGenerationSNS,
         IaCTemplateGenerationAutoScaling,
         IaCTemplateGenerationSQS,
-        IaCTemplateGenerationRoute53
+        IaCTemplateGenerationRoute53,
+        IaCTemplateGenerationKeyPair
        )
 
 from fastapi import Response
@@ -42,6 +43,7 @@
 from app.template_generators.terraform.aws.AutoScaling import (IaC_template_generator_autoscaling)
 from app.template_generators.terraform.aws.SQS import (IaC_template_generator_sqs)
 from app.template_generators.terraform.aws.Route53 import (IaC_template_generator_route53)
+from app.template_generators.terraform.aws.KeyPair import (IaC_template_generator_key_pair)
 from app.template_generators.terraform.Installation.main import (select_install)
 import os
 
@@ -225,3 +227,15 @@ async def IaC_template_generation_aws_route53(request:IaCTemplateGenerationRoute
 
         return FileResponse(dir, media_type='application/zip', filename=f"terraform.tfvars")
 
+
+@app.post("/api/IaC-template/aws/key_pair")
+async def IaC_template_generation_aws_key_pair(request:IaCTemplateGenerationKeyPair) -> Output:
+         
+        dir = 'app/media/terraform.tfvars'
+        
+        file_response = IaC_template_generator_key_pair(request)
+        with open(dir,'w')as f:
+            f.write(file_response)
+        
+        return FileResponse(dir, media_type='application/zip', filename=f"terraform.tfvars")
+

app/template_generators/terraform/aws/KeyPair.py

@@ -0,0 +1,9 @@
+def IaC_template_generator_key_pair(input) -> str:
+
+    aws_key_pair_create = 'true' if input.key_pair else 'false'
+    aws_key_pair_create_private_key = 'true' if input.private_key else 'false'
+    
+    tfvars_file = f"""create = {aws_key_pair_create}
+create_private_key = {aws_key_pair_create_private_key}
+"""
+    return tfvars_file