package friend import ( "fmt" "net/http" "strings" "git.kirsle.net/apps/gosocial/pkg/models" "git.kirsle.net/apps/gosocial/pkg/session" "git.kirsle.net/apps/gosocial/pkg/templates" ) // AddFriend controller to send a friend request. func AddFriend() http.HandlerFunc { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { // POST only. if r.Method != http.MethodPost { session.FlashError(w, r, "Unacceptable Request Method") templates.Redirect(w, "/") return } // Form fields var ( username = strings.ToLower(r.PostFormValue("username")) verdict = r.PostFormValue("verdict") ) // Get the current user. currentUser, err := session.CurrentUser(r) if err != nil { session.FlashError(w, r, "Couldn't get CurrentUser: %s", err) templates.Redirect(w, "/") return } // Get the target user. user, err := models.FindUser(username) if err != nil { session.FlashError(w, r, "User Not Found") templates.Redirect(w, "/") return } // Can't friend yourself. if currentUser.ID == user.ID { session.FlashError(w, r, "You can't send a friend request to yourself!") templates.Redirect(w, "/u/"+username) return } // Are we adding, or rejecting+removing? if verdict == "reject" || verdict == "remove" { err := models.RemoveFriend(currentUser.ID, user.ID) if err != nil { session.FlashError(w, r, "Failed to remove friend: %s", err) templates.Redirect(w, "/u/"+username) return } var message string if verdict == "reject" { message = fmt.Sprintf("Friend request from %s has been rejected.", username) } else { message = fmt.Sprintf("Removed friendship with %s.", username) } session.Flash(w, r, message) if verdict == "reject" { templates.Redirect(w, "/friends?view=requests") } templates.Redirect(w, "/friends") } else { // Post the friend request. if err := models.AddFriend(currentUser.ID, user.ID); err != nil { session.FlashError(w, r, "Couldn't send friend request: %s.", err) } else { if verdict == "approve" { session.Flash(w, r, "You accepted the friend request from %s!", username) templates.Redirect(w, "/friends?view=requests") return } session.Flash(w, r, "Friend request sent!") } } templates.Redirect(w, "/u/"+username) }) }