upload wallpaper, reuses media api with a tag

1 files changed, 11 insertions, 2 deletions

diff --git a/server/lib/api/media.js b/server/lib/api/media.js
index 16f9d41..1eb08c1 100644
--- a/server/lib/api/media.js
+++ b/server/lib/api/media.js
@@ -8,8 +8,13 @@ var _ = require('lodash'),
 	Media = require('../schemas/Media');
 
 var media = {
+
 	user: function(req, res){
-		Media.find({ user_id: req.user._id }, function(err, media){
+	  var query = { user_id: req.user._id }
+	  if (req.query.tag) {
+	    query.tag = req.query.tag
+	  }
+		Media.find(query, function(err, media){
 			res.json(media || [])
 		})
 	},
@@ -18,10 +23,14 @@ var media = {
 		var data = util.cleanQuery(req.body)
 		data.user_id = req.user._id
 		data.created_at = new Date ()
+		
+		if (data.tag) {
+		  data.tag = util.sanitize(data.tag)
+		}
 
 		new Media(data).save(function(err, rec){
 			if (err || ! rec) { return res.json({ error: err }) }
-			return res.json(rec) 
+			return res.json(rec)
 		})
 	},