296 lines
6.9 KiB
Go
296 lines
6.9 KiB
Go
package api
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/gin-gonic/gin"
|
||
_ "gorm.io/driver/sqlite"
|
||
db "main.go/database"
|
||
"main.go/tools"
|
||
. "main.go/type"
|
||
"regexp"
|
||
"unicode/utf8"
|
||
)
|
||
|
||
//连接数据库
|
||
func Link() {
|
||
err := db.Inimysql()
|
||
if err != nil{
|
||
panic(err)
|
||
}
|
||
}
|
||
|
||
|
||
|
||
|
||
func Login(c * gin.Context) {
|
||
var request LoginRequest
|
||
var user User
|
||
|
||
Link()
|
||
DB := db.DBsnctf
|
||
DB.AutoMigrate(&User{})
|
||
//用ShouldBindJSON解析绑定传入的Json数据。
|
||
if err := c.ShouldBindJSON(&request); err != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Bind json error!"})
|
||
return
|
||
}
|
||
//code 0用户名不存在
|
||
//code 1用户名或密码错误
|
||
|
||
err := DB.Take(&user,"username = ?",request.Username).Error
|
||
if err != nil {
|
||
c.JSON(200,gin.H{
|
||
"Error":true,
|
||
"Msg":"登陆失败",
|
||
"Code": 0,
|
||
})
|
||
return
|
||
}
|
||
//判断md5值与数据库内容是否相同
|
||
if user.Password != tools.MD5(request.Password){
|
||
c.JSON(200,gin.H{
|
||
"Error":true,
|
||
"Msg":"用户名或密码错误",
|
||
"Code": 1,
|
||
})
|
||
return
|
||
}
|
||
//身份认证结束
|
||
//session
|
||
// 设置session
|
||
session, _ := Store.Get(c.Request, "SNCTFSESSID")
|
||
user.Password = ""
|
||
// 根据remember值设置session有效期
|
||
if request.Remember {
|
||
session.Options.MaxAge = 7 * 24 * 60 * 60 // 7 days
|
||
} else {
|
||
session.Options.MaxAge = 24 * 60 * 60 // 1 day
|
||
}
|
||
//保存session
|
||
session.Values["user"] = user
|
||
|
||
err = session.Save(c.Request, c.Writer)
|
||
fmt.Println(err)
|
||
if err != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Save SNCTFSESSID error"})
|
||
return
|
||
}
|
||
//登录成功
|
||
c.JSON(200, gin.H{"code": 200, "username": user.Username, "role": user.Role, "msg": "Login success!"})
|
||
|
||
}
|
||
|
||
|
||
func Register(c *gin.Context) {
|
||
var request RegisterRequest
|
||
var user User
|
||
var score ScoreResponse
|
||
|
||
Link()
|
||
DB := db.DBsnctf
|
||
err := DB.AutoMigrate(&User{})
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
//用ShouldBindJSON解析绑定传入的Json数据。
|
||
if err := c.ShouldBindJSON(&request); err != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Bind json error!"})
|
||
return
|
||
}
|
||
//限制传入用户名为中文、数字、大小写字母下划线和横杠，1到10位
|
||
if !checkUsername(request.Username) {
|
||
c.JSON(200, gin.H{
|
||
"Error": true,
|
||
"Msg": "用户名不符合规范",
|
||
"Code": 2,
|
||
})
|
||
return
|
||
}
|
||
//限制密码长度6到20位
|
||
if !checkPassword(request.Password) {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Password format error!"})
|
||
return
|
||
}
|
||
//限制传入邮箱符合格式
|
||
if !checkEmail(request.Email) {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Email format error!"})
|
||
return
|
||
}
|
||
//判断用户名是否已被使用
|
||
if isNameExisted(user, request.Username) {
|
||
c.JSON(200, gin.H{"code": 1000, "msg": "Username has already been used!"})
|
||
return
|
||
}
|
||
//判断邮箱是否已被使用
|
||
if isEmailExisted(user, request.Email) {
|
||
c.JSON(200, gin.H{"code": 1001, "msg": "Email has already been used!"})
|
||
return
|
||
}
|
||
user.Token = tools.Token()
|
||
user.Username = request.Username
|
||
user.Password = tools.MD5(request.Password)
|
||
user.Email = request.Email
|
||
user.Created = tools.Timestamp()
|
||
score.Score = 0
|
||
score.Username =request.Username
|
||
//创建数据
|
||
err1 := DB.Table("user").Create(&user).Error
|
||
err2 := DB.Table("score").Create(&score).Error
|
||
if err1 != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Register error!"})
|
||
return
|
||
}
|
||
if err2 != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Register error!"})
|
||
return
|
||
}
|
||
|
||
c.JSON(200, gin.H{
|
||
"success": "注册成功",
|
||
})
|
||
|
||
|
||
}
|
||
|
||
// Logout 实现用户注销登陆
|
||
func Logout(c *gin.Context) {
|
||
var user User
|
||
|
||
session, err := Store.Get(c.Request, "SNCTFSESSID")
|
||
if err != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Get CTFGOSESSID error"})
|
||
return
|
||
}
|
||
user, ok := session.Values["user"].(User)
|
||
if !ok {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "No session"})
|
||
return
|
||
}
|
||
|
||
session.Options.MaxAge = -1
|
||
err = session.Save(c.Request, c.Writer)
|
||
if err != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Save CTFGOSESSID error"})
|
||
return
|
||
}
|
||
|
||
fmt.Sprintf("[%s] logout success!", user.Username)
|
||
c.JSON(200, gin.H{"code": 200, "msg": "Logout success!"})
|
||
}
|
||
//GetInfoByUserId 获取用户信息
|
||
func GetInfoByUserId(c *gin.Context) {
|
||
var info PublicInfoResponse
|
||
Link()
|
||
DB := db.DBsnctf
|
||
//获取用户id
|
||
id := c.Params.ByName("id")
|
||
if id == "" {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Need id!"})
|
||
return
|
||
}
|
||
//检查id是否合法
|
||
|
||
|
||
if !tools.CheckID(id) {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "ID format error!"})
|
||
return
|
||
}
|
||
|
||
err := DB.Debug().Raw("SELECT username,affiliation,country,team_id FROM user WHERE id = ? LIMIT 1", id).Scan(&info).Error
|
||
//err := DB.Where("id = ?", id).First(user).Error
|
||
if err != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Get info error!"})
|
||
return
|
||
}
|
||
c.JSON(200, gin.H{"code": 200, "data": info})
|
||
|
||
|
||
}
|
||
|
||
|
||
// checkUsername 验证用户名是否符合中文数字字母下划线横杠，长度1到10位，返回true或false
|
||
func checkUsername(username string) bool {
|
||
if !(utf8.RuneCountInString(username) > 0) || !(utf8.RuneCountInString(username) < 11) {
|
||
return false
|
||
}
|
||
pattern := `^[-\w\p{Han}]+$`
|
||
reg := regexp.MustCompile(pattern)
|
||
return reg.MatchString(username)
|
||
}
|
||
|
||
// checkEmail 验证是否符合邮箱格式，返回true或false
|
||
func checkEmail(email string) bool {
|
||
pattern := `^\w+([-+.]\w+)*@\w+([-.]\w+)*\.\w+([-.]\w+)*$`
|
||
reg := regexp.MustCompile(pattern)
|
||
return reg.MatchString(email)
|
||
}
|
||
|
||
|
||
// checkPassword 验证密码是否符合长度6到20位，返回true或false
|
||
func checkPassword(password string) bool {
|
||
if !(utf8.RuneCountInString(password) > 5) || !(utf8.RuneCountInString(password) < 21) {
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
// isNameExisted 判断用户名是否已经被占用，被占用返回true，未被占用则返回false
|
||
func isNameExisted(user User, username string) bool {
|
||
Link()
|
||
DB := db.DBsnctf
|
||
err := DB.First(&user,"Username = ?",username).Error
|
||
if err != nil {
|
||
return false
|
||
}
|
||
return true
|
||
|
||
}
|
||
// isNameExisted 判断邮箱是否已经被占用，被占用返回true，未被占用则返回false
|
||
func isEmailExisted(user User, email string) bool {
|
||
Link()
|
||
DB := db.DBsnctf
|
||
|
||
err := DB.First(&user,"Email = ?",email).Error
|
||
if err != nil {
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
// Session 获取当前用户session信息
|
||
func Session(c *gin.Context) {
|
||
var user User
|
||
|
||
// 默认在此之前已经通过了中间件的session权限验证
|
||
session, _ := Store.Get(c.Request, "SNCTFSESSID")
|
||
user = session.Values["user"].(User)
|
||
|
||
c.JSON(200, gin.H{"code": 200, "data": user})
|
||
}
|
||
|
||
|
||
|
||
//下面是身份认证用 AUTH部分
|
||
// AuthRequired 用于普通用户权限控制的中间件
|
||
func AuthRequired()gin.HandlerFunc{
|
||
return func(c *gin.Context) {
|
||
session, err := Store.Get(c.Request, "SNCTFSESSID")
|
||
if err != nil {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Get SNCTFSESSID error"})
|
||
c.Abort()
|
||
return
|
||
}
|
||
user, ok := session.Values["user"].(User)
|
||
if !ok {
|
||
c.JSON(400, gin.H{"code": 400, "msg": "No session"})
|
||
c.Abort()
|
||
return
|
||
}
|
||
if user.Role != 0&&user.Role!=1{
|
||
c.JSON(400, gin.H{"code": 400, "msg": "Permission denied"})
|
||
c.Abort()
|
||
return
|
||
}
|
||
c.Next()
|
||
}
|
||
} |