require 'English'

module Polyfill
  module V2_4
    module IO
      module Instance
        module Lines
          module Method
            def lines(*args)
              hash, others = args.partition { |arg| arg.is_a?(::Hash) }
              chomps = hash[0] && hash[0][:chomp]

              unless block_given?
                return super(*others) unless chomps

                separator = others.find do |other|
                  other.respond_to?(:to_str)
                end || $INPUT_RECORD_SEPARATOR
                return ::Enumerator.new do |yielder|
                  super(*others) do |line|
                    yielder.yield(line.chomp(separator))
                  end
                end
              end

              block =
                if chomps
                  separator = others.find do |other|
                    other.respond_to?(:to_str)
                  end || $INPUT_RECORD_SEPARATOR

                  proc do |line|
                    yield(line.chomp(separator))
                  end
                else
                  ::Proc.new
                end

              super(*others, &block)
            end
          end

          refine ::IO do
            include Method
          end

          def self.included(base)
            base.include Method
          end
        end
      end
    end
  end
end