lib/cucumber/formatter/color_io.rb


require 'forwardable'

module Cucumber
  module Formatter
    # Adapter to make #puts/#print/#flush work with win32console
    class ColorIO #:nodoc:
      extend Forwardable
      def_delegators :@kernel, :puts, :print # win32console colours only work when sent to Kernel
      def_delegators :@stdout, :flush, :tty?, :write, :close

      def initialize(kernel, stdout)
        @kernel = kernel
        @stdout = stdout
      end

      # Ensure using << still gets colours in win32console
      def <<(output)
        print(output)
        self
      end
    end
  end
end