89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package verification
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"git.sendico.io/tech/sxchnge/bff/edge/shared/storage/ratelimits"
|
|
ri "git.sendico.io/tech/sxchnge/bff/pkg/db/repository/index"
|
|
"git.sendico.io/tech/sxchnge/bff/pkg/db/template"
|
|
"git.sendico.io/tech/sxchnge/bff/pkg/db/transaction"
|
|
"git.sendico.io/tech/sxchnge/bff/pkg/mlogger"
|
|
"git.sendico.io/tech/sxchnge/bff/pkg/model"
|
|
"git.sendico.io/tech/sxchnge/bff/pkg/mservice"
|
|
"go.mongodb.org/mongo-driver/v2/mongo"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type verificationDB struct {
|
|
template.DBImp[*model.VerificationToken]
|
|
tf transaction.Factory
|
|
limits ratelimits.Store
|
|
}
|
|
|
|
func (db *verificationDB) WithIssuanceLimits(limits ratelimits.Store) *verificationDB {
|
|
db.limits = limits
|
|
return db
|
|
}
|
|
|
|
const verificationContextIdempotencyIndexName = "uniq_verification_context_idempotency_v2"
|
|
|
|
func Create(
|
|
logger mlogger.Logger,
|
|
db *mongo.Database,
|
|
tf transaction.Factory,
|
|
) (*verificationDB, error) {
|
|
return CreateWithCollection(logger, db, tf, "")
|
|
}
|
|
|
|
func CreateWithCollection(
|
|
logger mlogger.Logger,
|
|
db *mongo.Database,
|
|
tf transaction.Factory,
|
|
collection string,
|
|
) (*verificationDB, error) {
|
|
if strings.TrimSpace(collection) == "" {
|
|
collection = mservice.VerificationTokens
|
|
}
|
|
p := &verificationDB{
|
|
DBImp: *template.Create[*model.VerificationToken](logger, mservice.Type(collection), db),
|
|
tf: tf,
|
|
}
|
|
|
|
if err := p.Repository.CreateIndex(&ri.Definition{
|
|
Keys: []ri.Key{{Field: "verifyTokenHash", Sort: ri.Asc}},
|
|
Unique: true,
|
|
Name: "unique_token_hash",
|
|
}); err != nil {
|
|
p.Logger.Error("Failed to create unique verifyTokenHash index", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
if err := p.Repository.CreateIndex(&ri.Definition{
|
|
Keys: []ri.Key{
|
|
{Field: "accountRef", Sort: ri.Asc},
|
|
{Field: "purpose", Sort: ri.Asc},
|
|
{Field: "target", Sort: ri.Asc},
|
|
{Field: "contextRef", Sort: ri.Asc},
|
|
{Field: "idempotencyKey", Sort: ri.Asc},
|
|
},
|
|
Unique: true,
|
|
Sparse: true,
|
|
Name: verificationContextIdempotencyIndexName,
|
|
}); err != nil {
|
|
p.Logger.Error("Failed to create unique idempotency index on verification context", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
ttl := int32(2678400) // 30 days
|
|
if err := p.Repository.CreateIndex(&ri.Definition{
|
|
Keys: []ri.Key{{Field: "expiresAt", Sort: ri.Asc}},
|
|
TTL: &ttl,
|
|
Name: "ttl_expires_at",
|
|
}); err != nil {
|
|
p.Logger.Error("Failed to create TTL index on expiresAt", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
return p, nil
|
|
}
|