Merge pull request #74 from abolfazl8131/master

abolfazl8131 · web-flow · commit 6a664774ce1b · 2024-11-18T12:31:17.000+03:30
feat(argocd): add argocd to terraform
diff --git a/app/media/MyTerraform/main.tf b/app/media/MyTerraform/main.tf
@@ -0,0 +1,15 @@
+
+provider "aws" {
+  region = "us-east-1"
+}
+
+module "ec2" {
+  key_pair_create      = var.key_pair_create
+  key_pair_name        = var.key_pair_name
+  security_group_create = var.security_group_create
+  security_group_name   = var.security_group_name
+  security_group_ingress_rules = var.security_group_ingress_rules
+  security_group_egress_rule   = var.security_group_egress_rule
+  instance_create      = var.instance_create
+  instance_type        = var.instance_type
+}
diff --git a/app/media/MyTerraform/modules/ec2/main.tf b/app/media/MyTerraform/modules/ec2/main.tf
@@ -0,0 +1,57 @@
+
+data "aws_ami" "linux" {
+  most_recent = true
+  owners      = ["amazon"]
+
+  filter {
+    name   = "name"
+    values = ["al2023-ami-2023*kernel-6.1-x86_64"]
+  }
+
+  filter {
+    name   = "root-device-type"
+    values = ["ebs"]
+  }
+
+  filter {
+    name   = "virtualization-type"
+    values = ["hvm"]
+  }
+}
+
+resource "aws_key_pair" "key_pair" {
+  count       = var.key_pair_create ? 1 : 0
+  key_name    = var.key_pair_name
+  public_key  = file("${path.module}/terraform.pub")
+}
+
+resource "aws_security_group" "security_group" {
+  count = var.security_group_create ? 1 : 0
+  name  = var.security_group_name
+
+  dynamic "ingress" {
+    for_each = var.security_group_ingress_rules
+    content {
+      description = ingress.value["description"]
+      from_port   = ingress.value["from_port"]
+      to_port     = ingress.value["to_port"]
+      protocol    = ingress.value["protocol"]
+      cidr_blocks = ingress.value["cidr_blocks"]
+    }
+  }
+
+  egress {
+    from_port   = var.security_group_egress_rule["from_port"]
+    to_port     = var.security_group_egress_rule["to_port"]
+    protocol    = var.security_group_egress_rule["protocol"]
+    cidr_blocks = var.security_group_egress_rule["cidr_blocks"]
+  }
+}
+
+resource "aws_instance" "instance" {
+  count                  = var.instance_create ? 1 : 0
+  ami                    = data.aws_ami.linux.id
+  instance_type          = var.instance_type
+  key_name               = var.key_pair_create ? aws_key_pair.key_pair[0].key_name : null
+  vpc_security_group_ids = var.security_group_create ? [aws_security_group.security_group[0].id] : null
+}
diff --git a/app/media/MyTerraform/modules/ec2/terraform.pub b/app/media/MyTerraform/modules/ec2/terraform.pub
diff --git a/app/media/MyTerraform/modules/ec2/terraform.tfvars b/app/media/MyTerraform/modules/ec2/terraform.tfvars
@@ -0,0 +1,31 @@
+
+key_pair_create = true
+key_pair_name = "ec2"
+
+security_group_create = true
+security_group_name = "my_rules"
+security_group_ingress_rules = {
+  ssh_rule = {
+    description = "SSH Ingress"
+    from_port   = 22
+    to_port     = 22
+    protocol    = "tcp"
+    cidr_blocks = ["0.0.0.0/0"]
+  },
+  http_rule = {
+    description = "HTTP Ingress"
+    from_port   = 80
+    to_port     = 80
+    protocol    = "tcp"
+    cidr_blocks = ["0.0.0.0/0"]
+  }
+}
+security_group_egress_rule = {
+  from_port   = 0
+  to_port     = 0
+  protocol    = "-1"
+  cidr_blocks = ["0.0.0.0/0"]
+}
+
+instance_create = true
+instance_type = "t2.micro"
diff --git a/app/media/MyTerraform/modules/ec2/variables.tf b/app/media/MyTerraform/modules/ec2/variables.tf
@@ -0,0 +1,51 @@
+
+variable "key_pair_create" {
+  description = "Create Key Pair"
+  type        = bool
+}
+
+variable "key_pair_name" {
+  description = "Key Pair Name"
+  type        = string
+}
+
+variable "security_group_create" {
+  description = "Create Security Group"
+  type        = bool
+}
+
+variable "security_group_name" {
+  description = "Security Group Name"
+  type        = string
+}
+
+variable "security_group_ingress_rules" {
+  description = "Security Group Ingress Rules"
+  type        = map(object({
+    description = string
+    from_port   = number
+    to_port     = number
+    protocol    = string
+    cidr_blocks = list(string)
+  }))
+}
+
+variable "security_group_egress_rule" {
+  description = "Security Group Egress Rule"
+  type        = object({
+    from_port   = number
+    to_port     = number
+    protocol    = string
+    cidr_blocks = list(string)
+  })
+}
+
+variable "instance_create" {
+  description = "Create EC2 Instance"
+  type        = bool
+}
+
+variable "instance_type" {
+  description = "EC2 Instance Type"
+  type        = string
+}
diff --git a/app/media/MyTerraform/modules/ec2/versions.tf b/app/media/MyTerraform/modules/ec2/versions.tf
@@ -0,0 +1,11 @@
+
+terraform {
+  required_version = ">= 1.0"
+
+  required_providers {
+    aws = {
+      source  = "hashicorp/aws"
+      version = ">= 5.20"
+    }
+  }
+}
diff --git a/app/media/MyTerraform/terraform.tfvars b/app/media/MyTerraform/terraform.tfvars
@@ -0,0 +1,31 @@
+
+key_pair_create = true
+key_pair_name = "ec2"
+
+security_group_create = true
+security_group_name = "my_rules"
+security_group_ingress_rules = {
+  ssh_rule = {
+    description = "SSH Ingress"
+    from_port   = 22
+    to_port     = 22
+    protocol    = "tcp"
+    cidr_blocks = ["0.0.0.0/0"]
+  },
+  http_rule = {
+    description = "HTTP Ingress"
+    from_port   = 80
+    to_port     = 80
+    protocol    = "tcp"
+    cidr_blocks = ["0.0.0.0/0"]
+  }
+}
+security_group_egress_rule = {
+  from_port   = 0
+  to_port     = 0
+  protocol    = "-1"
+  cidr_blocks = ["0.0.0.0/0"]
+}
+
+instance_create = true
+instance_type = "t2.micro"
diff --git a/app/media/MyTerraform/variables.tf b/app/media/MyTerraform/variables.tf
@@ -0,0 +1,51 @@
+
+variable "key_pair_create" {
+  description = "Create Key Pair"
+  type        = bool
+}
+
+variable "key_pair_name" {
+  description = "Key Pair Name"
+  type        = string
+}
+
+variable "security_group_create" {
+  description = "Create Security Group"
+  type        = bool
+}
+
+variable "security_group_name" {
+  description = "Security Group Name"
+  type        = string
+}
+
+variable "security_group_ingress_rules" {
+  description = "Security Group Ingress Rules"
+  type        = map(object({
+    description = string
+    from_port   = number
+    to_port     = number
+    protocol    = string
+    cidr_blocks = list(string)
+  }))
+}
+
+variable "security_group_egress_rule" {
+  description = "Security Group Egress Rule"
+  type        = object({
+    from_port   = number
+    to_port     = number
+    protocol    = string
+    cidr_blocks = list(string)
+  })
+}
+
+variable "instance_create" {
+  description = "Create EC2 Instance"
+  type        = bool
+}
+
+variable "instance_type" {
+  description = "EC2 Instance Type"
+  type        = string
+}
diff --git a/app/media/MyTerraform/versions.tf b/app/media/MyTerraform/versions.tf
@@ -0,0 +1,11 @@
+
+terraform {
+  required_version = ">= 1.0"
+
+  required_providers {
+    aws = {
+      source  = "hashicorp/aws"
+      version = ">= 5.20"
+    }
+  }
+}
diff --git a/app/models/terraform_models.py b/app/models/terraform_models.py
@@ -79,4 +79,10 @@ class IaCTemplateGenerationS3(BaseModel):
 
 class IaCTemplateGenerationIAM(BaseModel):
     iam_user:bool = True
-    iam_group:bool = True
+    iam_group:bool = True
+
+class IaCTemplateGenerationArgoCD(BaseModel):
+    argocd_application:bool = True
+    argocd_project:bool = True
+    argocd_repository:bool = True
+    argocd_cluster:bool = True
diff --git a/app/routes/terraform.py b/app/routes/terraform.py
@@ -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')
diff --git a/app/template_generators/terraform/argocd.py b/app/template_generators/terraform/argocd.py
@@ -0,0 +1,2 @@
+def IaC_template_generator_argocd(input) -> str:
+    pass
