package data

import (
	"context"
	"git.usbharu.dev/usbharu/unos/illust"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

type IllustDataRepository interface {
	Save(illust *illust.Illust) error
}

type MongodbIllustDataRepository struct {
	mongo.Collection
}

func NewMongodbIllustDataRepository(database mongo.Database) *MongodbIllustDataRepository {
	return &MongodbIllustDataRepository{Collection: *database.Collection("illusts")}
}

func (m *MongodbIllustDataRepository) Save(illust *illust.Illust) error {
	if _, err := m.ReplaceOne(context.Background(), bson.D{{"_id", illust.Id}}, illust, options.Replace().SetUpsert(true)); err != nil {
		return err
	}
	return nil
}