59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package models
|
|
|
|
import (
|
|
"main/configs"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type Account struct {
|
|
ID int `json:"id"`
|
|
Name string `json:"name"`
|
|
Email string `json:"email"`
|
|
Admin bool `json:"admin"`
|
|
SessionID string `json:"session_id"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
func AccountRead(w http.ResponseWriter, r *http.Request, cb func(account *Account)) {
|
|
|
|
// 獲取Cookie
|
|
cookie, err := r.Cookie("session_id")
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
w.Write([]byte("401 - 未登錄, 請登錄後再進行操作"))
|
|
return
|
|
}
|
|
|
|
// 獲取會話
|
|
session := Session{ID: cookie.Value}
|
|
if err := configs.ORMDB().Take(&session).Error; err != nil {
|
|
http.SetCookie(w, &http.Cookie{Name: "session_id", Value: "", Path: "/", MaxAge: -1})
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
w.Write([]byte("401 - 會話已過期, 請重新登錄"))
|
|
return
|
|
}
|
|
|
|
// 獲取當前用戶
|
|
user := User{ID: session.UserID}
|
|
if err := configs.ORMDB().Model(&user).Select("id, name, email, created_at, updated_at").Find(&user).Error; err != nil {
|
|
http.SetCookie(w, &http.Cookie{Name: "session_id", Value: "", Path: "/", MaxAge: -1})
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
w.Write([]byte("401 - 用戶不存在, 請重新登錄"))
|
|
return
|
|
}
|
|
|
|
var account Account
|
|
account.ID = user.ID
|
|
account.Name = user.Name
|
|
account.Email = user.Email
|
|
account.Admin = user.Admin
|
|
account.SessionID = session.ID
|
|
account.CreatedAt = user.CreatedAt
|
|
account.UpdatedAt = user.UpdatedAt
|
|
|
|
cb(&account)
|
|
|
|
}
|