from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import (
    AuthProvider,
    OpenMetadataConnection,
)
from metadata.generated.schema.security.client.openMetadataJWTClientConfig import (
    OpenMetadataJWTClientConfig,
)
from metadata.ingestion.ometa.ometa_api import OpenMetadata
from metadata.generated.schema.entity.data.topic import Topic

metadata = OpenMetadata(
    OpenMetadataConnection(
        hostPort="https://your-company.open-metadata.org/api",
        authProvider=AuthProvider.openmetadata,
        securityConfig=OpenMetadataJWTClientConfig(jwtToken="your-jwt-token"),
    )
)

# List all versions
history = metadata.list_versions(
    entity_id="0a021819-982e-4ee4-a5a2-4af30eaa3016",
    entity=Topic
)
for v in history.versions:
    topic_version = Topic.model_validate_json(v)
    print(f"Version {topic_version.version.root}: {topic_version.description.root}")

# Get a specific version
topic_v1 = metadata.get_entity_version(
    entity=Topic,
    entity_id="0a021819-982e-4ee4-a5a2-4af30eaa3016",
    version="0.1"
)
print(f"Original description: {topic_v1.description.root}")

import org.openmetadata.schema.entity.data.Topic;
import org.openmetadata.schema.type.EntityHistory;

// client is an initialized OpenMetadataClient.

// List all versions
EntityHistory history = client.topics().getVersionList("0a021819-982e-4ee4-a5a2-4af30eaa3016");

for (Object version : history.getVersions()) {
    System.out.println("Version: " + version);
}

// Get a specific version
Topic v1 = client.topics().getVersion("0a021819-982e-4ee4-a5a2-4af30eaa3016", 0.1);
System.out.println("Description: " + v1.getDescription());

# List all versions
curl "{base_url}/api/v1/topics/0a021819-982e-4ee4-a5a2-4af30eaa3016/versions" \
  -H "Authorization: Bearer {access_token}"

# Get a specific version
curl "{base_url}/api/v1/topics/0a021819-982e-4ee4-a5a2-4af30eaa3016/versions/0.1" \
  -H "Authorization: Bearer {access_token}"

{
  "entityType": "topic",
  "versions": [
    "{\"id\":\"0a021819-982e-4ee4-a5a2-4af30eaa3016\",\"name\":\"address_book\",\"fullyQualifiedName\":\"sample_kafka.address_book\",\"version\":0.2,\"description\":\"Updated: All Protobuf record events captured with enhanced schema\",\"serviceType\":\"Kafka\"}",
    "{\"id\":\"0a021819-982e-4ee4-a5a2-4af30eaa3016\",\"name\":\"address_book\",\"fullyQualifiedName\":\"sample_kafka.address_book\",\"version\":0.1,\"description\":\"All Protobuf record related events gets captured in this topic\",\"serviceType\":\"Kafka\"}"
  ]
}