#!/bin/bash

# $1 Path to store built package

packagesidentity="org.chameleon"

packagename="Chameleon"

pkgroot="${0%/*}"

#version=$( grep I386BOOT_CHAMELEONVERSION sym/i386/vers.h | awk '{ print $3 }' | tr -d '\"' )
version=$( cat version )
stage=${version##*-}
revision=$( grep I386BOOT_CHAMELEONREVISION sym/i386/vers.h | awk '{ print $3 }' | tr -d '\"' )
builddate=$( grep I386BOOT_BUILDDATE sym/i386/vers.h | awk '{ print $3,$4 }' | tr -d '\"' )
timestamp=$( date -j -f "%Y-%m-%d %H:%M:%S" "${builddate}" "+%s" )

distributioncount=0
xmlindent=0

indent[0]="\t"
indent[1]="\t\t"
indent[2]="\t\t\t"
indent[3]="\t\t\t\t"

main ()
{

# clean up the destination path

	rm -R -f "${1}"
	
	echo "Building $packagename Install Package v${version%%-*} $stage r$revision $builddate"
	
	outline[$((outlinecount++))]="${indent[$xmlindent]}<choices-outline>"

# Start



# build Chameleon metapackages 
	outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"Chameleon\">"
	choices[$((choicescount++))]="<choice\n\tid=\"Chameleon\"\n\ttitle=\"Chameleon_title\"\n\tdescription=\"Chameleon_description\"\n>\n</choice>\n"
	((xmlindent++))

	# build core package
		mkdir -p ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/boot ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/boot0 ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/boot0hfs ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/boot1f32 ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/boot1h ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/boot1he ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/boot1hp ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/cdboot ${1}/Core/Root/i386
		cp -f ${1%/*}/i386/chain0 ${1}/Core/Root/i386
		cp -f ${pkgroot}/fdisk440 ${1}/Core/Root/i386
		fixperms "${1}/Core/Root/i386"
		local coresize=$( du -hkc "${1}/Core/Root/i386" | tail -n1 | awk {'print $1'} )
		echo "Building core package"
		buildpackage "${1}/Core" "/binaries" "0" "start_visible=\"false\" start_selected=\"true\""

	# End build core package

	# build standard package 
		mkdir -p ${1}/boot0/Root
		mkdir -p ${1}/boot0/Scripts
		cp -f ${pkgroot}/Scripts/boot0/postinstall ${1}/boot0/Scripts
		echo "Building boot0 package"
		buildpackage "${1}/boot0" "/binaries" "${coresize}" "start_enabled=\"true\" start_selected=\"false\" selected=\"exclusive(choices['EnhancedHFS']) &amp;&amp; exclusive(choices['boot0hfs']) &amp;&amp; exclusive(choices['EnhancedFAT']) &amp;&amp; exclusive(choices['noboot'])\""
	# End build standard package

	# build standard package with boot0hfs
		mkdir -p ${1}/boot0hfs/Root
		mkdir -p ${1}/boot0hfs/Scripts
		cp -f ${pkgroot}/Scripts/boot0hfs/postinstall ${1}/boot0hfs/Scripts
		echo "Building boot0hfs package"
		buildpackage "${1}/boot0hfs" "/binaries" "${coresize}" "start_enabled=\"true\" start_selected=\"false\" selected=\"exclusive(choices['EnhancedHFS']) &amp;&amp; exclusive(choices['boot0']) &amp;&amp; exclusive(choices['EnhancedFAT']) &amp;&amp; exclusive(choices['noboot'])\""
	# End build standard package with boot0hfs

	# build efi fat32 package 
		mkdir -p ${1}/EnhancedFAT/Root
		mkdir -p ${1}/EnhancedFAT/Scripts/Tools
		cp -f ${pkgroot}/Scripts/FAT/postinstall ${1}/EnhancedFAT/Scripts
		ditto --arch i386 `which SetFile` ${1}/EnhancedFAT/Scripts/Tools/SetFile
		echo "Building EnhacedFAT package"
		buildpackage "${1}/EnhancedFAT" "/binaries" "${coresize}" "start_visible=\"systemHasGPT()\" start_selected=\"false\" selected=\"exclusive(choices['boot0']) &amp;&amp; exclusive(choices['boot0hfs']) &amp;&amp; exclusive(choices['EnhancedHFS']) &amp;&amp; exclusive(choices['noboot'])\""
	# End build efi fat32 package 

	# build efi hfs package 
		mkdir -p ${1}/EnhancedHFS/Root
		mkdir -p ${1}/EnhancedHFS/Scripts/Tools
		cp -f ${pkgroot}/Scripts/HFS/postinstall ${1}/EnhancedHFS/Scripts
		ditto --arch i386 `which SetFile` ${1}/EnhancedHFS/Scripts/Tools/SetFile
		echo "Building EnhacedHFS package"
		buildpackage "${1}/EnhancedHFS" "/binaries" "${coresize}" "start_visible=\"systemHasGPT()\" start_selected=\"false\" selected=\"exclusive(choices['boot0']) &amp;&amp; exclusive(choices['boot0hfs']) &amp;&amp; exclusive(choices['EnhancedFAT']) &amp;&amp; exclusive(choices['noboot'])\""
	# End build efi hfs package 

	# build reset choice package 
		mkdir -p ${1}/noboot/Root
		echo "Building reset choice package"
		buildpackage "${1}/noboot" "/binaries" "${coresize}" "start_visible=\"true\" start_selected=\"true\" selected=\"exclusive(choices['boot0']) &amp;&amp; exclusive(choices['boot0hfs']) &amp;&amp; exclusive(choices['EnhancedFAT']) &amp;&amp; exclusive(choices['EnhancedHFS'])\""
	# End build reset choice package 

	((xmlindent--))
 	outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
# End Chameleon metapakage

# build utility package
	outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"Utility\">"
	choices[$((choicescount++))]="<choice\n\tid=\"Utility\"\n\ttitle=\"Utility_title\"\n\tdescription=\"Utility_description\"\n>\n</choice>\n"
	((xmlindent++))
	packagesidentity="org.chameleon"

	# build package for bdmesg
		mkdir -p "${1}/bdmesg/Root"
		ditto --noextattr --noqtn "${1%/*}/i386/bdmesg" "${1}/bdmesg/Root"
		echo "Building bdmesg package"
		buildpackage "${1}/bdmesg" "/Extra/Util" "" "start_selected=\"false\""
	# End build package for bdmesg

	# build package for Chameleon PrefPanel
		mkdir -p "${1}/prefPane/Root/"
		ditto --noextattr --noqtn "${pkgroot}/Configuration" "${1}/prefPane/Root"
		echo "Building Chameleon Preference Panel package"
		buildpackage "${1}/prefPane" "/Library/PreferencePanes" "" "start_selected=\"false\""
	# End build package for Chameleon PrefPanel
		
	((xmlindent--))
	outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
# End utility package

# build options packages

	outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"Options\">"
	choices[$((choicescount++))]="<choice\n\tid=\"Options\"\n\ttitle=\"Options_title\"\n\tdescription=\"Options_description\"\n>\n</choice>\n"
	((xmlindent++))
	packagesidentity="org.chameleon.options"
	options=($( find "${pkgroot}/Scripts/Options" -type d -depth 1 -not -name '.svn' ))
	for (( i = 0 ; i < ${#options[@]} ; i++)) 
	do
		packagesidentity="org.chameleon.flags"
		flagname=($( find "${options[$i]}" -type d -depth 1 -not -name '.svn' ))
		outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"${options[$i]##*/}\">"
		choices[$((choicescount++))]="<choice\n\tid=\"${options[$i]##*/}\"\n\ttitle=\"${options[$i]##*/}\"\n\tdescription=\"${options[$i]##*/}_description\"\n>\n</choice>\n"
		((xmlindent++))
		for (( j = 0 ; j < ${#flagname[@]} ; j++ )) 
			do
				outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"${options[$i]##*/}_${flagname[$j]##*/}\">"
				choices[$((choicescount++))]="<choice\n\tid=\"${options[$i]##*/}_${flagname[$j]##*/}\"\n\ttitle=\"${options[$i]##*/}_${flagname[$j]##*/}\"\n\tdescription=\"${options[$i]##*/}_${flagname[$j]##*/}_description\"\n>\n</choice>\n"
				((xmlindent++))
						mkdir -p "${1}/${flagname[$j]##*/}/Root"
						mkdir -p "${1}/${flagname[$j]##*/}/Scripts"
						ditto --noextattr --noqtn "${options[$i]}/${flagname[$j]##*/}/postinstall" "${1}/${options[$i]##*/}${flagname[$j]##*/}/Scripts/postinstall"
					#	cp -f "${options[$i]}/${flagname[$j]##*/}/*" "${1}/${options[$i]##*/}${flagname[$j]##*/}/Scripts/postinstall"
					#	echo "Building ${options[$i]##*/}${flagname[$j]##*/} package"
						buildpackage "${1}/${options[$i]##*/}_${flagname[$j]##*/}" "/" "${coresize}" "start_selected=\"false\""
						rm -R -f "${1}/${j##*/}"
				((xmlindent--))
				outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
		done
		((xmlindent--))
		outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
		echo "Building ${options[$i]##*/} package"
		buildpackage "${1}/${options[$i]##*/}" "/" "${coresize}" "start_selected=\"false\""
		rm -R -f "${1}/${i##*/}"
	done
	((xmlindent--))
	outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
# End build options packages

# build theme packages

	outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"Themes\">"
	choices[$((choicescount++))]="<choice\n\tid=\"Themes\"\n\ttitle=\"Themes_title\"\n\tdescription=\"Themes_description\"\n>\n</choice>\n"
	((xmlindent++))
		packagesidentity="org.chameleon.theme"
		artwork="${1%/*}"
		themes=($( find "${artwork%/*}/artwork/themes" -type d -depth 1 -not -name '.svn' ))
		for (( i = 0 ; i < ${#themes[@]} ; i++ )) 
		do
			theme=$( echo ${themes[$i]##*/} | awk 'BEGIN{OFS=FS=""}{$1=toupper($1);print}' )
			mkdir -p "${1}/${theme}/Root/"
			ditto --noextattr --noqtn "${themes[$i]}" "${1}/${themes[$i]##*/}/Root/${theme}"
			find "${1}/${themes[$i]##*/}" -name '.DS_Store' -or -name '.svn' -exec rm -R {} \+
			find "${1}/${themes[$i]##*/}" -type f -exec chmod 644 {} \+
			buildpackage "${1}/${theme}" "/Extra/Themes" "" "start_selected=\"false\""
			rm -R -f "${1}/${i##*/}"
		done
	((xmlindent--))
	outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
# End build theme packages



# build kext packages 
	outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"Extras\">"
	choices[$((choicescount++))]="<choice\n\tid=\"Extras\"\n\ttitle=\"Extras_title\"\n\tdescription=\"Extras_description\"\n>\n</choice>\n"
	((xmlindent++))
		outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"Kexts\">"
		choices[$((choicescount++))]="<choice\n\tid=\"Kexts\"\n\ttitle=\"Kexts_title\"\n\tdescription=\"Kexts_description\"\n>\n</choice>\n"
		((xmlindent++))
			packagesidentity="org.chameleon"
			kexts=($( find "${pkgroot}/Kexts" -type d -name '*.kext' -depth 1 ))
			for (( i = 0 ; i < ${#kexts[@]} ; i++ )) 
			do
				filename="${kexts[$i]##*/}"
				mkdir -p "${1}/${filename%.kext}/Root/"
				ditto --noextattr --noqtn --arch i386 "${kexts[$i]}" "${1}/${filename%.kext}/Root/${filename}"
				find "${1}/${filename%.kext}" -name '.DS_Store' -or -name '.svn' -exec rm -R -f {} \; 2>/dev/null
				fixperms "${1}/${filename%.kext}/Root/"
				chown 501:20 "${1}/${filename%.kext}/Root/"
				buildpackage "${1}/${filename%.kext}" "/Extra/Extensions" "" "start_selected=\"false\""
				rm -R -f "${1}/${filename%.kext}"
			done
		((xmlindent--))
		outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
	((xmlindent--))
	outline[$((outlinecount++))]="${indent[$xmlindent]}\t</line>"
# End build kext packages

# build post install package 
	mkdir -p ${1}/Post/Root
	mkdir -p ${1}/Post/Scripts
	cp -f ${pkgroot}/Scripts/Post/* ${1}/Post/Scripts
	buildpackage "${1}/Post" "/binaries" "" "start_visible=\"false\" start_selected=\"true\""
# End post install package 	

	outline[$((outlinecount++))]="${indent[$xmlindent]}</choices-outline>"
# End main package


# build meta package

	makedistribution "${1}" "${2}" "${3}" "${4}" "${5}"

# clean up 

	rm -R -f "${1}"

}

fixperms ()
{
	# $1 path
	find "${1}" -type f -exec chmod 644 {} \;
	find "${1}" -type d -exec chmod 755 {} \;
	chown -R 0:0 "${1}"
}

buildpackage ()
{
#  $1 Path to package to build containing Root and or Scripts
#  $2 Install Location
#  $3 Size
#  $4 Options

if [ -d "${1}/Root" ] && [ "${1}/Scripts" ]; then

	local packagename="${1##*/}"
	local identifier=$( echo ${packagesidentity}.${packagename//_/.} | tr [:upper:] [:lower:] )
	find "${1}" -name '.DS_Store' -delete
	local filecount=$( find "${1}/Root" | wc -l )
	if [ "${3}" ]; then
		local installedsize="${3}"
	else
		local installedsize=$( du -hkc "${1}/Root" | tail -n1 | awk {'print $1'} )
	fi
	local header="<?xml version=\"1.0\"?>\n<pkg-info format-version=\"2\" "

	#[ "${3}" == "relocatable" ] && header+="relocatable=\"true\" "		

	header+="identifier=\"${identifier}\" "
	header+="version=\"${version}\" "

	[ "${2}" != "relocatable" ] && header+="install-location=\"${2}\" "

	header+="auth=\"root\">\n"
	header+="\t<payload installKBytes=\"${installedsize##* }\" numberOfFiles=\"${filecount##* }\"/>\n"
	rm -R -f "${1}/Temp"

	[ -d "${1}/Temp" ] || mkdir -m 777 "${1}/Temp"
	[ -d "${1}/Root" ] && mkbom "${1}/Root" "${1}/Temp/Bom"

	if [ -d "${1}/Scripts" ]; then 
		header+="\t<scripts>\n"
		for script in $( find "${1}/Scripts" -type f \( -name 'pre*' -or -name 'post*' \) )
		do
			header+="\t\t<${script##*/} file=\"./${script##*/}\"/>\n"
		done
		header+="\t</scripts>\n"
		chown -R 0:0 "${1}/Scripts"
		pushd "${1}/Scripts" >/dev/null
		find . -print | cpio -o -z -H cpio > "../Temp/Scripts"
		popd >/dev/null
	fi

	header+="</pkg-info>"
	echo -e "${header}" > "${1}/Temp/PackageInfo"
	pushd "${1}/Root" >/dev/null
	find . -print | cpio -o -z -H cpio > "../Temp/Payload"
	popd >/dev/null
	pushd "${1}/Temp" >/dev/null

	xar -c -f "${1%/*}/${packagename// /}.pkg" --compression none .

	popd >/dev/null

	outline[$((outlinecount++))]="${indent[$xmlindent]}\t<line choice=\"${packagename// /}\"/>"

	if [ "${4}" ]; then
		local choiceoptions="${indent[$xmlindent]}${4}\n"	
	fi
	choices[$((choicescount++))]="<choice\n\tid=\"${packagename// /}\"\n\ttitle=\"${packagename}_title\"\n\tdescription=\"${packagename}_description\"\n${choiceoptions}>\n\t<pkg-ref id=\"${identifier}\" installKBytes='${installedsize}' version='${version}.0.0.${timestamp}' auth='root'>#${packagename// /}.pkg</pkg-ref>\n</choice>\n"

	rm -R -f "${1}"
fi
}

makedistribution ()
{
	rm -f "${1%/*}/${packagename// /}"*.pkg

	find "${1}" -type f -name '*.pkg' -depth 1 | while read component
	do
		mkdir -p "${1}/${packagename}/${component##*/}"
		pushd "${1}/${packagename}/${component##*/}" >/dev/null
		xar -x -f "${1%}/${component##*/}"
		popd >/dev/null
	done

	ditto --noextattr --noqtn "${pkgroot}/Distribution" "${1}/${packagename}/Distribution"
	ditto --noextattr --noqtn "${pkgroot}/Resources" "${1}/${packagename}/Resources"

	find "${1}/${packagename}/Resources" -type d -name '.svn' -exec rm -R -f {} \; 2>/dev/null

	for (( i=0; i < ${#outline[*]} ; i++));
		do
			echo -e "${outline[$i]}" >> "${1}/${packagename}/Distribution"
		done

	for (( i=0; i < ${#choices[*]} ; i++));
		do
			echo -e "${choices[$i]}" >> "${1}/${packagename}/Distribution"
		done

	echo "</installer-gui-script>"  >> "${1}/${packagename}/Distribution"

	perl -i -p -e "s/%CHAMELEONVERSION%/${version%%-*}/g" `find "${1}/${packagename}/Resources" -type f`
	perl -i -p -e "s/%CHAMELEONREVISION%/${revision}/g" `find "${1}/${packagename}/Resources" -type f`

	stage=${stage/RC/Release Candidate }
	stage=${stage/FINAL/2.0 Final}
	perl -i -p -e "s/%CHAMELEONSTAGE%/${stage}/g" `find "${1}/${packagename}/Resources" -type f`

	find "${1}/${packagename}" -name '.DS_Store' -delete
	pushd "${1}/${packagename}" >/dev/null
	xar -c -f "${1%/*}/${packagename// /}-${version}-r${revision}.pkg" --compression none .
	popd >/dev/null

	md5=$( md5 "${1%/*}/${packagename// /}-${version}-r${revision}.pkg" | awk {'print $4'} )
	echo "MD5 (${packagename// /}-${version}-r${revision}.pkg) = ${md5}" > "${1%/*}/${packagename// /}-${version}-r${revision}.pkg.md5"
   cp -R -f "${1}/${packagename}/Resources" ${1%/*}/Resources
}

main "${1}" "${2}" "${3}" "${4}" "${5}"