softdelete.go

package softdelete

import (
	"context"

	"entgo.io/contrib/entgql"
	"entgo.io/ent"
	"entgo.io/ent/entql"
	"entgo.io/ent/schema/field"
	"entgo.io/ent/schema/mixin"
)

// Mixin for all schemas in the graph.
type Mixin struct {
	mixin.Schema
}

// Fields of the SoftDeleteMixin mixin.
func (Mixin) Fields() []ent.Field {
	return []ent.Field{
		field.Time("delete_time").
			Optional().
			Annotations(
				entgql.Skip(),
			),
	}
}

// Policy defines the privacy policy of the BaseMixin.
func (Mixin) Policy() ent.Policy {
	return privacy.Policy{
		Query: privacy.QueryPolicy{
			NotDeleteRule(),
		},
	}
}

func NotDeleteRule() privacy.QueryMutationRule {
	type SoftDeleteFilter interface {
		WhereDeleteTime(p entql.TimeP)
	}

	return privacy.FilterFunc(func(ctx context.Context, f privacy.Filter) error {
		tf, ok := f.(SoftDeleteFilter)
		if !ok {
			return privacy.Denyf("unexpected filter type %T", f)
		}

		tf.WhereDeleteTime(entql.TimeNil())
		return privacy.Skip
	})
}