package mongo

import (
	"context"
	"fmt"
	"time"

	"git.acooldomain.co/server-manager/backend/models"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

func getMongoCollection(config models.MongoDBConfig) (*mongo.Collection, error) {
	clientOptions := options.Client().ApplyURI(config.Url)

	if config.Username != "" || config.Password != "" {
		clientOptions = clientOptions.SetAuth(options.Credential{
			Username: config.Username,
			Password: config.Password,
		})
	}

	ctx, cancel := context.WithTimeoutCause(context.Background(), 30*time.Second, fmt.Errorf("Timeout"))
	defer cancel()

	client, err := mongo.Connect(ctx, clientOptions)

	if err != nil {
		return nil, err
	}

	return client.Database(config.Database).Collection(config.Collection), nil
}