use login context with newmessage

api/api.go

@@ -52,7 +52,10 @@ func Start() {
 			r.Delete("/", DeleteMessage)
 			r.Post("/edit", EditMessage)
 		})
-		r.Post("/new", NewMessage)
+		r.Route("/new", func(r chi.Router) {
+			r.Use(LoginCtx)
+			r.Post("/", NewMessage)
+		})
 	})
 
 	r.Route("/users", func(r chi.Router) {

api/message.go

@@ -115,11 +115,14 @@ func NewMessage(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	userID := r.FormValue("user_id")
-	if userID == "" {
-		http.Error(w, "Invalid user ID", http.StatusBadRequest)
-		return
-	}
+	// userID := r.FormValue("user_id")
+	//if userID == "" {
+	//	http.Error(w, "Invalid user ID", http.StatusBadRequest)
+	//	return
+	//}
+
+	var user = r.Context().Value(userKey{}).(*User)
+
 	body := r.FormValue("body")
 
 	if body == "" {
@@ -129,7 +132,7 @@ func NewMessage(w http.ResponseWriter, r *http.Request) {
 
 	msg := Message{
 		ID:        newMessageID(),
-		UserID:    userID,
+		UserID:    user.ID,
 		Body:      body,
 		Timestamp: time.Now(),
 		Edited:    time.Time{},

api/user.go

@@ -52,6 +52,7 @@ func LoginCtx(next http.Handler) http.Handler {
 			render.Render(w, r, ErrNotFound)
 			return
 		}
+
 		if err != nil {
 			render.Render(w, r, ErrNotFound)
 			return