kafka

type BrokerConfig struct {
	Runtime      BrokerRuntime
	Ports        *portalloc.Allocator
	Probe        func(ctx context.Context, addr string) error
	CA           *devtls.CA
	ScramAdmin   ScramAdmin
	Host         string
	Image        string
	StartTimeout time.Duration
}

type BrokerEBSVolumeInfo struct {
	ProvisionedThroughput *ProvisionedThroughput `json:"provisionedThroughput,omitempty"`
	KafkaBrokerNodeID     string                 `json:"kafkaBrokerNodeId,omitempty"`
	VolumeSizeGB          int32                  `json:"volumeSizeGB,omitempty"`
}

type BrokerLogs struct {
	CloudWatchLogs *CloudWatchLogs `json:"cloudWatchLogs,omitempty"`
	Firehose       *Firehose       `json:"firehose,omitempty"`
	S3             *S3Logs         `json:"s3,omitempty"`
}

type BrokerNodeGroupInfo struct {
	ConnectivityInfo     *ConnectivityInfo `json:"connectivityInfo,omitempty"`
	StorageInfo          *StorageInfo      `json:"storageInfo,omitempty"`
	BrokerAZDistribution string            `json:"brokerAZDistribution,omitempty"`
	InstanceType         string            `json:"instanceType"`
	ZoneIDs              []string          `json:"zoneIds,omitempty"`
	ClientSubnets        []string          `json:"clientSubnets"`
	SecurityGroups       []string          `json:"securityGroups,omitempty"`
}

type BrokerRuntime interface {
	CreateAndStart(ctx context.Context, spec container.Spec) (string, error)
	StopAndRemove(ctx context.Context, containerID string) error
}

type Catalog struct {
	CatalogArn        string `json:"catalogArn,omitempty"`
	WarehouseLocation string `json:"warehouseLocation,omitempty"`
}

type Channel struct {
	Tags                            map[string]string                `json:"tags,omitempty"`
	EncryptionConfiguration         *ChannelEncryptionConfiguration  `json:"encryptionConfiguration,omitempty"`
	IcebergDestinationConfiguration *IcebergDestinationConfiguration `json:"icebergDestinationConfiguration,omitempty"`
	LoggingInfo                     *ChannelLoggingInfo              `json:"loggingInfo,omitempty"`
	S3DestinationConfiguration      *S3DestinationConfiguration      `json:"s3DestinationConfiguration,omitempty"`
	StateInfo                       *ChannelStateInfo                `json:"stateInfo,omitempty"`
	ClusterArn                      string                           `json:"clusterArn"`
	ChannelArn                      string                           `json:"channelArn"`
	ChannelName                     string                           `json:"channelName"`
	ClusterOperationArn             string                           `json:"clusterOperationArn,omitempty"`
	CreationTime                    string                           `json:"creationTime,omitempty"`
	DestinationType                 string                           `json:"destinationType"`
	Status                          string                           `json:"status"`
	TopicConfigurationList          []TopicConfiguration             `json:"topicConfigurationList,omitempty"`
}

type ChannelEncryptionConfiguration struct {
	KmsKeyArn string `json:"kmsKeyArn,omitempty"`
}

type ChannelLoggingInfo struct {
	CloudWatchLogs *CloudWatchLogs `json:"cloudWatchLogs,omitempty"`
	Firehose       *Firehose       `json:"firehose,omitempty"`
	S3             *S3Logs         `json:"s3,omitempty"`
}

type ChannelStateInfo struct {
	Code    string `json:"code,omitempty"`
	Message string `json:"message,omitempty"`
}

type ClientAuthentication struct {
	Sasl            *SaslSettings            `json:"sasl,omitempty"`
	TLS             *TLSSettings             `json:"tls,omitempty"`
	Unauthenticated *UnauthenticatedSettings `json:"unauthenticated,omitempty"`
}

type CloudWatchLogs struct {
	LogGroup string `json:"logGroup,omitempty"`
	Enabled  bool   `json:"enabled"`
}

type Cluster struct {
	Tags                 map[string]string      `json:"-"`
	ClientAuthentication *ClientAuthentication  `json:"clientAuthentication,omitempty"`
	EncryptionInfo       *EncryptionInfo        `json:"encryptionInfo,omitempty"`
	OpenMonitoring       *OpenMonitoring        `json:"openMonitoring,omitempty"`
	LoggingInfo          *LoggingInfo           `json:"loggingInfo,omitempty"`
	StateInfo            *StateInfo             `json:"stateInfo,omitempty"`
	Serverless           *ServerlessClusterInfo `json:"serverless,omitempty"`
	ConfigurationInfo    *ConfigurationInfo     `json:"configurationInfo,omitempty"`
	Rebalancing          *Rebalancing           `json:"rebalancing,omitempty"`
	KafkaVersion         string                 `json:"kafkaVersion,omitempty"`
	StorageMode          string                 `json:"storageMode,omitempty"`
	ClusterName          string                 `json:"clusterName"`
	State                string                 `json:"state"`
	CurrentVersion       string                 `json:"currentVersion"`
	ActiveOperationArn   string                 `json:"activeOperationArn,omitempty"`
	EnhancedMonitoring   string                 `json:"enhancedMonitoring,omitempty"`
	ClusterType          string                 `json:"clusterType"`
	CreationTime         string                 `json:"creationTime,omitempty"`
	ClusterArn           string                 `json:"clusterArn"`

	BrokerNodeGroupInfo BrokerNodeGroupInfo `json:"brokerNodeGroupInfo"`

	NumberOfBrokerNodes int32 `json:"numberOfBrokerNodes"`
	// contains filtered or unexported fields
}

type ClusterConfig struct {
	ClientAuthentication  *ReplicatorClientAuthentication `json:"clientAuthentication,omitempty"`
	EncryptionInTransit   *ReplicatorEncryptionInTransit  `json:"encryptionInTransit,omitempty"`
	MskClusterArn         string                          `json:"mskClusterArn"`
	ApacheKafkaClusterID  string                          `json:"apacheKafkaClusterId,omitempty"`
	BootstrapBrokerString string                          `json:"bootstrapBrokerString,omitempty"`
	Alias                 string                          `json:"-"`
	SubnetIDs             []string                        `json:"subnetIds,omitempty"`
	SecurityGroupIDs      []string                        `json:"securityGroupIds,omitempty"`
}

type ClusterCreateOptions struct {
	ConfigurationInfo  *ConfigurationInfo
	EncryptionInfo     *EncryptionInfo
	LoggingInfo        *LoggingInfo
	OpenMonitoring     *OpenMonitoring
	Rebalancing        *Rebalancing
	EnhancedMonitoring string
	StorageMode        string
}

type ClusterOperation struct {
	SourceClusterInfo   *MutableClusterInfo `json:"sourceClusterInfo,omitempty"`
	TargetClusterInfo   *MutableClusterInfo `json:"targetClusterInfo,omitempty"`
	ClusterOperationArn string              `json:"operationArn"`
	ClusterArn          string              `json:"clusterArn"`
	OperationType       string              `json:"operationType"`
	OperationState      string              `json:"operationState"`
	ClientRequestID     string              `json:"clientRequestId,omitempty"`
	CreationTime        string              `json:"creationTime,omitempty"`
	EndTime             string              `json:"endTime,omitempty"`
	// contains filtered or unexported fields
}

type CompatibleKafkaVersion struct {
	SourceVersion  string   `json:"sourceVersion"`
	TargetVersions []string `json:"targetVersions"`
}

type Configuration struct {
	Tags           map[string]string      `json:"-"`
	LatestRevision *ConfigurationRevision `json:"latestRevision"`
	// PriorRevisions holds superseded revisions, oldest first.
	PriorRevisions   []*ConfigurationRevision `json:"priorRevisions,omitempty"`
	Arn              string                   `json:"arn"`
	Name             string                   `json:"name"`
	Description      string                   `json:"description"`
	ServerProperties string                   `json:"serverProperties"`
	CreationTime     string                   `json:"creationTime"`
	State            string                   `json:"state"`
	KafkaVersions    []string                 `json:"kafkaVersions"`
}

type ConfigurationInfo struct {
	Arn      string `json:"arn"`
	Revision int64  `json:"revision"`
}

type ConfigurationRevision struct {
	ConfigurationArn string `json:"configurationArn"`
	Description      string `json:"description"`
	ServerProperties string `json:"serverProperties,omitempty"`
	CreationTime     string `json:"creationTime"`
	Revision         int64  `json:"revision"`
}

type ConnectivityInfo struct {
	PublicAccess    *PublicAccess    `json:"publicAccess,omitempty"`
	VpcConnectivity *VpcConnectivity `json:"vpcConnectivity,omitempty"`
}

type ConsumerGroupReplicationConfig struct {
	ConsumerGroupsToExclude         []string `json:"consumerGroupsToExclude,omitempty"`
	ConsumerGroupsToReplicate       []string `json:"consumerGroupsToReplicate,omitempty"`
	DetectAndCopyNewConsumerGroups  bool     `json:"detectAndCopyNewConsumerGroups"`
	SynchroniseConsumerGroupOffsets bool     `json:"synchroniseConsumerGroupOffsets"`
}

type ControllerNodeInfo struct {
	Endpoints []string `json:"endpoints,omitempty"`
}

type DeadLetterQueueS3 struct {
	BucketArn           string `json:"bucketArn,omitempty"`
	ErrorOutputPrefix   string `json:"errorOutputPrefix,omitempty"`
	ExpectedBucketOwner string `json:"expectedBucketOwner,omitempty"`
}

type DestinationTable struct {
	PartitionSpec           *PartitionSpec `json:"partitionSpec,omitempty"`
	DestinationDatabaseName string         `json:"destinationDatabaseName,omitempty"`
	DestinationTableName    string         `json:"destinationTableName,omitempty"`
}

type EBSStorageInfo struct {
	ProvisionedThroughput *ProvisionedThroughput `json:"provisionedThroughput,omitempty"`
	VolumeSize            int32                  `json:"volumeSize,omitempty"`
}

type EncryptionAtRest struct {
	DataVolumeKMSKeyID string `json:"dataVolumeKMSKeyId,omitempty"`
}

type EncryptionInTransit struct {
	ClientBroker string `json:"clientBroker,omitempty"`
	InCluster    bool   `json:"inCluster"`
}

type EncryptionInfo struct {
	EncryptionAtRest    *EncryptionAtRest    `json:"encryptionAtRest,omitempty"`
	EncryptionInTransit *EncryptionInTransit `json:"encryptionInTransit,omitempty"`
}

type EngineCADirConfig interface {
	GetKafkaCADir() string
}

type EngineConfig interface {
	GetKafkaEngine() string
}

type Firehose struct {
	DeliveryStream string `json:"deliveryStream,omitempty"`
	Enabled        bool   `json:"enabled"`
}

type Handler struct {
	Backend StorageBackend
}

type IAMAuthConfig struct {
	Identities IAMAuthIdentityProvider
	Sessions   IAMAuthSessionProvider
	// Enforce evaluates kafka-cluster:Connect against the caller's policies; off, any known principal connects.
	Enforce bool
}

type IAMAuthIdentityProvider interface {
	ListAllAccessKeys() []iam.AccessKey
	GetUserByAccessKeyID(accessKeyID string) (*iam.User, error)
	GetPoliciesForUser(userName string) ([]string, error)
	GetPoliciesForRole(roleName string) ([]string, error)
}

type IAMAuthResolver func() IAMAuthConfig

type IAMAuthSessionProvider interface {
	LookupSession(accessKeyID, sessionToken string) *sts.SessionInfo
}

type IcebergDestinationConfiguration struct {
	Catalog                 *Catalog           `json:"catalog,omitempty"`
	DeadLetterQueueS3       *DeadLetterQueueS3 `json:"deadLetterQueueS3,omitempty"`
	SchemaEvolution         *SchemaEvolution   `json:"schemaEvolution,omitempty"`
	TableCreation           *TableCreation     `json:"tableCreation,omitempty"`
	CompressionType         string             `json:"compressionType,omitempty"`
	ServiceExecutionRoleArn string             `json:"serviceExecutionRoleArn,omitempty"`
	DestinationTableList    []DestinationTable `json:"destinationTableList,omitempty"`
	DataFreshnessInSeconds  int32              `json:"dataFreshnessInSeconds,omitempty"`
	AppendOnly              bool               `json:"appendOnly"`
}

type IcebergDestinationUpdate struct {
	DataFreshnessInSeconds int32 `json:"dataFreshnessInSeconds"`
}

type InMemoryBackend struct {
	// contains filtered or unexported fields
}

type JmxExporter struct {
	EnabledInBroker bool `json:"enabledInBroker"`
}

type LogDelivery struct {
	ReplicatorLogDelivery *ReplicatorLogDelivery `json:"replicatorLogDelivery,omitempty"`
}

type LoggingInfo struct {
	BrokerLogs *BrokerLogs `json:"brokerLogs,omitempty"`
}

type MSKVersion struct {
	Version string `json:"version"`
	Status  string `json:"status"`
}

type ManagedEndpoints struct {
	Plaintext string
	TLS       string
	SaslScram string
	SaslIam   string
}

type MutableClusterInfo struct {
	ConnectivityInfo     *ConnectivityInfo     `json:"connectivityInfo,omitempty"`
	OpenMonitoring       *OpenMonitoring       `json:"openMonitoring,omitempty"`
	LoggingInfo          *LoggingInfo          `json:"loggingInfo,omitempty"`
	ClientAuthentication *ClientAuthentication `json:"clientAuthentication,omitempty"`
	EncryptionInfo       *EncryptionInfo       `json:"encryptionInfo,omitempty"`
	Rebalancing          *Rebalancing          `json:"rebalancing,omitempty"`
	ConfigurationInfo    *ConfigurationInfo    `json:"configurationInfo,omitempty"`
	ZookeeperAccess      *ZookeeperAccess      `json:"zookeeperAccess,omitempty"`
	StorageMode          string                `json:"storageMode,omitempty"`
	EnhancedMonitoring   string                `json:"enhancedMonitoring,omitempty"`
	InstanceType         string                `json:"instanceType,omitempty"`
	KafkaVersion         string                `json:"kafkaVersion,omitempty"`
	BrokerEBSVolumeInfo  []BrokerEBSVolumeInfo `json:"brokerEBSVolumeInfo,omitempty"`
	NumberOfBrokerNodes  int32                 `json:"numberOfBrokerNodes,omitempty"`
}

type NodeBrokerInfo struct {
	CurrentBrokerSoftwareInfo *brokerSoftwareInfo `json:"currentBrokerSoftwareInfo,omitempty"`
	ClientSubnet              string              `json:"clientSubnet,omitempty"`
	BrokerID                  float64             `json:"brokerId"`
}

type NodeExporter struct {
	EnabledInBroker bool `json:"enabledInBroker"`
}

type NodeInfo struct {
	BrokerNodeInfo     *NodeBrokerInfo     `json:"brokerNodeInfo,omitempty"`
	ControllerNodeInfo *ControllerNodeInfo `json:"controllerNodeInfo,omitempty"`
	ZookeeperNodeInfo  *ZookeeperNodeInfo  `json:"zookeeperNodeInfo,omitempty"`
	AddedToClusterTime string              `json:"addedToClusterTime,omitempty"`
	InstanceType       string              `json:"instanceType,omitempty"`
	NodeARN            string              `json:"nodeARN,omitempty"`
	NodeType           string              `json:"nodeType,omitempty"`
}

type OpenMonitoring struct {
	Prometheus *PrometheusInfo `json:"prometheus,omitempty"`
}

type PartitionSource struct {
	SourceName string `json:"sourceName,omitempty"`
}

type PartitionSpec struct {
	PartitionStrategy string            `json:"partitionStrategy,omitempty"`
	SourceList        []PartitionSource `json:"sourceList,omitempty"`
}

type PrometheusInfo struct {
	JmxExporter  *JmxExporter  `json:"jmxExporter,omitempty"`
	NodeExporter *NodeExporter `json:"nodeExporter,omitempty"`
}

type Provider struct{}

type ProvisionedThroughput struct {
	VolumeThroughput int32 `json:"volumeThroughput,omitempty"`
	Enabled          bool  `json:"enabled"`
}

type PublicAccess struct {
	Type string `json:"type,omitempty"`
}

type Rebalancing struct {
	Status string `json:"status,omitempty"`
}

type RecordConverter struct {
	ValueConverter string `json:"valueConverter,omitempty"`
}

type RecordSchema struct {
	GsrArn string `json:"gsrArn,omitempty"`
}

type ReplicationInfoConfig struct {
	SourceKafkaClusterArn    string                         `json:"sourceKafkaClusterArn"`
	SourceKafkaClusterID     string                         `json:"sourceKafkaClusterId,omitempty"`
	TargetKafkaClusterArn    string                         `json:"targetKafkaClusterArn"`
	TargetKafkaClusterID     string                         `json:"targetKafkaClusterId,omitempty"`
	TargetCompressionType    string                         `json:"targetCompressionType,omitempty"`
	SourceAlias              string                         `json:"-"`
	TargetAlias              string                         `json:"-"`
	TopicReplication         TopicReplicationConfig         `json:"topicReplication"`
	ConsumerGroupReplication ConsumerGroupReplicationConfig `json:"consumerGroupReplication"`
}

type Replicator struct {
	Tags                    map[string]string       `json:"-"`
	LogDelivery             *LogDelivery            `json:"logDelivery,omitempty"`
	ReplicatorArn           string                  `json:"replicatorArn"`
	ReplicatorName          string                  `json:"replicatorName"`
	Description             string                  `json:"description,omitempty"`
	ServiceExecutionRoleArn string                  `json:"serviceExecutionRoleArn"`
	ReplicatorState         string                  `json:"replicatorState"`
	CurrentVersion          string                  `json:"currentVersion,omitempty"`
	CreationTime            string                  `json:"creationTime,omitempty"`
	StateInfoCode           string                  `json:"stateInfoCode,omitempty"`
	StateInfoMessage        string                  `json:"stateInfoMessage,omitempty"`
	KafkaClusters           []ClusterConfig         `json:"kafkaClusters,omitempty"`
	ReplicationInfoList     []ReplicationInfoConfig `json:"replicationInfoList,omitempty"`
}

type ReplicatorClientAuthentication struct {
	MTLSSecretArn      string `json:"mtlsSecretArn,omitempty"`
	SaslScramMechanism string `json:"saslScramMechanism,omitempty"`
	SaslScramSecretArn string `json:"saslScramSecretArn,omitempty"`
}

type ReplicatorEncryptionInTransit struct {
	EncryptionType    string `json:"encryptionType,omitempty"`
	RootCaCertificate string `json:"rootCaCertificate,omitempty"`
}

type ReplicatorLogDelivery struct {
	CloudWatchLogs *CloudWatchLogs `json:"cloudWatchLogs,omitempty"`
	Firehose       *Firehose       `json:"firehose,omitempty"`
	S3             *S3Logs         `json:"s3,omitempty"`
}

type S3DestinationConfiguration struct {
	DeadLetterQueueS3       *DeadLetterQueueS3 `json:"deadLetterQueueS3,omitempty"`
	Storage                 *S3Storage         `json:"storage,omitempty"`
	ServiceExecutionRoleArn string             `json:"serviceExecutionRoleArn,omitempty"`
	DataFreshnessInSeconds  int32              `json:"dataFreshnessInSeconds,omitempty"`
}

type S3DestinationUpdate struct {
	DataFreshnessInSeconds int32 `json:"dataFreshnessInSeconds"`
}

type S3Logs struct {
	Bucket  string `json:"bucket,omitempty"`
	Prefix  string `json:"prefix,omitempty"`
	Enabled bool   `json:"enabled"`
}

type S3Storage struct {
	BucketArn           string `json:"bucketArn,omitempty"`
	CompressionType     string `json:"compressionType,omitempty"`
	StorageClass        string `json:"storageClass,omitempty"`
	ExpectedBucketOwner string `json:"expectedBucketOwner,omitempty"`
	OutputKeyTemplate   string `json:"outputKeyTemplate,omitempty"`
	OutputPrefix        string `json:"outputPrefix,omitempty"`
}

type SaslIam struct {
	Enabled bool `json:"enabled"`
}

type SaslScram struct {
	Enabled bool `json:"enabled"`
}

type SaslSettings struct {
	Scram *SaslScram `json:"scram,omitempty"`
	Iam   *SaslIam   `json:"iam,omitempty"`
}

type SchemaEvolution struct {
	EnableSchemaEvolution bool `json:"enableSchemaEvolution"`
}

type ScramAdmin func(ctx context.Context, addr string, upsert []ScramUser, remove []string) error

type ScramCredential struct {
	Username string
	Password string
	// SecretName is the Secrets Manager name; MSK requires the AmazonMSK_ prefix.
	SecretName string
}

type ScramSecretError struct {
	SecretArn    string `json:"secretArn"`
	ErrorCode    string `json:"errorCode,omitempty"`
	ErrorMessage string `json:"errorMessage,omitempty"`
}

type ScramSecretResolver interface {
	ResolveScramSecret(ctx context.Context, secretArn string) (ScramCredential, error)
}

type ScramUser struct {
	Username string
	Password string
}

type ServerlessClientAuthentication struct {
	Sasl *SaslSettings `json:"sasl,omitempty"`
}

type ServerlessClusterInfo struct {
	ClientAuthentication *ServerlessClientAuthentication `json:"clientAuthentication,omitempty"`
	VpcConfigs           []ServerlessVpcConfig           `json:"vpcConfigs,omitempty"`
}

type ServerlessVpcConfig struct {
	SubnetIDs        []string `json:"subnetIds,omitempty"`
	SecurityGroupIDs []string `json:"securityGroupIds,omitempty"`
}

type StateInfo struct {
	Code    string `json:"code,omitempty"`
	Message string `json:"message,omitempty"`
}

type StorageBackend interface {
	// Cluster operations
	CreateCluster(
		ctx context.Context,
		name, kafkaVersion string,
		numBrokers int32,
		brokerInfo BrokerNodeGroupInfo,
		clientAuth *ClientAuthentication,
		tags map[string]string,
		opts ...ClusterCreateOptions,
	) (*Cluster, error)
	CreateServerlessCluster(
		ctx context.Context, name string, serverless *ServerlessClusterInfo, tags map[string]string,
	) (*Cluster, error)
	DescribeCluster(ctx context.Context, clusterArn string) (*Cluster, error)
	ListClusters(ctx context.Context) []*Cluster
	DeleteCluster(ctx context.Context, clusterArn string) error

	// Configuration operations
	CreateConfiguration(
		ctx context.Context,
		name, description string,
		kafkaVersions []string,
		serverProperties string,
	) (*Configuration, error)
	DescribeConfiguration(ctx context.Context, configArn string) (*Configuration, error)
	ListConfigurations(ctx context.Context) []*Configuration
	DeleteConfiguration(ctx context.Context, configArn string) error

	// Tag operations
	TagResource(ctx context.Context, resourceArn string, tags map[string]string) error
	UntagResource(ctx context.Context, resourceArn string, tagKeys []string) error
	GetTags(ctx context.Context, resourceArn string) (map[string]string, error)

	// SCRAM secret operations
	BatchAssociateScramSecret(
		ctx context.Context,
		clusterArn string,
		secretArnList []string,
	) ([]ScramSecretError, error)
	BatchDisassociateScramSecret(
		ctx context.Context, clusterArn string, secretArnList []string,
	) ([]ScramSecretError, error)

	// Replicator operations
	CreateReplicator(
		ctx context.Context,
		name, description, serviceExecutionRoleArn string,
		kafkaClusters []ClusterConfig,
		replicationInfoList []ReplicationInfoConfig,
		tags map[string]string,
		logDelivery *LogDelivery,
	) (*Replicator, error)
	DeleteReplicator(ctx context.Context, replicatorArn string) error
	DescribeReplicator(ctx context.Context, replicatorArn string) (*Replicator, error)
	ListReplicators(ctx context.Context) []*Replicator
	UpdateReplicationInfo(
		ctx context.Context,
		replicatorArn, currentVersion, sourceKafkaClusterArn, targetKafkaClusterArn string,
		sourceKafkaClusterID, targetKafkaClusterID string,
		topicReplication *TopicReplicationConfig,
		consumerGroupReplication *ConsumerGroupReplicationConfig,
		logDelivery *LogDelivery,
	) (*Replicator, error)

	// Topic operations
	CreateTopic(
		ctx context.Context,
		clusterArn, topicName string,
		replicationFactor, partitionCount int32,
		configs string,
	) (*Topic, error)
	DeleteTopic(ctx context.Context, clusterArn, topicName string) error
	DescribeTopic(ctx context.Context, clusterArn, topicName string) (*Topic, error)
	DescribeTopicPartitions(ctx context.Context, clusterArn, topicName string) ([]*TopicPartitionInfo, error)
	ListTopics(ctx context.Context, clusterArn, topicNameFilter string) ([]*Topic, error)
	UpdateTopic(
		ctx context.Context, clusterArn, topicName string, partitionCount int32, configs string,
	) (*Topic, error)

	// VPC connection operations
	CreateVpcConnection(
		ctx context.Context,
		targetClusterArn, vpcID, authentication string,
		clientSubnets, securityGroups []string,
		tags map[string]string,
	) (*VpcConnection, error)
	DeleteVpcConnection(ctx context.Context, vpcConnectionArn string) error
	DescribeVpcConnection(ctx context.Context, vpcConnectionArn string) (*VpcConnection, error)
	ListVpcConnections(ctx context.Context) []*VpcConnection
	ListClientVpcConnections(ctx context.Context, clusterArn string) ([]*VpcConnection, error)
	RejectClientVpcConnection(ctx context.Context, vpcConnectionArn string) error

	// Cluster policy operations
	DeleteClusterPolicy(ctx context.Context, clusterArn string) error
	GetClusterPolicy(ctx context.Context, clusterArn string) (policy, version string, err error)
	PutClusterPolicy(ctx context.Context, clusterArn, policy, currentVersion string) (string, error)

	// Cluster operation operations
	DescribeClusterOperation(ctx context.Context, clusterOperationArn string) (*ClusterOperation, error)
	DescribeClusterOperationV2(ctx context.Context, clusterOperationArn string) (*ClusterOperation, error)
	ListClusterOperations(ctx context.Context, clusterArn string) ([]*ClusterOperation, error)
	ListClusterOperationsV2(ctx context.Context, clusterArn string) ([]*ClusterOperation, error)

	// Configuration revision operations
	DescribeConfigurationRevision(ctx context.Context, configArn string, revision int64) (*ConfigurationRevision, error)
	UpdateConfiguration(ctx context.Context, configArn, description, serverProperties string) (*Configuration, error)
	ListConfigurationRevisions(ctx context.Context, configArn string) ([]*ConfigurationRevision, error)

	// Broker / cluster update operations
	UpdateBrokerCount(ctx context.Context, clusterArn string, numBrokers int32) (*ClusterOperation, error)
	UpdateBrokerStorage(ctx context.Context, clusterArn string, volumeSize int32) (*ClusterOperation, error)
	UpdateBrokerType(ctx context.Context, clusterArn, instanceType string) (*ClusterOperation, error)
	UpdateClusterConfiguration(
		ctx context.Context,
		clusterArn, configArn string,
		revision int64,
	) (*ClusterOperation, error)
	UpdateClusterKafkaVersion(
		ctx context.Context,
		clusterArn, targetKafkaVersion string,
		configuration *ConfigurationInfo,
	) (*ClusterOperation, error)
	UpdateConnectivity(
		ctx context.Context,
		clusterArn string,
		settings UpdateConnectivitySettings,
	) (*ClusterOperation, error)
	UpdateMonitoring(
		ctx context.Context,
		clusterArn string,
		settings UpdateMonitoringSettings,
	) (*ClusterOperation, error)
	UpdateRebalancing(ctx context.Context, clusterArn, status string) (*ClusterOperation, error)
	UpdateSecurity(ctx context.Context, clusterArn string, settings UpdateSecuritySettings) (*ClusterOperation, error)
	UpdateStorage(ctx context.Context, clusterArn string, settings UpdateStorageSettings) (*ClusterOperation, error)
	RebootBroker(ctx context.Context, clusterArn string, brokerIDs []string) (*ClusterOperation, error)

	// SCRAM secret list
	ListScramSecrets(ctx context.Context, clusterArn string) ([]string, error)

	// Node / version ops
	ListNodes(ctx context.Context, clusterArn string) ([]*NodeInfo, error)
	ListKafkaVersions(ctx context.Context) []*MSKVersion
	GetCompatibleKafkaVersions(ctx context.Context, clusterArn string) ([]*CompatibleKafkaVersion, error)

	// Channel operations
	CreateChannel(
		ctx context.Context,
		clusterArn, channelName string,
		topicConfigurationList []TopicConfiguration,
		encryptionConfiguration *ChannelEncryptionConfiguration,
		icebergDestinationConfiguration *IcebergDestinationConfiguration,
		s3DestinationConfiguration *S3DestinationConfiguration,
		loggingInfo *ChannelLoggingInfo,
		tags map[string]string,
	) (*Channel, error)
	DeleteChannel(ctx context.Context, clusterArn, channelArn string) (*Channel, error)
	DescribeChannel(ctx context.Context, clusterArn, channelArn string) (*Channel, error)
	ListChannels(ctx context.Context, clusterArn, topicNameFilter string) ([]*Channel, error)
	UpdateChannel(
		ctx context.Context,
		clusterArn, channelArn string,
		icebergDestinationUpdate *IcebergDestinationUpdate,
		s3DestinationUpdate *S3DestinationUpdate,
	) (*Channel, error)

	// Lifecycle
	Reset()
	Region() string
	AccountID() string
	Snapshot(ctx context.Context) []byte
	Restore(ctx context.Context, data []byte) error
}

type StorageInfo struct {
	EbsStorageInfo *EBSStorageInfo `json:"ebsStorageInfo,omitempty"`
}

type TLSSettings struct {
	CertificateAuthorityArnList []string `json:"certificateAuthorityArnList,omitempty"`
	Enabled                     bool     `json:"enabled"`
}

type TableCreation struct {
	EnableTableCreation bool `json:"enableTableCreation"`
}

type TaggedEntry struct {
	Tags map[string]string
	ARN  string
}

type Topic struct {
	ClusterArn        string `json:"clusterArn"`
	Configs           string `json:"configs,omitempty"`
	Status            string `json:"status"`
	TopicArn          string `json:"topicArn"`
	TopicName         string `json:"topicName"`
	PartitionCount    int32  `json:"partitionCount"`
	ReplicationFactor int32  `json:"replicationFactor"`
}

type TopicConfiguration struct {
	RecordConverter *RecordConverter `json:"recordConverter,omitempty"`
	RecordSchema    *RecordSchema    `json:"recordSchema,omitempty"`
	TopicArn        string           `json:"topicArn,omitempty"`
}

type TopicInfo struct {
	TopicArn              string `json:"topicArn"`
	TopicName             string `json:"topicName"`
	PartitionCount        int32  `json:"partitionCount"`
	ReplicationFactor     int32  `json:"replicationFactor"`
	OutOfSyncReplicaCount int32  `json:"outOfSyncReplicaCount"`
}

type TopicPartitionInfo struct {
	Isr       []int32 `json:"isr"`
	Replicas  []int32 `json:"replicas"`
	Partition int32   `json:"partition"`
	Leader    int32   `json:"leader"`
}

type TopicReplicationConfig struct {
	StartingPositionType            string   `json:"startingPositionType,omitempty"`
	TopicNameConfigurationType      string   `json:"topicNameConfigurationType,omitempty"`
	TopicsToExclude                 []string `json:"topicsToExclude,omitempty"`
	TopicsToReplicate               []string `json:"topicsToReplicate,omitempty"`
	CopyAccessControlListsForTopics bool     `json:"copyAccessControlListsForTopics"`
	CopyTopicConfigurations         bool     `json:"copyTopicConfigurations"`
	DetectAndCopyNewTopics          bool     `json:"detectAndCopyNewTopics"`
}

type UnauthenticatedSettings struct {
	Enabled bool `json:"enabled"`
}

type UpdateConnectivitySettings struct {
	ConnectivityInfo *ConnectivityInfo
	ZookeeperAccess  *ZookeeperAccess
}

type UpdateMonitoringSettings struct {
	OpenMonitoring     *OpenMonitoring
	LoggingInfo        *LoggingInfo
	EnhancedMonitoring string
}

type UpdateSecuritySettings struct {
	ClientAuthentication *ClientAuthentication
	EncryptionInfo       *EncryptionInfo
}

type UpdateStorageSettings struct {
	ProvisionedThroughput *ProvisionedThroughput
	StorageMode           string
	VolumeSizeGB          int32
}

type VpcConnection struct {
	Tags             map[string]string `json:"-"`
	VpcConnectionArn string            `json:"vpcConnectionArn"`
	TargetClusterArn string            `json:"targetClusterArn"`
	VpcID            string            `json:"vpcId"`
	Authentication   string            `json:"authentication,omitempty"`
	State            string            `json:"state"`
	CreationTime     string            `json:"creationTime,omitempty"`
	SubnetIDs        []string          `json:"subnetIds,omitempty"`
	SecurityGroupIDs []string          `json:"securityGroupIds,omitempty"`
}

type VpcConnectivity struct {
	ClientAuthentication *VpcConnectivityClientAuthentication `json:"clientAuthentication,omitempty"`
}

type VpcConnectivityClientAuthentication struct {
	Sasl *VpcConnectivitySasl `json:"sasl,omitempty"`
	TLS  *VpcConnectivityTLS  `json:"tls,omitempty"`
}

type VpcConnectivitySasl struct {
	Iam   *VpcConnectivitySaslIam   `json:"iam,omitempty"`
	Scram *VpcConnectivitySaslScram `json:"scram,omitempty"`
}

type VpcConnectivitySaslIam struct {
	Enabled bool `json:"enabled"`
}

type VpcConnectivitySaslScram struct {
	Enabled bool `json:"enabled"`
}

type VpcConnectivityTLS struct {
	Enabled bool `json:"enabled"`
}

type ZookeeperAccess struct {
	Enabled *bool `json:"enabled,omitempty"`
}

type ZookeeperNodeInfo struct {
	AttachedENIId      string   `json:"attachedENIId,omitempty"`
	ClientVpcIPAddress string   `json:"clientVpcIpAddress,omitempty"`
	ZookeeperVersion   string   `json:"zookeeperVersion,omitempty"`
	Endpoints          []string `json:"endpoints,omitempty"`
	ZookeeperID        float64  `json:"zookeeperId,omitempty"`
}