package mongo

import (
	"context"
	"time"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"go.mongodb.org/mongo-driver/mongo/readpref"
)

func initClient(config *MongoConfig) (*mongo.Client, error) {
	option := options.Client()
	uri := config.Uri()
	option.ApplyURI(uri)

	return mongoConnect(option)
}

func mongoConnect(opt *options.ClientOptions) (*mongo.Client, error) {
	ctx := context.Background()
	client, err := mongo.Connect(ctx, opt)
	if err != nil {
		return nil, err
	}

	pCtx, cancel := context.WithTimeout(ctx, 15*time.Second)
	defer cancel()
	if err := client.Ping(pCtx, readpref.Primary()); err != nil {
		return nil, err
	}
	return client, nil
}