kafka

type Connect interface {
	// GetInfo gets information about the Kafka Connect cluster
	GetInfo(ctx context.Context) (map[string]interface{}, error)
	// ListConnectors lists all connectors
	ListConnectors(ctx context.Context) ([]string, error)
	// GetConnector gets information about a connector
	GetConnector(ctx context.Context, name string) (*ConnectorInfo, error)
	// CreateConnector creates a new connector
	CreateConnector(ctx context.Context, name string, config map[string]string) (*ConnectorInfo, error)
	// UpdateConnector updates a connector
	UpdateConnector(ctx context.Context, name string, config map[string]string) (*ConnectorInfo, error)
	// DeleteConnector deletes a connector
	DeleteConnector(ctx context.Context, name string) error
	// PauseConnector pauses a connector
	PauseConnector(ctx context.Context, name string) error
	// ResumeConnector resumes a connector
	ResumeConnector(ctx context.Context, name string) error
	// RestartConnector restarts a connector
	RestartConnector(ctx context.Context, name string) error
	// GetConnectorConfig gets the configuration of a connector
	GetConnectorConfig(ctx context.Context, name string) (map[string]string, error)
	// GetConnectorStatus gets the status of a connector
	GetConnectorStatus(ctx context.Context, name string) (*ConnectorInfo, error)
	// GetConnectorTasks gets the tasks of a connector
	GetConnectorTasks(ctx context.Context, name string) ([]TaskInfo, error)
	// ListPlugins lists all connector plugins
	ListPlugins(ctx context.Context) ([]PluginInfo, error)
	// ValidateConfig validates a connector configuration
	ValidateConfig(ctx context.Context, pluginClass string, config map[string]string) (map[string]interface{}, error)
}

type ConnectorInfo struct {
	// Name is the connector name
	Name string
	// Config is the connector configuration
	Config map[string]string
	// Tasks is the list of tasks
	Tasks []TaskInfo
	// Type is the connector type (source or sink)
	Type string
	// State is the connector state
	State ConnectorState
}

type ConnectorState string

type KafkaContext struct {
	BootstrapServers  string
	AuthType          string
	AuthMechanism     string
	AuthUser          string
	AuthPass          string
	UseTLS            bool
	ClientKeyFile     string
	ClientCertFile    string
	CaFile            string
	SchemaRegistryURL string
	ConnectURL        string

	SchemaRegistryAuthUser    string
	SchemaRegistryAuthPass    string
	SchemaRegistryBearerToken string

	ConnectAuthUser string
	ConnectAuthPass string
}

type PluginInfo struct {
	// Class is the plugin class
	Class string
	// Type is the plugin type (source or sink)
	Type string
	// Version is the plugin version
	Version string
}

type SASLConfig struct {
	Mechanism string
	Username  string
	Password  string
}

type Session struct {
	Ctx                  KafkaContext
	Client               *kgo.Client
	AdminClient          *kadm.Client
	SchemaRegistryClient *sr.Client
	ConnectClient        Connect
	Options              []kgo.Opt
	// contains filtered or unexported fields
}

type TLSConfig struct {
	Enabled        bool
	ClientKeyFile  string
	ClientCertFile string
	CaFile         string
}

type TaskInfo struct {
	// ID is the task ID
	ID int
	// State is the task state
	State ConnectorState
	// WorkerID is the worker ID
	WorkerID string
	// Trace is the error trace (if any)
	Trace string
}