85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
package share
|
|
|
|
import (
|
|
"context"
|
|
|
|
"dynatron.me/x/stillbox/internal/jsontypes"
|
|
"dynatron.me/x/stillbox/pkg/database"
|
|
)
|
|
|
|
type Store interface {
|
|
// Get retreives a share record.
|
|
Get(ctx context.Context, id string) (*Share, error)
|
|
|
|
// Create stores a new share record.
|
|
Create(ctx context.Context, share *Share) error
|
|
|
|
// Delete deletes a share record.
|
|
Delete(ctx context.Context, id string) error
|
|
|
|
// Prune removes expired share records.
|
|
Prune(ctx context.Context) error
|
|
}
|
|
|
|
type postgresStore struct {
|
|
}
|
|
|
|
func recToShare(share database.Share) *Share {
|
|
return &Share{
|
|
ID: share.ID,
|
|
Type: EntityType(share.EntityType),
|
|
EntityID: share.EntityID,
|
|
Expiration: jsontypes.TimePtrFromTSTZ(share.Expiration),
|
|
}
|
|
}
|
|
|
|
func (s *postgresStore) Get(ctx context.Context, id string) (*Share, error) {
|
|
db := database.FromCtx(ctx)
|
|
rec, err := db.GetShare(ctx, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return recToShare(rec), nil
|
|
}
|
|
|
|
func (s *postgresStore) Create(ctx context.Context, share *Share) error {
|
|
db := database.FromCtx(ctx)
|
|
err := db.CreateShare(ctx, database.CreateShareParams{
|
|
ID: share.ID,
|
|
EntityType: string(share.Type),
|
|
EntityID: share.EntityID,
|
|
Expiration: share.Expiration.PGTypeTSTZ(),
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
func (s *postgresStore) Delete(ctx context.Context, id string) error {
|
|
return database.FromCtx(ctx).DeleteShare(ctx, id)
|
|
}
|
|
|
|
func (s *postgresStore) Prune(ctx context.Context) error {
|
|
return database.FromCtx(ctx).PruneShares(ctx)
|
|
}
|
|
|
|
func NewStore() *postgresStore {
|
|
return new(postgresStore)
|
|
}
|
|
|
|
type storeCtxKey string
|
|
|
|
const StoreCtxKey storeCtxKey = "store"
|
|
|
|
func CtxWithStore(ctx context.Context, s Store) context.Context {
|
|
return context.WithValue(ctx, StoreCtxKey, s)
|
|
}
|
|
|
|
func FromCtx(ctx context.Context) Store {
|
|
s, ok := ctx.Value(StoreCtxKey).(Store)
|
|
if !ok {
|
|
return NewStore()
|
|
}
|
|
|
|
return s
|
|
}
|