IaC Memory MCP Server

Instructions

Implementation Reference

• src/iac_memory_mcp_server/db/terraform.py:344-386 (handler)

  The core handler function that updates the schema of a Terraform resource in the database. This implements the main logic for the 'update_resource_schema' tool.

  def update_resource_schema(
      db: DatabaseManager,
      resource_id: str,
      new_schema: str,
      new_version: Optional[str] = None,
      new_doc_url: Optional[str] = None,
  ) -> bool:
      """Update a Terraform resource's schema and optional fields."""
      logger.info(
          "Updating resource schema",
          extra={
              "resource_id": resource_id,
              "has_new_version": bool(new_version),
              "has_new_doc_url": bool(new_doc_url),
              "operation": "update_resource_schema",
          },
      )
      try:
          updates = ["schema = ?"]
          params = [new_schema]
  
          if new_version:
              updates.append("version = ?")
              params.append(new_version)
          if new_doc_url:
              updates.append("doc_url = ?")
              params.append(new_doc_url)
  
          updates.append("updated_at = CURRENT_TIMESTAMP")
          params.append(resource_id)
  
          with db.get_connection() as conn:
              cursor = conn.execute(
                  f"""UPDATE terraform_resources
                  SET {', '.join(updates)}
                  WHERE id = ?""",
                  tuple(params),
              )
              return cursor.rowcount > 0
      except sqlite3.Error as e:
          error_msg = f"Failed to update resource schema: {str(e)}"
          logger.error(error_msg)
          raise DatabaseError(error_msg)

• src/iac_memory_mcp_server/db/tools.py:222-232 (schema)

  JSON schema defining the input parameters and validation for the update_resource_schema tool.

  "update_resource_schema": {
      "type": "object",
      "description": "Update an existing Terraform resource's schema and related information",
      "required": ["resource_id", "new_schema"],
      "properties": {
          "resource_id": {"type": "string", "description": "Resource ID"},
          "new_schema": {"type": "string", "description": "New schema"},
          "new_version": {"type": "string", "description": "New version"},
          "new_doc_url": {"type": "string", "description": "New documentation URL"},
      },
  },

• src/iac_memory_mcp_server/db/__init__.py:29-35 (registration)

  Import and export of the update_resource_schema function in the db package __init__.py, making it available for use in tool handlers.

      add_terraform_provider,
      add_terraform_resource,
      get_provider_resources,
      get_resource_info,
      update_provider_version,
      update_resource_schema,
  )

• src/iac_memory_mcp_server/db/connection.py:282-319 (helper)

  Supporting method in DatabaseManager class that duplicates the resource schema update logic.

  def update_resource_schema(
      self,
      resource_id: str,
      new_schema: str,
      new_version: str | None = None,
      new_doc_url: str | None = None,
  ) -> bool:
      """Update a resource's schema."""
      try:
          with self.get_connection() as conn:
              conn.execute("BEGIN IMMEDIATE")
              try:
                  # Build update query dynamically
                  updates = ["schema = ?"]
                  params = [new_schema]
  
                  if new_version:
                      updates.append("version = ?")
                      params.append(new_version)
                  if new_doc_url:
                      updates.append("doc_url = ?")
                      params.append(new_doc_url)
  
                  updates.append("updated_at = CURRENT_TIMESTAMP")
                  params.append(resource_id)  # Add resource_id last
  
                  query = f"""UPDATE terraform_resources 
                            SET {', '.join(updates)}
                            WHERE id = ?"""
  
                  cursor = conn.execute(query, tuple(params))
                  conn.commit()
                  return cursor.rowcount > 0
              except Exception:
                  conn.rollback()
                  raise
      except sqlite3.Error as e:
          raise DatabaseError(f"Failed to update resource schema: {str(e)}")