conn

type AvailableCloudConnectionType struct {
	Subtype string
}

type AvailableDatabaseConnectionType struct {
	Subtype string
	Driver  string
}

type AvailableKubernetesConnectionType struct {
	Subtype string
}

type AzureConnection struct {
	BaseCloudConnection

	ResourceGroups []AzureResourceGroup
}

type AzureResourceGroup struct {
	Name string `json:"name"`
}

type BaseCloudConnection struct {
	Connection Connection
}

type BaseDatabaseConnection struct {
	Connection Connection
}

type BaseRDBMSConnection struct {
	BaseDatabaseConnection

	// TablesAndColumns is a map of tables and their columns.
	// This will be set via SetContext.
	TablesAndColumns map[string][]ColumnDetail

	// DB is the database connection.
	DB *sql.DB
}

type CloudConnectionDetails struct {
}

type CloudConnectionType struct {
	// MainType of the connection type.
	// Example: "cloud".
	MainType string `json:"mainType"`

	// Subtype of the cloud connection type.
	// Example: "aws", "gcp", "azure".
	Subtype string `json:"subtype"`
}

type ColumnDetail struct {
	Name     string
	DataType string
}

type Connection struct {
	// Name of the connection.
	Name string `json:"name"`

	// Type of the connection.
	// Example: "database", "cloud", "kubernetes".
	Type ConnectionType `json:"type"`

	// Details of the connection.
	// Can include different details based on the connection type.
	// Like database connection string, cloud credentials, etc.
	Details ConnectionDetails `json:"details"`
}

type ConnectionDetails interface {
	// GetDriver returns the driver name for the connection.
	// Example: "postgres", "mysql", "mongodb".
	// Can be empty if there is no driver.
	GetDriver() string
}

type ConnectionInterface interface {
	GetConnection() Connection
	CheckAuthentication() error

	// SetContext gets the necessary information for the connection.
	// For example, for a database connection, it can get the list of tables and columns.
	// For a cloud connection, it can get the list of resources.
	// For a kubernetes connection, it can get the list of deployments, services, etc.
	// This information will be sent to the AI model which will use it to generate the queries/commands.
	SetContext() error

	// GetContext returns the information set by the SetContext method.
	// This information will be sent to the AI model which will use it to generate the queries/commands.
	GetContext() string

	// GetFormattedContext returns the formatted context for the AI model.
	GetFormattedContext() (string, error)

	// ExecuteCommand executes the given command and returns the output as byte array.
	ExecuteCommand(command string) ([]byte, error)

	// FormatResultAsTable formats the result as a table.
	FormatResultAsTable(result []byte) (string, error)

	// GetCommand gets the command from AI using context and the user prompt.
	GetCommand(prompt string) (string, error)

	// GetAnswer gets the answer from AI using context and the user prompt.
	GetAnswer(prompt string) (string, error)

	// CommandType returns the type of the command.
	// Example: "psql", "az", "kubectl".
	CommandType() string
}

type ConnectionType interface {
	// GetMainType returns the main type of the connection.
	// Example: "database", "cloud", "kubernetes".
	GetMainType() string

	// GetSubtype returns the subtype of the connection.
	// Example: "postgres", "mysql", "aws", "gcp", "azure".
	// Can be empty if there is no subtype.
	GetSubtype() string
}

type DatabaseConnectionDetails struct {
	// ConnectionString is the connection string for the database.
	ConnectionString string `json:"connectionString"`

	// Driver is the driver name for the database.
	// Example: "postgres", "mysql", "mongodb".
	Driver string `json:"driver"`
}

type DatabaseConnectionType struct {
	// MainType of the connection type.
	// Example: "database".
	MainType string `json:"mainType"`

	// Subtype is the subtype of the database connection type.
	// Example: "PostgreSQL", "MySQL", "MongoDB".
	Subtype string `json:"subtype"`
}

type Deployment struct {
	Name      string `json:"name"`
	Namespace string `json:"namespace"`
}

type KubernetesConnectionDetails struct {
	// SelectedContext is the selected context for the kubernetes connection.
	SelectedContext string `json:"selectedContext"`
}

type KubernetesConnectionImpl struct {
	Connection Connection

	Namespaces  []Namespace
	Pods        []Pod
	Deployments []Deployment
	Services    []Service
}

type KubernetesConnectionType struct {
	// MainType of the connection type.
	// Example: "kubernetes".
	MainType string `json:"mainType"`
}

type Namespace struct {
	Name string `json:"name"`
}

type Pod struct {
	Name      string `json:"name"`
	Namespace string `json:"namespace"`
}

type PostgreSQLConnection struct {
	BaseRDBMSConnection
}

type Service struct {
	Name      string `json:"name"`
	Namespace string `json:"namespace"`
}