orm

type AfterDeleteHook interface {
	AfterDelete(ctx context.Context) error
}

type AfterFindHook interface {
	AfterFind(ctx context.Context) error
}

type AfterInsertHook interface {
	AfterInsert(ctx context.Context) error
}

type AfterInterceptor func(ctx context.Context, info OperationInfo, opErr error) error

type AfterUpdateHook interface {
	AfterUpdate(ctx context.Context) error
}

type BeforeDeleteHook interface {
	BeforeDelete(ctx context.Context) error
}

type BeforeInsertHook interface {
	BeforeInsert(ctx context.Context) error
}

type BeforeInterceptor func(ctx context.Context, info OperationInfo) (context.Context, error)

type BeforeUpdateHook interface {
	BeforeUpdate(ctx context.Context) error
}

type ClusterResolvePolicy struct {
	PreferReadReplicas     bool
	AllowUnhealthyFallback bool
	AllowWriteToReadOnly   bool
}

type ClusterResolver struct {
	Policy ClusterResolvePolicy
	// contains filtered or unexported fields
}

type Codec interface {
	Encode(value any) (any, error)
	Decode(value any) (any, error)
}

type CodegenConfig struct {
	Package       string             `json:"package"`
	OutputDir     string             `json:"output_dir"`
	ORMImport     string             `json:"orm_import"`
	Header        string             `json:"header"`
	Models        []CodegenModelSpec `json:"models"`
	GenerateIndex bool               `json:"generate_index"`
}

type CodegenModelSpec struct {
	Name               string   `json:"name"`
	Columns            []string `json:"columns"`
	PKType             string   `json:"pk_type"`
	GenerateColumns    bool     `json:"generate_columns"`
	GenerateRepository bool     `json:"generate_repository"`
}

type ColumnSchema struct {
	Name       string `json:"name"`
	Type       string `json:"type"`
	Nullable   bool   `json:"nullable"`
	PrimaryKey bool   `json:"primary_key"`
}

type DB interface {
	NewQuery(sql string) *dbx.Query
	Select(cols ...string) *dbx.SelectQuery
	Insert(table string, cols dbx.Params) *dbx.Query
	InsertMany(table string, rows []dbx.Params) *dbx.Query
	InsertReturning(table string, cols dbx.Params, returning ...string) *dbx.Query
	Update(table string, cols dbx.Params, where dbx.Expression) *dbx.Query
	Delete(table string, where dbx.Expression) *dbx.Query
	Upsert(table string, cols dbx.Params, constraints ...string) *dbx.Query
	UpsertOnConflict(table string, cols dbx.Params, conflict dbx.OnConflict) *dbx.Query
}

type DBXObserverOptions struct {
	RedactSQL func(sql string) string
	OnEvent   func(QueryEvent)
}

type DefaultNamingStrategy struct{}

type Error struct {
	Code  ErrorCode
	Op    string
	Model string
	Field string
	Err   error
}

type ErrorCode string

type ErrorDetails struct {
	Op    string
	Model string
	Field string
}

type FieldExtender interface {
	ExtendField(sf reflect.StructField, field *FieldMeta) error
}

type FieldMeta struct {
	GoName       string
	DBName       string
	JSONName     string
	Type         reflect.Type
	Index        []int
	EmbeddedPath []string

	IsPK           bool
	IsNullable     bool
	IsIgnored      bool
	IsReadOnly     bool
	IsWriteOnly    bool
	HasDefault     bool
	IsGenerated    bool
	IsSoftDelete   bool
	IsCreatedAt    bool
	IsUpdatedAt    bool
	IsRelation     bool
	IsEmbeddedRoot bool
	Attributes     map[string]string
	Codec          Codec
}

type JSONValue[T any] struct {
	V T
}

type ListResult[T any] struct {
	Items   []T
	Total   int64
	Page    int64
	PerPage int64
}

type MetricsObserverOptions struct {
	OnQueryCount   func(kind string, op Operation, model, table string)
	OnError        func(kind string, op Operation, model, table string, err error)
	OnLatency      func(kind string, op Operation, model, table string, d time.Duration)
	OnRowsAffected func(op Operation, model, table string, rows int64)
	RedactSQL      func(sql string) string
}

type Migration struct {
	Version int64
	Name    string
	UpSQL   string
	DownSQL string
}

type MigrationPlan struct {
	CurrentVersion int64
	TargetVersion  int64
	Up             []Migration
	Down           []Migration
	Dirty          bool
	LastError      string
}

type MigrationRunner struct {
	DB              DB
	TableName       string
	LockName        string
	LockTTL         time.Duration
	UseDBLock       bool
	UseProcessMutex bool
}

type MigrationStatus struct {
	CurrentVersion int64
	Applied        []int64
	Pending        []Migration
	Dirty          bool
	LastError      string
	Locked         bool
	LockOwner      string
}

type ModelConfig struct {
	Table         string
	Naming        NamingStrategy
	Relations     map[string]RelationConfig
	Extenders     []FieldExtender
	TenantField   string
	FieldCodecs   map[string]Codec
	RequireTenant *bool
}

type ModelMeta struct {
	Type  reflect.Type
	Name  string
	Table string

	Fields      []*FieldMeta
	FieldsByGo  map[string]*FieldMeta
	FieldsByDB  map[string]*FieldMeta
	PrimaryKeys []*FieldMeta

	SoftDeleteField *FieldMeta
	CreatedAtField  *FieldMeta
	UpdatedAtField  *FieldMeta
	TenantField     *FieldMeta
	RequireTenant   bool
	Relations       map[string]*RelationMeta
}

type ModelQuery[T any] struct {
	// contains filtered or unexported fields
}

type NamingStrategy interface {
	TableName(typeName string) string
	ColumnName(fieldName string) string
}

type Nullable[T any] = optional.Nullable[T]

type Operation string

type OperationInfo struct {
	Operation      Operation
	Model          string
	Table          string
	Fields         []string
	ConflictFields []string
	Relations      []string
	Preloads       []string
	HasWhere       bool
	Limit          int64
	Offset         int64
}

type Optional[T any] = optional.Value[T]

type Plugin interface {
	Name() string
	Install(host *PluginHost) error
}

type PluginHost struct{}

type PreloadOption func(*preloadOptions)

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

type QueryEvent struct {
	Kind         string
	Operation    Operation
	Model        string
	Table        string
	SQL          string
	Duration     time.Duration
	RowsAffected int64
	Err          error
}

type QueryScope[T any] func(*ModelQuery[T]) *ModelQuery[T]

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

type RegistryOption func(*Registry)

type RelationConfig struct {
	Kind           RelationKind
	LocalField     string
	ForeignRef     string
	JoinTable      string
	JoinLocalKey   string
	JoinForeignKey string
}

type RelationKind string

type RelationMeta struct {
	Name           string
	Kind           RelationKind
	FieldIndex     []int
	TargetType     reflect.Type
	LocalField     string
	ForeignRef     string
	JoinTable      string
	JoinLocalKey   string
	JoinForeignKey string
}

type Router struct {
	Base     DB
	Resolver ShardResolver
	Policy   RoutingPolicy
}

type RoutingPolicy struct {
	RequireShard      bool
	RequireFor        map[Operation]bool
	UseTenantFallback bool
	TenantToShard     func(tenant any) (string, error)
}

type RuntimeMetricsOptions struct {
	OnRowsScanned func(op Operation, model, table string, rows int64)
	OnTxCount     func(event string)
}

type SQLDialect string

type SchemaChange struct {
	Kind       SchemaChangeKind `json:"kind"`
	Table      string           `json:"table"`
	Column     string           `json:"column,omitempty"`
	FromType   string           `json:"from_type,omitempty"`
	ToType     string           `json:"to_type,omitempty"`
	UpSQL      string           `json:"up_sql"`
	DownSQL    string           `json:"down_sql"`
	Breaking   bool             `json:"breaking"`
	Reversible bool             `json:"reversible"`
}

type SchemaChangeKind string

type SchemaDiff struct {
	Dialect  SQLDialect     `json:"dialect"`
	Changes  []SchemaChange `json:"changes"`
	Warnings []string       `json:"warnings,omitempty"`
}

type SchemaDiffOptions struct {
	Dialect            SQLDialect
	IncludeDestructive bool
}

type SchemaSnapshot struct {
	Dialect SQLDialect              `json:"dialect"`
	Tables  map[string]*TableSchema `json:"tables"`
}

type ShardReplica struct {
	Name     string
	DB       DB
	Weight   int
	ReadOnly bool
	Healthy  bool
}

type ShardResolver interface {
	ResolveShard(ctx context.Context, shardKey string, info OperationInfo) (DB, error)
}

type SortDirection string

type TableNamer interface {
	TableName() string
}

type TableSchema struct {
	Name       string                   `json:"name"`
	Columns    map[string]*ColumnSchema `json:"columns"`
	PrimaryKey []string                 `json:"primary_key,omitempty"`
}

type TenantPlugin struct {
	NameStr     string
	Required    bool
	Column      string
	PerModel    bool
	RegistryRef *Registry
}

type TraceObserverOptions struct {
	OnStart  func(ctx context.Context, info OperationInfo) context.Context
	OnFinish func(ctx context.Context, info OperationInfo, opErr error)
}

type Tx = dbx.Tx

type UpsertOptions struct {
	ConflictFields []string
	UpdateFields   []string
	DoNothing      bool
}

type ZeroDowntimePlan struct {
	Dialect  SQLDialect     `json:"dialect"`
	Expand   []SchemaChange `json:"expand"`
	Backfill []string       `json:"backfill"`
	Contract []SchemaChange `json:"contract"`
	Warnings []string       `json:"warnings,omitempty"`
}