# Copyright (c) 2006 Zach Tibbitts ('zachtib') <zach@collegegeek.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program.  If not, write to:
# 	The Free Software Foundation, Inc.,
# 	51 Franklin Street, Fifth Floor
# 	Boston, MA  02110-1301, USA.

import os, platform
print "Attempting to detect your system information"
if platform.machine() == "i386" or platform.machine() == "i686":
	print "32bit x86 system detected"
	ARCH = "x86"
elif platform.machine() == "x86_64" or platform.machine() == "amd64":
	print "64bit x86_64 system detected"
	ARCH = "x64"
elif platform.machine() == "ppc":
	print "PowerPC system detected"
	ARCH = "ppc"
else:
	print "Couldn't detect CPU architecture"
	ARCH = ""
if platform.system() == "Linux":
	print "Linux operating system detected"
	OS = "linux"
elif platform.system() == "Darwin" :
	print "Darwin / OS X system detected"
	OS = "osx"
elif platform.system() == "FreeBSD" :
	print "FreeBSD operating system detected"
	OS = "freebsd"
elif platform.system() == "Windows":
	print "Windows system detected"
	OS = "win"
elif os.name == "posix":
	print "Unix system detected"
	OS = "nix"
else:
	print "Couldn't detect operating system"
	OS = ""
import os.path, glob
from distutils.core import setup, Extension
from distutils import sysconfig
import shutil
from distutils import cmd
from distutils.command.install import install as _install
from distutils.command.install_data import install_data as _install_data
from distutils.command.build import build as _build
import msgfmt

python_version = platform.python_version()[0:3]

NAME		= "deluge"
FULLNAME	= "Deluge BitTorrent Client"
VERSION		= "0.5.0.99.2"
AUTHOR		= "Zach Tibbitts, Alon Zakai, Marcos Pinto"
EMAIL		= "zach@collegegeek.org, kripkensteiner@gmail.com, marcospinto@dipconsultants.com"
DESCRIPTION	= "A bittorrent client written in PyGTK"
URL		= "http://deluge-torrent.org"
LICENSE		= "GPLv2"

EXTRA_COMPILE_ARGS = ["-Wno-missing-braces"]
if ARCH == "x64":
	EXTRA_COMPILE_ARGS.append("-DAMD64")


# NOTE: The following "hack" removes the -g and -Wstrict-prototypes
# build options from the command that will compile the C++ module,
# deluge_core.  While we understand that you aren't generally
# encouraged to do this, we have done so for the following reasons:
# 1) The -g compiler option produces debugging information about
#	the compiled module.  However, this option increases the 
#	size of deluge_core.so from ~1.9MB to 13.6MB and slows down
#	the program's execution without offering any benefits 
#	whatsoever.
# 2) -Wstrict-prototypes is not a valid C++ build option, and the
#	compiler will throw a number of warnings at compile time.
#	While this does not really impact anything, it makes it
#	seem as if something is going wrong with the compile, and
#	it has been removed to prevent confusion.

if not OS == "win":
	if OS == "linux":
		if os.WEXITSTATUS(os.system('grep -iq "Debian GNU/Linux 4.0\|Ubuntu 7.04\|Ubuntu 6.06\|Fedora Core release 6\|openSUSE 10.2\|Mandriva Linux release 2007.1" /etc/issue')) == 0:
			boosttype = 'nomt'
		else:
			boosttype = 'mt'
	elif OS == "freebsd":
		boosttype = 'nomt'
	else:
		boosttype = 'mt'
	removals = ['-g', '-DNDEBUG', '-O2', '-Wstrict-prototypes']
        additions = ['-DNDEBUG', '-O2']

	if python_version == '2.5':
		cv_opt = sysconfig.get_config_vars()["CFLAGS"]
		for removal in removals:
			cv_opt = cv_opt.replace(removal, " ")
		for addition in additions:
			cv_opt = cv_opt + " " + addition
		sysconfig.get_config_vars()["CFLAGS"] = ' '.join(cv_opt.split())
	else:
		cv_opt = sysconfig.get_config_vars()["OPT"]
		for removal in removals:
			cv_opt = cv_opt.replace(removal, " ")
		for addition in additions:
			cv_opt = cv_opt + " " + addition
		sysconfig.get_config_vars()["OPT"] = ' '.join(cv_opt.split())
else:
	boosttype = 'mt'

# NOTE: The Rasterbar Libtorrent source code is in the libtorrent/ directory
# inside of Deluge's source tarball.  On several occasions, it has been 
# pointed out to us that we should build against the system's installed 
# libtorrent rather than our internal copy, and a few people even submitted
# patches to do just that. However, as of now, this version
# of libtorrent is not available in Debian, and as a result, Ubuntu. Once
# libtorrent-rasterbar is available in the repositories of these distributions,
# we will probably begin to build against a system libtorrent, but at the
# moment, we are including the source code to make packaging on Debian and
# Ubuntu possible.
if boosttype == "nomt":
	librariestype = ['boost_filesystem', 'boost_date_time',
			'boost_thread', 'z', 'pthread', 'ssl']
	print 'Libraries nomt' 
elif boosttype == "mt":
	librariestype = ['boost_filesystem-mt', 'boost_date_time-mt',
			'boost_thread-mt', 'z', 'pthread', 'ssl']
	print 'Libraries mt'

deluge_core = Extension('deluge_core',
                    include_dirs = ['./libtorrent', './libtorrent/include', 
                    			'./libtorrent/include/libtorrent', 
                    			'/usr/include/python' + python_version],
		    libraries = librariestype,
                    extra_compile_args = EXTRA_COMPILE_ARGS,
                    sources = ['src/deluge_core.cpp',
					'libtorrent/src/alert.cpp',
					'libtorrent/src/allocate_resources.cpp',
					'libtorrent/src/bandwidth_manager.cpp',
					'libtorrent/src/bt_peer_connection.cpp',
					'libtorrent/src/connection_queue.cpp',
					'libtorrent/src/entry.cpp',
					'libtorrent/src/escape_string.cpp',
					'libtorrent/src/file.cpp',
					'libtorrent/src/file_pool.cpp',
					'libtorrent/src/http_connection.cpp',
					'libtorrent/src/http_stream.cpp',
					'libtorrent/src/http_tracker_connection.cpp',
					'libtorrent/src/identify_client.cpp',
					'libtorrent/src/instantiate_connection.cpp',
					'libtorrent/src/ip_filter.cpp',
					'libtorrent/src/logger.cpp',
					'libtorrent/src/lsd.cpp',
					'libtorrent/src/metadata_transfer.cpp',
					'libtorrent/src/natpmp.cpp',
					'libtorrent/src/pe_crypto.cpp',
					'libtorrent/src/peer_connection.cpp',
					'libtorrent/src/piece_picker.cpp',     
					'libtorrent/src/policy.cpp',       
					'libtorrent/src/session.cpp',   
					'libtorrent/src/session_impl.cpp',
					'libtorrent/src/sha1.cpp',
					'libtorrent/src/socks5_stream.cpp',
					'libtorrent/src/stat.cpp',
					'libtorrent/src/storage.cpp',
					'libtorrent/src/torrent.cpp',
					'libtorrent/src/torrent_handle.cpp',
					'libtorrent/src/torrent_info.cpp',
					'libtorrent/src/tracker_manager.cpp',
					'libtorrent/src/udp_tracker_connection.cpp',
					'libtorrent/src/upnp.cpp',
					'libtorrent/src/ut_pex.cpp',
					'libtorrent/src/web_peer_connection.cpp',
						'libtorrent/src/kademlia/closest_nodes.cpp',
						'libtorrent/src/kademlia/dht_tracker.cpp',
						'libtorrent/src/kademlia/find_data.cpp',
						'libtorrent/src/kademlia/node.cpp',
						'libtorrent/src/kademlia/node_id.cpp',
						'libtorrent/src/kademlia/refresh.cpp',
						'libtorrent/src/kademlia/routing_table.cpp',
						'libtorrent/src/kademlia/rpc_manager.cpp',
						'libtorrent/src/kademlia/traversal_algorithm.cpp'])

# Thanks to Iain Nicol for code to save the location for installed prefix
# At runtime, we need to know where we installed the data to.

class write_data_install_path(cmd.Command):
	description = 'saves the data installation path for access at runtime'
	
	def initialize_options(self):
		self.prefix = None
		self.lib_build_dir = None

	def finalize_options(self):
		self.set_undefined_options('install',
			('prefix', 'prefix')
		)
		self.set_undefined_options('build',
			('build_lib', 'lib_build_dir')
		)

	def run(self):
		conf_filename = os.path.join(self.lib_build_dir,
			'deluge', 'common.py')

		conf_file = open(conf_filename, 'r')
		data = conf_file.read()
		conf_file.close()
		data = data.replace('@datadir@', self.prefix)

		conf_file = open(conf_filename, 'w')
		conf_file.write(data)
		conf_file.close()

class unwrite_data_install_path(cmd.Command):
	description = 'undoes write_data_install_path'

	def initialize_options(self):
		self.lib_build_dir = None

	def finalize_options(self):		
		self.set_undefined_options('build',
			('build_lib', 'lib_build_dir')
		)

	def run(self):
		dest = os.path.join(self.lib_build_dir,
			'deluge', 'common.py')
		shutil.copyfile('src/common.py', dest)

class build_trans(cmd.Command):
	description = 'Compile .po files into .mo files'
	
	def initialize_options(self):
		pass

	def finalize_options(self):		
		pass

	def run(self):
		po_dir = os.path.join(os.path.dirname(__file__), 'po')
		for path, names, filenames in os.walk(po_dir):
			for f in filenames:
				if f.endswith('.po'):
					lang = f[:len(f) - 3]
					src = os.path.join(path, f)
					dest_path = os.path.join('build', 'locale', lang, 'LC_MESSAGES')
					dest = os.path.join(dest_path, 'deluge.mo')
					if not os.path.exists(dest_path):
						os.makedirs(dest_path)
					if not os.path.exists(dest):
						print 'Compiling %s' % src
						msgfmt.make(src, dest)
					else:
						src_mtime = os.stat(src)[8]
						dest_mtime = os.stat(dest)[8]
						if src_mtime > dest_mtime:
							print 'Compiling %s' % src
							msgfmt.make(src, dest)

class build(_build):
	sub_commands = _build.sub_commands + [('build_trans', None)]
	def run(self):
		_build.run(self)

class install(_install):
	sub_commands = [('write_data_install_path', None)] + \
		_install.sub_commands + [('unwrite_data_install_path', None)]
	def run(self):
		_install.run(self)

class install_data(_install_data):
	def run(self):
		for lang in os.listdir('build/locale/'):
			lang_dir = os.path.join('share', 'locale', lang, 'LC_MESSAGES')
			lang_file = os.path.join('build', 'locale', lang, 'LC_MESSAGES', 'deluge.mo')
			self.data_files.append( (lang_dir, [lang_file]) )
		_install_data.run(self)

cmdclass = {
	'build': build,
	'install': install,
	'build_trans': build_trans,
	'install_data': install_data,
	'write_data_install_path': write_data_install_path,
	'unwrite_data_install_path': unwrite_data_install_path,
}

data = [('share/deluge/glade',  glob.glob('glade/*.glade')),
        ('share/deluge/pixmaps', glob.glob('pixmaps/*.png')),
        ('share/applications' , ['deluge.desktop']),
        ('share/pixmaps' , ['deluge.xpm'])]

for plugin in glob.glob('plugins/*'):
	data.append( ('share/deluge/' + plugin, glob.glob(plugin + '/*')) )

setup(name=NAME, fullname=FULLNAME, version=VERSION,
	author=AUTHOR, author_email=EMAIL, description=DESCRIPTION,
	url=URL, license=LICENSE,
	scripts=["scripts/deluge"],
	packages=['deluge'],
	package_dir = {'deluge': 'src'},
	data_files=data,
	ext_package='deluge',
	ext_modules=[deluge_core],
	cmdclass=cmdclass
)