Feat: login / register setting in dashboard

pkg/email/mail.go

@@ -1,6 +1,9 @@
 package email
 
-import "errors"
+import (
+	"errors"
+	"strings"
+)
 
 // Driver 邮件发送驱动
 type Driver interface {
@@ -19,6 +22,11 @@ var (
 
 // Send 发送邮件
 func Send(to, title, body string) error {
+	// 忽略通过QQ登录的邮箱
+	if strings.HasSuffix(to, "@login.qq.com") {
+		return nil
+	}
+
 	if Client == nil {
 		return ErrNoActiveDriver
 	}

pkg/qq/connect.go

@@ -25,6 +25,12 @@ type UserCredentials struct {
 	AccessToken string
 }
 
+// UserInfo 用户信息
+type UserInfo struct {
+	Nick   string
+	Avatar string
+}
+
 var (
 	// ErrNotEnabled 未开启登录功能
 	ErrNotEnabled = serializer.NewError(serializer.CodeNoPermissionErr, "QQ登录功能未开启", nil)
@@ -89,6 +95,20 @@ func getAccessTokenURL(code string) string {
 	return api.String()
 }
 
+func getUserInfoURL(openid, ak string) string {
+	// 获取相关设定
+	options := model.GetSettingByNames("qq_login_id", "qq_login_key")
+
+	api, _ := url.Parse("https://graph.qq.com/user/get_user_info")
+	queries := api.Query()
+	queries.Add("oauth_consumer_key", options["qq_login_id"])
+	queries.Add("openid", openid)
+	queries.Add("access_token", ak)
+	api.RawQuery = queries.Encode()
+
+	return api.String()
+}
+
 func getResponse(body string) (map[string]interface{}, error) {
 	var res map[string]interface{}
 
@@ -157,3 +177,35 @@ func Callback(code string) (*UserCredentials, error) {
 
 	return nil, ErrDecodeResponse
 }
+
+// GetUserInfo 使用凭证获取用户信息
+func GetUserInfo(credential *UserCredentials) (*UserInfo, error) {
+	api := getUserInfoURL(credential.OpenID, credential.AccessToken)
+
+	// 获取用户信息
+	client := request.HTTPClient{}
+	res := client.Request("GET", api, nil)
+	resp, err := res.GetResponse()
+	if err != nil {
+		return nil, ErrObtainAccessToken.WithError(err)
+	}
+
+	var resSerialized map[string]interface{}
+	if err := json.Unmarshal([]byte(resp), &resSerialized); err != nil {
+		return nil, ErrDecodeResponse.WithError(err)
+	}
+
+	// 如果服务端返回错误
+	if msg, ok := resSerialized["msg"]; ok && msg.(string) != "" {
+		return nil, ErrObtainAccessToken.WithError(errors.New(msg.(string)))
+	}
+
+	if avatar, ok := resSerialized["figureurl_qq_2"]; ok {
+		return &UserInfo{
+			Nick:   resSerialized["nickname"].(string),
+			Avatar: avatar.(string),
+		}, nil
+	}
+
+	return nil, ErrDecodeResponse
+}

pkg/thumb/image.go

@@ -24,7 +24,7 @@ type Thumb struct {
 
 // NewThumbFromFile 从文件数据获取新的Thumb对象，
 // 尝试通过文件名name解码图像
-func NewThumbFromFile(file io.ReadSeeker, name string) (*Thumb, error) {
+func NewThumbFromFile(file io.Reader, name string) (*Thumb, error) {
 	ext := strings.ToLower(filepath.Ext(name))
 	// 无扩展名时
 	if len(ext) == 0 {