package utils

import (
	"errors"
	"io"
	"strconv"

	"github.com/99designs/gqlgen/graphql"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

func MarshalObjectID(value primitive.ObjectID) graphql.Marshaler {

	return graphql.WriterFunc(func(writer io.Writer) {

		io.WriteString(writer, strconv.Quote(value.Hex()))
	})

}

func UnmarshalObjectID(value any) (primitive.ObjectID, error) {

	if str, ok := value.(string); ok {

		return primitive.ObjectIDFromHex(str)
	}

	return primitive.NilObjectID, errors.New("invalid Object ID string")
}