128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
package user
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"html/template"
|
||
"io"
|
||
"log"
|
||
"net/http"
|
||
"net/smtp"
|
||
"strings"
|
||
"time"
|
||
|
||
"math/rand"
|
||
|
||
"github.com/go-redis/redis/v8"
|
||
)
|
||
|
||
type Userregveri struct {
|
||
Email string `json:"email"`
|
||
}
|
||
|
||
func randomCode() string {
|
||
source := rand.NewSource(time.Now().UnixNano())
|
||
r := rand.New(source)
|
||
letters := []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
|
||
b := make([]rune, 6)
|
||
for i := range b {
|
||
b[i] = letters[r.Intn(len(letters))]
|
||
}
|
||
return string(b)
|
||
}
|
||
|
||
func Registerverify(w http.ResponseWriter, r *http.Request) {
|
||
w.Header().Set("Access-Control-Allow-Origin", "http://localhost:5173")
|
||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, OPTIONS")
|
||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||
w.Header().Set("Access-Control-Allow-Credentials", "true")
|
||
|
||
if r.Method == "OPTIONS" {
|
||
w.WriteHeader(http.StatusOK)
|
||
return
|
||
}
|
||
bodyrec, error := io.ReadAll(r.Body)
|
||
if error != nil {
|
||
fmt.Println("接受数据错误")
|
||
return
|
||
}
|
||
var bodyjx Userregveri
|
||
error = json.Unmarshal(bodyrec, &bodyjx)
|
||
if error != nil {
|
||
fmt.Println("数据解析失败")
|
||
return
|
||
}
|
||
|
||
var useremail = bodyjx.Email
|
||
log.Println("邮箱是", useremail)
|
||
|
||
redisdb := redis.NewClient(&redis.Options{
|
||
Addr: "localhost:6379",
|
||
Password: "",
|
||
DB: 0,
|
||
})
|
||
var redisverifyhash = "userverify"
|
||
|
||
_, error = redisdb.Ping(context.Background()).Result()
|
||
if error != nil {
|
||
log.Println("redis连接失败")
|
||
return
|
||
}
|
||
|
||
result, _ := redisdb.Exists(context.Background(), redisverifyhash, useremail).Result()
|
||
if result == 1 {
|
||
log.Println("用户验证码已经存在")
|
||
redisdb.HDel(context.Background(), redisverifyhash, useremail)
|
||
}
|
||
log.Println(result)
|
||
codesend := randomCode()
|
||
|
||
error = redisdb.HSet(context.Background(), redisverifyhash, map[string]string{
|
||
"useremail": useremail,
|
||
"userver": codesend,
|
||
}).Err()
|
||
if error != nil {
|
||
log.Println("redis键值设置失败")
|
||
return
|
||
}
|
||
|
||
_, error = redisdb.Expire(context.Background(), redisverifyhash, 3*time.Minute).Result()
|
||
if error != nil {
|
||
log.Println("有效时长设置失败")
|
||
return
|
||
}
|
||
|
||
from := "mayiming_person@163.com"
|
||
password := "VChadvNigu6TdzYH"
|
||
to := []string{useremail}
|
||
subject := "MS用户验证码 --MS助手"
|
||
|
||
htmlContent := `以下为您的验证码，3分钟内有效，请勿分享给别人. 验证码：` + string(codesend) + ``
|
||
t, err := template.New("emailBody").Parse(htmlContent)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
var bodyBuilder strings.Builder
|
||
err = t.Execute(&bodyBuilder, nil)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
body := bodyBuilder.String()
|
||
|
||
msg := "From: " + from + "\r\n" +
|
||
"To: " + to[0] + "\r\n" +
|
||
"Subject: " + subject + "\r\n" +
|
||
"\r\n" +
|
||
body
|
||
|
||
smtpServer := "smtp.163.com:25"
|
||
auth := smtp.PlainAuth("", from, password, "smtp.163.com")
|
||
err = smtp.SendMail(smtpServer, auth, from, to, []byte(msg))
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
log.Println("邮箱发送成功!")
|
||
|
||
}
|