cloudreve/models/user_authn.go

package model

import (
	"encoding/base64"
	"encoding/binary"
	"encoding/json"
	"fmt"
	"github.com/HFO4/cloudreve/pkg/hashid"
	"github.com/duo-labs/webauthn/webauthn"
	"net/url"
)

/*
	`webauthn.User` 接口的实现
*/

// WebAuthnID 返回用户ID
func (user User) WebAuthnID() []byte {
	bs := make([]byte, 8)
	binary.LittleEndian.PutUint64(bs, uint64(user.ID))
	return bs
}

// WebAuthnName 返回用户名
func (user User) WebAuthnName() string {
	return user.Email
}

// WebAuthnDisplayName 获得用于展示的用户名
func (user User) WebAuthnDisplayName() string {
	return user.Nick
}

// WebAuthnIcon 获得用户头像
func (user User) WebAuthnIcon() string {
	avatar, _ := url.Parse("/api/v3/user/avatar/" + hashid.HashID(user.ID, hashid.UserID) + "/l")
	base := GetSiteURL()
	base.Scheme = "https"
	return base.ResolveReference(avatar).String()
}

// WebAuthnCredentials 获得已注册的验证器凭证
func (user User) WebAuthnCredentials() []webauthn.Credential {
	var res []webauthn.Credential
	err := json.Unmarshal([]byte(user.Authn), &res)
	if err != nil {
		fmt.Println(err)
	}
	return res
}

// RegisterAuthn 添加新的验证器
func (user *User) RegisterAuthn(credential *webauthn.Credential) error {
	exists := user.WebAuthnCredentials()
	exists = append(exists, *credential)
	res, err := json.Marshal(exists)
	if err != nil {
		return err
	}

	return DB.Model(user).Update("authn", string(res)).Error
}

// RemoveAuthn 删除验证器
func (user *User) RemoveAuthn(id string) {
	exists := user.WebAuthnCredentials()
	for i := 0; i < len(exists); i++ {
		idEncoded := base64.StdEncoding.EncodeToString(exists[i].ID)
		if idEncoded == id {
			exists[len(exists)-1], exists[i] = exists[i], exists[len(exists)-1]
			exists = exists[:len(exists)-1]
			break
		}
	}

	res, _ := json.Marshal(exists)
	DB.Model(user).Update("authn", string(res))
}