no message

models/profile.js

@@ -169,15 +169,11 @@ module.exports = {
 	allMessageImages: (e, profileId) => {
 		const promise = new Promise((resolve, reject) => {	
 			ProfileModel
-				.findById(profileId, (err, profile) => {
-					if (err) {
-						reject(err);
-					}
-
-					if (profile) {
-            profile.messages
-              .find({ 'image': { $exists: true } })
-              .select('image -_id')
+			  .aggregate(
+			    { $match: { _id: profileId } },
+			    { $unwind: '$messages' },
+			    { $match: { 'messages.image': { $exists: true } } }
+			  )
         .exec((err, result) => {
           if (err) {
             reject(err);
@@ -186,8 +182,6 @@ module.exports = {
           if (result) {
             resolve(result);
           }              
-              });
-					}
         });			
 		});