gitea
Зеркало из https://github.com/go-gitea/gitea
93 строки · 3.0 Кб
1// Copyright 2021 The Gitea Authors. All rights reserved.
2// SPDX-License-Identifier: MIT
3
4package oauth2_provider //nolint
5
6import (
7"fmt"
8"time"
9
10"code.gitea.io/gitea/modules/timeutil"
11
12"github.com/golang-jwt/jwt/v5"
13)
14
15// Token represents an Oauth grant
16
17// TokenKind represents the type of token for an oauth application
18type TokenKind int
19
20const (
21// KindAccessToken is a token with short lifetime to access the api
22KindAccessToken TokenKind = 0
23// KindRefreshToken is token with long lifetime to refresh access tokens obtained by the client
24KindRefreshToken = iota
25)
26
27// Token represents a JWT token used to authenticate a client
28type Token struct {
29GrantID int64 `json:"gnt"`
30Kind TokenKind `json:"tt"`
31Counter int64 `json:"cnt,omitempty"`
32jwt.RegisteredClaims
33}
34
35// ParseToken parses a signed jwt string
36func ParseToken(jwtToken string, signingKey JWTSigningKey) (*Token, error) {
37parsedToken, err := jwt.ParseWithClaims(jwtToken, &Token{}, func(token *jwt.Token) (any, error) {
38if token.Method == nil || token.Method.Alg() != signingKey.SigningMethod().Alg() {
39return nil, fmt.Errorf("unexpected signing algo: %v", token.Header["alg"])
40}
41return signingKey.VerifyKey(), nil
42})
43if err != nil {
44return nil, err
45}
46if !parsedToken.Valid {
47return nil, fmt.Errorf("invalid token")
48}
49var token *Token
50var ok bool
51if token, ok = parsedToken.Claims.(*Token); !ok || !parsedToken.Valid {
52return nil, fmt.Errorf("invalid token")
53}
54return token, nil
55}
56
57// SignToken signs the token with the JWT secret
58func (token *Token) SignToken(signingKey JWTSigningKey) (string, error) {
59token.IssuedAt = jwt.NewNumericDate(time.Now())
60jwtToken := jwt.NewWithClaims(signingKey.SigningMethod(), token)
61signingKey.PreProcessToken(jwtToken)
62return jwtToken.SignedString(signingKey.SignKey())
63}
64
65// OIDCToken represents an OpenID Connect id_token
66type OIDCToken struct {
67jwt.RegisteredClaims
68Nonce string `json:"nonce,omitempty"`
69
70// Scope profile
71Name string `json:"name,omitempty"`
72PreferredUsername string `json:"preferred_username,omitempty"`
73Profile string `json:"profile,omitempty"`
74Picture string `json:"picture,omitempty"`
75Website string `json:"website,omitempty"`
76Locale string `json:"locale,omitempty"`
77UpdatedAt timeutil.TimeStamp `json:"updated_at,omitempty"`
78
79// Scope email
80Email string `json:"email,omitempty"`
81EmailVerified bool `json:"email_verified,omitempty"`
82
83// Groups are generated by organization and team names
84Groups []string `json:"groups,omitempty"`
85}
86
87// SignToken signs an id_token with the (symmetric) client secret key
88func (token *OIDCToken) SignToken(signingKey JWTSigningKey) (string, error) {
89token.IssuedAt = jwt.NewNumericDate(time.Now())
90jwtToken := jwt.NewWithClaims(signingKey.SigningMethod(), token)
91signingKey.PreProcessToken(jwtToken)
92return jwtToken.SignedString(signingKey.SignKey())
93}
94