Added user path

wybiral · wybiral · commit e49eb96a54c6 · 2017-11-11T13:55:47.000-06:00
diff --git a/models/update.go b/models/update.go
@@ -19,6 +19,7 @@ func NewUpdate(userId int64, body string) (*Update, error) {
 	pipe.HSet(key, "user_id", userId)
 	pipe.HSet(key, "body", body)
 	pipe.LPush("updates", id)
+	pipe.LPush(fmt.Sprintf("user:%d:updates", userId), id)
 	_, err = pipe.Exec()
 	if err != nil {
 		return nil, err
@@ -38,8 +39,7 @@ func (update *Update) GetUser() (*User, error) {
 	return GetUserById(userId)
 }
 
-
-func GetUpdates() ([]*Update, error) {
+func GetAllUpdates() ([]*Update, error) {
 	updateIds, err := client.LRange("updates", 0, 10).Result()
 	if err != nil {
 		return nil, err
@@ -52,6 +52,20 @@ func GetUpdates() ([]*Update, error) {
 	return updates, nil
 }
 
+func GetUpdates(userId int64) ([]*Update, error) {
+	key := fmt.Sprintf("user:%d:updates", userId)
+	updateIds, err := client.LRange(key, 0, 10).Result()
+	if err != nil {
+		return nil, err
+	}
+	updates := make([]*Update, len(updateIds))
+	for i, id := range updateIds {
+		key := "update:" + id
+		updates[i] = &Update{key}
+	}
+	return updates, nil
+}
+
 func PostUpdate(userId int64, body string) error {
 	_, err := NewUpdate(userId, body)
 	return err
diff --git a/routes/routes.go b/routes/routes.go
@@ -19,17 +19,25 @@ func NewRouter() *mux.Router {
 	r.HandleFunc("/register", registerPostHandler).Methods("POST")
 	fs := http.FileServer(http.Dir("./static/"))
 	r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", fs))
+	r.HandleFunc("/{username}",
+		middleware.AuthRequired(userGetHandler)).Methods("GET")
 	return r
 }
 
 func indexGetHandler(w http.ResponseWriter, r *http.Request) {
-	updates, err := models.GetUpdates()
+	updates, err := models.GetAllUpdates()
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)
 		w.Write([]byte("Internal server error"))
 		return
 	}
-	utils.ExecuteTemplate(w, "index.html", updates)
+	utils.ExecuteTemplate(w, "index.html", struct {
+		Title string
+		Updates []*models.Update
+	} {
+		Title: "All updates",
+		Updates: updates,
+	})
 }
 
 func indexPostHandler(w http.ResponseWriter, r *http.Request) {
@@ -52,6 +60,36 @@ func indexPostHandler(w http.ResponseWriter, r *http.Request) {
 	http.Redirect(w, r, "/", 302)
 }
 
+func userGetHandler(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	username := vars["username"]
+	user, err := models.GetUserByUsername(username)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte("Internal server error"))
+		return
+	}
+	userId, err := user.GetId()
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte("Internal server error"))
+		return
+	}
+	updates, err := models.GetUpdates(userId)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte("Internal server error"))
+		return
+	}
+	utils.ExecuteTemplate(w, "index.html", struct {
+		Title string
+		Updates []*models.Update
+	} {
+		Title: username,
+		Updates: updates,
+	})
+}
+
 func loginGetHandler(w http.ResponseWriter, r *http.Request) {
 	utils.ExecuteTemplate(w, "login.html", nil)
 }
diff --git a/templates/index.html b/templates/index.html
@@ -1,17 +1,17 @@
 <html>
 	<head>
-		<title>Updates</title>
+		<title>{{ .Title }}</title>
 		<link rel="stylesheet" type="text/css" href="/static/index.css">
 	</head>
 	<body>
-		<h1>Updates</h1>
+		<h1>{{ .Title }}</h1>
 		<form method="POST">
 			<textarea name="update"></textarea>
 			<div>
 				<button type="submit">Post Update</button>
 			</div>
 		</form>
-		{{ range . }}
+		{{ range .Updates }}
 		<div>
 			<div>
 				<strong>{{ .GetUser.GetUsername }} wrote:</strong>
