Set flag in preview mode and re-generate files before deployment if necessary

2 files changed, 11 insertions, 1 deletions

diff --git a/Rakefile b/Rakefile
index c3ad68be..ebc386e6 100644
--- a/Rakefile
+++ b/Rakefile
@@ -200,6 +200,13 @@ end
 
 desc "Default deploy task"
 task :deploy do
+  # Check if preview posts exist, which should not be published
+  if File.exists?(".preview-mode")
+    puts "## Found posts in preview mode, regenerating files ..."
+    File.delete(".preview-mode")
+    Rake::Task[:generate].execute
+  end
+
   Rake::Task[:copydot].invoke(source_dir, public_dir)
   Rake::Task["#{deploy_default}"].execute
 end
diff --git a/plugins/preview_unpublished.rb b/plugins/preview_unpublished.rb
index 20b27d6b..321ffd6f 100644
--- a/plugins/preview_unpublished.rb
+++ b/plugins/preview_unpublished.rb
@@ -24,8 +24,11 @@ module Jekyll
 
           # Monkeypatch:
           # On preview environment (localhost), publish all posts
-          if ENV.has_key?('OCTOPRESS_ENV') && ENV['OCTOPRESS_ENV'] == 'preview'
+          if ENV.has_key?('OCTOPRESS_ENV') && ENV['OCTOPRESS_ENV'] == 'preview' && post.data.has_key?('published') && post.data['published'] == false
             post.published = true
+            # Set preview mode flag (if necessary), `rake generate` will check for it
+            # to prevent pushing preview posts to productive environment
+            File.open(".preview-mode", "w") {}
           end
 
           if post.published && (self.future || post.date <= self.time)