lib/yamg/cli.rb in yamg-0.3.5 vs lib/yamg/cli.rb in yamg-0.3.7

@@ -31,10 +31,17 @@
       path = setup['path']
       FileUtils.mkdir_p path unless File.exist?(path)
       File.join(path, asset)
     end
 
+    def compile_screenshots(ss, size, setup)
+      return unless YAMG.config['screenshot'].respond_to?(:[])
+      fail 'No url provided' unless url = YAMG.config['screenshot']['url']
+      Screenshot.new(ss, { 'size' => size, 'url' => url } ).work(setup['path'])
+      puts Rainbow("[o]SS #{ss}").black
+    end
+
     def compile_media(i, size, setup)
     end
 
     def compile_icon(i, size, setup)
       folder = setup['icon'] || YAMG.config['icon']['path']
@@ -54,14 +61,15 @@
       YAMG.info("Splash #{size.join('x')}px #{setup['path']}#{s}", :black)
     end
 
     def compile_work(job, opts)
       task = YAMG::TEMPLATES[job] || (works[job] && works[job]['export'])
-      %w(icon splash media).each do |key|
-        next unless (work = task[key])
+      %w(icon logo splash media screenshots).each do |subtask|
+        next unless (work = task[subtask])
+
         work.each do |asset, size|
           Thread.new do # 500% speed up with 8 cores
-            send(:"compile_#{key}", asset, size, setup_for(opts))
+            send(:"compile_#{subtask}", asset, size, setup_for(opts))
           end
         end
       end
     end