kirillius
/
Cloak
mirror of https://github.com/cbeuw/Cloak
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix CORS and POST handler

This commit is contained in:
Qian Wang 2019-07-30 23:49:22 +01:00
parent 6af97e2c22
commit e1c879a478
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
internal/server/um_local.go
View File

@ -56,6 +56,9 @@ func (manager *localManager) registerMux() *gmux.Router {
r.HandleFunc("/admin/users/{UID}", manager.getUserInfoHlr).Methods("GET") r.HandleFunc("/admin/users/{UID}", manager.getUserInfoHlr).Methods("GET")
r.HandleFunc("/admin/users/{UID}", manager.writeUserInfoHlr).Methods("POST") r.HandleFunc("/admin/users/{UID}", manager.writeUserInfoHlr).Methods("POST")
r.HandleFunc("/admin/users/{UID}", manager.deleteUserHlr).Methods("DELETE") r.HandleFunc("/admin/users/{UID}", manager.deleteUserHlr).Methods("DELETE")
r.Methods("OPTIONS").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Access-Control-Allow-Methods", "GET,POST,DELETE,OPTIONS")
})
r.Use(corsMiddleware) r.Use(corsMiddleware)
return r return r
} }

4
internal/server/um_local_hlr.go
View File

@ -96,7 +96,7 @@ func (manager *localManager) writeUserInfoHlr(w http.ResponseWriter, r *http.Req
return return
} }
jsonUinfo := gmux.Vars(r)["UserInfo"] jsonUinfo := r.FormValue("UserInfo")
if jsonUinfo == "" { if jsonUinfo == "" {
http.Error(w, "UserInfo cannot be empty", http.StatusBadRequest) http.Error(w, "UserInfo cannot be empty", http.StatusBadRequest)
return return
@ -112,7 +112,7 @@ func (manager *localManager) writeUserInfoHlr(w http.ResponseWriter, r *http.Req
} }
err = manager.db.Update(func(tx *bolt.Tx) error { err = manager.db.Update(func(tx *bolt.Tx) error {
bucket, err := tx.CreateBucket(uinfo.UID) bucket, err := tx.CreateBucketIfNotExists(uinfo.UID)
if err != nil { if err != nil {
return err return err
} }