#! /usr/local/bin/ruby

$".push 'mkmf.rb'

if ARGV[0] == 'static'
  $force_static = TRUE
  ARGV.shift
elsif ARGV[0] == 'install'
  $install = TRUE
  $destdir = ARGV[1] || ''
  ARGV.shift
elsif ARGV[0] == 'clean'
  $clean = TRUE
  ARGV.shift
end

SRC_EXT = ["c", "cc", "cxx", "C"]
$extlist = []

$cache_mod = FALSE;
$lib_cache = {}
$func_cache = {}
$hdr_cache = {}
$top_srcdir = "@top_srcdir@"
if $top_srcdir !~ "^/"
  # get absolute path
  $top_srcdir = File.expand_path($top_srcdir)
end
# get absolute path
$topdir = File.expand_path("..")

load "#{$top_srcdir}/lib/find.rb"

## drive letter
if PLATFORM == "i386-os2_emx" then
  $dots = ""
else
  $dots = if "@INSTALL@" =~ /^\// then "" else "#{$topdir}/ext/" end
end

if File.exist?("config.cache") then
  f = open("config.cache", "r")
  while f.gets
    case $_
    when /^lib: (.+) (yes|no)/
      $lib_cache[$1] = $2
    when /^func: ([\w_]+) (yes|no)/
      $func_cache[$1] = $2
    when /^hdr: (.+) (yes|no)/
      $hdr_cache[$1] = $2
    end
  end
  f.close
end

def older(file1, file2)
  if !File.exist?(file1) then
    return true
  end
  if !File.exist?(file2) then
    return false
  end
  if File.mtime(file1) < File.mtime(file2)
    return true
  end
  return false
end

if PLATFORM == "m68k-human"
CFLAGS = "@CFLAGS@".gsub(/-c..-stack=[0-9]+ */, '')
else
CFLAGS = "@CFLAGS@"
end
LINK = "@CC@ -o conftest -I#$topdir -I#$top_srcdir -I@includedir@ #{CFLAGS} @LDFLAGS@ %s %s conftest.c %s %s @LIBS@"
CPP = "@CPP@ @CPPFLAGS@ -I#$topdir -I#$top_srcdir -I@includedir@ #{CFLAGS} %s %s conftest.c"

if /cygwin|mswin32|djgpp|mingw32|m68k-human|i386-os2_emx/i =~ PLATFORM
  $null = open("nul", "w")
else
  $null = open("/dev/null", "w")
end

$orgerr = $stderr.dup
$orgout = $stdout.dup
def xsystem command
  if $DEBUG
    return system(command)
  end
  $stderr.reopen($null) 
  $stdout.reopen($null) 
  r = system(command)
  $stderr.reopen($orgerr)
  $stdout.reopen($orgout)
  return r
end

def try_link0(src, opt="")
  cfile = open("conftest.c", "w")
  cfile.print src
  cfile.close
  xsystem(format(LINK, $CFLAGS, $LDFLAGS, opt, $LOCAL_LIBS))
end

def try_link(src, opt="")
  begin
    try_link0(src, opt)
  ensure
    system "rm -f conftest*"
  end
end

def try_cpp(src, opt="")
  cfile = open("conftest.c", "w")
  cfile.print src
  cfile.close
  begin
    xsystem(format(CPP, $CFLAGS, opt))
  ensure
    system "rm -f conftest*"
  end
end

def egrep_cpp(pat, src, opt="")
  cfile = open("conftest.c", "w")
  cfile.print src
  cfile.close
  begin
    xsystem(format(CPP+"|egrep #{pat}", $CFLAGS, opt))
  ensure
    system "rm -f conftest*"
  end
end

def try_run(src, opt="")
  begin
    if try_link0(src, opt)
      if xsystem("./conftest")
	true
      else
	false
      end
    else
      nil
    end
  ensure
    system "rm -f conftest*"
  end
end

def install_rb(mfile, srcdir)
  libdir = srcdir + "/lib"
  path = []
  dir = []
  Find.find(libdir) do |f|
    next unless /\.rb$/ =~ f
    f = f[libdir.length+1..-1]
    path.push f
    dir |= File.dirname(f)
  end
  for f in dir
    next if f == "."
    mfile.printf "\t@test -d $(DESTDIR)$(pkglibdir)/%s || mkdir $(DESTDIR)$(pkglibdir)/%s\n", f, f
  end
  for f in path
    mfile.printf "\t$(INSTALL_DATA) $(srcdir)/lib/%s $(DESTDIR)$(pkglibdir)/%s\n", f, f
  end
end

def have_library(lib, func="main")
  if $lib_cache[lib]
    if $lib_cache[lib] == "yes"
      if $libs
        $libs = "-l" + lib + " " + $libs 
      else
	$libs = "-l" + lib
      end
      return true
    else
      return false
    end
  end

  if func && func != ""
    if $libs
      libs = "-l" + lib + " " + $libs 
    else
      libs = "-l" + lib
    end
    unless try_link(<<"SRC", libs)
int main() { return 0; }
int t() { #{func}(); return 0; }
SRC
      $lib_cache[lib] = 'no'
      $cache_mod = true
      return false
    end
  else
    if $libs
      libs = "-l" + lib + " " + $libs 
    else
      libs = "-l" + lib
    end
  end

  $libs = libs
  $lib_cache[lib] = 'yes'
  $cache_mod = true
  return true
end

def have_func(func)
  if $func_cache[func]
    if $func_cache[func] == "yes"
      $defs.push(format("-DHAVE_%s", func.upcase))
      return true
    else
      return false
    end
  end

  libs = $libs
  libs = "" if libs == nil

  unless try_link(<<"SRC", libs)
char #{func}();
int main() { return 0; }
int t() { #{func}(); return 0; }
SRC
    $func_cache[func] = 'no'
    $cache_mod = true
    return false
  end
  $defs.push(format("-DHAVE_%s", func.upcase))
  $func_cache[func] = 'yes'
  $cache_mod = true
  return true
end

def have_header(header)
  if $hdr_cache[header]
    if $hdr_cache[header] == "yes"
      header.tr!("a-z./\055", "A-Z___")
      $defs.push(format("-DHAVE_%s", header))
      return true
    else
      return false
    end
  end

  unless try_cpp(<<"SRC")
#include <#{header}>
SRC
    $hdr_cache[header] = 'no'
    $cache_mod = true
    return false
  end
  $hdr_cache[header] = 'yes'
  header.tr!("a-z./\055", "A-Z___")
  $defs.push(format("-DHAVE_%s", header))
  $cache_mod = true
  return true
end

def arg_config(config, default=nil)
  unless defined? $configure_args
    $configure_args = {}
    for arg in "@configure_args@".split
      next unless /^--/ =~ arg
      if /=/ =~ arg
	$configure_args[$`] = $'
      else
	$configure_args[arg] = true
      end
    end
  end
  $configure_args.fetch(config, default)
end

def with_config(config, default=nil)
  unless /^--with-/ =~ config
    config = '--with-' + config
  end
  arg_config(config, default)
end

def enable_config(config, default=nil)
  if arg_config("--enable-"+config, default)
    true
  elsif arg_config("--disable-"+config, false)
    false
  else
    default
  end
end

def create_header()
  if $defs.length > 0
    hfile = open("extconf.h", "w")
    for line in $defs
      line =~ /^-D(.*)/
      hfile.printf "#define %s 1\n", $1
    end
    hfile.close
  end
end

def create_makefile(target)
  system "rm -f conftest*"
  if $libs and "@DLEXT@" == "o"
    libs = $libs.split
    for lib in libs
      lib.sub!(/-l(.*)/, '"lib\1.a"')
    end
    $defs.push(format("-DEXTLIB='%s'", libs.join(",")))
  end

  $DLDFLAGS = '@DLDFLAGS@'

  if PLATFORM =~ /beos/
    if $libs
      $libs = $libs + " -lruby"
    else
      $libs = "-lruby"
    end
    $DLDFLAGS = $DLDFLAGS + " -L" + $topdir
  end

  defflag = ''
  if PLATFORM =~ /cygwin/ and not $static
    if File.exist? target + ".def"
      defflag = "--def=" + target + ".def"
    end
    if $libs
      $libs = $libs + " @LIBRUBYARG@"
    else
      $libs = "@LIBRUBYARG@"
    end
    $DLDFLAGS = $DLDFLAGS + " -L" + $topdir
  end

  $srcdir = $top_srcdir + "/ext/" + $mdir
  mfile = open("Makefile", "w")
  mfile.printf "\
SHELL = /bin/sh

#### Start of system configuration section. ####

srcdir = #{$srcdir}
VPATH = #{$srcdir}

topdir = #{$topdir}
hdrdir = #{$top_srcdir}
DESTDIR = 

CC = @CC@

prefix = @prefix@
CFLAGS   = %s -I$(topdir) -I$(hdrdir) -I@includedir@ #{CFLAGS} #$CFLAGS %s
DLDFLAGS = #$DLDFLAGS #$LDFLAGS
LDSHARED = @LDSHARED@ #{defflag}
", if $static then "" else "@CCDLFLAGS@" end, $defs.join(" ")

  mfile.printf "\

RUBY_INSTALL_NAME = @RUBY_INSTALL_NAME@

prefix = @prefix@
exec_prefix = @exec_prefix@
libdir = @libdir@
#pkglibdir = $(libdir)/$(RUBY_INSTALL_NAME)/@MAJOR@.@MINOR@
pkglibdir = $(libdir)/ruby/@MAJOR@.@MINOR@
archdir = $(pkglibdir)/@arch@
@SET_MAKE@

#### End of system configuration section. ####

"
  mfile.printf "LOCAL_LIBS = %s\n", $LOCAL_LIBS unless $LOCAL_LIBS == ""
  mfile.printf "LIBS = %s\n", $libs
  mfile.printf "OBJS = "
  if !$objs then
    $objs = []
    for f in Dir["#{$top_srcdir}/ext/#{$mdir}/*.{#{SRC_EXT.join(%q{,})}}"]
      f = File.basename(f)
      f.sub!(/(#{SRC_EXT.join(%q{|})})$/, "o")
      $objs.push f
    end
  end
  mfile.printf $objs.join(" ")
  mfile.printf "\n"

  mfile.printf <<EOS
TARGET = #{target}
DLLIB = $(TARGET).#{$static ? "a" : "@DLEXT@"}

INSTALL = #{$dots}@INSTALL@
INSTALL_DLLIB = @INSTALL_DLLIB@
INSTALL_DATA = @INSTALL_DATA@

binsuffix = @binsuffix@

all:		$(DLLIB)

clean:;		@rm -f *.o *.a *.so *.sl *.a
		@rm -f Makefile extconf.h conftest.*
		@rm -f core ruby$(binsuffix) *~

realclean:	clean
EOS

  mfile.printf <<EOS

install:
	@test -d $(DESTDIR)$(libdir) || mkdir $(DESTDIR)$(libdir)
	@test -d $(DESTDIR)$(pkglibdir) || mkdir $(DESTDIR)$(pkglibdir)
	@test -d $(DESTDIR)$(archdir) || mkdir $(DESTDIR)$(archdir)
EOS
  if !$static
    mfile.printf "\
	$(INSTALL_DLLIB) $(DLLIB) $(DESTDIR)$(archdir)/$(DLLIB)
"
  end
  install_rb(mfile, $srcdir)
  mfile.printf "\n"

  if $static
      mfile.printf "\
$(DLLIB): $(OBJS)
	@AR@ cru $(DLLIB) $(OBJS)
	@-@RANLIB@ $(DLLIB) 2> /dev/null || true
"
  elsif "@DLEXT@" != "o"
    mfile.printf "\
$(DLLIB): $(OBJS)
	$(LDSHARED) $(DLDFLAGS) -o $(DLLIB) $(OBJS) $(LIBS) $(LOCAL_LIBS)
"
  elsif PLATFORM == "m68k-human"
    mfile.printf "\
$(DLLIB): $(OBJS)
	ar cru $(DLLIB) $(OBJS)
"
  else
    mfile.printf "\
$(DLLIB): $(OBJS)
	ld $(DLDFLAGS) -r -o $(DLLIB) $(OBJS)
"
  end

  if File.exist?("#{$srcdir}/depend")
    dfile = open("#{$srcdir}/depend", "r")
    mfile.printf "###\n"
    while line = dfile.gets()
      mfile.printf "%s", line.gsub('\$\(hdrdir\)/config.h', '$(topdir)/config.h')
    end
    dfile.close
  end
  mfile.close
  
  if PLATFORM =~ /beos/
    if PLATFORM =~ /^powerpc/ then
      deffilename = "ruby.exp"
    else
      deffilename = "ruby.def"
    end
    print "creating ruby.def\n"
    open(deffilename, "w") do |file|
      file.print("EXPORTS\n") if PLATFORM =~ /^i/
      file.print("Init_#{target}\n")
    end
  end
end

def extmake(target)
  if $force_static or $static_ext[target]
    $static = target
  else
    $static = FALSE
  end

  return if $nodynamic and not $static

  $objs = nil
  $libs = PLATFORM =~ /cygwin|beos|openstep|nextstep|rhapsody/ ? nil : "-lc"
  $LOCAL_LIBS = ""		# to be assigned in extconf.rb
  $CFLAGS = ""
  $LDFLAGS = ""

  begin
    system "mkdir", target unless File.directory?(target)
    Dir.chdir target
    $mdir = target
    if $static_ext.size > 0 ||
      !File.exist?("./Makefile") ||
      older("./Makefile", "#{$top_srcdir}/ext/@setup@") ||
      older("./Makefile", "../extmk.rb") ||
      older("./Makefile", "#{$top_srcdir}/ext/#{target}/extconf.rb")
    then
      $defs = []
      if File.exist?("#{$top_srcdir}/ext/#{target}/extconf.rb")
	load "#{$top_srcdir}/ext/#{target}/extconf.rb"
      else
	create_makefile(target);
      end
    end
    if File.exist?("./Makefile")
      if $static
	$extlist.push [$static,target]
      end
      if $install
	system "#{$make} install DESTDIR=#{$destdir}"
      elsif $clean
	system "#{$make} clean"
      else
	system "#{$make} all" or exit
      end
    end
    if $static
      $extlibs ||= ""
      $extlibs += " " + $LDFLAGS unless $LDFLAGS == ""
      $extlibs += " " + $libs if $libs
      $extlibs += " " + $LOCAL_LIBS unless $LOCAL_LIBS == ""
    end
  ensure
    system "rm -f conftest*"
    Dir.chdir ".."
  end
end

$make = ENV["MAKE"]
$make ||= with_config("make-prog", "make")

# get static-link modules
$static_ext = {}
for setup in ["@setup@", "#{$top_srcdir}/ext/@setup@"]
  if File.file? setup
    f = open(setup) 
    while f.gets()
      $_.chomp!
      sub!(/#.*$/, '')
      next if /^\s*$/
      if /^option +nodynamic/
	$nodynamic = TRUE
	next
      end
      $static_ext[$_.split[0]] = TRUE
    end
    f.close
    break
  end
end

for d in Dir["#{$top_srcdir}/ext/*"]
  File.directory?(d) || next
  File.file?(d + "/MANIFEST") || next
  
  d = File.basename(d)
  if $install
    print "installing ", d, "\n"
  elsif $clean
    print "cleaning ", d, "\n"
  else
    print "compiling ", d, "\n"
    if PLATFORM =~ /-aix/ and older("../ruby.imp", "../miniruby")
      load "#{$top_srcdir}/ext/aix_mksym.rb"
    end
  end
  extmake(d)
end

if $cache_mod
  f = open("config.cache", "w")
  for k,v in $lib_cache
    f.printf "lib: %s %s\n", k, v
  end
  for k,v in $func_cache
    f.printf "func: %s %s\n", k, v
  end
  for k,v in $hdr_cache
    f.printf "hdr: %s %s\n", k, v
  end
  f.close
end

exit if $install or $clean
$extinit = "" unless $extinit

ruby = "@RUBY_INSTALL_NAME@@binsuffix@"
miniruby = "miniruby@binsuffix@"

if $extlist.size > 0
  for s,t in $extlist
    f = format("%s/%s.a", s, t)
    if File.exist?(f)
      $extinit += format("\
\tInit_%s();\n\
\trb_provide(\"%s.so\");\n\
", t, t)
      $extobjs = "" unless $extobjs
      $extobjs += "ext/"
      $extobjs += f
      $extobjs += " "
    else
      FALSE
    end
  end

  if older("extinit.c", "#{$top_srcdir}/ext/@setup@")
    f = open("extinit.c", "w")
    f.printf "void Init_ext() {\n"
    f.printf $extinit
    f.printf "}\n"
    f.close
  end
  if older("extinit.o", "extinit.c")
    cmd = "@CC@ " + CFLAGS + " -c extinit.c"
    print cmd, "\n"
    system cmd or exit 1
  end

  Dir.chdir ".."

  if older(ruby, "#{$top_srcdir}/ext/@setup@") or older(ruby, miniruby)
    system("rm -f #{ruby}")
  end

  if $extobjs
    $extobjs = "ext/extinit.o " + $extobjs
  else
    $extobjs = "ext/extinit.o "
  end
  if PLATFORM =~ /m68k-human|beos/
    $extlibs.gsub!("-L/usr/local/lib", "") if $extlibs
  end
  system format(%[#{$make} #{ruby} EXTOBJS="%s" EXTLIBS="%s"], $extobjs, $extlibs)
else
  Dir.chdir ".."
  if older(ruby, miniruby)
    system("rm -f #{ruby}")
    system("make #{ruby}")
  end
end

#Local variables:
# mode: ruby
#end: