From e231709fbf7525f547b6d9546b61f3055dab2c06 Mon Sep 17 00:00:00 2001
From: ZJY <764763903@qq.com>
Date: Thu, 2 Sep 2021 19:47:00 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=AE=89=E5=85=A8=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98=20=E8=AF=B7=E5=8F=8A=E6=97=B6=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controllers/base.go  |  1 +
 controllers/login.go |  1 +
 models/db.go         | 32 ++++++++++++++++++++++++++++++++
 models/toekn.go      | 11 +++++++++++
 4 files changed, 45 insertions(+)
 create mode 100644 models/toekn.go

diff --git a/controllers/base.go b/controllers/base.go
index 4f534d4..0383c1b 100644
--- a/controllers/base.go
+++ b/controllers/base.go
@@ -114,6 +114,7 @@ func (c *BaseController) Logined() *BaseController {
 	} else {
 		logs.Warn("登录成功")
 		c.PtPin = v.(string)
+		logs.Info(models.Config.Master)
 		if strings.EqualFold(models.Config.Master, v.(string)) {
 			c.Master = true
 		}
diff --git a/controllers/login.go b/controllers/login.go
index 046a019..b674689 100644
--- a/controllers/login.go
+++ b/controllers/login.go
@@ -330,6 +330,7 @@ func (c *LoginController) IsAdmin() {
 		c.Ctx.Redirect(302, "/")
 		c.StopRun()
 	} else {
+
 		c.SetSession("pin", pin)
 		c.Ctx.WriteString("登录")
 	}
diff --git a/models/db.go b/models/db.go
index b8c10de..17d1bcd 100644
--- a/models/db.go
+++ b/models/db.go
@@ -42,6 +42,7 @@ func initDB() {
 		&UserAgent{},
 		&Env{},
 		&Wish{},
+		&Token{},
 	)
 	keys = make(map[string]bool)
 	pins = make(map[string]bool)
@@ -273,3 +274,34 @@ func CheckIn(pin, key string) int {
 	}
 	return 2
 }
+
+func setToken(token *Token) error {
+	tx := db.Begin()
+	if err := tx.Create(token).Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+	return tx.Commit().Error
+}
+
+//func getToken(pin string) (*Token, error) {
+//	token := &Token{}
+//	db.Where(CreateAt+" = ?", pin).First(token)
+//	format := "2006-01-02 15:04:05"
+//	sqlUpdatedAt, _ := time.ParseInLocation(format, "2021-03-24 15:00:00", time.Local)
+//	fmt.Println("测试时间:", sqlUpdatedAt)
+//
+//	t := time.Now()
+//	t_zero := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, t.Location())
+//	fmt.Println("当天凌晨时间:", t_zero)
+//
+//	t_ := sqlUpdatedAt.Sub(t_zero)
+//	fmt.Println("测试时间到当天凌晨时间:", t_)
+//	if t_ > 0 {
+//		fmt.Println("未超有效时间!!!")
+//	} else {
+//		fmt.Println("超过有效时间!!!")
+//	}
+//
+//	return token, db.Where(CreateAt+" = ?", pin).First(ck).Error
+//}
diff --git a/models/toekn.go b/models/toekn.go
new file mode 100644
index 0000000..597d8b8
--- /dev/null
+++ b/models/toekn.go
@@ -0,0 +1,11 @@
+package models
+
+import (
+	"time"
+)
+
+type Token struct {
+	ID        int
+	CreatedAt time.Time
+	Token     string
+}