feat(ansible): add nginx template generator to ansible. (#97)

* feat(nginx_ansible): recreate nginx ansible model

* fix(nginx_ansible): fix nginx validator

* refactor(ansible): refactor ansible function for more flexibility

app/models/ansible_models.py

@@ -1,6 +1,16 @@
 from typing import List, Optional
 from pydantic import BaseModel, validator, ValidationError
 
-class AnsibleInstall(BaseModel):
+
+
+class AnsibleInstallNginx(BaseModel):
     os: str = 'ubuntu'
-    tool: str = 'nginx'
+    hosts:List[str] = ['www.example.com']
+    version:str = 'latest'
+
+    @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/routes/ansible.py

@@ -1,15 +1,16 @@
 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 (AnsibleInstall,Output)
+from app.models import (AnsibleInstallNginx,Output)
 from app.template_generators.ansible.install.main import ansible_install_template
 import os
 
-@app.post("/ansible-install/")
-async def ansible_install_generation(request:AnsibleInstall) -> Output:
+@app.post("/ansible-install/nginx/")
+async def ansible_install_generation_nginx(request:AnsibleInstallNginx) -> Output:
+
         if os.environ.get("TEST"):
             return Output(output='output')
-        generated_prompt = ansible_install_template(request)
+        generated_prompt = ansible_install_template(request,"nginx")
         output = gpt_service(generated_prompt)
         edit_directory_generator("ansible_generator",output)
         execute_pythonfile("MyAnsible","ansible_generator")

app/template_generators/ansible/install/main.py

@@ -1,8 +1,9 @@
 from .docker import ansible_docker_install
 from .nginx import ansible_nginx_install
 from fastapi import HTTPException
-def ansible_install_template(input_):
-    match input_.tool:
+
+def ansible_install_template(input_, tool:str):
+    match tool:
         case 'nginx':
             return ansible_nginx_install(input_)
         case 'docker':