// 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
}
// Returns simulations favorited by and comments from the user id passed in the url
func InteractionsHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
userKeyName := vars["userID"]
ctx := appengine.NewContext(r)
var simulations []models.SimulationData
var comments []models.CommentData
// Check to see if the page user is the same as the logged in user
userIsOwner := utils.IsOwner(userKeyName, ctx)
// Only get favorited and commented information if it's the proper user
// Don't display interaction data to any user except for the user who owns it
if userIsOwner {
// Get 50 of the most recent ratings made by the logged in user
var ratingKeys []*datastore.Key
var ratingObjs []models.Rating
q := datastore.NewQuery("Rating").KeysOnly().Filter("AuthorKeyName =", userKeyName).Order("-CreationDate").Limit(50)
ratingKeys, err := q.GetAll(ctx, ratingObjs)
// Get the parent keys of the ratings made (these are the keys of the simulations the ratings were for)
var simulationRateKeys []*datastore.Key
for _, key := range ratingKeys {
simulationRateKeys = append(simulationRateKeys, key.Parent())
}
// Get all of the simulation objects from the simulation keys
simulationRateObjs := make([]models.Simulation, len(simulationRateKeys))
err = datastore.GetMulti(ctx, simulationRateKeys, simulationRateObjs)
if err != nil {
controllers.ErrorHandler(w, r, err.Error(), http.StatusInternalServerError)
return
}
// Build the proper simulation data objects from the simulation models
simulations, err = utils.BuildSimulationDataSlice(ctx, simulationRateObjs, simulationRateKeys)
if err != nil {
controllers.ErrorHandler(w, r, "Could not load user simulations: "+err.Error(), http.StatusInternalServerError)
return
}
// Get 50 of the most recent comments made by the logged in user
q = datastore.NewQuery("Comment").Filter("AuthorKeyName =", userKeyName).Order("-CreationDate").Limit(50)
comments, err = utils.GetCommentDataSlice(r, q)
if err != nil {
controllers.ErrorHandler(w, r, "Error fetching comments: "+err.Error(), http.StatusInternalServerError)
return
}
}
data := map[string]interface{}{
"simulations": simulations,
"comments": comments,
"userIsOwner": false,
"userOwnsPage": userIsOwner,
}
controllers.BaseHandler(w, r, "user/interactions", data)
}
// GET returns JSON all comments associated with the simulationID passed in the url
// POST saves the comment to datastore with the simulationID as the ancestor key
func CommentHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
simKeyName := vars["simulationID"]
ctx := appengine.NewContext(r)
simulationKey := datastore.NewKey(ctx, "Simulation", simKeyName, 0, nil)
if r.Method == "GET" {
q := datastore.NewQuery("Comment").Ancestor(simulationKey).Order("-CreationDate")
comments, err := utils.GetCommentDataSlice(r, q)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
// Return comments as json
json.NewEncoder(w).Encode(comments)
}
if r.Method == "POST" {
formMessage := r.FormValue("Contents")
if len(formMessage) == 0 || len(formMessage) > 500 {
ApiErrorResponse(w, "Cannot create empty comments or comments longer than 500 characters", http.StatusBadRequest)
return
}
user, err := utils.GetCurrentUser(ctx)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
// Get an ID as a simulation descendant
key, keyName := utils.GenerateUniqueKey(ctx, "Comment", user, simulationKey)
// Build the comment object
comment := models.Comment{
KeyName: keyName,
AuthorKeyName: user.KeyName,
Contents: formMessage,
CreationDate: time.Now(),
}
// Put the comment in the datastore
_, err = datastore.Put(ctx, key, &comment)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
}
}
// Returns simulations tied to the user id passed in the url
func SimulationsHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
pageUserKeyName := vars["userID"]
ctx := appengine.NewContext(r)
// Get the page user information
var pageUser models.User
pageUserKey := datastore.NewKey(ctx, "User", pageUserKeyName, 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(pageUserKeyName, ctx)
// Build a query to get the 50 most recent simulations that belong to the user
q := datastore.NewQuery("Simulation").Filter("AuthorKeyName =", pageUserKeyName)
if !userIsOwner { // Only get public simulations if the pageUser is not the logged in user
q = q.Filter("IsPrivate =", false)
}
q = q.Order("-CreationDate").Limit(50)
// Get the simulations for the page user from the query
simulations, err := utils.GetSimulationDataSlice(r, q)
if err != nil {
controllers.ErrorHandler(w, r, "Could not load user simulations: "+err.Error(), http.StatusInternalServerError)
return
}
data := map[string]interface{}{
"user": pageUser,
"simulations": simulations,
"userIsOwner": userIsOwner,
}
controllers.BaseHandler(w, r, "user/simulations", data)
}
// GET returns JSON all ratings associated with the simulationID passed in the url
// POST saves the rating to datastore with the simulationID as the ancestor key
func RatingHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
simKeyName := vars["simulationID"]
ctx := appengine.NewContext(r)
simulationKey := datastore.NewKey(ctx, "Simulation", simKeyName, 0, nil)
var ratings []models.Rating
q := datastore.NewQuery("Rating").Ancestor(simulationKey).Order("-CreationDate")
_, err := q.GetAll(ctx, &ratings)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
if r.Method == "GET" {
totalScore := 0
for i := 0; i < len(ratings); i++ {
totalScore += int(ratings[i].Score)
}
// Return rating as json
json.NewEncoder(w).Encode(struct {
Ratings []models.Rating
TotalScore int
}{
Ratings: ratings,
TotalScore: totalScore,
})
}
if r.Method == "POST" {
user, err := utils.GetCurrentUser(ctx)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
for i := 0; i < len(ratings); i++ {
if user.KeyName == ratings[i].AuthorKeyName {
ratingsKey := datastore.NewKey(ctx, "Rating", ratings[i].KeyName, 0, simulationKey)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
datastore.Delete(ctx, ratingsKey)
return
}
}
// Get an ID as a simulation descendant
key, keyName := utils.GenerateUniqueKey(ctx, "Rating", user, simulationKey)
score, err := strconv.ParseInt(r.FormValue("Score"), 10, 8)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
// Build the rating object
rating := models.Rating{
KeyName: keyName,
AuthorKeyName: user.KeyName,
Score: int8(score),
CreationDate: time.Now(),
}
// Put the rating in the datastore
_, err = datastore.Put(ctx, key, &rating)
if err != nil {
ApiErrorResponse(w, err.Error(), http.StatusInternalServerError)
return
}
}
}
// Takes a blobstore key and serves the file with the appropriate headers
func BlobstoreFileServer(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
blobKey := vars["blobKey"]
blobstore.Send(w, appengine.BlobKey(blobKey))
}
// 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
}
}
}