// Generates an image upload url for the blobstore and returns it as a string
func GetBlobstoreUploadPath(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
userKeyName := vars["userID"]
ctx := appengine.NewContext(r)
// Check to see if the page user is the same as the logged in user
userIsOwner := utils.IsOwner(userKeyName, ctx)
if !userIsOwner {
api.ApiErrorResponse(w, "You cannot edit other profiles.", http.StatusInternalServerError)
return
}
returnPath := "/user/" + userKeyName
// The autosaved thumbnail images need to be POSTed to specific appengine blobstore "action" paths.
// Have to specify a path to return to after the post succeeds
imageUploadUrl, err := blobstore.UploadURL(ctx, returnPath, nil)
if err != nil {
api.ApiErrorResponse(w, "Could not generate blobstore upload: "+err.Error(), http.StatusInternalServerError)
return
}
// Need to return the uploadUrl to use to post the image to
uploadUrl := bytes.NewBufferString(imageUploadUrl.Path)
io.Copy(w, uploadUrl)
return
}
// Displays a users profile page and handles updates to a logged in users profile information
func ProfileHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
userKeyName := vars["userID"]
ctx := appengine.NewContext(r)
// Get the page user information
var pageUser models.User
pageUserKey := datastore.NewKey(ctx, "User", userKeyName, 0, nil)
err := datastore.Get(ctx, pageUserKey, &pageUser)
if err != nil {
controllers.ErrorHandler(w, r, "User was not found: "+err.Error(), http.StatusNotFound)
return
}
// Check to see if the logged in user matches the page user
userIsOwner := utils.IsOwner(userKeyName, ctx)
// If a user is just viewing the page
if r.Method == "GET" {
var simulations []models.SimulationData
// If viewing someone else's profile page, get their 8 most recent public simultions to display
if !userIsOwner {
// Build a query
q := datastore.NewQuery("Simulation").Filter("AuthorKeyName =", userKeyName).Filter("IsPrivate =", false).Order("-CreationDate").Limit(8)
simulations, err = utils.GetSimulationDataSlice(r, q)
if err != nil {
controllers.ErrorHandler(w, r, err.Error(), http.StatusInternalServerError)
return
}
}
// Nicely format the join date
prettyJoinDate := pageUser.JoinDate.Format("January _2, 2006")
var empty []models.Simulation
totalFavoritesReceived := 0
// Get the profile image they may or may not
var userProfileImageSrc string
if len(string(pageUser.ImageBlobKey)) > 0 {
userProfileImageSrc = "/api/img/" + string(pageUser.ImageBlobKey)
}
// Only want to generate an image upload user if it is the user's profile page
var imageUploadUrl string
if userIsOwner {
// The user's profile images need to be POSTed to specific appengine blobstore "action" paths.
// Have to specify a path to return to after the post succeeds
imageUpload, err := blobstore.UploadURL(ctx, r.URL.Path, nil)
if err != nil {
api.ApiErrorResponse(w, "Could not generate blobstore upload: "+err.Error(), http.StatusInternalServerError)
return
}
imageUploadUrl = imageUpload.Path
}
// Get a count of all favorites received on all simulations created by this user
q := datastore.NewQuery("Simulation").KeysOnly().Filter("AuthorKeyName =", userKeyName)
simKeys, err := q.GetAll(ctx, &empty) // Get all simulation keys made by this user
if err != nil {
controllers.ErrorHandler(w, r, err.Error(), http.StatusInternalServerError)
return
}
// Get a count of all of the favorites received for each simulation and add to total
for _, key := range simKeys {
q := datastore.NewQuery("Rating").Ancestor(key)
simFaves, err := q.Count(ctx)
if err != nil {
controllers.ErrorHandler(w, r, err.Error(), http.StatusInternalServerError)
return
}
totalFavoritesReceived += simFaves
}
data := map[string]interface{}{
"user": pageUser,
"userJoinDate": prettyJoinDate,
"userProfileImageSrc": userProfileImageSrc,
"imageUploadUrl": imageUploadUrl,
"userIsOwner": userIsOwner,
"simulations": simulations,
"totalFavoritesReceived": totalFavoritesReceived,
}
controllers.BaseHandler(w, r, "user/profile", data)
return
}
// When a user tries to post information
if r.Method == "POST" {
// Make sure only the owner is trying to update the information
if !userIsOwner {
controllers.ErrorHandler(w, r, "Unauthorized update attempt: "+err.Error(), http.StatusInternalServerError)
return
}
// Get all of the form values and blob image from the post
blobs, formValues, err := blobstore.ParseUpload(r)
if err != nil {
controllers.ErrorHandler(w, r, "Bad blobstore form parse: "+err.Error(), http.StatusInternalServerError)
return
}
// Only update the profile image if they posted a new one
newImage := blobs["ProfileImage"]
if len(newImage) != 0 {
// Delete the old profile photo if they already had one
err = blobstore.Delete(ctx, pageUser.ImageBlobKey)
if err != nil {
api.ApiErrorResponse(w, "Can't delete the blobstore image: "+err.Error(), http.StatusInternalServerError)
return
}
pageUser.ImageBlobKey = newImage[0].BlobKey
}
displayName := formValues["DisplayName"][0]
if len(displayName) > 50 {
api.ApiErrorResponse(w, "Your Display Name must be shorter than 50 characters.", http.StatusInternalServerError)
return
}
interests := formValues["Interests"][0]
if len(interests) > 1500 {
api.ApiErrorResponse(w, "Your Interests must be shorter than 1500 characters.", http.StatusInternalServerError)
return
}
// Update user information
pageUser.DisplayName = displayName
pageUser.Interests = interests
_, err = datastore.Put(ctx, pageUserKey, &pageUser)
if err != nil {
// Could not place the user in the datastore
controllers.ErrorHandler(w, r, "Could not save user data: "+err.Error(), http.StatusInternalServerError)
return
}
}
}