@@ -0,0 +1,282 @@
|
||||
package routers
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"saas/models"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func Def_router(r *gin.RouterGroup) {
|
||||
r.Use(func(ctx *gin.Context) {
|
||||
ctx.Set("is_login", false)
|
||||
//读取用户cookie，判断用户是否已登录
|
||||
cookie_vel, _ := ctx.Cookie("user")
|
||||
//fmt.Println(cookie_vel)
|
||||
if cookie_vel != "" {
|
||||
var cookie models.Cookie
|
||||
cookie.Value = cookie_vel
|
||||
if models.DB.Where(&cookie).First(&cookie).Error == nil {
|
||||
// 有数据
|
||||
//有cookie，判断cookie有效性
|
||||
if cookie.ExpiresAt.After(time.Now()) {
|
||||
// ExpiresAt 在当前时间之后（未过期）
|
||||
//fmt.Println("Cookie 未过期")
|
||||
//cookie有效，说明已经登录，cookie过期时间延长，避免大量写入数据库，先判断还有多久过期，小于一天才刷新
|
||||
// 计算过期时间与当前时间的差值
|
||||
remaining := time.Until(cookie.ExpiresAt) // 直接使用 time.Until
|
||||
|
||||
if remaining < 24*time.Hour {
|
||||
//fmt.Println("剩余时间不足 1 天")
|
||||
var cookie_up models.Cookie
|
||||
cookie_up.UpdatedAt = time.Now()
|
||||
cookie_up.ExpiresAt = time.Now().Add(time.Duration(models.User_configs["cookie_timeout"].(int)) * time.Second) //计算过期时间
|
||||
models.DB.Model(&models.Cookie{}).Where(&cookie).Updates(&cookie_up)
|
||||
} else {
|
||||
//fmt.Println("cookie时间大于一天")
|
||||
}
|
||||
|
||||
//读取用户权限信息
|
||||
var user models.User
|
||||
user.ID = cookie.UserID
|
||||
if models.DB.Where(&user).First(&user).Error == nil {
|
||||
//找到登录权限
|
||||
//清除一些重要数据，避免传递的时候泄露
|
||||
user.Pass = ""
|
||||
// 读取用户info
|
||||
var user_info models.User_info
|
||||
user_info.UserID = cookie.UserID
|
||||
if models.DB.Where(&user_info).First(&user_info).Error == nil {
|
||||
// 有数据
|
||||
//fmt.Println(user_info)
|
||||
|
||||
} else {
|
||||
// 无数据
|
||||
//创建一个默认info
|
||||
user_info.AvatarPath = models.User_configs["def_avatar_path"].(string)
|
||||
user_info.UserID = cookie.UserID
|
||||
models.DB.Create(&user_info) // 传入指针
|
||||
}
|
||||
|
||||
ctx.Set("is_login", true)
|
||||
ctx.Set("user_info", &user_info)
|
||||
ctx.Set("user", &user)
|
||||
|
||||
} else {
|
||||
//找不到登录权限？？ 可能被封号？
|
||||
//删除前端cookie
|
||||
ctx.SetCookie("user", "", -1, "/", models.Wed_configs["host"].(string), models.Wed_configs["tls"].(bool), true)
|
||||
}
|
||||
|
||||
} else {
|
||||
// ExpiresAt 在当前时间之前或等于（已过期）
|
||||
//fmt.Println("Cookie 已过期")
|
||||
//删除数据库的cookie
|
||||
models.DB.Delete(&cookie)
|
||||
//删除前端cookie
|
||||
ctx.SetCookie("user", "", -1, "/", models.Wed_configs["host"].(string), models.Wed_configs["tls"].(bool), true)
|
||||
}
|
||||
} else {
|
||||
//找不到cookie，未登录
|
||||
//删除前端cookie
|
||||
ctx.SetCookie("user", "", -1, "/", models.Wed_configs["host"].(string), models.Wed_configs["tls"].(bool), true)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
})
|
||||
|
||||
Api_router(r.Group("/api/")) //分组路由传递到api_routers。go
|
||||
|
||||
//无需权限的页面
|
||||
r.GET("/", func(ctx *gin.Context) {
|
||||
is_login, _ := ctx.Get("is_login")
|
||||
user_info, _ := ctx.Get("user_info")
|
||||
ctx.HTML(http.StatusOK, "index.html", gin.H{
|
||||
"is_login": is_login,
|
||||
"user_info": user_info,
|
||||
})
|
||||
})
|
||||
|
||||
r.GET("/sign-up", func(ctx *gin.Context) {
|
||||
ctx.HTML(http.StatusOK, "sign-up.html", gin.H{})
|
||||
})
|
||||
r.GET("/sign-in", func(ctx *gin.Context) {
|
||||
ctx.HTML(http.StatusOK, "sign-in-01.html", gin.H{})
|
||||
})
|
||||
r.GET("/test", func(ctx *gin.Context) {
|
||||
ctx.HTML(http.StatusOK, "empty.html", gin.H{})
|
||||
})
|
||||
|
||||
//需要权限的页面
|
||||
// r.Use(func(ctx *gin.Context) {
|
||||
// is_login, _ := ctx.Get("is_login")
|
||||
// //判断是否登录
|
||||
// if is_login == true {
|
||||
|
||||
// } else {
|
||||
// ctx.Redirect(302, "/sign-in")
|
||||
// }
|
||||
|
||||
// })
|
||||
r.GET("/workorders", func(ctx *gin.Context) {
|
||||
is_login, _ := ctx.Get("is_login")
|
||||
user_info, _ := ctx.Get("user_info")
|
||||
//判断是否登录
|
||||
if is_login == true {
|
||||
|
||||
ctx.HTML(http.StatusOK, "workorders.html", gin.H{
|
||||
"is_login": is_login,
|
||||
"user_info": user_info,
|
||||
})
|
||||
} else {
|
||||
ctx.Redirect(302, "/sign-in")
|
||||
}
|
||||
})
|
||||
|
||||
r.GET("/warehouses", func(ctx *gin.Context) {
|
||||
is_login, _ := ctx.Get("is_login")
|
||||
user_info, _ := ctx.Get("user_info")
|
||||
//判断是否登录
|
||||
if is_login == true {
|
||||
total_pages := models.Warehouse_get_total_pages() //获取总页数
|
||||
disabled_next_page := false
|
||||
if total_pages <= 1 {
|
||||
disabled_next_page = true
|
||||
} else {
|
||||
disabled_next_page = false
|
||||
}
|
||||
|
||||
ctx.HTML(http.StatusOK, "warehouses.html", gin.H{
|
||||
"total_pages": total_pages,
|
||||
"this_page": 1,
|
||||
"prev_page": fmt.Sprintf("/warehouses/%d", 1),
|
||||
"next_page": fmt.Sprintf("/warehouses/%d", 2),
|
||||
"disabled_prev_page": true,
|
||||
"disabled_next_page": disabled_next_page,
|
||||
"page_range": models.Page_range(1, total_pages, 1, "/warehouses/"),
|
||||
"Warehouses": models.Warehouse_get_warehouses(1),
|
||||
"is_login": is_login,
|
||||
"user_info": user_info,
|
||||
})
|
||||
} else {
|
||||
ctx.Redirect(302, "/sign-in")
|
||||
}
|
||||
})
|
||||
r.GET("/warehouses/:id", func(ctx *gin.Context) {
|
||||
is_login, _ := ctx.Get("is_login")
|
||||
user_info, _ := ctx.Get("user_info")
|
||||
//判断是否登录
|
||||
if is_login == true {
|
||||
id := ctx.Param("id")
|
||||
id_int64, err := strconv.ParseInt(id, 10, 64)
|
||||
if err == nil {
|
||||
if id_int64 > 0 {
|
||||
total_pages := models.Warehouse_get_total_pages() //获取总页数
|
||||
disabled_next_page := false
|
||||
disabled_prev_page := false
|
||||
if total_pages == id_int64 {
|
||||
disabled_next_page = true
|
||||
} else {
|
||||
disabled_next_page = false
|
||||
}
|
||||
if id_int64 > 1 {
|
||||
disabled_prev_page = false
|
||||
} else {
|
||||
disabled_prev_page = true
|
||||
}
|
||||
|
||||
//制作页码标签
|
||||
|
||||
ctx.HTML(http.StatusOK, "warehouses.html", gin.H{
|
||||
"total_pages": total_pages,
|
||||
"this_page": id_int64,
|
||||
"prev_page": fmt.Sprintf("/warehouses/%d", id_int64-1),
|
||||
"next_page": fmt.Sprintf("/warehouses/%d", id_int64+1),
|
||||
"disabled_prev_page": disabled_prev_page,
|
||||
"disabled_next_page": disabled_next_page,
|
||||
"page_range": models.Page_range(1, total_pages, id_int64, "/warehouses/"),
|
||||
"Warehouses": models.Warehouse_get_warehouses(id_int64),
|
||||
"is_login": is_login,
|
||||
"user_info": user_info,
|
||||
})
|
||||
} else {
|
||||
ctx.Redirect(302, "/warehouses")
|
||||
}
|
||||
|
||||
} else {
|
||||
ctx.Redirect(302, "/warehouses")
|
||||
}
|
||||
|
||||
} else {
|
||||
ctx.Redirect(302, "/sign-in")
|
||||
}
|
||||
})
|
||||
|
||||
r.GET("/warehouse/:id", func(ctx *gin.Context) {
|
||||
is_login, _ := ctx.Get("is_login")
|
||||
user_info, _ := ctx.Get("user_info")
|
||||
//判断是否登录
|
||||
if is_login == true {
|
||||
id := ctx.Param("id")
|
||||
id_int64, err := strconv.ParseInt(id, 10, 64)
|
||||
if err == nil {
|
||||
if id_int64 > 0 {
|
||||
|
||||
ctx.HTML(http.StatusOK, "warehouses_show_item.html", gin.H{
|
||||
"warehouse_id": id_int64,
|
||||
"is_login": is_login,
|
||||
"user_info": user_info,
|
||||
"items": models.Warehouse_get_items_from_whid(uint(id_int64)),
|
||||
})
|
||||
} else {
|
||||
ctx.Redirect(302, "/warehouses")
|
||||
}
|
||||
|
||||
} else {
|
||||
ctx.Redirect(302, "/warehouses")
|
||||
}
|
||||
|
||||
} else {
|
||||
ctx.Redirect(302, "/sign-in")
|
||||
}
|
||||
})
|
||||
r.GET("/setting-my", func(ctx *gin.Context) {
|
||||
is_login, _ := ctx.Get("is_login")
|
||||
user_info, _ := ctx.Get("user_info")
|
||||
user, _ := ctx.Get("user")
|
||||
//判断是否登录
|
||||
if is_login == true {
|
||||
|
||||
ctx.HTML(http.StatusOK, "setting-my.html", gin.H{
|
||||
"is_login": is_login,
|
||||
"user_info": user_info,
|
||||
"user": user,
|
||||
})
|
||||
} else {
|
||||
ctx.HTML(404, "error_404.html", gin.H{})
|
||||
}
|
||||
|
||||
})
|
||||
|
||||
r.GET("/setting-security", func(ctx *gin.Context) {
|
||||
is_login, _ := ctx.Get("is_login")
|
||||
user_info, _ := ctx.Get("user_info")
|
||||
user, _ := ctx.Get("user")
|
||||
//判断是否登录
|
||||
if is_login == true {
|
||||
|
||||
ctx.HTML(http.StatusOK, "setting-security.html", gin.H{
|
||||
"is_login": is_login,
|
||||
"user_info": user_info,
|
||||
"user": user,
|
||||
})
|
||||
} else {
|
||||
ctx.HTML(404, "error_404.html", gin.H{})
|
||||
}
|
||||
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user