database

type AggregateOptions struct {
	AllowDiskUse             *bool
	BatchSize                *int32
	BypassDocumentValidation *bool
	MaxTime                  *time.Duration
	UseCursor                *bool
}

type BulkWriteOptions struct {
	BypassDocumentValidation *bool
	Ordered                  *bool
}

type ChangeStream interface {
	Next(ctx context.Context) bool
	TryNext(ctx context.Context) bool
	Decode(val any) error
	Err() error
	Close(ctx context.Context) error
	ResumeToken() bson.Raw
}

type ChangeStreamOptions struct {
	BatchSize            *int32
	FullDocument         *string
	MaxAwaitTime         *time.Duration
	ResumeAfter          any
	StartAtOperationTime any
	StartAfter           any
}

type ChangeStreamPreAndPostImages struct {
	Enabled *bool
}

type Collation struct {
	Locale          *string
	CaseLevel       *bool
	CaseFirst       *string
	Strength        *int32
	NumericOrdering *bool
	Alternate       *string
	MaxVariable     *string
	Backwards       *bool
}

type CountOptions struct {
	Skip    *int64
	Limit   *int64
	MaxTime *time.Duration
}

type CreateCollectionOptions struct {
	Capped                       *bool
	SizeInBytes                  *int64
	MaxDocuments                 *int64
	StorageEngine                any
	Validator                    any
	ValidationLevel              *string
	ValidationAction             *string
	IndexOptionDefaults          any
	ViewOn                       *string
	Pipeline                     any
	Collation                    *Collation
	ChangeStreamPreAndPostImages *ChangeStreamPreAndPostImages
}

type DeleteOptions struct {
}

type DistinctOptions struct {
	MaxTime *time.Duration
}

type DocumentBulkWriteResult interface {
	InsertedCount() int64
	MatchedCount() int64
	ModifiedCount() int64
	DeletedCount() int64
	UpsertedCount() int64
	UpsertedIDs() map[int64]any
}

type DocumentCollection interface {
	// Single document operations
	InsertOne(ctx context.Context, document any, opts *InsertOneOptions) (any, error)
	FindOne(ctx context.Context, filter any, opts *FindOneOptions) DocumentResult
	UpdateOne(ctx context.Context, filter any, update any, opts *UpdateOptions) (DocumentUpdateResult, error)
	ReplaceOne(ctx context.Context, filter any, replacement any, opts *ReplaceOptions) (DocumentUpdateResult, error)
	DeleteOne(ctx context.Context, filter any, opts *DeleteOptions) (DocumentDeleteResult, error)

	// Multiple document operations
	InsertMany(ctx context.Context, documents []any, opts *InsertManyOptions) ([]any, error)
	Find(ctx context.Context, filter any, opts *FindOptions) (DocumentCursor, error)
	UpdateMany(ctx context.Context, filter any, update any, opts *UpdateOptions) (DocumentUpdateResult, error)
	DeleteMany(ctx context.Context, filter any, opts *DeleteOptions) (DocumentDeleteResult, error)

	// Count operations
	CountDocuments(ctx context.Context, filter any, opts *CountOptions) (int64, error)
	EstimatedDocumentCount(ctx context.Context, opts *EstimatedCountOptions) (int64, error)

	// Aggregation operations
	Aggregate(ctx context.Context, pipeline any, opts *AggregateOptions) (DocumentCursor, error)
	Distinct(ctx context.Context, fieldName string, filter any, opts *DistinctOptions) ([]any, error)

	// Index operations
	CreateIndex(ctx context.Context, model IndexModel) error
	CreateIndexes(ctx context.Context, models []IndexModel) error
	DropIndex(ctx context.Context, name string) error
	ListIndexes(ctx context.Context) (DocumentCursor, error)

	// Bulk operations
	BulkWrite(ctx context.Context, models []WriteModel, opts *BulkWriteOptions) (DocumentBulkWriteResult, error)

	// Watch changes (if supported)
	Watch(ctx context.Context, pipeline any, opts *ChangeStreamOptions) (ChangeStream, error)
}

type DocumentCursor interface {
	// Iteration
	Next(ctx context.Context) bool
	TryNext(ctx context.Context) bool
	Decode(val any) error
	All(ctx context.Context, results any) error

	// Cursor management
	Close(ctx context.Context) error
	Err() error
	ID() int64

	// Current document access
	Current() bson.Raw
}

type DocumentDeleteResult interface {
	DeletedCount() int64
}

type DocumentInterface interface {
	// Collection operations
	Collection(name string) DocumentCollection

	// Database-level operations
	CreateCollection(ctx context.Context, name string, opts *CreateCollectionOptions) error
	DropCollection(ctx context.Context, name string) error

	// Index operations
	CreateIndex(ctx context.Context, collection string, model IndexModel) error
	DropIndex(ctx context.Context, collection string, name string) error
	ListIndexes(ctx context.Context, collection string) ([]IndexModel, error)

	// Aggregation operations
	Aggregate(ctx context.Context, collection string, pipeline any, opts *AggregateOptions) (DocumentCursor, error)

	// Database administration
	RunCommand(ctx context.Context, command any) DocumentResult
}

type DocumentResult interface {
	Decode(v any) error
	Err() error
}

type DocumentUpdateResult interface {
	MatchedCount() int64
	ModifiedCount() int64
	UpsertedCount() int64
	UpsertedID() any
}

type EstimatedCountOptions struct {
	MaxTime *time.Duration
}

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

type FindOneOptions struct {
	Sort                any
	Skip                *int64
	Projection          any
	MaxTime             *time.Duration
	ShowRecordID        *bool
	AllowPartialResults *bool
}

type FindOptions struct {
	Sort                any
	Skip                *int64
	Limit               *int64
	Projection          any
	MaxTime             *time.Duration
	ShowRecordID        *bool
	AllowPartialResults *bool
	BatchSize           *int32
	NoCursorTimeout     *bool
}

type IndexModel struct {
	Keys    any
	Options *IndexOptions
}

type IndexOptions struct {
	Background              *bool
	ExpireAfterSeconds      *int32
	Name                    *string
	Sparse                  *bool
	StorageEngine           any
	Unique                  *bool
	Version                 *int32
	DefaultLanguage         *string
	LanguageOverride        *string
	TextVersion             *int32
	Weights                 any
	SphereVersion           *int32
	Bits                    *int32
	Max                     *float64
	Min                     *float64
	BucketSize              *float64
	PartialFilterExpression any
	Collation               *Collation
	WildcardProjection      any
	Hidden                  *bool
}

type InsertManyOptions struct {
	BypassDocumentValidation *bool
	Ordered                  *bool
}

type InsertOneOptions struct {
	BypassDocumentValidation *bool
}

type Interface = types.Interface

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

type ReplaceOptions struct {
	BypassDocumentValidation *bool
	Upsert                   *bool
}

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

type Statement = types.Statement

type Tx = types.Tx

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

type UpdateOptions struct {
	ArrayFilters             []any
	BypassDocumentValidation *bool
	Upsert                   *bool
}

type WriteModel interface {
	GetModel() any
}