1 files changed, 27 insertions, 19 deletions
diff --git a/server/lib/api/media.js b/server/lib/api/media.js
index 089e327..9898727 100644
--- a/server/lib/api/media.js
+++ b/server/lib/api/media.js
@@ -14,30 +14,38 @@ var media = {
 		})
 	},
 	
+	create: function(req, res){
+		var data = util.cleanQuery(req.body)
+		data.user_id = req.user._id
+		data.created_at = new Date ()
+
+		new Media(data).save(function(err, rec){
+			if (err || ! rec) { return res.json({ error: err }) }
+			return res.json(rec) 
+		})
+	},
+	
 	upload: function(req, res){
 		var data = util.cleanQuery(req.body)
-		data.updated_at = new Date ()
+		data.user_id = req.user._id
+		data.created_at = new Date ()
+		data.type = "image"
 		
-		if (req.files.avatar) {
-			upload.put("avatars", req.files.avatar, {
-				unacceptable: function(err){
-					res.json({ error: { errors: { avatar: { message: "Problem saving avatar: " + err } } } })
-				},
-				success: function(url){
-					data.photo = url
-					done()
-				}
-			})
-		}
-		else {
-			done()
-		}
+		upload.put("media", req.files.image, {
+			unacceptable: function(err){
+				res.json({ error: { errors: { avatar: { message: "Problem saving image: " + err } } } })
+			},
+			success: function(url){
+				data.url = url
+				done()
+			}
+		})
 		
 		function done () {
-			_.extend( req.user, data )
-			req.user.save(function(err, msg) {
-				err ? res.json({ status: "FAIL", error: err })
-					: res.json({ status: "OK", payload: req.user })
+			console.log(data)
+			new Media(data).save(function(err, rec) {
+				if (err || ! rec) { return res.json({ error: err }) }
+				res.json(rec)
 			})
 		}
 	}