ruby/test/mkmf/base.rb

# frozen_string_literal: false
$extmk = true
require 'rbconfig'
RbConfig.fire_update!("top_srcdir", File.expand_path("../..", __dir__))
File.foreach(RbConfig::CONFIG["topdir"]+"/Makefile") do |line|
  if /^CC_WRAPPER\s*=\s*/ =~ line
    RbConfig.fire_update!('CC_WRAPPER', $'.strip)
    break
  end
end

require 'test/unit'
require 'mkmf'
require 'tmpdir'

$extout = '$(topdir)/'+RbConfig::CONFIG["EXTOUT"]
RbConfig::CONFIG['topdir'] = CONFIG['topdir'] = File.expand_path(CONFIG['topdir'])
RbConfig::CONFIG["extout"] = CONFIG["extout"] = $extout
$INCFLAGS << " -I."
$extout_prefix = "$(extout)$(target_prefix)/"

class TestMkmf < Test::Unit::TestCase
  module Base
    MKMFLOG = proc {File.read("mkmf.log") rescue ""}

    class Capture
      attr_accessor :origin
      def initialize
        @buffer = ""
        @filter = nil
        @out = true
        @origin = nil
      end
      def clear
        @buffer.clear
      end
      def flush
        STDOUT.print @filter ? @filter.call(@buffer) : @buffer
        clear
      end
      def reopen(io)
        case io
        when Capture
          initialize_copy(io)
        when File
          @out = false
          @origin.reopen(io) if @origin
        when IO
          @out = true
          @origin.reopen(io) if @origin
        else
          @out = false
        end
      end
      def filter(&block)
        @filter = block
      end
      def write(*s)
        if @out
          @buffer.concat(*s)
        elsif @origin
          @origin.write(*s)
        end
      end
    end

    attr_reader :stdout

    def mkmflog(msg)
      proc {MKMFLOG[] << msg}
    end

    def setup
      @rbconfig = rbconfig0 = RbConfig::CONFIG
      @mkconfig = mkconfig0 = RbConfig::MAKEFILE_CONFIG
      rbconfig = {
        "hdrdir" => $hdrdir,
        "srcdir" => $srcdir,
        "topdir" => $topdir,
      }
      mkconfig = {
        "hdrdir" => "$(top_srcdir)/include",
        "srcdir" => "$(top_srcdir)",
        "topdir" => $topdir,
      }
      rbconfig0.each_pair {|key, val| rbconfig[key] ||= val.dup}
      mkconfig0.each_pair {|key, val| mkconfig[key] ||= val.dup}
      RbConfig.module_eval {
        remove_const(:CONFIG)
        const_set(:CONFIG, rbconfig)
        remove_const(:MAKEFILE_CONFIG)
        const_set(:MAKEFILE_CONFIG, mkconfig)
      }
      MakeMakefile.class_eval {
        remove_const(:CONFIG)
        const_set(:CONFIG, mkconfig)
      }
      @tmpdir = Dir.mktmpdir
      @curdir = Dir.pwd
      @mkmfobj = Object.new
      @stdout = Capture.new
      Dir.chdir(@tmpdir)
      @quiet, Logging.quiet = Logging.quiet, true
      init_mkmf
      $INCFLAGS[0, 0] = "-I. "
    end

    def teardown
      return if @omitted
      rbconfig0 = @rbconfig
      mkconfig0 = @mkconfig
      RbConfig.module_eval {
        remove_const(:CONFIG)
        const_set(:CONFIG, rbconfig0)
        remove_const(:MAKEFILE_CONFIG)
        const_set(:MAKEFILE_CONFIG, mkconfig0)
      }
      MakeMakefile.class_eval {
        remove_const(:CONFIG)
        const_set(:CONFIG, mkconfig0)
      }
      Logging.quiet = @quiet
      Logging.log_close
      FileUtils.rm_f("mkmf.log")
      Dir.chdir(@curdir)
      FileUtils.rm_rf(@tmpdir)
    end

    def mkmf(*args, &block)
      @stdout.clear
      stdout, @stdout.origin, $stdout = @stdout.origin, $stdout, @stdout
      verbose, $VERBOSE = $VERBOSE, false
      @mkmfobj.instance_eval(*args, &block)
    ensure
      $VERBOSE = verbose
      $stdout, @stdout.origin = @stdout.origin, stdout
    end

    def config_value(name)
      create_tmpsrc("---config-value=#{name}")
      xpopen(cpp_command('')) do |f|
        f.grep(/^---config-value=(.*)/) {return $1}
      end
      nil
    end
  end

  include Base

  def assert_separately(args, extra_args, src, *rest, **options)
    super(args + ["-r#{__FILE__}"] + %w[- --] + extra_args,
          "extend TestMkmf::Base; setup\nEND{teardown}\n#{src}",
          *rest,
          **options)
  end
end