engine

type ChatMessage struct {
	Type   string
	Result *GetRowsResult
	Text   string
}

type Column struct {
	Type             string
	Name             string
	IsPrimary        bool
	IsForeignKey     bool
	ReferencedTable  *string
	ReferencedColumn *string
	Length           *int // For VARCHAR(n), CHAR(n) types
	Precision        *int // For DECIMAL(p,s) types
	Scale            *int // For DECIMAL(p,s) types
}

type Credentials struct {
	Id          *string
	Type        string
	Hostname    string
	Username    string
	Password    string
	Database    string
	Advanced    []Record
	AccessToken *string
	IsProfile   bool
}

type DatabaseMetadata struct {
	DatabaseType    DatabaseType
	TypeDefinitions []TypeDefinition
	Operators       []string
	AliasMap        map[string]string
}

type DatabaseType string

type Engine struct {
	Plugins           []*Plugin
	LoginProfiles     []types.DatabaseCredentials
	ProfileRetrievers []LoginProfileRetriever
}

type ExternalModel struct {
	Type     string // Provider type: "OpenAI", "Anthropic", "Ollama", etc.
	Token    string // API key
	Model    string // User-selected model: "gpt-4o", "claude-sonnet-4", etc.
	Endpoint string // Base URL (for Ollama/generic providers)
}

type ForeignKeyRelationship struct {
	ColumnName       string
	ReferencedTable  string
	ReferencedColumn string
}

type GetRowsResult struct {
	Columns       []Column
	Rows          [][]string
	DisableUpdate bool
	TotalCount    int64
}

type GraphUnit struct {
	Unit      StorageUnit
	Relations []GraphUnitRelationship
}

type GraphUnitRelationship struct {
	Name             string
	RelationshipType GraphUnitRelationshipType
	SourceColumn     *string
	TargetColumn     *string
}

type GraphUnitRelationshipType string

type LoginProfileRetriever func() ([]types.DatabaseCredentials, error)

type Plugin struct {
	PluginFunctions
	Type DatabaseType
}

type PluginConfig struct {
	Credentials   *Credentials
	ExternalModel *ExternalModel
}

type PluginFunctions interface {
	GetDatabases(config *PluginConfig) ([]string, error)
	IsAvailable(config *PluginConfig) bool
	GetAllSchemas(config *PluginConfig) ([]string, error)
	GetStorageUnits(config *PluginConfig, schema string) ([]StorageUnit, error)
	StorageUnitExists(config *PluginConfig, schema string, storageUnit string) (bool, error)
	AddStorageUnit(config *PluginConfig, schema string, storageUnit string, fields []Record) (bool, error)
	UpdateStorageUnit(config *PluginConfig, schema string, storageUnit string, values map[string]string, updatedColumns []string) (bool, error)
	AddRow(config *PluginConfig, schema string, storageUnit string, values []Record) (bool, error)
	DeleteRow(config *PluginConfig, schema string, storageUnit string, values map[string]string) (bool, error)
	GetRows(config *PluginConfig, schema string, storageUnit string, where *model.WhereCondition, sort []*model.SortCondition, pageSize int, pageOffset int) (*GetRowsResult, error)
	GetRowCount(config *PluginConfig, schema string, storageUnit string, where *model.WhereCondition) (int64, error)
	GetGraph(config *PluginConfig, schema string) ([]GraphUnit, error)
	RawExecute(config *PluginConfig, query string) (*GetRowsResult, error)
	Chat(config *PluginConfig, schema string, previousConversation string, query string) ([]*ChatMessage, error)
	ExportData(config *PluginConfig, schema string, storageUnit string, writer func([]string) error, selectedRows []map[string]any) error
	FormatValue(val any) string
	GetColumnsForTable(config *PluginConfig, schema string, storageUnit string) ([]Column, error)

	// Mock data generation methods
	GetColumnConstraints(config *PluginConfig, schema string, storageUnit string) (map[string]map[string]any, error)
	ClearTableData(config *PluginConfig, schema string, storageUnit string) (bool, error)

	// Foreign key detection
	GetForeignKeyRelationships(config *PluginConfig, schema string, storageUnit string) (map[string]*ForeignKeyRelationship, error)

	// Transaction support
	WithTransaction(config *PluginConfig, operation func(tx any) error) error

	// Database metadata for frontend type/operator configuration
	GetDatabaseMetadata() *DatabaseMetadata
}

type Record struct {
	Key   string
	Value string
	Extra map[string]string
}

type StorageUnit struct {
	Name       string
	Attributes []Record
}

type TypeCategory string

type TypeDefinition struct {
	ID               string       // Canonical type name (e.g., "VARCHAR", "INTEGER")
	Label            string       // Display label for UI (e.g., "varchar", "integer")
	HasLength        bool         // Shows length input when selected (VARCHAR, CHAR)
	HasPrecision     bool         // Shows precision/scale inputs (DECIMAL, NUMERIC)
	DefaultLength    *int         // Default length for types with HasLength
	DefaultPrecision *int         // Default precision for types with HasPrecision
	Category         TypeCategory // Type category for grouping
}

type UnsupportedTypeError struct {
	TypeName     string
	DatabaseType string
}