import os import subprocess from typing import Optional from pydantic import BaseModel, Field from langchain.tools import Tool import shutil class Tools: def create_resource_group(resource_group_name: str, location: str) -> str: """ Modify rg-main.tf file and create a Resource Group by running Terraform.""" missingRG = False missingLocation = False # Error handling for empty or None inputs if not resource_group_name or resource_group_name.lower() in ["unknown", "none", "none"]: missingRG = True if not location or location.lower() in ["unknown", "none", "none"]: missingLocation = True if missingRG and missingLocation: return "Both Resource Group name and Location are missing." elif missingRG: return "Resource Group name is missing." elif missingLocation: return "Location is missing." try: new_directory = os.path.expanduser(f"~/autogen/RG/{resource_group_name}") if not os.path.exists(new_directory): os.makedirs(new_directory) else: return f"Resource Group '{resource_group_name}' already exists." original_tf_file_path = os.path.expanduser("~/autogen/RG/rg-main.tf") new_tf_file_path = os.path.join(new_directory, "rg-main.tf") shutil.copy(original_tf_file_path, new_tf_file_path) with open(new_tf_file_path, "r") as file: tf_code = file.read() tf_code = tf_code.replace("REPLACE_WITH_RG_NAME", resource_group_name) tf_code = tf_code.replace("REPLACE_WITH_LOCATION", location) with open(new_tf_file_path, "w") as file: file.write(tf_code) subprocess.run(["terraform", "init"], check=True, cwd=new_directory) plan_result = subprocess.run(["terraform", "plan"], capture_output=True, text=True, cwd=new_directory) if "0 to destroy" not in plan_result.stdout: return "There are resources to be destroyed. Please check the plan output." subprocess.run(["terraform", "apply", "-auto-approve"], check=True, cwd=new_directory) return f"Resource group '{resource_group_name}' in '{location}' created successfully." except subprocess.CalledProcessError as e: return f"Error executing Terraform code: {str(e)}" except Exception as e: return f"Error: {str(e)}"