ext/mpfi/extconf.rb in ruby-mpfi-0.0.6 vs ext/mpfi/extconf.rb in ruby-mpfi-0.0.7

@@ -18,7 +18,18 @@
 
 dir_config("mpfr")
 dir_config("mpfi")
 dir_config("gmp")
 if have_header('mpfr.h') && have_library('mpfr') && have_header('mpfi.h') && have_library('mpfi') && have_header('gmp.h') && have_library('gmp')
+
+  ruby_mpfr_header_dir = nil
+  begin
+    require "rubygems"
+    spec = Gem::Specification.find_by_name("ruby-mpfr")
+    ruby_mpfr_header_dir = File.join(spec.full_gem_path, 'ext/mpfr')
+  rescue LoadError
+  end
+  unless find_header("ruby_mpfr.h", ruby_mpfr_header_dir)
+    header_not_found("ruby_mpfr.h")
+  end
   create_makefile("mpfi")
 end