Merge pull request #89 from abolfazl8131/master

feat(elb&efs):declare elb and efs

app/models/terraform_models.py

@@ -93,4 +93,25 @@ class ArgoApplication(BaseModel):
 class IaCTemplateGenerationArgoCD(BaseModel):
     argocd_application:ArgoApplication | None = None
     argocd_repository:bool = True
-
+
+
+class IaCTemplateGenerationELB(BaseModel):
+    security_group:bool = True
+    lb_target_group:bool = True
+    lb:bool = True
+    lb_listener:bool = True
+    lb_listener_rule:bool = True
+    key_pair:bool = True
+    launch_configuration:bool = True
+    autoscaling_group:bool = True
+    autoscaling_attachment:bool = True
+    autoscaling_policy:bool = True
+
+
+
+class IaCTemplateGenerationEFS(BaseModel):
+    security_group:bool = True
+    efs_file_system:bool = True
+    efs_mount_target:bool = True
+    efs_backup_policy:bool = True
+

app/routes/terraform.py

@@ -11,7 +11,9 @@
         IaCTemplateGenerationEC2,
         IaCTemplateGenerationS3,
         IaCTemplateGenerationIAM,
-        IaCTemplateGenerationArgoCD
+        IaCTemplateGenerationArgoCD,
+        IaCTemplateGenerationELB,
+        IaCTemplateGenerationEFS
        )
 
 from fastapi import Response
@@ -25,6 +27,8 @@
 from app.template_generators.terraform.aws.IAM import (IaC_template_generator_iam)
 from app.template_generators.terraform.aws.s3 import (IaC_template_generator_s3)
 from app.template_generators.terraform.argocd import (IaC_template_generator_argocd)
+from app.template_generators.terraform.aws.ELB import (IaC_template_generator_elb)
+from app.template_generators.terraform.aws.EFS import (IaC_template_generator_efs)
 import os
 
 @app.post("/IaC-basic/")
@@ -102,4 +106,26 @@ async def IaC_template_generation_argocd(request:IaCTemplateGenerationArgoCD) ->
         output = gpt_service(generated_prompt)
         edit_directory_generator("terraform_generator",output)
         execute_pythonfile("MyTerraform","terraform_generator")
-        return Output(output='output')
+        return Output(output='output')
+
+
+
+@app.post("/IaC-template/aws/elb")
+async def IaC_template_generation_aws_elb(request:IaCTemplateGenerationELB) -> Output:
+        if os.environ.get("TEST"):
+            return Output(output='output (nothing special)')
+        generated_prompt = IaC_template_generator_elb(request)
+        output = gpt_service(generated_prompt)
+        edit_directory_generator("terraform_generator",output)
+        execute_pythonfile("MyTerraform","terraform_generator")
+        return Output(output='output')
+
+@app.post("/IaC-template/aws/efs")
+async def IaC_template_generation_aws_efs(request:IaCTemplateGenerationEFS) -> Output:
+        if os.environ.get("TEST"):
+            return Output(output='output (nothing special)')
+        generated_prompt = IaC_template_generator_efs(request)
+        output = gpt_service(generated_prompt)
+        edit_directory_generator("terraform_generator",output)
+        execute_pythonfile("MyTerraform","terraform_generator")
+        return Output(output='output')

app/template_generators/terraform/aws/EFS.py

@@ -0,0 +1,2 @@
+def IaC_template_generator_efs(input) -> str:
+    pass

app/template_generators/terraform/aws/ELB.py

@@ -0,0 +1,2 @@
+def IaC_template_generator_elb(input) -> str:
+    pass