-
Notifications
You must be signed in to change notification settings - Fork 23
/
user.go
109 lines (93 loc) · 2.25 KB
/
user.go
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package ds
import (
"bytes"
dsr3 "github.com/aserto-dev/go-directory/aserto/directory/reader/v3"
"github.com/aserto-dev/go-directory/pkg/convert"
"github.com/aserto-dev/topaz/resolvers"
"github.com/open-policy-agent/opa/ast"
"github.com/open-policy-agent/opa/rego"
"github.com/open-policy-agent/opa/types"
"github.com/pkg/errors"
"github.com/rs/zerolog"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"google.golang.org/protobuf/proto"
)
// RegisterUser - ds.user
//
// v3 (latest) request format:
//
// ds.user({
// "id": ""
// })
//
// v2 request format:
//
// ds.user({
// "key": ""
// })
func RegisterUser(logger *zerolog.Logger, fnName string, dr resolvers.DirectoryResolver) (*rego.Function, rego.Builtin1) {
return ®o.Function{
Name: fnName,
Decl: types.NewFunction(types.Args(types.A), types.A),
Memoize: true,
},
func(bctx rego.BuiltinContext, op1 *ast.Term) (*ast.Term, error) {
var (
args struct {
ID string `json:"id"`
Key string `json:"key"`
}
outputV2 bool
)
if err := ast.As(op1.Value, &args); err != nil {
return nil, err
}
if args.ID == "" && args.Key != "" {
args.ID = args.Key
outputV2 = true
}
if args.ID == "" && args.Key == "" {
type argsV3 struct {
ID string `json:"id"`
}
return help(fnName, argsV3{})
}
client, err := dr.GetDS(bctx.Context)
if err != nil {
return nil, errors.Wrapf(err, "get directory client")
}
resp, err := client.GetObject(bctx.Context, &dsr3.GetObjectRequest{
ObjectType: "user",
ObjectId: args.ID,
WithRelations: false,
})
switch {
case status.Code(err) == codes.NotFound:
traceError(&bctx, fnName, err)
astVal, err := ast.InterfaceToValue(map[string]any{})
if err != nil {
return nil, err
}
return ast.NewTerm(astVal), nil
case err != nil:
return nil, err
}
buf := new(bytes.Buffer)
var result proto.Message
if resp.Result != nil {
result = resp.Result
if outputV2 {
result = convert.ObjectToV2(resp.Result)
}
}
if err := ProtoToBuf(buf, result); err != nil {
return nil, err
}
v, err := ast.ValueFromReader(buf)
if err != nil {
return nil, err
}
return ast.NewTerm(v), nil
}
}