db

type DB interface {
	DbQuerier
	DbClient
}

type DbClient interface {
	// SetDbInContext stores the database connection in the given context
	SetDbInContext(ctx context.Context) context.Context

	// Disconnect closes the database connection
	Disconnect(ctx context.Context) error
}

type DbCollection interface {
	Name() string   // Returns the collection name
	DbName() string // Returns the database name
}

type DbCtxKey struct{}

type DbQuerier interface {
	// FindOne executes a find operation that returns a single document
	FindOne(ctx context.Context, col DbCollection, filter interface{}, opts ...*options.FindOneOptions) *mongo.SingleResult

	// Find executes a find operation that can return multiple documents
	Find(ctx context.Context, col DbCollection, filter interface{}, opts ...*options.FindOptions) (*mongo.Cursor, error)

	// InsertOne inserts a single document into the collection
	InsertOne(ctx context.Context, col DbCollection, document interface{}, opts ...*options.InsertOneOptions) (*mongo.InsertOneResult, error)

	// UpdateOne updates a single document in the collection
	UpdateOne(ctx context.Context, col DbCollection, filter interface{}, update interface{}, opts ...*options.UpdateOptions) (*mongo.UpdateResult, error)

	// DeleteOne deletes a single document from the collection
	DeleteOne(ctx context.Context, col DbCollection, filter interface{}, opts ...*options.DeleteOptions) (*mongo.DeleteResult, error)

	// Aggregate executes an aggregation pipeline
	Aggregate(ctx context.Context, col DbCollection, filter interface{}, opts ...*options.AggregateOptions) (*mongo.Cursor, error)

	// CountDocuments counts the number of documents matching the filter
	CountDocuments(ctx context.Context, col DbCollection, filter interface{}, opts ...*options.CountOptions) (int64, error)

	// BulkWrite executes multiple write operations in bulk
	BulkWrite(ctx context.Context, col DbCollection, models []mongo.WriteModel, opts ...*options.BulkWriteOptions) (*mongo.BulkWriteResult, error)

	// UpdateMany updates multiple documents in the collection
	UpdateMany(ctx context.Context, col DbCollection, filter interface{}, update interface{}, opts ...*options.UpdateOptions) (*mongo.UpdateResult, error)
}

type MigrationFunc func(ctx context.Context, db DB) error

type MigrationInfo struct {
	Version     string        // Unique version identifier for the migration
	Name        string        // Human-readable name of the migration
	Description string        // Detailed description of what the migration does
	Up          MigrationFunc // Function to apply the migration
	Down        MigrationFunc // Function to rollback the migration
}

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