268 lines
7.9 KiB
Go
268 lines
7.9 KiB
Go
package photo
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strconv"
|
|
|
|
"code.nonshy.com/nonshy/website/pkg/config"
|
|
"code.nonshy.com/nonshy/website/pkg/log"
|
|
"code.nonshy.com/nonshy/website/pkg/models"
|
|
pphoto "code.nonshy.com/nonshy/website/pkg/photo"
|
|
"code.nonshy.com/nonshy/website/pkg/session"
|
|
"code.nonshy.com/nonshy/website/pkg/templates"
|
|
)
|
|
|
|
// Edit controller (/photo/edit?id=N) to change properties about your picture.
|
|
func Edit() http.HandlerFunc {
|
|
// Reuse the upload page but with an EditPhoto variable.
|
|
tmpl := templates.Must("photo/upload.html")
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Query params.
|
|
photoID, err := strconv.Atoi(r.FormValue("id"))
|
|
if err != nil {
|
|
session.FlashError(w, r, "Photo 'id' parameter required.")
|
|
templates.Redirect(w, "/")
|
|
return
|
|
}
|
|
|
|
// Find this photo by ID.
|
|
photo, err := models.GetPhoto(uint64(photoID))
|
|
if err != nil {
|
|
templates.NotFoundPage(w, r)
|
|
return
|
|
}
|
|
|
|
// Load the current user.
|
|
currentUser, err := session.CurrentUser(r)
|
|
if err != nil {
|
|
session.FlashError(w, r, "Unexpected error: couldn't get CurrentUser")
|
|
templates.Redirect(w, "/")
|
|
return
|
|
}
|
|
|
|
// Do we have permission for this photo?
|
|
if photo.UserID != currentUser.ID {
|
|
if !currentUser.IsAdmin {
|
|
templates.ForbiddenPage(w, r)
|
|
return
|
|
}
|
|
|
|
// Find the owner of this photo and assume currentUser is them for the remainder
|
|
// of this controller.
|
|
if user, err := models.GetUser(photo.UserID); err != nil {
|
|
session.FlashError(w, r, "Couldn't get the owner User for this photo!")
|
|
templates.Redirect(w, "/")
|
|
return
|
|
} else {
|
|
currentUser = user
|
|
}
|
|
}
|
|
|
|
// Is the user throttled for Site Gallery photo uploads?
|
|
var SiteGalleryThrottled = models.IsSiteGalleryThrottled(currentUser, photo)
|
|
|
|
// Are we saving the changes?
|
|
if r.Method == http.MethodPost {
|
|
var (
|
|
caption = r.FormValue("caption")
|
|
isExplicit = r.FormValue("explicit") == "true"
|
|
isGallery = r.FormValue("gallery") == "true"
|
|
visibility = models.PhotoVisibility(r.FormValue("visibility"))
|
|
|
|
// Profile pic fields
|
|
setProfilePic = r.FormValue("intent") == "profile-pic"
|
|
crop = pphoto.ParseCropCoords(r.FormValue("crop"))
|
|
|
|
// Are we GOING private or changing to Inner Circle?
|
|
goingPrivate = visibility == models.PhotoPrivate && visibility != photo.Visibility
|
|
goingCircle = visibility == models.PhotoInnerCircle && visibility != photo.Visibility
|
|
)
|
|
|
|
// Respect the Site Gallery throttle in case the user is messing around.
|
|
if SiteGalleryThrottled {
|
|
isGallery = false
|
|
}
|
|
|
|
photo.Caption = caption
|
|
photo.Explicit = isExplicit
|
|
photo.Gallery = isGallery
|
|
photo.Visibility = visibility
|
|
|
|
// Can not use a GIF as profile pic.
|
|
if setProfilePic && filepath.Ext(photo.Filename) == ".gif" {
|
|
session.FlashError(w, r, "You can not use a GIF as your profile picture.")
|
|
templates.Redirect(w, "/")
|
|
return
|
|
}
|
|
|
|
// Are we cropping ourselves a new profile pic?
|
|
log.Error("Profile pic? %+v and crop is: %+v", setProfilePic, crop)
|
|
if setProfilePic && crop != nil && len(crop) >= 4 {
|
|
cropFilename, err := pphoto.ReCrop(photo.Filename, crop[0], crop[1], crop[2], crop[3])
|
|
log.Error("ReCrop got: %s, %s", cropFilename, err)
|
|
if err != nil {
|
|
session.FlashError(w, r, "Couldn't re-crop for profile picture: %s", err)
|
|
} else {
|
|
// If there was an old profile pic, remove it from disk.
|
|
if photo.CroppedFilename != "" {
|
|
pphoto.Delete(photo.CroppedFilename)
|
|
}
|
|
photo.CroppedFilename = cropFilename
|
|
log.Warn("HERE WE SET (%s) ON PHOTO (%+v)", cropFilename, photo)
|
|
}
|
|
} else {
|
|
setProfilePic = false
|
|
}
|
|
|
|
log.Error("SAVING PHOTO: %+v", photo)
|
|
|
|
if err := photo.Save(); err != nil {
|
|
session.FlashError(w, r, "Couldn't save photo: %s", err)
|
|
}
|
|
|
|
// Set their profile pic to this one.
|
|
if setProfilePic {
|
|
currentUser.ProfilePhoto = *photo
|
|
log.Error("Set user ProfilePhotoID=%d", photo.ID)
|
|
if err := currentUser.Save(); err != nil {
|
|
session.FlashError(w, r, "Couldn't save user: %s", err)
|
|
}
|
|
}
|
|
|
|
// Flash success.
|
|
session.Flash(w, r, "Photo settings updated!")
|
|
|
|
// If this picture has moved to Private, revoke any notification we gave about it before.
|
|
if goingPrivate || goingCircle {
|
|
log.Info("The picture is GOING PRIVATE (to %s), revoke any notifications about it", photo.Visibility)
|
|
models.RemoveNotification("photos", photo.ID)
|
|
}
|
|
|
|
// Return the user to their gallery.
|
|
templates.Redirect(w, "/u/"+currentUser.Username+"/photos")
|
|
return
|
|
}
|
|
|
|
var vars = map[string]interface{}{
|
|
"EditPhoto": photo,
|
|
"SiteGalleryThrottled": SiteGalleryThrottled,
|
|
"SiteGalleryThrottleLimit": config.SiteGalleryRateLimitMax,
|
|
}
|
|
|
|
if err := tmpl.Execute(w, r, vars); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
})
|
|
}
|
|
|
|
// Delete controller (/photo/Delete?id=N) to change properties about your picture.
|
|
func Delete() http.HandlerFunc {
|
|
// Reuse the upload page but with an EditPhoto variable.
|
|
tmpl := templates.Must("photo/delete.html")
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Query params.
|
|
photoID, err := strconv.Atoi(r.FormValue("id"))
|
|
if err != nil {
|
|
log.Error("photo.Delete: failed to parse `id` param (%s) as int: %s", r.FormValue("id"), err)
|
|
session.FlashError(w, r, "Photo 'id' parameter required.")
|
|
templates.Redirect(w, "/")
|
|
return
|
|
}
|
|
|
|
// Page to redirect to in case of errors.
|
|
redirect := fmt.Sprintf("%s?id=%d", r.URL.Path, photoID)
|
|
|
|
// Find this photo by ID.
|
|
photo, err := models.GetPhoto(uint64(photoID))
|
|
if err != nil {
|
|
templates.NotFoundPage(w, r)
|
|
return
|
|
}
|
|
|
|
// Load the current user.
|
|
currentUser, err := session.CurrentUser(r)
|
|
if err != nil {
|
|
session.FlashError(w, r, "Unexpected error: couldn't get CurrentUser")
|
|
templates.Redirect(w, "/")
|
|
return
|
|
}
|
|
|
|
// Do we have permission for this photo?
|
|
if photo.UserID != currentUser.ID {
|
|
if !currentUser.IsAdmin {
|
|
templates.ForbiddenPage(w, r)
|
|
return
|
|
}
|
|
|
|
// Find the owner of this photo and assume currentUser is them for the remainder
|
|
// of this controller.
|
|
if user, err := models.GetUser(photo.UserID); err != nil {
|
|
session.FlashError(w, r, "Couldn't get the owner User for this photo!")
|
|
templates.Redirect(w, "/")
|
|
return
|
|
} else {
|
|
currentUser = user
|
|
}
|
|
}
|
|
|
|
// Confirm deletion?
|
|
if r.Method == http.MethodPost {
|
|
confirm := r.PostFormValue("confirm") == "true"
|
|
if !confirm {
|
|
session.FlashError(w, r, "Confirm you want to delete this photo.")
|
|
templates.Redirect(w, redirect)
|
|
return
|
|
}
|
|
|
|
// Was this our profile picture?
|
|
if currentUser.ProfilePhotoID != nil && *currentUser.ProfilePhotoID == photo.ID {
|
|
log.Debug("Delete Photo: was the user's profile photo, unset ProfilePhotoID")
|
|
if err := currentUser.RemoveProfilePhoto(); err != nil {
|
|
session.FlashError(w, r, "Error unsetting your current profile photo: %s", err)
|
|
templates.Redirect(w, redirect)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Remove the images from disk.
|
|
for _, filename := range []string{
|
|
photo.Filename,
|
|
photo.CroppedFilename,
|
|
} {
|
|
if len(filename) > 0 {
|
|
if err := pphoto.Delete(filename); err != nil {
|
|
log.Error("Delete Photo: couldn't remove file from disk: %s: %s", filename, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Take back notifications on it.
|
|
models.RemoveNotification("photos", photo.ID)
|
|
|
|
if err := photo.Delete(); err != nil {
|
|
session.FlashError(w, r, "Couldn't delete photo: %s", err)
|
|
templates.Redirect(w, redirect)
|
|
return
|
|
}
|
|
|
|
session.Flash(w, r, "Photo deleted!")
|
|
|
|
// Return the user to their gallery.
|
|
templates.Redirect(w, "/u/"+currentUser.Username+"/photos")
|
|
return
|
|
}
|
|
|
|
var vars = map[string]interface{}{
|
|
"Photo": photo,
|
|
}
|
|
|
|
if err := tmpl.Execute(w, r, vars); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
})
|
|
}
|