support set gravatar as user avatar provider

pkg/models/user.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"time"
 
+	"github.com/mayswind/ezbookkeeping/pkg/settings"
 	"github.com/mayswind/ezbookkeeping/pkg/utils"
 )
 
@@ -77,6 +78,7 @@ type UserBasicInfo struct {
 	Username             string               `json:"username"`
 	Email                string               `json:"email"`
 	Nickname             string               `json:"nickname"`
+	AvatarUrl            string               `json:"avatar"`
 	DefaultAccountId     int64                `json:"defaultAccountId,string"`
 	TransactionEditScope TransactionEditScope `json:"transactionEditScope"`
 	Language             string               `json:"language"`
@@ -133,6 +135,7 @@ type UserProfileResponse struct {
 	Username             string               `json:"username"`
 	Email                string               `json:"email"`
 	Nickname             string               `json:"nickname"`
+	AvatarUrl            string               `json:"avatar"`
 	DefaultAccountId     int64                `json:"defaultAccountId,string"`
 	TransactionEditScope TransactionEditScope `json:"transactionEditScope"`
 	Language             string               `json:"language"`
@@ -193,6 +196,7 @@ func (u *User) ToUserBasicInfo() *UserBasicInfo {
 		Username:             u.Username,
 		Email:                u.Email,
 		Nickname:             u.Nickname,
+		AvatarUrl:            u.getAvatarUrl(),
 		DefaultAccountId:     u.DefaultAccountId,
 		TransactionEditScope: u.TransactionEditScope,
 		Language:             u.Language,
@@ -211,6 +215,7 @@ func (u *User) ToUserProfileResponse() *UserProfileResponse {
 		Username:             u.Username,
 		Email:                u.Email,
 		Nickname:             u.Nickname,
+		AvatarUrl:            u.getAvatarUrl(),
 		DefaultAccountId:     u.DefaultAccountId,
 		TransactionEditScope: u.TransactionEditScope,
 		Language:             u.Language,
@@ -223,3 +228,13 @@ func (u *User) ToUserProfileResponse() *UserProfileResponse {
 		LastLoginAt:          u.LastLoginUnixTime,
 	}
 }
+
+func (u *User) getAvatarUrl() string {
+	avatarProvider := settings.Container.Current.AvatarProvider
+
+	if avatarProvider == settings.GravatarProvider {
+		return utils.GetGravatarUrl(u.Email)
+	}
+
+	return ""
+}

pkg/settings/setting.go

@@ -62,6 +62,11 @@ const (
 	InternalUuidGeneratorType string = "internal"
 )
 
+// User avatar provider types
+const (
+	GravatarProvider string = "gravatar"
+)
+
 // Map provider types
 const (
 	OpenStreetMapProvider                  string = "openstreetmap"
@@ -185,6 +190,7 @@ type Config struct {
 
 	// User
 	EnableUserRegister bool
+	AvatarProvider     string
 
 	// Data
 	EnableDataExport bool
@@ -444,6 +450,12 @@ func loadSecurityConfiguration(config *Config, configFile *ini.File, sectionName
 func loadUserConfiguration(config *Config, configFile *ini.File, sectionName string) error {
 	config.EnableUserRegister = getConfigItemBoolValue(configFile, sectionName, "enable_register", false)
 
+	if getConfigItemStringValue(configFile, sectionName, "avatar_provider") == "" {
+		config.AvatarProvider = ""
+	} else if getConfigItemStringValue(configFile, sectionName, "avatar_provider") == GravatarProvider {
+		config.AvatarProvider = GravatarProvider
+	}
+
 	return nil
 }
 

pkg/utils/avatar.go

@@ -0,0 +1,11 @@
+package utils
+
+import "fmt"
+
+const gravatarUrlFormat = "https://www.gravatar.com/avatar/%s"
+
+// GetGravatarUrl returns the Gravatar url according to the specified user email address
+func GetGravatarUrl(email string) string {
+	emailMd5 := MD5EncodeToString([]byte(email))
+	return fmt.Sprintf(gravatarUrlFormat, emailMd5)
+}

pkg/utils/strings.go

@@ -7,6 +7,7 @@ import (
 	"crypto/rand"
 	"crypto/sha256"
 	"encoding/base64"
+	"encoding/hex"
 	"strings"
 	"unicode"
 
@@ -133,6 +134,12 @@ func MD5Encode(data []byte) []byte {
 	return m.Sum(nil)
 }
 
+// MD5EncodeToString returns a hashed string by md5
+func MD5EncodeToString(data []byte) string {
+	hash := MD5Encode(data)
+	return hex.EncodeToString(hash)
+}
+
 // AESGCMEncrypt returns a encrypted string by aes-gcm
 func AESGCMEncrypt(key []byte, plainText []byte) ([]byte, error) {
 	block, err := aes.NewCipher(key)