activity.go

package stores

import (
	"context"

	"github.com/sol-armada/sol-bot/utils"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

type ActivityStore struct {
	*store
}

const ACTIVITY Collection = "activity"

func newActivityStore(ctx context.Context, client *mongo.Client, database string) *ActivityStore {
	_ = client.Database(database).CreateCollection(ctx, string(ACTIVITY), &options.CreateCollectionOptions{
		TimeSeriesOptions: &options.TimeSeriesOptions{
			TimeField: "when",
			MetaField: utils.StringPointer("meta"),
		},
	})
	s := &store{
		Collection: client.Database(database).Collection(string(ACTIVITY)),
		ctx:        ctx,
	}
	return &ActivityStore{s}
}

func (c *Client) GetActivityStore() (*ActivityStore, bool) {
	storeInterface, ok := c.GetCollection(ACTIVITY)
	if !ok {
		return nil, false
	}
	return storeInterface.(*ActivityStore), ok
}

func (s *ActivityStore) Create(activity any) error {
	_, err := s.InsertOne(s.ctx, activity)
	return err
}