From 4feee6f25775ed7d075f0e4d9bcd1077ab616370 Mon Sep 17 00:00:00 2001
From: Andrew Davidson <andrew@amdavidson.com>
Date: Sat, 23 Jan 2016 18:20:21 -0500
Subject: [PATCH] Changing rebuild route so that it redirects back with a flash
 message rather than pinning to the 'view' page

---
 routes-post.js | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/routes-post.js b/routes-post.js
index aa023bf..45e3737 100644
--- a/routes-post.js
+++ b/routes-post.js
@@ -15,7 +15,12 @@ module.exports = function(app) {
                 var start = 0;
             }
             Post.getPosts(count, start, function(err, posts){
-                res.render('admin-post-list', {posts, user: req.user});
+                res.render('admin-post-list', {
+                    successNotice: req.flash('successNotice'),
+                    failureNotice: req.flash('failureNotice'),
+                    posts: posts,
+                    user: req.user
+                });
             });
         }
     );
@@ -156,12 +161,12 @@ module.exports = function(app) {
                     post.buildPost(function (err) {
                         if (!err) {
                             req.flash('successNotice', 'Post rebuilt successfully.');
-                            res.redirect('/admin/post/view/'+req.params.uuid);
+                            res.redirect(req.header('Referer') || '/admin');
                         }
                         else {
                             console.log(err);
                             req.flash('failureNotice', 'Post regeneration failed, check logs.');
-                            res.redirect('/admin/post/view/'+req.params.uuid);
+                            res.redirect(req.header('Referer') || '/admin');
                         }
                     });
                 });