70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package auth
|
|
|
|
import (
|
|
"fusenapi/utils/encryption_decryption"
|
|
"net/url"
|
|
)
|
|
|
|
type OperateType int8
|
|
|
|
const (
|
|
OpTypeRegister OperateType = 1 //注册的操作类型
|
|
OpTypeResetToken OperateType = 2 //重置密码类型
|
|
)
|
|
|
|
type ConfirmationLink[T any] struct {
|
|
// Secret []byte
|
|
SecretGCM *encryption_decryption.SecretGCM[T]
|
|
|
|
DefaultQueryKey string // 默认key 是 token
|
|
link *url.URL
|
|
}
|
|
|
|
func NewConfirmationLink[T any](key string, UrlStr string) *ConfirmationLink[T] {
|
|
u, err := url.Parse(UrlStr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return &ConfirmationLink[T]{
|
|
SecretGCM: encryption_decryption.NewSecretGCM[T](key),
|
|
DefaultQueryKey: "token",
|
|
link: u,
|
|
}
|
|
}
|
|
|
|
// Generate 序列化链接传入需求的obj
|
|
func (cl *ConfirmationLink[T]) Generate(obj *T) (string, error) {
|
|
|
|
token, err := cl.Encrypt(obj)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return cl.GenerateWithToken(token)
|
|
}
|
|
|
|
// GenerateWithToken 序列化url带token
|
|
func (cl *ConfirmationLink[T]) GenerateWithToken(token string) (string, error) {
|
|
|
|
q := cl.link.Query()
|
|
if q.Has(cl.DefaultQueryKey) {
|
|
q.Set(cl.DefaultQueryKey, token)
|
|
} else {
|
|
q.Add(cl.DefaultQueryKey, token)
|
|
}
|
|
|
|
// 生成确认链接
|
|
cl.link.RawQuery = q.Encode()
|
|
|
|
return cl.link.String(), nil
|
|
}
|
|
|
|
func (cl *ConfirmationLink[T]) Encrypt(obj *T) (string, error) {
|
|
return cl.SecretGCM.Encrypt(obj)
|
|
}
|
|
|
|
func (cl *ConfirmationLink[T]) Decrypt(ciphertext string) (*T, error) {
|
|
return cl.SecretGCM.Decrypt(ciphertext)
|
|
}
|