snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)
This commit is contained in:
667
entc/integration/customid/ent/privacy/privacy.go
Normal file
667
entc/integration/customid/ent/privacy/privacy.go
Normal file
@@ -0,0 +1,667 @@
|
||||
// Copyright 2019-present Facebook Inc. All rights reserved.
|
||||
// This source code is licensed under the Apache 2.0 license found
|
||||
// in the LICENSE file in the root directory of this source tree.
|
||||
|
||||
// Code generated by ent, DO NOT EDIT.
|
||||
|
||||
package privacy
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"entgo.io/ent/entc/integration/customid/ent"
|
||||
|
||||
"entgo.io/ent/entql"
|
||||
"entgo.io/ent/privacy"
|
||||
)
|
||||
|
||||
var (
|
||||
// Allow may be returned by rules to indicate that the policy
|
||||
// evaluation should terminate with allow decision.
|
||||
Allow = privacy.Allow
|
||||
|
||||
// Deny may be returned by rules to indicate that the policy
|
||||
// evaluation should terminate with deny decision.
|
||||
Deny = privacy.Deny
|
||||
|
||||
// Skip may be returned by rules to indicate that the policy
|
||||
// evaluation should continue to the next rule.
|
||||
Skip = privacy.Skip
|
||||
)
|
||||
|
||||
// Allowf returns a formatted wrapped Allow decision.
|
||||
func Allowf(format string, a ...any) error {
|
||||
return privacy.Allowf(format, a...)
|
||||
}
|
||||
|
||||
// Denyf returns a formatted wrapped Deny decision.
|
||||
func Denyf(format string, a ...any) error {
|
||||
return privacy.Denyf(format, a...)
|
||||
}
|
||||
|
||||
// Skipf returns a formatted wrapped Skip decision.
|
||||
func Skipf(format string, a ...any) error {
|
||||
return privacy.Skipf(format, a...)
|
||||
}
|
||||
|
||||
// DecisionContext creates a new context from the given parent context with
|
||||
// a policy decision attach to it.
|
||||
func DecisionContext(parent context.Context, decision error) context.Context {
|
||||
return privacy.DecisionContext(parent, decision)
|
||||
}
|
||||
|
||||
// DecisionFromContext retrieves the policy decision from the context.
|
||||
func DecisionFromContext(ctx context.Context) (error, bool) {
|
||||
return privacy.DecisionFromContext(ctx)
|
||||
}
|
||||
|
||||
type (
|
||||
// Policy groups query and mutation policies.
|
||||
Policy = privacy.Policy
|
||||
|
||||
// QueryRule defines the interface deciding whether a
|
||||
// query is allowed and optionally modify it.
|
||||
QueryRule = privacy.QueryRule
|
||||
// QueryPolicy combines multiple query rules into a single policy.
|
||||
QueryPolicy = privacy.QueryPolicy
|
||||
|
||||
// MutationRule defines the interface which decides whether a
|
||||
// mutation is allowed and optionally modifies it.
|
||||
MutationRule = privacy.MutationRule
|
||||
// MutationPolicy combines multiple mutation rules into a single policy.
|
||||
MutationPolicy = privacy.MutationPolicy
|
||||
// MutationRuleFunc type is an adapter which allows the use of
|
||||
// ordinary functions as mutation rules.
|
||||
MutationRuleFunc = privacy.MutationRuleFunc
|
||||
|
||||
// QueryMutationRule is an interface which groups query and mutation rules.
|
||||
QueryMutationRule = privacy.QueryMutationRule
|
||||
)
|
||||
|
||||
// QueryRuleFunc type is an adapter to allow the use of
|
||||
// ordinary functions as query rules.
|
||||
type QueryRuleFunc func(context.Context, ent.Query) error
|
||||
|
||||
// Eval returns f(ctx, q).
|
||||
func (f QueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
return f(ctx, q)
|
||||
}
|
||||
|
||||
// AlwaysAllowRule returns a rule that returns an allow decision.
|
||||
func AlwaysAllowRule() QueryMutationRule {
|
||||
return privacy.AlwaysAllowRule()
|
||||
}
|
||||
|
||||
// AlwaysDenyRule returns a rule that returns a deny decision.
|
||||
func AlwaysDenyRule() QueryMutationRule {
|
||||
return privacy.AlwaysDenyRule()
|
||||
}
|
||||
|
||||
// ContextQueryMutationRule creates a query/mutation rule from a context eval func.
|
||||
func ContextQueryMutationRule(eval func(context.Context) error) QueryMutationRule {
|
||||
return privacy.ContextQueryMutationRule(eval)
|
||||
}
|
||||
|
||||
// OnMutationOperation evaluates the given rule only on a given mutation operation.
|
||||
func OnMutationOperation(rule MutationRule, op ent.Op) MutationRule {
|
||||
return privacy.OnMutationOperation(rule, op)
|
||||
}
|
||||
|
||||
// DenyMutationOperationRule returns a rule denying specified mutation operation.
|
||||
func DenyMutationOperationRule(op ent.Op) MutationRule {
|
||||
rule := MutationRuleFunc(func(_ context.Context, m ent.Mutation) error {
|
||||
return Denyf("ent/privacy: operation %s is not allowed", m.Op())
|
||||
})
|
||||
return OnMutationOperation(rule, op)
|
||||
}
|
||||
|
||||
// The AccountQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type AccountQueryRuleFunc func(context.Context, *ent.AccountQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f AccountQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.AccountQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.AccountQuery", q)
|
||||
}
|
||||
|
||||
// The AccountMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type AccountMutationRuleFunc func(context.Context, *ent.AccountMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f AccountMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.AccountMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.AccountMutation", m)
|
||||
}
|
||||
|
||||
// The BlobQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type BlobQueryRuleFunc func(context.Context, *ent.BlobQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f BlobQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.BlobQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.BlobQuery", q)
|
||||
}
|
||||
|
||||
// The BlobMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type BlobMutationRuleFunc func(context.Context, *ent.BlobMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f BlobMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.BlobMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.BlobMutation", m)
|
||||
}
|
||||
|
||||
// The BlobLinkQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type BlobLinkQueryRuleFunc func(context.Context, *ent.BlobLinkQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f BlobLinkQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.BlobLinkQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.BlobLinkQuery", q)
|
||||
}
|
||||
|
||||
// The BlobLinkMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type BlobLinkMutationRuleFunc func(context.Context, *ent.BlobLinkMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f BlobLinkMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.BlobLinkMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.BlobLinkMutation", m)
|
||||
}
|
||||
|
||||
// The CarQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type CarQueryRuleFunc func(context.Context, *ent.CarQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f CarQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.CarQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.CarQuery", q)
|
||||
}
|
||||
|
||||
// The CarMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type CarMutationRuleFunc func(context.Context, *ent.CarMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f CarMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.CarMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.CarMutation", m)
|
||||
}
|
||||
|
||||
// The DeviceQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type DeviceQueryRuleFunc func(context.Context, *ent.DeviceQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f DeviceQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.DeviceQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.DeviceQuery", q)
|
||||
}
|
||||
|
||||
// The DeviceMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type DeviceMutationRuleFunc func(context.Context, *ent.DeviceMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f DeviceMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.DeviceMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.DeviceMutation", m)
|
||||
}
|
||||
|
||||
// The DocQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type DocQueryRuleFunc func(context.Context, *ent.DocQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f DocQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.DocQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.DocQuery", q)
|
||||
}
|
||||
|
||||
// The DocMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type DocMutationRuleFunc func(context.Context, *ent.DocMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f DocMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.DocMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.DocMutation", m)
|
||||
}
|
||||
|
||||
// The GroupQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type GroupQueryRuleFunc func(context.Context, *ent.GroupQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f GroupQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.GroupQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.GroupQuery", q)
|
||||
}
|
||||
|
||||
// The GroupMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type GroupMutationRuleFunc func(context.Context, *ent.GroupMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f GroupMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.GroupMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.GroupMutation", m)
|
||||
}
|
||||
|
||||
// The IntSIDQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type IntSIDQueryRuleFunc func(context.Context, *ent.IntSIDQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f IntSIDQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.IntSIDQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.IntSIDQuery", q)
|
||||
}
|
||||
|
||||
// The IntSIDMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type IntSIDMutationRuleFunc func(context.Context, *ent.IntSIDMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f IntSIDMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.IntSIDMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.IntSIDMutation", m)
|
||||
}
|
||||
|
||||
// The LinkQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type LinkQueryRuleFunc func(context.Context, *ent.LinkQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f LinkQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.LinkQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.LinkQuery", q)
|
||||
}
|
||||
|
||||
// The LinkMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type LinkMutationRuleFunc func(context.Context, *ent.LinkMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f LinkMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.LinkMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.LinkMutation", m)
|
||||
}
|
||||
|
||||
// The MixinIDQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type MixinIDQueryRuleFunc func(context.Context, *ent.MixinIDQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f MixinIDQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.MixinIDQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.MixinIDQuery", q)
|
||||
}
|
||||
|
||||
// The MixinIDMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type MixinIDMutationRuleFunc func(context.Context, *ent.MixinIDMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f MixinIDMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.MixinIDMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.MixinIDMutation", m)
|
||||
}
|
||||
|
||||
// The NoteQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type NoteQueryRuleFunc func(context.Context, *ent.NoteQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f NoteQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.NoteQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.NoteQuery", q)
|
||||
}
|
||||
|
||||
// The NoteMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type NoteMutationRuleFunc func(context.Context, *ent.NoteMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f NoteMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.NoteMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.NoteMutation", m)
|
||||
}
|
||||
|
||||
// The OtherQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type OtherQueryRuleFunc func(context.Context, *ent.OtherQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f OtherQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.OtherQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.OtherQuery", q)
|
||||
}
|
||||
|
||||
// The OtherMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type OtherMutationRuleFunc func(context.Context, *ent.OtherMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f OtherMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.OtherMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.OtherMutation", m)
|
||||
}
|
||||
|
||||
// The PetQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type PetQueryRuleFunc func(context.Context, *ent.PetQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f PetQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.PetQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.PetQuery", q)
|
||||
}
|
||||
|
||||
// The PetMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type PetMutationRuleFunc func(context.Context, *ent.PetMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f PetMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.PetMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.PetMutation", m)
|
||||
}
|
||||
|
||||
// The RevisionQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type RevisionQueryRuleFunc func(context.Context, *ent.RevisionQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f RevisionQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.RevisionQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.RevisionQuery", q)
|
||||
}
|
||||
|
||||
// The RevisionMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type RevisionMutationRuleFunc func(context.Context, *ent.RevisionMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f RevisionMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.RevisionMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.RevisionMutation", m)
|
||||
}
|
||||
|
||||
// The SessionQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type SessionQueryRuleFunc func(context.Context, *ent.SessionQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f SessionQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.SessionQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.SessionQuery", q)
|
||||
}
|
||||
|
||||
// The SessionMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type SessionMutationRuleFunc func(context.Context, *ent.SessionMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f SessionMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.SessionMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.SessionMutation", m)
|
||||
}
|
||||
|
||||
// The TokenQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type TokenQueryRuleFunc func(context.Context, *ent.TokenQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f TokenQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.TokenQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.TokenQuery", q)
|
||||
}
|
||||
|
||||
// The TokenMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type TokenMutationRuleFunc func(context.Context, *ent.TokenMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f TokenMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.TokenMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.TokenMutation", m)
|
||||
}
|
||||
|
||||
// The UserQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type UserQueryRuleFunc func(context.Context, *ent.UserQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f UserQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.UserQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.UserQuery", q)
|
||||
}
|
||||
|
||||
// The UserMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type UserMutationRuleFunc func(context.Context, *ent.UserMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f UserMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.UserMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.UserMutation", m)
|
||||
}
|
||||
|
||||
// The ValueScanQueryRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a query rule.
|
||||
type ValueScanQueryRuleFunc func(context.Context, *ent.ValueScanQuery) error
|
||||
|
||||
// EvalQuery return f(ctx, q).
|
||||
func (f ValueScanQueryRuleFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
if q, ok := q.(*ent.ValueScanQuery); ok {
|
||||
return f(ctx, q)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected query type %T, expect *ent.ValueScanQuery", q)
|
||||
}
|
||||
|
||||
// The ValueScanMutationRuleFunc type is an adapter to allow the use of ordinary
|
||||
// functions as a mutation rule.
|
||||
type ValueScanMutationRuleFunc func(context.Context, *ent.ValueScanMutation) error
|
||||
|
||||
// EvalMutation calls f(ctx, m).
|
||||
func (f ValueScanMutationRuleFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
if m, ok := m.(*ent.ValueScanMutation); ok {
|
||||
return f(ctx, m)
|
||||
}
|
||||
return Denyf("ent/privacy: unexpected mutation type %T, expect *ent.ValueScanMutation", m)
|
||||
}
|
||||
|
||||
type (
|
||||
// Filter is the interface that wraps the Where function
|
||||
// for filtering nodes in queries and mutations.
|
||||
Filter interface {
|
||||
// Where applies a filter on the executed query/mutation.
|
||||
Where(entql.P)
|
||||
}
|
||||
|
||||
// The FilterFunc type is an adapter that allows the use of ordinary
|
||||
// functions as filters for query and mutation types.
|
||||
FilterFunc func(context.Context, Filter) error
|
||||
)
|
||||
|
||||
// EvalQuery calls f(ctx, q) if the query implements the Filter interface, otherwise it is denied.
|
||||
func (f FilterFunc) EvalQuery(ctx context.Context, q ent.Query) error {
|
||||
fr, err := queryFilter(q)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return f(ctx, fr)
|
||||
}
|
||||
|
||||
// EvalMutation calls f(ctx, q) if the mutation implements the Filter interface, otherwise it is denied.
|
||||
func (f FilterFunc) EvalMutation(ctx context.Context, m ent.Mutation) error {
|
||||
fr, err := mutationFilter(m)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return f(ctx, fr)
|
||||
}
|
||||
|
||||
var _ QueryMutationRule = FilterFunc(nil)
|
||||
|
||||
func queryFilter(q ent.Query) (Filter, error) {
|
||||
switch q := q.(type) {
|
||||
case *ent.AccountQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.BlobQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.BlobLinkQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.CarQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.DeviceQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.DocQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.GroupQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.IntSIDQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.LinkQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.MixinIDQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.NoteQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.OtherQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.PetQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.RevisionQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.SessionQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.TokenQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.UserQuery:
|
||||
return q.Filter(), nil
|
||||
case *ent.ValueScanQuery:
|
||||
return q.Filter(), nil
|
||||
default:
|
||||
return nil, Denyf("ent/privacy: unexpected query type %T for query filter", q)
|
||||
}
|
||||
}
|
||||
|
||||
func mutationFilter(m ent.Mutation) (Filter, error) {
|
||||
switch m := m.(type) {
|
||||
case *ent.AccountMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.BlobMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.BlobLinkMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.CarMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.DeviceMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.DocMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.GroupMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.IntSIDMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.LinkMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.MixinIDMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.NoteMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.OtherMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.PetMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.RevisionMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.SessionMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.TokenMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.UserMutation:
|
||||
return m.Filter(), nil
|
||||
case *ent.ValueScanMutation:
|
||||
return m.Filter(), nil
|
||||
default:
|
||||
return nil, Denyf("ent/privacy: unexpected mutation type %T for mutation filter", m)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user