feat: added ability to use the client for sessions and mocking

[happilymarrieddad]

Hello, I would like to add the ability to use a session wrapper and I need access to the underlying client. Please accept this pull request so I can get access to it. Also, I would like the ability to add mock testing. Thanks!

• Nick

package repos

import (
	"context"
	"fmt"

	"github.com/jensteichert/colt"
	"go.mongodb.org/mongo-driver/mongo"
)

// wrapInTransaction handles the boilerplate of starting a session,
// managing the transaction lifecycle, and type-casting the result.
func wrapInTransaction[T any](db *colt.Database, fn func(sCtx mongo.SessionContext) (T, error)) (T, error) {
	var result T

	// Use db.Client() to access the exported Database's Client method
	err := db.Client().UseSession(context.Background(), func(sessionContext mongo.SessionContext) error {

		out, err := sessionContext.WithTransaction(sessionContext, func(sCtx mongo.SessionContext) (interface{}, error) {
			return fn(sCtx)
		})

		if err != nil {
			return err
		}

		if val, ok := out.(T); ok {
			result = val
		}
		return nil
	})

	if err != nil {
		return result, fmt.Errorf("transaction failed: %w", err)
	}

	return result, nil
}