methods for creating and getting users

This commit is contained in:
2025-03-27 21:07:33 -04:00
parent 9d7ad260f2
commit 25ee1d3299
3 changed files with 81 additions and 5 deletions

View File

@@ -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",

View File

@@ -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)
} }

View File

@@ -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"`