dynamicguy · April 6, 2023 18:17 · Apr 6, 2023 · Apr 6, 2023 · Apr 5, 2023 · Apr 5, 2023
diff --git a/s3batch.rb b/s3batch.rb
@@ -10,7 +10,7 @@ def initialize(bucket)
     @bucket = bucket
   end
 
-  # Lists object in a bucket update it's mime type.
+  # Lists object in a bucket and update it's mime type.
   #
   # @param start [Integer] The maximum number of objects to list.
   # @param max_objects [Integer] The maximum number of objects to list.

diff --git a/s4batch.rb → s3batch.rb b/s4batch.rb → s3batch.rb
diff --git a/s4batch.rb b/s4batch.rb
@@ -38,7 +38,7 @@ def list_objects(start, max_objects)
 
 def run_script
   bucket_name = "oscar-dev-document-service"
-  puts ARGV[1]
+
   if ARGV.length != 2
     puts "We need exactly two arguments. a start and limit. Example: ruby s3batch.rb 0 2"
     exit

diff --git a/s4batch.rb b/s4batch.rb
@@ -23,7 +23,7 @@ def list_objects(start, max_objects)
         source = Aws::S3::Object.new(obj_summary.bucket_name, obj_summary.key)
         target = source
         mime = Rack::Mime.mime_type('.' + obj_summary.key.split('.').last)
-        puts "processing key: \t#{obj_summary.key} to set content type as: \t#{mime}"
+        puts "processing key: #{obj_summary.key} to set content type as: #{mime}"
         target.copy_from(source, content_type: mime, metadata_directive: 'REPLACE')
       end
       count += 1
@@ -38,13 +38,15 @@ def list_objects(start, max_objects)
 
 def run_script
   bucket_name = "oscar-dev-document-service"
+  puts ARGV[1]
   if ARGV.length != 2
     puts "We need exactly two arguments. a start and limit. Example: ruby s3batch.rb 0 2"
     exit
   end
-  wrapper = BucketListObjectsWrapper.new(Aws::S3::Bucket.new(bucket_name))
+  bkt = Aws::S3::Bucket.new(bucket_name)
+  wrapper = BucketListObjectsWrapper.new(bkt)
   count = wrapper.list_objects(ARGV[0].to_i, ARGV[1].to_i)
-  puts "Listed #{count} objects."
+  puts "Updated #{count} objects out of #{bkt.objects.count}"
 end
 
 run_script if $PROGRAM_NAME == __FILE__
diff --git a/s4batch.rb b/s4batch.rb
@@ -0,0 +1,50 @@
+#!/usr/bin/env ruby
+
+require 'aws-sdk-s3'
+require 'rack'
+
+class BucketListObjectsWrapper
+  attr_reader :bucket
+
+  def initialize(bucket)
+    @bucket = bucket
+  end
+
+  # Lists object in a bucket update it's mime type.
+  #
+  # @param start [Integer] The maximum number of objects to list.
+  # @param max_objects [Integer] The maximum number of objects to list.
+  # @return [Integer] The number of objects listed.
+  def list_objects(start, max_objects)
+    count = 0
+    puts "The objects in #{@bucket.name} are:"
+    @bucket.objects.each do |obj_summary|
+      if count >= start
+        source = Aws::S3::Object.new(obj_summary.bucket_name, obj_summary.key)
+        target = source
+        mime = Rack::Mime.mime_type('.' + obj_summary.key.split('.').last)
+        puts "processing key: \t#{obj_summary.key} to set content type as: \t#{mime}"
+        target.copy_from(source, content_type: mime, metadata_directive: 'REPLACE')
+      end
+      count += 1
+      break if count == max_objects
+    end
+    count
+  rescue Aws::Errors::ServiceError => e
+    puts "Couldn't list objects in bucket #{bucket.name}. Here's why: #{e.message}"
+    0
+  end
+end
+
+def run_script
+  bucket_name = "oscar-dev-document-service"
+  if ARGV.length != 2
+    puts "We need exactly two arguments. a start and limit. Example: ruby s3batch.rb 0 2"
+    exit
+  end
+  wrapper = BucketListObjectsWrapper.new(Aws::S3::Bucket.new(bucket_name))
+  count = wrapper.list_objects(ARGV[0].to_i, ARGV[1].to_i)
+  puts "Listed #{count} objects."
+end
+
+run_script if $PROGRAM_NAME == __FILE__