Merge pull request #111 from abolfazl8131/master

feat(ansible kuber): add the route and config adding to add configs t…

Author: abolfazl8131

app/directory_generators/ansible_generator.py

@@ -1,6 +1,7 @@
 import os
 
 project_name = "app/media/MyAnsible"
+
 ansible_dir = project_name
 group_vars_dir = os.path.join(ansible_dir, "group_vars")
 host_vars_dir = os.path.join(ansible_dir, "host_vars")
@@ -25,27 +26,30 @@
 os.makedirs(templates_dir, exist_ok=True)
 
 # Create ansible.cfg
-with open(os.path.join(ansible_dir, "ansible.cfg"), "w") as cfg_file:
-    cfg_file.write("[defaults]\n")
-    cfg_file.write("host_key_checking=false\n")
+with open(os.path.join(ansible_dir, "ansible.cfg"), "w") as ansible_cfg:
+    ansible_cfg.write("[defaults]\n")
+    ansible_cfg.write("host_key_checking=false\n")
 
 # Create group_vars/docker_nodes
-with open(os.path.join(group_vars_dir, "docker_nodes"), "w") as gv_file:
-    gv_file.write("ansible_port: 28\n")
-    gv_file.write("ansible_user: root\n")
+with open(os.path.join(group_vars_dir, "docker_nodes"), "w") as docker_nodes:
+    docker_nodes.write("ansible_port: 22\n")
+    docker_nodes.write("ansible_user: root\n")
 
 # Create hosts
 with open(os.path.join(ansible_dir, "hosts"), "w") as hosts_file:
     hosts_file.write("[docker_nodes]\n")
     hosts_file.write("www.example.com\n")
 
+# Create empty host_vars directory
+open(os.path.join(host_vars_dir, ".gitkeep"), 'a').close()
+
 # Create docker_playbook.yml
-with open(os.path.join(ansible_dir, "docker_playbook.yml"), "w") as playbook_file:
-    playbook_file.write("- hosts: all\n")
-    playbook_file.write("  roles:\n")
-    playbook_file.write("    - install_docker\n")
+with open(os.path.join(ansible_dir, "docker_playbook.yml"), "w") as playbook:
+    playbook.write("- hosts: all\n")
+    playbook.write("  roles:\n")
+    playbook.write("    - install_docker\n")
 
-# Create tasks/main.yml
+# Create install_docker/tasks/main.yml
 with open(os.path.join(tasks_dir, "main.yml"), "w") as tasks_file:
     tasks_file.write("---\n")
     tasks_file.write("- name: Install prerequisite packages\n")
@@ -83,7 +87,7 @@
     tasks_file.write("    enabled: yes\n")
     tasks_file.write("  loop: \"{{ docker_services }}\"\n")
 
-# Create vars/main.yml
+# Create install_docker/vars/main.yml
 with open(os.path.join(vars_dir, "main.yml"), "w") as vars_file:
     vars_file.write("prerequisite_packages:\n")
     vars_file.write("  - ca-certificates\n")
@@ -96,16 +100,4 @@
     vars_file.write("  - docker-ce-cli\n")
     vars_file.write("  - containerd.io\n")
     vars_file.write("  - docker-buildx-plugin\n")
-    vars_file.write("  - docker-compose-plugin\n")
-
-# Create empty host_vars directory
-os.makedirs(host_vars_dir, exist_ok=True)
-
-# Create empty files directory
-os.makedirs(files_dir, exist_ok=True)
-
-# Create empty handlers directory
-os.makedirs(handlers_dir, exist_ok=True)
-
-# Create empty templates directory
-os.makedirs(templates_dir, exist_ok=True)
+    vars_file.write("  - docker-compose-plugin\n")

app/media/MyAnsible/group_vars/docker_nodes

@@ -1,2 +1,2 @@
-ansible_port: 28
+ansible_port: 22
 ansible_user: root

app/media/MyAnsible/host_vars/.gitkeep

@@ -29,4 +29,22 @@ def validator_os(cls, value):
         valid_oss = ['ubuntu']
         if value not in valid_oss:
             raise ValueError(f"your selected OS must be in {valid_oss}")
-        return value
+        return value
+    
+    
+    
+class AnsibleInstallKuber(AnsibleBase):
+    os: str = 'ubuntu'
+    k8s_worker_nodes: List[str]
+    k8s_master_nodes: List[str]
+    lb_nodes: List[str]
+    version:str = "1.31.2"
+    
+    
+    @validator("os")
+    def validator_os(cls, value):
+        valid_oss = ['ubuntu']
+        if value not in valid_oss:
+            raise ValueError(f"your selected OS must be in {valid_oss}")
+        return value
+    

app/models/ansible_models.py

@@ -1,8 +1,8 @@
 from app.app_instance import app
 from app.gpt_services import gpt_service
 from app.services import (write_installation,edit_directory_generator,execute_pythonfile)
-
-from app.models import (AnsibleInstallNginx,AnsibleInstallDocker,Output)
+from app.routes.utils import add_files_to_folder
+from app.models import (AnsibleInstallNginx,AnsibleInstallDocker,Output,AnsibleInstallKuber)
 
 from app.models import (AnsibleInstallNginx,Output)
 
@@ -32,4 +32,25 @@ async def ansible_install_generation_docker(request:AnsibleInstallDocker) -> Out
         output = gpt_service(generated_prompt)
         edit_directory_generator("ansible_generator",output)
         execute_pythonfile("MyAnsible","ansible_generator")
+        return Output(output='output')
+    
+    
+@app.post("/ansible-install/kuber/")
+async def ansible_install_generation_kuber(request:AnsibleInstallKuber) -> Output:
+    
+        if os.environ.get("TEST"):
+            return Output(output='output')
+        generated_prompt = ansible_install_template(request,"kuber")
+
+        output = gpt_service(generated_prompt)
+        edit_directory_generator("ansible_generator",output)
+        execute_pythonfile("MyAnsible","ansible_generator")
+        add_files_to_folder(files = ['app/media/kuber_configs/resolv.conf.j2'] , folder='app/media/MyAnsible/roles/preinstall/templates/')
+        add_files_to_folder(files = ['app/media/kuber_configs/kubeadmcnf.yml.j2'] , folder='app/media/MyAnsible/roles/init_k8s/templates')
+        add_files_to_folder(files = ['app/media/kuber_configs/kubeadmcnf-join.yml.j2'] , folder='app/media/MyAnsible/roles/join_master/templates')
+        add_files_to_folder(files = ['app/media/kuber_configs/check_apiserveer.sh.j2',
+                                     'app/media/kuber_configs/haproxy.cfg.j2',
+                                     'app/media/kuber_configs/keepalived.conf.j2'
+                                     ] , folder='app/media/MyAnsible/roles/lb/templates')
+        
         return Output(output='output')

app/routes/ansible.py

@@ -14,7 +14,12 @@ def zip_folder(folder_path: str, output_zip_path: str):
                 # Add file to the zip file
                 zip_file.write(file_path, os.path.relpath(file_path, folder_path))
 
+def add_files_to_folder(files:list,folder:str):
+    
+    os.makedirs(folder, exist_ok=True)
 
+    for filename in files:
+        os.path.join(folder, filename)
 
 
 @app.get("/download-folder{folder_name}/{source}")

app/routes/utils.py

@@ -0,0 +1,3 @@
+
+def ansible_kuber_install(input):
+    pass

app/template_generators/ansible/install/kuber.py

@@ -1,13 +1,20 @@
 from .docker import ansible_docker_install
 from .nginx import ansible_nginx_install
+from .kuber import ansible_kuber_install
 from fastapi import HTTPException
 
 def ansible_install_template(input_, tool:str):
+    
     match tool:
+        
         case 'nginx':
             return ansible_nginx_install(input_)
+        
         case 'docker':
             return ansible_docker_install(input_)
 
+        case 'kuber':
+            return ansible_kuber_install(input_)
+        
         case _: 
             raise HTTPException(400,"please select a valid tool for installation")