zip up compressed files

Author: Ubuntu

Gemfile

@@ -43,6 +43,9 @@ gem 'less'
 gem 'less-rails'
 gem 'twitter-bootstrap-rails'
 
+# zip
+gem 'rubyzip'
+
 group :development, :test do
   # Call 'byebug' anywhere in the code to stop execution and get a debugger console
   gem 'byebug'

Gemfile.lock

@@ -126,6 +126,7 @@ GEM
       json (~> 1.4)
     ref (1.0.5)
     rmagick (2.14.0)
+    rubyzip (1.1.7)
     sass (3.4.13)
     sass-rails (5.0.3)
       railties (>= 4.0.0, < 5.0)
@@ -184,6 +185,7 @@ DEPENDENCIES
   less-rails
   rails (= 4.2.0)
   rmagick
+  rubyzip
   sass-rails (~> 5.0)
   sdoc (~> 0.4.0)
   spring

lib/tasks/import.rake

@@ -1,7 +1,14 @@
 namespace :pictures do
   require 'fileutils'
+  require 'rubygems'
+  require 'zip'
 
   S3_BUCKET_NAME = ENV['S3_BUCKET_NAME'] || raise('please set environment var S3_BUCKET_NAME')
+
+  desc 'Zip up compressed images'
+  task :zip => :environment do
+    generate_zip Rails.root.join('app','assets','images', 'compressed'), Rails.root.join('test.zip')
+  end
 
   desc 'Download pictures from a S3 bucket to assets'
   task :import => :environment do
@@ -39,6 +46,9 @@ namespace :pictures do
       generate_compressed File.join(compress_dir, img_name), img
     end
 
+    # generate zip file
+    generate_zip compress_dir, Rails.root.join('app', 'assets', 'images.zip')
+
     # done!
     puts "All done!"
     puts "Total time taken was #{Time.now - start_time} seconds"
@@ -54,4 +64,19 @@ namespace :pictures do
   def generate_compressed path, img
     img.write(path) { self.quality = 70 }
   end
+
+  # zip files up so they can be downloaded all at once
+  def generate_zip folder, zip_path
+    puts File.join(folder, '*')
+    input_files = Dir.glob(File.join(folder, '*'))
+    
+    Zip::File.open(zip_path, Zip::File::CREATE) do |zipfile|
+      input_files.each do |file|
+        # Two arguments:
+        # - The name of the file as it will appear in the archive
+        # - The original file, including the path to find it
+        zipfile.add(File.basename(file), file)
+      end
+    end
+  end
 end

lib/tasks/import.rb

@@ -1,7 +1,14 @@
 namespace :pictures do
   require 'fileutils'
+  require 'rubygems'
+  require 'zip'
 
   S3_BUCKET_NAME = ENV['S3_BUCKET_NAME'] || raise('please set environment var S3_BUCKET_NAME')
+
+  desc 'Zip up compressed images'
+  task :zip => :environment do
+    generate_zip Rails.root.join('app','assets','images', 'compressed'), Rails.root.join('test.zip')
+  end
 
   desc 'Download pictures from a S3 bucket to assets'
   task :import => :environment do
@@ -39,6 +46,9 @@
       generate_compressed File.join(compress_dir, img_name), img
     end
 
+    # generate zip file
+    generate_zip compress_dir, Rails.root.join('app', 'assets', 'images.zip')
+
     # done!
     puts "All done!"
     puts "Total time taken was #{Time.now - start_time} seconds"
@@ -54,4 +64,19 @@ def generate_thumbnail path, img
   def generate_compressed path, img
     img.write(path) { self.quality = 70 }
   end
+
+  # zip files up so they can be downloaded all at once
+  def generate_zip folder, zip_path
+    puts File.join(folder, '*')
+    input_files = Dir.glob(File.join(folder, '*'))
+    
+    Zip::File.open(zip_path, Zip::File::CREATE) do |zipfile|
+      input_files.each do |file|
+        # Two arguments:
+        # - The name of the file as it will appear in the archive
+        # - The original file, including the path to find it
+        zipfile.add(File.basename(file), file)
+      end
+    end
+  end
 end