package model

import (
	"go.mongodb.org/mongo-driver/bson"
)

type AttrOption struct {
	Attr  string
	Value interface{}
}

type ModelBuilder[T interface{}] struct {
	// attrValues map[string]interface{}
	attrValues []AttrOption
}

// for others
func (mb *ModelBuilder[T]) GetMap() map[string]interface{} {
	m := map[string]interface{}{}
	for _, attr := range mb.attrValues {
		m[attr.Attr] = attr.Value
	}
	return m
}

// for mongo
func (mb *ModelBuilder[T]) GetBsonM() bson.M {
	m := bson.M{}
	for _, attr := range mb.attrValues {
		m[attr.Attr] = attr.Value
	}
	return m
}
func (mb *ModelBuilder[T]) GetSlice() []AttrOption {
	return mb.attrValues
}

func (mb *ModelBuilder[T]) AddAttr(attr string, value interface{}) *ModelBuilder[T] {
	mb.attrValues = append(mb.attrValues, AttrOption{attr, value})
	// mb.attrValues[attr] = value
	return mb
}