lib/polyfill/v2_4/io/instance/readlines.rb in polyfill-0.3.0 vs lib/polyfill/v2_4/io/instance/readlines.rb in polyfill-0.4.0

@@ -1,5 +1,7 @@
+require 'English'
+
 module Polyfill
   module V2_4
     module IO
       module Instance
         module Readlines
@@ -8,29 +10,26 @@
               hash, others = args.partition { |arg| arg.is_a?(::Hash) }
 
               inputs = super(*others)
 
               if hash[0] && hash[0][:chomp]
-                separator = others.find { |other| other.respond_to?(:to_str) }
-                if separator
-                  inputs.each { |input| input.chomp!(separator) }
-                else
-                  inputs.each(&:chomp!)
-                end
+                separator = others.find do |other|
+                  other.respond_to?(:to_str)
+                end || $INPUT_RECORD_SEPARATOR
+
+                inputs.each { |input| input.chomp!(separator) }
               end
 
               inputs
             end
           end
 
-          if RUBY_VERSION < '2.4.0'
-            refine ::IO do
-              include Method
-            end
+          refine ::IO do
+            include Method
+          end
 
-            def self.included(base)
-              base.include Method
-            end
+          def self.included(base)
+            base.include Method
           end
         end
       end
     end
   end