methods for creating and getting users
This commit is contained in:
@@ -47,6 +47,14 @@ func Start() {
|
|||||||
r.Post("/new", NewMessage)
|
r.Post("/new", NewMessage)
|
||||||
})
|
})
|
||||||
|
|
||||||
|
r.Route("/users", func(r chi.Router) {
|
||||||
|
r.Route("/{userID}", func(r chi.Router) {
|
||||||
|
r.Use(UserCtx) // Load user
|
||||||
|
r.Get("/", GetUser)
|
||||||
|
})
|
||||||
|
r.Post("/new", NewUser)
|
||||||
|
})
|
||||||
|
|
||||||
if *routes {
|
if *routes {
|
||||||
fmt.Println(docgen.MarkdownRoutesDoc(r, docgen.MarkdownOpts{
|
fmt.Println(docgen.MarkdownRoutesDoc(r, docgen.MarkdownOpts{
|
||||||
ProjectPath: "git.dubyatp.xyz/chat-api-server",
|
ProjectPath: "git.dubyatp.xyz/chat-api-server",
|
||||||
|
10
api/db.go
10
api/db.go
@@ -100,7 +100,7 @@ func dbGetAllMessages() ([]*Message, error) {
|
|||||||
return result, nil
|
return result, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func dbAddUser(id int64, name string) error {
|
func dbAddUser(user *User) error {
|
||||||
currentData := db.ExecDB("users")
|
currentData := db.ExecDB("users")
|
||||||
if currentData == nil {
|
if currentData == nil {
|
||||||
return fmt.Errorf("error reading users database")
|
return fmt.Errorf("error reading users database")
|
||||||
@@ -111,12 +111,12 @@ func dbAddUser(id int64, name string) error {
|
|||||||
return fmt.Errorf("users data is in an unexpected format")
|
return fmt.Errorf("users data is in an unexpected format")
|
||||||
}
|
}
|
||||||
|
|
||||||
user := map[string]interface{}{
|
dbUser := map[string]interface{}{
|
||||||
"ID": float64(id), // JSON numbers are float64 by default
|
"ID": user.ID,
|
||||||
"Name": name,
|
"Name": user.Name,
|
||||||
}
|
}
|
||||||
|
|
||||||
users = append(users, user)
|
users = append(users, dbUser)
|
||||||
return db.WriteDB("users", users)
|
return db.WriteDB("users", users)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
68
api/user.go
68
api/user.go
@@ -1,13 +1,81 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/go-chi/chi/v5"
|
||||||
|
"github.com/go-chi/render"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func UserCtx(next http.Handler) http.Handler {
|
||||||
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
var user *User
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if userID := chi.URLParam(r, "userID"); userID != "" {
|
||||||
|
user, err = dbGetUser(userID)
|
||||||
|
} else {
|
||||||
|
render.Render(w, r, ErrNotFound)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
render.Render(w, r, ErrNotFound)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx := context.WithValue(r.Context(), userKey{}, user)
|
||||||
|
next.ServeHTTP(w, r.WithContext(ctx))
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetUser(w http.ResponseWriter, r *http.Request) {
|
||||||
|
user, ok := r.Context().Value(userKey{}).(*User)
|
||||||
|
if !ok || user == nil {
|
||||||
|
render.Render(w, r, ErrNotFound)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := render.Render(w, r, NewUserPayloadResponse(user)); err != nil {
|
||||||
|
render.Render(w, r, ErrRender(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func newUserID() string {
|
func newUserID() string {
|
||||||
return "user_" + uuid.New().String()
|
return "user_" + uuid.New().String()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NewUser(w http.ResponseWriter, r *http.Request) {
|
||||||
|
err := r.ParseMultipartForm(64 << 10)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "Unable to parse form", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
newUserName := r.FormValue("name")
|
||||||
|
|
||||||
|
newUser := User{
|
||||||
|
ID: newUserID(),
|
||||||
|
Name: newUserName,
|
||||||
|
}
|
||||||
|
|
||||||
|
err = dbAddUser(&newUser)
|
||||||
|
if err != nil {
|
||||||
|
render.Render(w, r, ErrRender(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
render.Render(w, r, NewUserPayloadResponse(&newUser))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *UserPayload) Render(w http.ResponseWriter, r *http.Request) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type userKey struct{}
|
||||||
|
|
||||||
type User struct {
|
type User struct {
|
||||||
ID string `json:"id"`
|
ID string `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
|
Reference in New Issue
Block a user