package routers import ( "encoding/json" "fmt" "io/ioutil" "main/models" "main/utils" "net/http" "github.com/google/uuid" "github.com/gorilla/mux" ) // 獲取會話列表 func SessionsGet(w http.ResponseWriter, r *http.Request) { var listview models.ListView listview.Page = utils.ParamInt(r.URL.Query().Get("page"), 1) listview.PageSize = utils.ParamInt(r.URL.Query().Get("pageSize"), 10) listview.List = models.QuerySessions(listview.Page, listview.PageSize) listview.Total = models.CountSessions() listview.Next = listview.Page*listview.PageSize < listview.Total listview.WriteJSON(w) } func GetForm(r *http.Request) (form []interface{}) { body, err := ioutil.ReadAll(r.Body) if err != nil { fmt.Println(err) return } defer r.Body.Close() if err = json.Unmarshal(body, &form); err != nil { fmt.Println(err) return } return } // 創建會話 func SessionsPost(w http.ResponseWriter, r *http.Request) { var form struct { Email string `json:"email"` Password string `json:"password"` } body, err := ioutil.ReadAll(r.Body) if err != nil { fmt.Println(err) return } defer r.Body.Close() if err = json.Unmarshal(body, &form); err != nil { fmt.Println(err) return } // 使用Email獲取用戶 user, err := models.GetUserByEmail(form.Email) if err != nil { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("404 - User Not Found")) return } // 使用密碼驗證登錄 if !user.CheckPassword(form.Password) { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("401 - Unauthorized")) return } // 創建會話(生成一個不重複的 uuid 作爲 sid) session := &models.Session{ID: uuid.New().String(), UserID: user.ID} session.Create() // 寫入Cookie cookie := http.Cookie{Name: "session_id", Value: session.ID, Path: "/", HttpOnly: true} http.SetCookie(w, &cookie) // 返回信息 w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(utils.ToJSON(session)) } // 獲取會話 func SessionsItemGet(w http.ResponseWriter, r *http.Request) { session := models.Session{ID: mux.Vars(r)["session_id"]} session.Get() w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(utils.ToJSON(session)) } // 更新會話 func SessionsItemPatch(w http.ResponseWriter, r *http.Request) { session := models.Session{ID: mux.Vars(r)["session_id"]} session.Get() session.Update() w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(utils.ToJSON(session)) } // 刪除會話 func SessionsItemDelete(w http.ResponseWriter, r *http.Request) { // 需要先驗證身份才能執行刪除操作 // 從Cookie中獲取session_id cookie, err := r.Cookie("session_id") if err != nil { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("401 - 未登錄")) return } // 獲取當前session session := models.Session{ID: cookie.Value} session.Get() // 獲取當前用戶 user := models.User{ID: session.UserID} user.Get() sessionx := models.Session{ID: mux.Vars(r)["id"]} sessionx.Get() if user.ID != sessionx.UserID { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("401 - 沒有權限")) return } sessionx.Delete() w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(utils.ToJSON(sessionx)) }