database

type Config struct {
	// Core database settings
	Driver          string
	DSN             string
	Host            string
	Port            int
	Database        string
	Username        string
	Password        string
	SSLMode         string
	MaxOpenConns    int
	MaxIdleConns    int
	ConnMaxLifetime time.Duration

	// TLS Configuration
	TLS *securitytls.Config

	// Timeout configurations (best practice)
	QueryTimeout   time.Duration // Default: 30s
	ConnectTimeout time.Duration // Default: 10s

	// AWS RDS specific settings (optional)
	UseAWS     bool
	UseIAM     bool
	AWSRegion  string
	AWSRoleARN string

	// RDS-specific configuration
	RDSHost              string
	RDSPort              int
	RDSDatabase          string
	RDSUsername          string
	RDSTokenExpiration   int // seconds
	RDSEnablePooling     bool
	RDSMinPoolSize       int
	RDSMaxPoolSize       int
	RDSConnectionTimeout int // seconds

	// Migration settings
	AutoMigrate          bool
	MigrationsPath       string
	FailOnMigrationError bool
}

type ContextReference struct {
	ID           string    `db:"id" json:"id"`
	AgentID      string    `db:"agent_id" json:"agent_id"`
	ModelID      string    `db:"model_id" json:"model_id"`
	SessionID    string    `db:"session_id" json:"session_id"`
	CreatedAt    time.Time `db:"created_at" json:"created_at"`
	UpdatedAt    time.Time `db:"updated_at" json:"updated_at"`
	ExpiresAt    time.Time `db:"expires_at" json:"expires_at"`
	TokenCount   int       `db:"token_count" json:"token_count"`
	MessageCount int       `db:"message_count" json:"message_count"`
	StoragePath  string    `db:"storage_path" json:"storage_path"`
}

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

type Embedding struct {
	ID          string         `json:"id" db:"id"`
	Vector      []float32      `json:"vector" db:"vector"`
	Dimensions  int            `json:"dimensions" db:"dimensions"`
	ModelID     string         `json:"model_id" db:"model_id"`
	ContentType string         `json:"content_type" db:"content_type"`
	ContentID   string         `json:"content_id" db:"content_id"`
	Namespace   string         `json:"namespace" db:"namespace"`
	ContextID   string         `json:"context_id" db:"context_id"`
	CreatedAt   time.Time      `json:"created_at" db:"created_at"`
	UpdatedAt   time.Time      `json:"updated_at" db:"updated_at"`
	Metadata    map[string]any `json:"metadata" db:"metadata"`
	Similarity  float64        `json:"similarity" db:"similarity"`
}

type EntityRelationshipRecord struct {
	ID          string    `db:"id"`
	Type        string    `db:"relationship_type"`
	Direction   string    `db:"direction"`
	SourceType  string    `db:"source_type"`
	SourceOwner string    `db:"source_owner"`
	SourceRepo  string    `db:"source_repo"`
	SourceID    string    `db:"source_id"`
	TargetType  string    `db:"target_type"`
	TargetOwner string    `db:"target_owner"`
	TargetRepo  string    `db:"target_repo"`
	TargetID    string    `db:"target_id"`
	Strength    float64   `db:"strength"`
	Context     string    `db:"context"`
	Metadata    []byte    `db:"metadata"`
	CreatedAt   time.Time `db:"created_at"`
	UpdatedAt   time.Time `db:"updated_at"`
}

type GitHubContentMetadata struct {
	ID          string    `db:"id"`
	Owner       string    `db:"owner"`
	Repo        string    `db:"repo"`
	ContentType string    `db:"content_type"`
	ContentID   string    `db:"content_id"`
	Checksum    string    `db:"checksum"`
	URI         string    `db:"uri"`
	Size        int64     `db:"size"`
	CreatedAt   time.Time `db:"created_at"`
	UpdatedAt   time.Time `db:"updated_at"`
	ExpiresAt   time.Time `db:"expires_at"`
	Metadata    []byte    `db:"metadata"`
}

type MockDatabase struct {
	TransactionFn func(ctx context.Context, fn func(*sqlx.Tx) error) error
}

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

type Repository interface {
	FindByID(ctx context.Context, id string) (any, error)
	FindAll(ctx context.Context, options models.QueryOptions) (any, error)
	Create(ctx context.Context, entity any) error
	Update(ctx context.Context, entity any) error
	Delete(ctx context.Context, id string) error
}

type Transaction interface {
	// Exec executes a query without returning any rows
	Exec(query string, args ...interface{}) (sql.Result, error)

	// ExecContext executes a query with context
	ExecContext(ctx context.Context, query string, args ...interface{}) (sql.Result, error)

	// Query executes a query that returns rows
	Query(query string, args ...interface{}) (*sql.Rows, error)

	// QueryContext executes a query with context
	QueryContext(ctx context.Context, query string, args ...interface{}) (*sql.Rows, error)

	// QueryRow executes a query that returns at most one row
	QueryRow(query string, args ...interface{}) *sql.Row

	// QueryRowContext executes a query with context that returns at most one row
	QueryRowContext(ctx context.Context, query string, args ...interface{}) *sql.Row

	// Get executes a query and scans the result into dest (sqlx)
	Get(dest interface{}, query string, args ...interface{}) error

	// Select executes a query and scans the results into dest (sqlx)
	Select(dest interface{}, query string, args ...interface{}) error

	// NamedExec executes a named query (sqlx)
	NamedExec(query string, arg interface{}) (sql.Result, error)

	// Savepoint creates a savepoint with the given name
	Savepoint(name string) error

	// RollbackToSavepoint rolls back to the named savepoint
	RollbackToSavepoint(name string) error

	// ReleaseSavepoint releases the named savepoint
	ReleaseSavepoint(name string) error

	// Commit commits the transaction
	Commit() error

	// Rollback rolls back the transaction
	Rollback() error

	// ID returns the unique transaction ID for tracking
	ID() string
}

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

type UnitOfWork interface {
	// BeginTx starts a new transaction with the given context and options
	BeginTx(ctx context.Context, opts *sql.TxOptions) (Transaction, error)

	// Execute runs a function within a transaction, automatically handling commit/rollback
	Execute(ctx context.Context, fn func(tx Transaction) error) error

	// ExecuteWithOptions runs a function within a transaction with custom options
	ExecuteWithOptions(ctx context.Context, opts *sql.TxOptions, fn func(tx Transaction) error) error
}

type VectorConfig struct {
	Enabled           bool
	ExtensionSchema   string
	IndexType         string
	DistanceMetric    string
	MaxDimensions     int
	DefaultDimensions int
}

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