'''
Download appropriate CMTK binary archive and install to fiji bin/cmtk directory
'''

import urllib2,os,sys,tarfile,shutil,tempfile

from fiji.util.gui import GenericDialogPlus

from java.lang.System import getProperty
sys.path.append(getProperty("fiji.dir") + "/plugins/CMTK_Registration")
import cmtkgui

def bin_files(members):
	'''
	Quick utility function to select binary files to install from CMTK Archive.
	'''
	for tarinfo in members:
		lastdir=os.path.basename(os.path.dirname(tarinfo.name))
		if lastdir == "bin":
#			print 'keeping '+tarinfo.name
			yield tarinfo

def license_files(members):
	'''
	Quick utility function to select license files to install from CMTK Archive.
	Do this by copying files from doc 
	'''
	for tarinfo in members:
		if re.search(os.path.join(os.path.sep,"doc",""),tarinfo.name):
			yield tarinfo

def untar_binaries(f,target_dir):
	'''
	untar binary files in a subfolder bin of tarfile f
	moving them from a temporary location to a final location in target_dir
	target_dir is created if it does not already exist
	'''
	tar=tarfile.open(f)
	td=tempfile.mkdtemp()

	cmtkgui.log("Extracting tar file")
	tar.extractall(path=td,members=bin_files(tar))
	# now move each binary to where we actually wanted it to go
	binaries=bin_files(tar)
	cmtkgui.log("Installing binaries to "+target_dir)
	if not os.path.isdir(target_dir):
		os.makedirs(target_dir)
	for binary in binaries:
		frompath = os.path.join(td,binary.name)
		cmtkgui.log('installing from = '+frompath+' to = '+ target_dir)
		cmtkgui.movefile(frompath,target_dir)
	print("Cleaning up!")
	shutil.rmtree(td)

download_urls=cmtkgui.downloads()
# download_urls=['http://www.nitrc.org/frs/download.php/4814/CMTK-2.2.3-CYGWIN-i686.tar.gz', 'http://www.nitrc.org/frs/download.php/4812/CMTK-2.2.3-Linux-x86_64.tar.gz', 'http://www.nitrc.org/frs/download.php/4820/CMTK-2.2.3-MacOSX-10.4-i686.tar.gz', 'http://www.nitrc.org/frs/download.php/4822/CMTK-2.2.3-MacOSX-10.5-x86_64.tar.gz', 'http://www.nitrc.org/frs/download.php/4824/CMTK-2.2.3-MacOSX-10.6-x86_64.tar.gz', 'http://www.nitrc.org/frs/download.php/4604/CMTK-2.2.1-CYGWIN-i686.tar.gz', 'http://www.nitrc.org/frs/download.php/4596/CMTK-2.2.1-Linux-x86_64.tar.gz', 'http://www.nitrc.org/frs/download.php/4608/CMTK-2.2.1-MacOSX-10.4-i686.tar.gz', 'http://www.nitrc.org/frs/download.php/4610/CMTK-2.2.1-MacOSX-10.5-x86_64.tar.gz', 'http://www.nitrc.org/frs/download.php/4611/CMTK-2.2.1-MacOSX-10.6-x86_64.tar.gz']
download_files=map(os.path.basename,download_urls)
download_dict=dict(zip(download_files,download_urls))
print cmtkgui.install_dir()

gd = GenericDialogPlus('Install CMTK')
gd.addMessage('Currently installed CMTK version: '+cmtkgui.installed_version())
recommended_file=cmtkgui.recommended_file(download_files)
if recommended_file is None:
	recommended_file = download_files[0]
gd.addChoice("Download file",download_files,recommended_file)
gd.addMessage('By downloading this file you agree to the following (if you '+
	'do not agree to this, please press "Cancel" below):')
gd.addMessage('Core CMTK code is licensed under the GPLv3.\nBundled software'+
	' may be licensed under different terms - see licences/ directory for details')
gd.showDialog()
if gd.wasOKed():
	download_file=gd.getNextChoice()
	# nb url has a suffix to indicate that user agreed to license
	download_url=download_dict[download_file]+'/?i_agree=1&download_now=1'
	print "Downloading "+download_file+' from url '+download_url+' to '+cmtkgui.install_dir()
	cmtkgui.download_and_untar_url(download_url,cmtkgui.install_dir(),untar_binaries,download_file)