package mongo

import (
	"context"
	"log"

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

type indexInfo struct {
	Key    map[string]any `bson:"key"`
	Name   string         `bson:"name"`
	Unique bool           `bson:"unique"`
}

func (c *Collection[T]) GetIndexes() (indexes map[string]any) {
	indexes = make(map[string]any)
	cursor, err := c.Indexes().List(context.TODO())
	if err != nil {
		log.Println("GetIndexes.err: " + err.Error())
		return
	}

	ret := make([]indexInfo, 0)
	if err = cursor.All(context.TODO(), &ret); err != nil {
		log.Println("cursor.All.err: " + err.Error())
	}

	for _, i := range ret {
		if len(i.Key) == 1 {
			for key := range i.Key {
				indexes[key] = i.Key[key]
			}
		} else {
			// not supported yet: compound indexes
			log.Printf("not supported yet: compound indexes\n")
		}
	}
	return
}

func (c *Collection[T]) SetIndexes(key string, unique bool) error {
	indexModel := mongo.IndexModel{
		Keys:    bson.D{{Key: key, Value: 1}},
		Options: options.Index().SetUnique(unique),
	}
	if _, err := c.Indexes().CreateOne(context.TODO(), indexModel); err != nil {
		return err
	}

	return nil
}