Merge branch 'master' of github.com:devopshobbies/devops-gpt into ec2

Author: abolfazl8131

app/media/MyHelm_zip.zip

@@ -0,0 +1,16 @@
+
+resource "aws_s3_bucket" "s3_bucket" {
+  count         = var.s3_create_bucket ? 1 : 0
+  bucket        = var.s3_bucket_name
+  force_destroy = var.s3_bucket_force_destroy
+  tags          = var.s3_bucket_tags
+}
+
+resource "aws_s3_bucket_versioning" "s3_bucket_versioning" {
+  count = var.s3_create_bucket && var.s3_create_bucket_versioning ? 1 : 0
+  bucket = aws_s3_bucket.s3_bucket[0].id
+
+  versioning_configuration {
+    status = var.s3_bucket_versioning_status
+  }
+}

app/media/MyTerraform/modules/s3/main.tf

@@ -0,0 +1,10 @@
+
+s3_create_bucket = true
+s3_bucket_name = "UniqueName"
+s3_bucket_force_destroy = false
+s3_bucket_tags = {
+  Name        = "My bucket"
+  Environment = "Dev"
+}
+s3_create_bucket_versioning = false
+s3_bucket_versioning_status = "Enabled"

app/media/MyTerraform/modules/s3/terraform.tfvars

@@ -0,0 +1,24 @@
+
+variable "s3_create_bucket" {
+  type = bool
+}
+
+variable "s3_bucket_name" {
+  type = string
+}
+
+variable "s3_bucket_force_destroy" {
+  type = bool
+}
+
+variable "s3_bucket_tags" {
+  type = map(string)
+}
+
+variable "s3_create_bucket_versioning" {
+  type = bool
+}
+
+variable "s3_bucket_versioning_status" {
+  type = string
+}

app/media/MyTerraform/modules/s3/variables.tf

@@ -0,0 +1,11 @@
+
+terraform {
+  required_version = ">= 1.0"
+
+  required_providers {
+    aws = {
+      source  = "hashicorp/aws"
+      version = ">= 5.20"
+    }
+  }
+}

app/media/MyTerraform/modules/s3/versions.tf

@@ -79,4 +79,22 @@ class IaCTemplateGenerationS3(BaseModel):
 
 class IaCTemplateGenerationIAM(BaseModel):
     iam_user:bool = True
-    iam_group:bool = True
+    iam_group:bool = True
+
+class SyncPolicy(BaseModel):
+    auto_prune: bool = True
+    self_heal: bool = True
+
+class SyncOptions(BaseModel):
+    apply_out_of_sync_only: bool = True
+    create_namespace: bool = True
+    fail_or_share_resource:bool = True
+
+class ArgoApplication(BaseModel):
+    sync_policy: SyncPolicy | None = None
+    sync_options: SyncOptions | None = None
+
+class IaCTemplateGenerationArgoCD(BaseModel):
+    argocd_application:ArgoApplication | None = None
+    argocd_repository:bool = True
+    argocd_cluster:bool = True

app/media/MyTerraform_zip.zip

@@ -11,6 +11,7 @@
         IaCTemplateGenerationEC2,
         IaCTemplateGenerationS3,
         IaCTemplateGenerationIAM,
+        IaCTemplateGenerationArgoCD
        )
 
 from fastapi import Response
@@ -23,6 +24,7 @@
 from app.template_generators.terraform.aws.ec2 import (IaC_template_generator_ec2)
 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)
 import os
 
 @app.post("/IaC-basic/")
@@ -92,3 +94,12 @@ async def IaC_template_generation_aws_iam(request:IaCTemplateGenerationIAM) -> O
         return Output(output='output')
 
 
+@app.post("/IaC-template/argocd")
+async def IaC_template_generation_argocd(request:IaCTemplateGenerationArgoCD) -> Output:
+        if os.environ.get("TEST"):
+            return Output(output='output (nothing special)')
+        generated_prompt = IaC_template_generator_argocd(request)
+        output = gpt_service(generated_prompt)
+        edit_directory_generator("terraform_generator",output)
+        execute_pythonfile("MyTerraform","terraform_generator")
+        return Output(output='output')

app/models/terraform_models.py

@@ -2,39 +2,33 @@
 from fastapi import FastAPI, HTTPException,Response
 from fastapi.responses import FileResponse
 import os
+import zipfile
 
 
-@app.get("/download-helm/{filename}")
-def download_file_helm(filename: str):
+def zip_folder(folder_path: str, output_zip_path: str):
+    """Zip the entire folder."""
+    with zipfile.ZipFile(output_zip_path, 'w', zipfile.ZIP_DEFLATED) as zip_file:
+        for root, dirs, files in os.walk(folder_path):
+            for file in files:
+                file_path = os.path.join(root, file)
+                # Add file to the zip file
+                zip_file.write(file_path, os.path.relpath(file_path, folder_path))
 
-    folder = 'app/media/MyHelm'
-    file_path = os.path.join(folder, filename)
-   
-    if not os.path.isfile(file_path):
-        raise HTTPException(status_code=404, detail="File not found.")
 
-   
-    return FileResponse(file_path, media_type='application/octet-stream', filename=filename)
 
 
-@app.get("/download-terraform/{filename}")
-def download_file_terraform(filename: str):
+@app.get("/download-folder{folder_name}/{source}")
+async def download_folder_MyHelm(folder_name: str,source:str):
+    folder_path = f"app/media/{folder_name}"  # Adjust the path as needed
+    if not os.path.exists(folder_path):
+        raise HTTPException(status_code=404, detail="Folder not found")
 
-    folder = 'app/media/MyTerraform'
-    file_path = os.path.join(folder, filename)
-    
-    if not os.path.isfile(file_path):
-        raise HTTPException(status_code=404, detail="File not found.")
+    zip_file_path = f"app/media/{folder_name}_zip.zip"
 
-   
-    return FileResponse(file_path, media_type='application/octet-stream', filename=filename)
+    # Zip the folder
+    zip_folder(folder_path, zip_file_path)
+
+    # Return the zip file as a response
+    return FileResponse(zip_file_path, media_type='application/zip', filename=f"{folder_name}_{source}.zip")
 
-@app.get("/list-directory")
-def list_directory(folder: str):
-    # Ensure the folder exists
-    if not os.path.isdir(folder):
-        raise HTTPException(status_code=404, detail=f"{folder} does not exist.")
 
-    # List the contents of the directory
-    contents = os.listdir(folder)
-    return {"folder": folder, "contents": contents}

app/routes/terraform.py

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