1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
// 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.
package viewer
import (
"context"
"github.com/facebook/ent/examples/privacytenant/ent"
)
// Role for viewer actions.
type Role int
// List of roles.
const (
_ Role = 1 << iota
Admin
View
)
// Viewer describes the query/mutation viewer-context.
type Viewer interface {
Admin() bool // If viewer is admin.
Tenant() string // Tenant name.
}
// UserViewer describes a user-viewer.
type UserViewer struct {
T *ent.Tenant
Role Role // Attached roles.
}
func (v UserViewer) Admin() bool {
return v.Role&Admin != 0
}
func (v UserViewer) Tenant() string {
if v.T != nil {
return v.T.Name
}
return ""
}
type ctxKey struct{}
// FromContext returns the Viewer stored in a context.
func FromContext(ctx context.Context) Viewer {
v, _ := ctx.Value(ctxKey{}).(Viewer)
return v
}
// NewContext returns a copy of parent context with the given Viewer attached with it.
func NewContext(parent context.Context, v Viewer) context.Context {
return context.WithValue(parent, ctxKey{}, v)
}
|
