// Generic Create handler func Create(w http.ResponseWriter, r *http.Request) { vars := mux.Vars(r) user, permissions := auth.Challenge(w, r, true) if (user == nil || permissions < 1) && vars["datatype"] != "users" { // Allow anyone to create users for now http.Error(w, "Please Login", http.StatusUnauthorized) return } w.Header().Set("Access-Control-Allow-Origin", "*") w.Header().Set("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept, X-PINGOTHER") var val interface{} // Generic container for the new object // Build a URI like representation of the datatype types := []string{vars["datatype"]} if childtype, ok := vars["childtype"]; ok { types = append(types, childtype) } // Switch based on that URI like representation and instantiate something in the generic container switch strings.Join(types, "/") { case "items": val = new(data.Item) case "items/comments": val = new(data.ItemComment) case "items/geolocations": val = new(data.Geolocation) case "users": val = new(data.User) case "roles": val = new(data.Role) case "taxonomy": val = new(data.Term) default: http.NotFound(w, r) return } // Perform the JSON decode decoder := json.NewDecoder(r.Body) err := decoder.Decode(&val) if err != nil { log.Println(err) http.Error(w, "Malformed json.", http.StatusBadRequest) return } // Perform post decode actions, setting automated field, validate values, exectute hooks, etc ... switch v := val.(type) { case *data.Item: v.Author = user.Username case *data.ItemComment: v.ItemID, err = strconv.ParseInt(vars["key"], 10, 64) v.Author = user.Username case *data.Geolocation: v.ItemID, err = strconv.ParseInt(vars["key"], 10, 64) case *data.User: err := v.Validate() if err != nil { log.Println(err) http.Error(w, err.Error(), http.StatusBadRequest) return } v.HashAndSalt() v.Role = "public" // TODO: Temporary while anyone can sign up maybe this will change? case *data.Term: v.Author = user.Username } if err != nil { log.Println(err) http.Error(w, "Malformed key in URI", http.StatusBadRequest) return } // Perform the DB insert err = storage.Insert(val) if err != nil { log.Println(err) var errorMsg string switch dbErr := err.(type) { case *storage.StorageError: switch dbErr.Code() { case 1062: switch val.(type) { case *data.User: errorMsg = "User already exists" default: errorMsg = "Database entry already exists" } default: errorMsg = "Database error (" + dbErr.Error() + ")" } default: errorMsg = "Insert Database error, likely due to malformed request" } http.Error(w, errorMsg, http.StatusInternalServerError) return } // Exectute post insert hooks, etc ... switch v := val.(type) { case *data.Item: if v.Type == "streaming-video-v1" { c, err := conf.ReadConfigFile("rter.config") if err != nil { log.Println(err) return } baseUrl, err := c.GetString("videoserver", "base-url") if err != nil { log.Println(err) return } v.UploadURI = baseUrl + "/v1/ingest/" + strconv.FormatInt(v.ID, 10) v.ThumbnailURI = baseUrl + "/v1/videos/" + strconv.FormatInt(v.ID, 10) + "/thumb/000000001.jpg" v.ContentURI = baseUrl + "/v1/videos/" + strconv.FormatInt(v.ID, 10) host, _, err := net.SplitHostPort(r.RemoteAddr) if err != nil { log.Println(err) http.Error(w, "Problem building streaming tokens, not remote addresse available.", http.StatusBadRequest) return } t, err := token.GenerateToken(v.UploadURI, host, time.Duration(3600)*time.Second, "1122AABBCCDDEEFF") if err != nil { log.Println(err) http.Error(w, "Problem building streaming tokens, likely due to malformed request.", http.StatusInternalServerError) return } v.Token = t err = storage.Update(v) //FIXME: This is awful, but probably not workaroundable? if err != nil { log.Println(err) http.Error(w, "Update Database error, likely due to malformed request.", http.StatusInternalServerError) return } } } w.Header().Set("Content-Type", "application/json") // Header are important when GZIP is enabled w.WriteHeader(http.StatusCreated) // Return the object we've inserted in the database. encoder := json.NewEncoder(w) err = encoder.Encode(val) if err != nil { log.Println(err) } }
// Handle a POST request with an image, phone_id, lat, lng and heading. Return a target heading from the web UI func MultiUploadHandler(rterDir string, uploadPath string, w http.ResponseWriter, r *http.Request) { imageFile, header, err := r.FormFile("image") if err != nil { return } user := new(data.User) user.Username = r.FormValue("phone_id") err = storage.Select(user) if err == storage.ErrZeroAffected { user.Role = "public" storage.Insert(user) // log.Println("upload failed, phone_id invalid:", user.Username) // http.Error(w, "Invalid credentials.", http.StatusUnauthorized) // return } else if err != nil { log.Println(err) http.Error(w, "Database error, likely due to malformed request.", http.StatusInternalServerError) return } os.Mkdir(filepath.Join(uploadPath, user.Username), os.ModeDir|0775) matchingItems := make([]*data.Item, 0) err = storage.SelectWhere(&matchingItems, "WHERE Type=\"streaming-video-v0\" AND Author=?", user.Username) exists := true if err == storage.ErrZeroAffected { exists = false } else if err != nil { log.Println(err) http.Error(w, "Database error, likely due to malformed request.", http.StatusInternalServerError) return } var item *data.Item if exists { item = matchingItems[0] } else { item = new(data.Item) } item.Author = user.Username item.Type = "streaming-video-v0" item.Live = true item.HasGeo = true item.HasHeading = true item.Lat, err = strconv.ParseFloat(r.FormValue("lat"), 64) if err != nil { item.HasGeo = false } item.Lng, err = strconv.ParseFloat(r.FormValue("lng"), 64) if err != nil { item.HasGeo = false } item.Heading, err = strconv.ParseFloat(r.FormValue("heading"), 64) if err != nil { item.HasHeading = false } item.StartTime = time.Now() item.StopTime = item.StartTime path := uploadPath if strings.HasSuffix(header.Filename, ".png") { path = filepath.Join(path, fmt.Sprintf("%v/%v.png", user.Username, item.StopTime.UnixNano())) } else if strings.HasSuffix(header.Filename, ".jpg") || strings.HasSuffix(header.Filename, "jpeg") { path = filepath.Join(path, fmt.Sprintf("%v/%v.jpg", user.Username, item.StopTime.UnixNano())) } outputFile, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE, 0664) if err != nil { log.Println(err) http.Error(w, "Error, likely due to malformed request.", http.StatusInternalServerError) } defer outputFile.Close() io.Copy(outputFile, imageFile) path = path[len(rterDir):] item.ContentURI = path item.ThumbnailURI = path if exists { storage.Update(item) } else { storage.Insert(item) } userDirection := new(data.UserDirection) userDirection.Username = user.Username err = storage.Select(userDirection) if err != nil { log.Println(err) http.Error(w, "Database error, likely due to malformed request.", http.StatusInternalServerError) return } w.Write([]byte(strconv.FormatFloat(userDirection.Heading, 'f', 6, 64))) }