cucumber-ruby/lib/cucumber/formatter/unicode.rb

# frozen_string_literal: true

# Require this file if you need Unicode support.
# Tips for improvement - esp. ruby 1.9: http://www.ruby-forum.com/topic/184730
require 'cucumber/platform'
require 'cucumber/formatter/ansicolor'

if Cucumber::WINDOWS
  if ENV['CUCUMBER_OUTPUT_ENCODING']
    Cucumber::CODEPAGE = ENV['CUCUMBER_OUTPUT_ENCODING']
  elsif `cmd /c chcp` =~ /(\d+)/
    if [65_000, 65_001].include? Regexp.last_match(1).to_i
      Cucumber::CODEPAGE = 'UTF-8'.freeze
      ENV['ANSICON_API'] = 'ruby'
    else
      Cucumber::CODEPAGE = "cp#{Regexp.last_match(1).to_i}".freeze
    end
  else
    Cucumber::CODEPAGE = 'cp1252'.freeze
    $stderr.puts("WARNING: Couldn't detect your output codepage. Assuming it is 1252. You may have to chcp 1252 or SET CUCUMBER_OUTPUT_ENCODING=cp1252.")
  end

  module Cucumber
    # @private
    module WindowsOutput
      def self.extended(output)
        output.instance_eval do
          def cucumber_preprocess_output(*out)
            out.map { |arg| arg.to_s.encode(Encoding.default_external) }
          rescue Encoding::UndefinedConversionError => e
            $stderr.cucumber_puts("WARNING: #{e.message}")
            out
          end

          alias cucumber_print print
          def print(*out)
            cucumber_print(*cucumber_preprocess_output(*out))
          end

          alias cucumber_puts puts
          def puts(*out)
            cucumber_puts(*cucumber_preprocess_output(*out))
          end
        end
      end

      Kernel.extend(self)
      $stdout.extend(self)
      $stderr.extend(self)
    end
  end
end