mirror of
https://github.com/superseriousbusiness/gotosocial
synced 2024-12-11 16:20:16 +01:00
39 lines
1.1 KiB
Go
39 lines
1.1 KiB
Go
|
package generates
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"encoding/base64"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/google/uuid"
|
||
|
|
"github.com/superseriousbusiness/oauth2/v4"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NewAccessGenerate create to generate the access token instance
|
||
|
|
func NewAccessGenerate() *AccessGenerate {
|
||
|
|
return &AccessGenerate{}
|
||
|
|
}
|
||
|
|
|
||
|
|
// AccessGenerate generate the access token
|
||
|
|
type AccessGenerate struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
// Token based on the UUID generated token
|
||
|
|
func (ag *AccessGenerate) Token(ctx context.Context, data *oauth2.GenerateBasic, isGenRefresh bool) (string, string, error) {
|
||
|
|
buf := bytes.NewBufferString(data.Client.GetID())
|
||
|
|
buf.WriteString(data.UserID)
|
||
|
|
buf.WriteString(strconv.FormatInt(data.CreateAt.UnixNano(), 10))
|
||
|
|
|
||
|
|
access := base64.URLEncoding.EncodeToString([]byte(uuid.NewMD5(uuid.Must(uuid.NewRandom()), buf.Bytes()).String()))
|
||
|
|
access = strings.ToUpper(strings.TrimRight(access, "="))
|
||
|
|
refresh := ""
|
||
|
|
if isGenRefresh {
|
||
|
|
refresh = base64.URLEncoding.EncodeToString([]byte(uuid.NewSHA1(uuid.Must(uuid.NewRandom()), buf.Bytes()).String()))
|
||
|
|
refresh = strings.ToUpper(strings.TrimRight(refresh, "="))
|
||
|
|
}
|
||
|
|
|
||
|
|
return access, refresh, nil
|
||
|
|
}
|