// Get a list of visualizations
func listVisualizations(w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
q := datastore.NewQuery("visualization").Order("-Date")
// Parse filter parameters
r.ParseForm()
fUser := r.Form["user"]
if len(fUser) > 0 && fUser[0] == "me" {
u := user.Current(c)
if u == nil {
common.Serve401(w)
return
}
q = q.Ancestor(model.GetAccountKey(c, u))
}
// Get visualizations
var e []model.Visualization
keys, err := q.GetAll(c, &e)
if err != nil {
common.ServeError(c, w, err)
return
}
// Prepare output
output := make([]map[string]interface{}, 0)
for i := range keys {
output = append(output, map[string]interface{}{"Key": keys[i], "Title": e[i].Title, "Date": e[i].Date})
}
common.WriteJson(c, w, output)
}
// Seat a title for a visualization
func setVisualizationTitle(w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
vars := mux.Vars(r)
u := user.Current(c)
// Check if user is logged in
if u == nil {
common.Serve401(w)
return
}
title := r.PostFormValue("title")
key, err := datastore.DecodeKey(vars["key"])
if err != nil {
common.Serve404(w)
return
}
// Check if user is the owner
if key.Parent().StringID() != u.ID {
common.Serve403(w)
return
}
// Start a datastore transaction
var e model.Visualization
err = datastore.RunInTransaction(c, func(c appengine.Context) error {
// Get the visualization object
err = datastore.Get(c, key, &e)
if err != nil {
return err
}
// Change the title
e.Title = title
// Save the visualization object
key, err = datastore.Put(c, key, &e)
return err
}, nil)
if err != nil {
if err == datastore.ErrNoSuchEntity {
common.Serve404(w)
return
}
common.ServeError(c, w, err)
return
}
}
// Create a new visualization
func newVisualization(w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
u := user.Current(c)
if u == nil {
common.Serve401(w)
return
}
e := model.Visualization{"Untitled", time.Now(), nil}
key, err := datastore.Put(c, datastore.NewIncompleteKey(c, "visualization", model.GetAccountKey(c, u)), &e)
if err != nil {
common.ServeError(c, w, err)
return
}
common.WriteJson(c, w, map[string]*datastore.Key{"key": key})
}
// Create a new module
func newModule(w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
u := user.Current(c)
acc, err := model.GetAccount(c, u)
if err != nil {
common.ServeError(c, w, err)
return
}
if acc == nil {
common.Serve401(w)
return
}
// Parse form data
devname := r.PostFormValue("devname")
name := r.PostFormValue("name")
// Set user developer name
if acc.Devname == "" {
if devname == "" {
common.Serve404(w) // TODO: maybe 400 instead
return
}
acc.Devname = devname
err := model.SaveAccount(c, u, acc)
if err != nil {
common.ServeError(c, w, err)
return
}
} else {
devname = acc.Devname
}
e := model.Module{devname, name, time.Now(), nil}
_, err = datastore.Put(c, datastore.NewKey(c, "module", name, 0, model.GetAccountKey(c, u)), &e)
if err != nil {
common.ServeError(c, w, err)
return
}
common.WriteJson(c, w, map[string]string{"Devname": devname, "Name": name})
}
// Upload a file to a visualization
func postVisualizationFile(w http.ResponseWriter, r *http.Request) {
c := appengine.NewContext(r)
vars := mux.Vars(r)
u := user.Current(c)
// Check if user is logged in
if u == nil {
common.Serve401(w)
return
}
key, err := datastore.DecodeKey(vars["key"])
if err != nil {
common.Serve404(w)
return
}
// Check if user is the owner
if key.Parent().StringID() != u.ID {
common.Serve403(w)
return
}
// Start a datastore transaction
var e model.Visualization
err = datastore.RunInTransaction(c, func(c appengine.Context) error {
// Get the visualization object
err = datastore.Get(c, key, &e)
if err != nil {
return err
}
if vars["action"] == "delete" {
e.Files, err = deleteFile(c, r, e.Files)
if err != nil {
return err
}
} else {
e.Files, err = uploadFile(c, r, e.Files)
if err != nil {
return err
}
}
// Save the visualization object
key, err = datastore.Put(c, key, &e)
return err
}, nil)
if err != nil {
if err == datastore.ErrNoSuchEntity {
common.Serve404(w)
return
}
common.ServeError(c, w, err)
return
}
res := map[string]interface{}{
"Owner": key.Parent().StringID(),
"Title": e.Title,
"Date": e.Date,
"Files": e.Files,
}
common.WriteJson(c, w, res)
}