Cleanup the thumbnail generator script

2015-03-17 01:47:29 +02:00
parent 7eed3d91de
commit 9bf44b01d3
2 changed files with 23 additions and 17 deletions
@@ -5,12 +5,14 @@ class ScreenshotGenerator < ::Middleman::Extension

  def initialize(app, options_hash={}, &block)
    super
-    app.before_build do
+    app.before_build do |builder|
+      builder.say "==> Begin - Theme thumbnail generator", :green
      Screenshot.prepare

      themes.each do |theme|
-        Screenshot.new(theme).generate_theme_image
+        Screenshot.new(theme).generate_theme_image(builder)
      end
+      builder.say "==> End   - Theme thumbnail generator", :green
    end
  end
 end