通过验证码注册

routers/users.go

@@ -2,6 +2,7 @@ package routers
 
 import (
 	"crypto/md5"
+	"encoding/json"
 	"fmt"
 	"main/configs"
 	"main/models"
@@ -13,7 +14,7 @@ import (
 	"github.com/gorilla/mux"
 )
 
-// 用戶列表
+// 获取用戶列表
 func UsersGet(w http.ResponseWriter, r *http.Request) {
 	var listview models.ListView
 	listview.Page = utils.ParamInt(r.URL.Query().Get("page"), 1)
@@ -28,34 +29,59 @@ func UsersGet(w http.ResponseWriter, r *http.Request) {
 
 // 創建用戶
 func UsersPost(w http.ResponseWriter, r *http.Request) {
-	var form map[string]interface{} = utils.BodyRead(r)
-	if form["name"] == nil || form["email"] == nil || form["password"] == nil {
-		w.WriteHeader(http.StatusBadRequest)
-		w.Write([]byte("400 - name, email, password cannot be empty"))
+	var data struct {
+		Name     string `json:"name"`
+		Email    string `json:"email"`
+		Mobile   string `json:"mobile"`
+		Password string `json:"password"`
+		Code     string `json:"code"`
+	}
+	if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
 		return
 	}
-	// 創建用戶
-	var slat string = uuid.New().String()
-	var user models.User = models.User{
-		Name:     form["name"].(string),
-		Email:    form["email"].(string),
-		Password: fmt.Sprintf("%x", md5.Sum([]byte(form["password"].(string)+slat))),
-		Slat:     slat,
-	}
-	// 檢查郵箱是否已經存在, 郵箱不能重複
+	var user models.User
 	var count int64
-	configs.ORMDB().Model(&models.User{}).Where("email = ?", user.Email).Count(&count)
-	if count > 0 {
-		w.WriteHeader(http.StatusBadRequest)
-		w.Write([]byte("400 - email already exists"))
-		return
+	// 如果是帐号密码注册
+	if data.Name != "" && data.Password != "" {
+		user.Name = data.Name
+		user.Slat = uuid.New().String()
+		user.Password = fmt.Sprintf("%x", md5.Sum([]byte(data.Password+user.Slat)))
+		configs.ORMDB().Model(&models.User{}).Where("name = ?", user.Name).Count(&count)
+		if count > 0 {
+			http.Error(w, "用户名已存在", http.StatusBadRequest)
+			return
+		}
 	}
-	// 檢查用戶名是否已經存在, 用戶名不能重複
-	configs.ORMDB().Model(&models.User{}).Where("name = ?", user.Name).Count(&count)
-	if count > 0 {
-		w.WriteHeader(http.StatusBadRequest)
-		w.Write([]byte("400 - name already exists"))
-		return
+	// 如果是邮箱验证码注册
+	if data.Email != "" && data.Code != "" {
+		// 检查验证码是否正确
+		if err := models.EmailCheck(data.Email, data.Code); err != nil {
+			http.Error(w, err.Error(), http.StatusBadRequest)
+			return
+		}
+		user.Email = data.Email
+		user.Name = fmt.Sprintf("user_%s", uuid.New().String()) // 设置一个随机用户名
+		configs.ORMDB().Model(&models.User{}).Where("email = ?", user.Email).Count(&count)
+		if count > 0 {
+			http.Error(w, "邮箱已存在", http.StatusBadRequest)
+			return
+		}
+	}
+	// 如果是短信验证码注册
+	if data.Mobile != "" && data.Code != "" {
+		// 检查验证码是否正确
+		if err := models.MobileCheck(data.Mobile, data.Code); err != nil {
+			http.Error(w, err.Error(), http.StatusBadRequest)
+			return
+		}
+		user.Mobile = data.Mobile
+		user.Name = fmt.Sprintf("user_%s", uuid.New().String()) // 设置一个随机用户名
+		configs.ORMDB().Model(&models.User{}).Where("mobile = ?", user.Mobile).Count(&count)
+		if count > 0 {
+			http.Error(w, "手机号已存在", http.StatusBadRequest)
+			return
+		}
 	}
 	// 寫入數據庫
 	if err := configs.ORMDB().Create(&user).Error; err != nil {