add_terraform_provider

Instructions

Implementation Reference

• src/iac_memory_mcp_server/tools/terraform.py:164-212 (handler)

  MCP tool handler: validates input arguments (name, version, source_url, doc_url), calls the database helper function, logs operations, and returns TextContent with success message containing provider ID or error.

  async def handle_add_terraform_provider(db: Any, arguments: Dict[str, Any], operation_id: str) -> list[types.TextContent | types.ImageContent | types.EmbeddedResource]:
      """Handle add_terraform_provider tool."""
      # Validate required arguments
      required_args = ["name", "version", "source_url", "doc_url"]
      missing_args = [arg for arg in required_args if arg not in arguments]
      if missing_args:
          error_msg = f"Missing required arguments for add_terraform_provider: {', '.join(missing_args)}"
          logger.error(error_msg, extra={"operation_id": operation_id})
          return [types.TextContent(type="text", text=error_msg)]
  
      # Validate version format (x.y.z)
      version_pattern = re.compile(r'^\d+\.\d+\.\d+$')
      if not version_pattern.match(arguments["version"]):
          error_msg = "Invalid version format. Version must be in x.y.z format (e.g. 1.0.0)"
          logger.error(error_msg, extra={"operation_id": operation_id})
          return [types.TextContent(
              type="text",
              text=error_msg
          )]
  
      try:
          # Add provider
          logger.info(
              "Adding Terraform provider",
              extra={
                  "provider_name": arguments["name"],
                  "version": arguments["version"],
                  "operation_id": operation_id,
              },
          )
  
          provider_id = add_terraform_provider(
              db,
              arguments["name"],
              arguments["version"],
              arguments["source_url"],
              arguments["doc_url"],
          )
  
          return [types.TextContent(
              type="text",
              text=f"Added provider {arguments['name']} with ID: {provider_id}"
          )]
  
      except Exception as e:
          error_msg = f"Failed to add provider: {str(e)}"
          logger.error(error_msg, extra={"operation_id": operation_id})
          return [types.TextContent(type="text", text=error_msg)]

• src/iac_memory_mcp_server/tools/terraform.py:311-318 (registration)

  Tool handler registration dictionary mapping 'add_terraform_provider' to its async handler function.

  terraform_tool_handlers = {
      "get_terraform_provider_info": handle_get_terraform_provider_info,
      "list_provider_resources": handle_list_provider_resources,
      "get_terraform_resource_info": handle_get_terraform_resource_info,
      "add_terraform_provider": handle_add_terraform_provider,
      "add_terraform_resource": handle_add_terraform_resource,
      "update_provider_version": handle_update_provider_version,
  }

• src/iac_memory_mcp_server/db/tools.py:179-189 (schema)

  JSON schema defining required inputs (name, version, source_url, doc_url) and properties for the tool.

  "add_terraform_provider": {
      "type": "object",
      "description": "Add a new Terraform provider to the memory store with version and documentation information",
      "required": ["name", "version", "source_url", "doc_url"],
      "properties": {
          "name": {"type": "string", "description": "Provider name"},
          "version": {"type": "string", "description": "Provider version"},
          "source_url": {"type": "string", "description": "Source repository URL"},
          "doc_url": {"type": "string", "description": "Documentation URL"},
      },
  },

• src/iac_memory_mcp_server/db/terraform.py:13-52 (helper)

  Core database helper: inserts new provider record into 'terraform_providers' table using transaction, returns generated provider_id.

  def add_terraform_provider(
      db: DatabaseManager,
      name: str,
      version: str,
      source_url: str,
      doc_url: str,
  ) -> str:
      """Add a new Terraform provider with proper error handling."""
      logger.info(
          "Adding Terraform provider",
          extra={"provider_name": name, "version": version, "source_url": source_url},
      )
  
      try:
          with db.get_connection() as conn:
              conn.execute("BEGIN IMMEDIATE")
              try:
                  cursor = conn.execute(
                      """INSERT INTO terraform_providers 
                      (name, version, source_url, doc_url, updated_at)
                      VALUES (?, ?, ?, ?, CURRENT_TIMESTAMP)""",
                      (name, version, source_url, doc_url),
                  )
  
                  conn.commit()
                  provider_id = str(cursor.lastrowid)
  
                  logger.info(
                      f"Successfully added provider {name}",
                      extra={"provider_id": provider_id, "provider_name": name},
                  )
                  return provider_id
              except Exception:
                  conn.rollback()
                  raise
      except sqlite3.Error as e:
          error_msg = f"Failed to add Terraform provider: {str(e)}"
          logger.error(error_msg)
          raise DatabaseError(error_msg)