#!/usr/bin/env ruby
#  Phusion Passenger - http://www.modrails.com/
#  Copyright (c) 2010-2012 Phusion
#
#  "Phusion Passenger" is a trademark of Hongli Lai & Ninh Bui.
#
#  Permission is hereby granted, free of charge, to any person obtaining a copy
#  of this software and associated documentation files (the "Software"), to deal
#  in the Software without restriction, including without limitation the rights
#  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
#  copies of the Software, and to permit persons to whom the Software is
#  furnished to do so, subject to the following conditions:
#
#  The above copyright notice and this permission notice shall be included in
#  all copies or substantial portions of the Software.
#
#  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
#  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
#  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
#  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
#  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
#  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
#  THE SOFTWARE.

ESSENTIALS = [
	"boost/detail/{limits,endian}.hpp",
	"boost/config",
	"boost/mpl",
	"boost/preprocessor/stringize.hpp",
	"boost/smart_ptr/detail/sp_counted_*",
	"boost/smart_ptr/detail/atomic_count*",
	"boost/smart_ptr/detail/spinlock*",
	"boost/thread",
	"libs/thread/src",
	"libs/system/src",
	"boost/date_time/gregorian/formatters_limited.hpp",
	"boost/date_time/date_formatting_limited.hpp",
	"boost/non_type.hpp",
	"boost/detail/fenv.hpp",
	"boost/foreach.hpp"
]
EXCLUDE = [
	"libs/thread/src/win32/*"
]
PROGRAM_SOURCE = %q{
	#include <boost/shared_ptr.hpp>
	#include <boost/shared_array.hpp>
	#include <boost/weak_ptr.hpp>
	#include <boost/make_shared.hpp>
	#include <boost/enable_shared_from_this.hpp>
	#include <boost/thread.hpp>
	#include <boost/noncopyable.hpp>
	#include <boost/function.hpp>
	#include <boost/bind.hpp>
	#include <boost/date_time/posix_time/posix_time.hpp>
	#include <boost/foreach.hpp>
	#include <boost/lambda/lambda.hpp>
}

require 'fileutils'
BOOST_DIR = ARGV[0]
Dir.chdir(File.dirname(__FILE__) + "/../ext")

# Run the given command, and abort on error.
def sh(*command)
	puts command.join(" ")
	if !system(*command)
		puts "*** ERROR"
		exit 1
	end
end

def install(source_filename, target_filename)
	dir = File.dirname(target_filename)
	if !File.exist?(dir)
		sh "mkdir", "-p", dir
	end
	command = ["install", "-m", "u+rw,g+r,o+r", source_filename, target_filename]
	sh(*command)
end

def copy_boost_files(patterns, exclude = nil)
	patterns.each do |pattern|
		files = Dir["#{BOOST_DIR}/#{pattern}"]
		files -= exclude if exclude
		files.each do |source|
			if File.directory?(source)
				source.slice!(0 .. BOOST_DIR.size)
				copy_boost_files(["#{source}/*"], exclude)
			else
				target = source.slice(BOOST_DIR.size + 1 .. source.size - 1)
				if target =~ /^libs\//
					target = "boost/#{target}"
				end
				if !File.exist?(target)
					install(source, target)
				end
			end
		end
	end
end

def copy_essential_files
	exclude = []
	EXCLUDE.each do |pattern|
		exclude.concat(Dir["#{BOOST_DIR}/#{pattern}"])
	end
	copy_boost_files(ESSENTIALS, exclude)
end

def prepare
	File.open("test.cpp", "w") do |f|
		f.write(PROGRAM_SOURCE)
	end
end

def cleanup
	FileUtils.rm_rf("boost/thread/win32")
	FileUtils.rm_rf("boost/src/win32")
	FileUtils.rm_rf("boost/asio/win32")
	FileUtils.rm_rf("boost/smart_ptr/detail/spinlock_w32.hpp")
	FileUtils.rm_rf("boost/smart_ptr/detail/sp_counted_base_w32.hpp")
	FileUtils.rm_rf("boost/smart_ptr/detail/atomic_count_win32.hpp")
	FileUtils.rm_rf("boost/config/platform/win32.hpp")
	File.unlink("test.cpp") rescue nil
end

# Compile PROGRAM_SOURCE and copy whatever missing header files the compiler needs.
def copy_dependencies
	done = false
	while !done
		missing_headers = `g++ -DBOOST_MPL_PREPROCESSING_MODE test.cpp -c -I. 2>&1`.
		  split("\n").
		  grep(/error: .*: No such file/).
		  map do |line|
			file = line.sub(/.*error: (.*): .*/, '\1')
			if file =~ /^boost\//
				file
			else
				line =~ /(.*?):/
				source = $1
				File.dirname(source) + "/" + file
			end
		end
		missing_headers.each do |header|
			install("#{BOOST_DIR}/#{header}", header)
		end
		done = missing_headers.empty?
	end
end

def start
	if BOOST_DIR.nil? || BOOST_DIR.empty?
		puts "Usage: copy_boost_headers.rb <boost source directory>"
		exit 1
	end
	begin
		prepare
		copy_essential_files
		copy_dependencies
	ensure
		cleanup
	end
end

start
