rename GetUserFromName -> GetUserByName
This commit is contained in:
@@ -151,7 +151,7 @@ func (c *Controller) ServeChangePassword(r *http.Request) *Response {
|
|||||||
code: 400,
|
code: 400,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
user := c.DB.GetUserFromName(username)
|
user := c.DB.GetUserByName(username)
|
||||||
if user == nil {
|
if user == nil {
|
||||||
return &Response{
|
return &Response{
|
||||||
err: "couldn't find a user with that name",
|
err: "couldn't find a user with that name",
|
||||||
@@ -176,7 +176,7 @@ func (c *Controller) ServeChangePasswordDo(r *http.Request) *Response {
|
|||||||
flashW: []string{err.Error()},
|
flashW: []string{err.Error()},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
user := c.DB.GetUserFromName(username)
|
user := c.DB.GetUserByName(username)
|
||||||
user.Password = passwordOne
|
user.Password = passwordOne
|
||||||
c.DB.Save(user)
|
c.DB.Save(user)
|
||||||
return &Response{redirect: "/admin/home"}
|
return &Response{redirect: "/admin/home"}
|
||||||
@@ -190,7 +190,7 @@ func (c *Controller) ServeDeleteUser(r *http.Request) *Response {
|
|||||||
code: 400,
|
code: 400,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
user := c.DB.GetUserFromName(username)
|
user := c.DB.GetUserByName(username)
|
||||||
if user == nil {
|
if user == nil {
|
||||||
return &Response{
|
return &Response{
|
||||||
err: "couldn't find a user with that name",
|
err: "couldn't find a user with that name",
|
||||||
@@ -207,7 +207,7 @@ func (c *Controller) ServeDeleteUser(r *http.Request) *Response {
|
|||||||
|
|
||||||
func (c *Controller) ServeDeleteUserDo(r *http.Request) *Response {
|
func (c *Controller) ServeDeleteUserDo(r *http.Request) *Response {
|
||||||
username := r.URL.Query().Get("user")
|
username := r.URL.Query().Get("user")
|
||||||
user := c.DB.GetUserFromName(username)
|
user := c.DB.GetUserByName(username)
|
||||||
if user.IsAdmin {
|
if user.IsAdmin {
|
||||||
return &Response{
|
return &Response{
|
||||||
redirect: "/admin/home",
|
redirect: "/admin/home",
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ func (c *Controller) ServeLoginDo(w http.ResponseWriter, r *http.Request) {
|
|||||||
http.Redirect(w, r, r.Referer(), http.StatusSeeOther)
|
http.Redirect(w, r, r.Referer(), http.StatusSeeOther)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
user := c.DB.GetUserFromName(username)
|
user := c.DB.GetUserByName(username)
|
||||||
if user == nil || password != user.Password {
|
if user == nil || password != user.Password {
|
||||||
sessAddFlashW(session, []string{"invalid username / password"})
|
sessAddFlashW(session, []string{"invalid username / password"})
|
||||||
sessLogSave(session, w, r)
|
sessLogSave(session, w, r)
|
||||||
@@ -26,7 +26,7 @@ func (c *Controller) ServeLoginDo(w http.ResponseWriter, r *http.Request) {
|
|||||||
// put the user name into the session. future endpoints after this one
|
// put the user name into the session. future endpoints after this one
|
||||||
// are wrapped with WithUserSession() which will get the name from the
|
// are wrapped with WithUserSession() which will get the name from the
|
||||||
// session and put the row into the request context
|
// session and put the row into the request context
|
||||||
session.Values["user"] = user.Name
|
session.Values["user"] = user.ID
|
||||||
sessLogSave(session, w, r)
|
sessLogSave(session, w, r)
|
||||||
http.Redirect(w, r, c.Path("/admin/home"), http.StatusSeeOther)
|
http.Redirect(w, r, c.Path("/admin/home"), http.StatusSeeOther)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,7 +21,7 @@ func (c *Controller) WithUserSession(next http.Handler) http.Handler {
|
|||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
// session exists at this point
|
// session exists at this point
|
||||||
session := r.Context().Value(CtxSession).(*sessions.Session)
|
session := r.Context().Value(CtxSession).(*sessions.Session)
|
||||||
username, ok := session.Values["user"].(string)
|
userID, ok := session.Values["user"].(int)
|
||||||
if !ok {
|
if !ok {
|
||||||
sessAddFlashW(session, []string{"you are not authenticated"})
|
sessAddFlashW(session, []string{"you are not authenticated"})
|
||||||
sessLogSave(session, w, r)
|
sessLogSave(session, w, r)
|
||||||
@@ -29,7 +29,7 @@ func (c *Controller) WithUserSession(next http.Handler) http.Handler {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
// take username from sesion and add the user row to the context
|
// take username from sesion and add the user row to the context
|
||||||
user := c.DB.GetUserFromName(username)
|
user := c.DB.GetUserByID(userID)
|
||||||
if user == nil {
|
if user == nil {
|
||||||
// the username in the client's session no longer relates to a
|
// the username in the client's session no longer relates to a
|
||||||
// user in the database (maybe the user was deleted)
|
// user in the database (maybe the user was deleted)
|
||||||
|
|||||||
@@ -67,7 +67,7 @@ func (c *Controller) WithUser(next http.Handler) http.Handler {
|
|||||||
"please provide `t` and `s`, or just `p`"))
|
"please provide `t` and `s`, or just `p`"))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
user := c.DB.GetUserFromName(username)
|
user := c.DB.GetUserByName(username)
|
||||||
if user == nil {
|
if user == nil {
|
||||||
_ = writeResp(w, r, spec.NewError(40,
|
_ = writeResp(w, r, spec.NewError(40,
|
||||||
"invalid username `%s`", username))
|
"invalid username `%s`", username))
|
||||||
|
|||||||
@@ -111,7 +111,19 @@ func (db *DB) GetOrCreateKey(key string) string {
|
|||||||
return value
|
return value
|
||||||
}
|
}
|
||||||
|
|
||||||
func (db *DB) GetUserFromName(name string) *User {
|
func (db *DB) GetUserByID(id int) *User {
|
||||||
|
user := &User{}
|
||||||
|
err := db.
|
||||||
|
Where("id=?", id).
|
||||||
|
First(user).
|
||||||
|
Error
|
||||||
|
if gorm.IsRecordNotFoundError(err) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return user
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *DB) GetUserByName(name string) *User {
|
||||||
user := &User{}
|
user := &User{}
|
||||||
err := db.
|
err := db.
|
||||||
Where("name=?", name).
|
Where("name=?", name).
|
||||||
|
|||||||
Reference in New Issue
Block a user