113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
package callbacksdb
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/tech/sendico/pkg/auth"
|
|
"github.com/tech/sendico/pkg/db/callbacks"
|
|
"github.com/tech/sendico/pkg/db/policy"
|
|
ri "github.com/tech/sendico/pkg/db/repository/index"
|
|
"github.com/tech/sendico/pkg/db/storable"
|
|
"github.com/tech/sendico/pkg/merrors"
|
|
"github.com/tech/sendico/pkg/mlogger"
|
|
"github.com/tech/sendico/pkg/model"
|
|
"github.com/tech/sendico/pkg/mservice"
|
|
"go.mongodb.org/mongo-driver/v2/mongo"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type CallbacksDB struct {
|
|
auth.ProtectedDBImp[*model.Callback]
|
|
auth.ArchivableDB[*model.Callback]
|
|
}
|
|
|
|
func Create(
|
|
ctx context.Context,
|
|
logger mlogger.Logger,
|
|
enforcer auth.Enforcer,
|
|
pdb policy.DB,
|
|
db *mongo.Database,
|
|
) (*CallbacksDB, error) {
|
|
if err := ensureBuiltInPolicy(ctx, logger, pdb); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
p, err := auth.CreateDBImp[*model.Callback](ctx, logger, pdb, enforcer, mservice.Callbacks, db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, definition := range []*ri.Definition{
|
|
{
|
|
Name: "uq_callbacks_client_url",
|
|
Keys: []ri.Key{
|
|
{Field: storable.OrganizationRefField, Sort: ri.Asc},
|
|
{Field: "client_id", Sort: ri.Asc},
|
|
{Field: "url", Sort: ri.Asc},
|
|
},
|
|
Unique: true,
|
|
},
|
|
{
|
|
Name: "idx_callbacks_lookup",
|
|
Keys: []ri.Key{
|
|
{Field: storable.OrganizationRefField, Sort: ri.Asc},
|
|
{Field: "status", Sort: ri.Asc},
|
|
{Field: "event_types", Sort: ri.Asc},
|
|
},
|
|
},
|
|
} {
|
|
if err := p.DBImp.Repository.CreateIndex(definition); err != nil {
|
|
p.DBImp.Logger.Warn("Failed to create callbacks index", zap.String("index", definition.Name), zap.Error(err))
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
createEmpty := func() *model.Callback {
|
|
return &model.Callback{}
|
|
}
|
|
getArchivable := func(callback *model.Callback) model.Archivable {
|
|
return &callback.ArchivableBase
|
|
}
|
|
|
|
return &CallbacksDB{
|
|
ProtectedDBImp: *p,
|
|
ArchivableDB: auth.NewArchivableDB(
|
|
p.DBImp,
|
|
p.DBImp.Logger,
|
|
enforcer,
|
|
createEmpty,
|
|
getArchivable,
|
|
),
|
|
}, nil
|
|
}
|
|
|
|
func ensureBuiltInPolicy(ctx context.Context, logger mlogger.Logger, pdb policy.DB) error {
|
|
var existing model.PolicyDescription
|
|
if err := pdb.GetBuiltInPolicy(ctx, mservice.Callbacks, &existing); err == nil {
|
|
return nil
|
|
} else if !errors.Is(err, merrors.ErrNoData) {
|
|
return err
|
|
}
|
|
|
|
description := "Callbacks subscription management"
|
|
resourceTypes := []mservice.Type{mservice.Callbacks}
|
|
policyDescription := &model.PolicyDescription{
|
|
Describable: model.Describable{
|
|
Name: "Callbacks",
|
|
Description: &description,
|
|
},
|
|
ResourceTypes: &resourceTypes,
|
|
}
|
|
if err := pdb.Create(ctx, policyDescription); err != nil && !errors.Is(err, merrors.ErrDataConflict) {
|
|
if logger != nil {
|
|
logger.Warn("Failed to create built-in callbacks policy", zap.Error(err))
|
|
}
|
|
return err
|
|
}
|
|
|
|
return pdb.GetBuiltInPolicy(ctx, mservice.Callbacks, &existing)
|
|
}
|
|
|
|
var _ callbacks.DB = (*CallbacksDB)(nil)
|