lib/imap/backup/serializer/mbox_enumerator.rb in imap-backup-6.3.0 vs lib/imap/backup/serializer/mbox_enumerator.rb in imap-backup-7.0.0.rc1

@@ -6,12 +6,12 @@
 
     def initialize(mbox_pathname)
       @mbox_pathname = mbox_pathname
     end
 
-    def each
-      return enum_for(:each) if !block_given?
+    def each(&block)
+      return enum_for(:each) if !block
 
       File.open(mbox_pathname, "rb") do |f|
         lines = []
 
         loop do
@@ -24,10 +24,16 @@
           else
             lines << line
           end
         end
 
-        yield lines.join if lines.count.positive?
+        block.call(lines.join) if lines.count.positive?
       end
+    end
+
+    def map(&block)
+      return enum_for(:map) if !block
+
+      each.map { |line| block.call(line) }
     end
   end
 end