-
Notifications
You must be signed in to change notification settings - Fork 1
/
token.go
93 lines (74 loc) · 2.07 KB
/
token.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
package cmd
import (
"encoding/json"
"fmt"
"os"
"github.com/spf13/cobra"
"github.com/gomeet/gomeet/utils/jwt"
)
const (
// issuer (the principal that issues tokens)
tokenIssuer = "https://gomeet.com/"
// default token lifetime expressed in hours
defaultTokenLifetimeHours = 24
)
var (
secretSigningKey string
tokenLifetimeHours int
subjectID string
customClaims string
tokenCmd = &cobra.Command{
Use: "token",
Short: "Creates a JSON Web Token",
Run: createToken,
}
)
func init() {
RootCmd.AddCommand(tokenCmd)
// JSON Web Token secret signing key
tokenCmd.PersistentFlags().StringVarP(&secretSigningKey, "secret-key", "k", "", "JSON Web Token secret signing key")
// JSON Web Token lifetime
tokenCmd.PersistentFlags().IntVarP(&tokenLifetimeHours, "lifetime", "l", defaultTokenLifetimeHours, "JSON Web Token lifetime expressed in hours")
// JSON Web Token subject identifier
tokenCmd.PersistentFlags().StringVarP(&subjectID, "subject", "s", "", "JSON Web Token subject identifier")
// JSON Web Token claims
tokenCmd.PersistentFlags().StringVarP(&customClaims, "claims", "c", "", "JSON Web Token custom claims encoded as a JSON object")
}
func createToken(cmd *cobra.Command, args []string) {
// ensure the user provided the secret signing key
if secretSigningKey == "" {
fmt.Printf("missing the secret signing key\n")
os.Exit(1)
}
// set custom token claims
var claimsMap jwt.Claims
if customClaims != "" {
var (
claimsData interface{}
ok bool
)
err := json.Unmarshal([]byte(customClaims), &claimsData)
if err != nil {
fmt.Printf("JSON parsing error: %v\n", err)
os.Exit(1)
}
claimsMap, ok = claimsData.(map[string]interface{})
if !ok {
fmt.Printf("JSON parsing error: failed type assertion on JSON object - %v\n", err)
os.Exit(1)
}
}
token, err := jwt.Create(
"github.com/gomeet/gomeet",
secretSigningKey,
tokenLifetimeHours,
subjectID,
claimsMap,
)
if err != nil {
fmt.Printf("failed to create JWT : %v\n", err)
os.Exit(1)
}
// display the token
fmt.Printf("%s\n", token)
}