using MongoDB.Bson;
using MongoDB.Driver;
using StellaOps.Provenance.Mongo;

namespace StellaOps.Events.Mongo;

public sealed class EventWriter
{
    private readonly IMongoCollection<BsonDocument> _events;

    public EventWriter(IMongoDatabase db, string collectionName = "events")
    {
        _events = db.GetCollection<BsonDocument>(collectionName);
    }

    public async Task AppendEventAsync(
        BsonDocument eventDoc,
        DsseProvenance dsse,
        TrustInfo trust,
        CancellationToken ct = default)
    {
        eventDoc.AttachDsseProvenance(dsse, trust);
        await _events.InsertOneAsync(eventDoc, cancellationToken: ct);
    }
}