101 lines
2.7 KiB
Go
101 lines
2.7 KiB
Go
package friend
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"code.nonshy.com/nonshy/website/pkg/log"
|
|
"code.nonshy.com/nonshy/website/pkg/models"
|
|
"code.nonshy.com/nonshy/website/pkg/session"
|
|
"code.nonshy.com/nonshy/website/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" {
|
|
// Notify the requestor they'd been approved.
|
|
notif := &models.Notification{
|
|
UserID: user.ID,
|
|
AboutUser: *currentUser,
|
|
Type: models.NotificationFriendApproved,
|
|
}
|
|
if err := models.CreateNotification(notif); err != nil {
|
|
log.Error("Couldn't create approved notification: %s", err)
|
|
}
|
|
|
|
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)
|
|
})
|
|
}
|