-
-
Notifications
You must be signed in to change notification settings - Fork 160
/
builder_gen.go
71 lines (57 loc) · 1.82 KB
/
builder_gen.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
// This file is auto-generated by jwt/internal/cmd/gentoken/main.go. DO NOT EDIT
package jwt
import (
"time"
"github.com/pkg/errors"
)
// Builder is a convenience wrapper around the New() constructor
// and the Set() methods to assign values to Token claims.
// Users can successively call Claim() on the Builder, and have it
// construct the Token when Build() is called. This alleviates the
// need for the user to check for the return value of every single
// Set() method call.
// Note that each call to Claim() overwrites the value set from the
// previous call.
type Builder struct {
claims []*ClaimPair
}
func NewBuilder() *Builder {
return &Builder{}
}
func (b *Builder) Claim(name string, value interface{}) *Builder {
b.claims = append(b.claims, &ClaimPair{Key: name, Value: value})
return b
}
func (b *Builder) Audience(v []string) *Builder {
return b.Claim(AudienceKey, v)
}
func (b *Builder) Expiration(v time.Time) *Builder {
return b.Claim(ExpirationKey, v)
}
func (b *Builder) IssuedAt(v time.Time) *Builder {
return b.Claim(IssuedAtKey, v)
}
func (b *Builder) Issuer(v string) *Builder {
return b.Claim(IssuerKey, v)
}
func (b *Builder) JwtID(v string) *Builder {
return b.Claim(JwtIDKey, v)
}
func (b *Builder) NotBefore(v time.Time) *Builder {
return b.Claim(NotBeforeKey, v)
}
func (b *Builder) Subject(v string) *Builder {
return b.Claim(SubjectKey, v)
}
// Build creates a new token based on the claims that the builder has received
// so far. If a claim cannot be set, then the method returns a nil Token with
// a en error as a second return value
func (b *Builder) Build() (Token, error) {
tok := New()
for _, claim := range b.claims {
if err := tok.Set(claim.Key.(string), claim.Value); err != nil {
return nil, errors.Wrapf(err, `failed to set claim %q`, claim.Key.(string))
}
}
return tok, nil
}