File: jwx_example_test.go

package info (click to toggle)
golang-github-lestrrat-go-jwx 2.1.4-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,872 kB
  • sloc: sh: 222; makefile: 86; perl: 62
file content (103 lines) | stat: -rw-r--r-- 2,213 bytes parent folder | download
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
package examples_test

import (
	"crypto/rand"
	"crypto/rsa"
	"fmt"
	"time"

	"github.com/lestrrat-go/jwx/v2/internal/json"
	"github.com/lestrrat-go/jwx/v2/jwa"
	"github.com/lestrrat-go/jwx/v2/jwk"
	"github.com/lestrrat-go/jwx/v2/jwt"
)

var payloadLoremIpsum []byte
var jwtSignedWithHS256 []byte
var rawRSAPrivateKey *rsa.PrivateKey
var rawRSAPublicKey *rsa.PublicKey
var jwkRSAPrivateKey jwk.Key
var jwkRSAPublicKey jwk.Key
var jwkSymmetricKey jwk.Key
var jsonRSAPrivateKey []byte
var jsonRSAPublicKey []byte

func init() {
	if err := Setup(); err != nil {
		panic(err.Error())
	}
}

// Create some variables that would be repeatedly used in the examples
func Setup() error {
	payloadLoremIpsum = []byte(`Lorem ipsum`)

	tok, err := jwt.NewBuilder().
		Issuer(`github.com/lestrrat-go/jwx`).
		IssuedAt(time.Now().Add(-5 * time.Minute)).
		Expiration(time.Now().Add(time.Hour)).
		Build()
	if err != nil {
		return fmt.Errorf(`failed to build token: %w`, err)
	}

	{
		v, err := rsa.GenerateKey(rand.Reader, 2048)
		if err != nil {
			return fmt.Errorf(`failed to create RSA private key: %w`, err)
		}

		rawRSAPrivateKey = v
		rawRSAPublicKey = &v.PublicKey
	}

	{
		v, err := jwk.FromRaw(rawRSAPrivateKey)
		if err != nil {
			return fmt.Errorf(`failed to create jwk.Key from RSA private key: %w`, err)
		}
		jwkRSAPrivateKey = v
	}

	{
		v, err := json.Marshal(jwkRSAPrivateKey)
		if err != nil {
			return fmt.Errorf(`failed to marshal RSA private jwk.Key: %w`, err)
		}
		jsonRSAPrivateKey = v
	}

	{
		v, err := jwk.FromRaw(rawRSAPublicKey)
		if err != nil {
			return fmt.Errorf(`failed to create jwk.Key from RSA public key: %w`, err)
		}
		jwkRSAPublicKey = v
	}

	{
		v, err := json.Marshal(jwkRSAPublicKey)
		if err != nil {
			return fmt.Errorf(`failed to marshal RSA public jwk.Key: %w`, err)
		}
		jsonRSAPublicKey = v
	}

	{
		v, err := jwk.FromRaw([]byte(`abracadabra`))
		if err != nil {
			return fmt.Errorf(`failed to create jwk.Key from symmetric key: %w`, err)
		}
		jwkSymmetricKey = v
	}

	{
		v, err := jwt.Sign(tok, jwt.WithKey(jwa.HS256, jwkSymmetricKey))
		if err != nil {
			return fmt.Errorf(`failed to sign token with HS256: %w`, err)
		}
		jwtSignedWithHS256 = v
	}

	return nil
}