#!/bin/sh
#
# author: hackbard@hackdaworld.dyndns.org
#
# hdw-tools: 
# this is hdw-get script, used to install/remove/(update) binary packages
# 

usage()	{
	echo
	echo "usage:"
	echo "$0 [install/remove] [root] [package]"
	echo "$0 [dist-install] [root] [dist file]"
	echo
	echo "options:"
	echo "-v	tell me whats going on"
	echo "-a	auto resolve deps, do not prompt user"
	echo "-no-dep	do not care for dependencies (used by hdw-get itself)"
	echo "-s	simulation only, do not install/remove packages"
	echo
}

package="" ; inst_root=""
update=""
d_install="" ; install=""
verbose="" ; no_dep=""
auto_resolve_deps=""
simulate=""

FL_DIR="var/adm/flists"
DEPS_DIR="var/adm/deps/run"
CONF="/etc/hdw-get.conf"
DBF="/etc/hdw-get.db"

while [ "$1" ] ; do
	case "$1" in
		update)		update="1" ; package="fake" ; inst_root="fake"
				shift 1 ;;
		install)	install="1" 
				inst_root="$2" ; package="$3" ; shift 3 ;;
		remove)		remove="1" 
				inst_root="$2" ; package="$3" ; shift 3 ;;
		dist-install)	d_install="1"  ; package="fake"
				inst_root="$2" ; dist_file="$3" ; shift 3 ;;
		-v)		verbose=1 ; shift 1 ;;
		-a)		auto_resolve_deps="1" ; shift 1 ;;
		-no-dep)	no_dep="1" ; shift 1 ;;
		-s)		simulate="1" ; shift 1 ;;
		*)		usage ; exit 1 ;;
	esac
done
			
if [ -z "inst_root" -o -z "$package" ] ; then
	usage ; exit 1
fi
	
# read hdw-get configs
net_addr="" ; bin_dir="" ; distf_dir=""
if [ -f $CONF ] ; then
	. $CONF
	if [ -z "$hdw_version" -o -z "$hdw_arch" -o -z "$hdw_arch_opt" ] ; then
		echo "version, arch and arch_opt need to be specified."
		echo "adjust $CONF file"
		exit
	else
		[ "$verbose" ] && echo "hdw-get.conf file looks good ..."
	fi
else
	echo "$CONF file not found, aborting"
	exit 1
fi

if [ "$update" = "1" ] ; then
	echo -en "updating hdw-get.db file ..."
	wget -q $net_addr/hdw-linux/hdw-linux-$hdw_version/hdw-get/hdw-get.db \
		-O $DBF
	echo -en " done. (`cat $DBF | wc -l` packages)\n"
fi

# create temp directory
mkdir -p /tmp/.hdw-get

# remove
if [ "$remove" = "1" ] ; then
	# check if packages depend on package to be removed
	pkgs=""
	((isdep = 0))
	for fl_file in $inst_root/$DEPS_DIR/*; do
		if [ "`grep "^$package$" $fl_file`" != "" ] ; then
			pkg=`echo $fl_file | sed "s%$inst_root/$DEPS_DIR/%%"`
			if [ "$pkg" != "$package" ] ; then
				((isdep += 1))
				pkgs="$pkg $pkgs"
			fi
		fi
	done
	if [ "$isdep" != "0" ] ; then
		echo "info:"
		echo "run time dependencies were autegenerated by build system."
		echo "they may be wrong. edit $inst_root/$DEPS_DIR/* files."
		echo "deps:"
		echo -en "$pkgs\n"
		echo "package $package should not be removed ($isdep deps)"
		[ "$no_dep" != "1" ] && exit 1
	fi
	# looking for flist file
	fl_pris="unknown"
	for fl_file in $inst_root/$FL_DIR/$package-[x,0-9]*; do
		[ -f $fl_file ] && \
		fl_pris="`echo $fl_file | awk -F- '{ print $NF }'` $fl_pris"
	done
	fl_max="0"
	for pri in $fl_pris; do
		if [ "$pri" = "x" ] ; then
			fl_max="x"
			break
		fi
		[ "$pri" = "unknown" ] && break
		[ "$pri" -gt "$fl_max" ] && fl_max=$pri
	done
	fl_file="$inst_root/$FL_DIR/$package-$fl_max"
	[ "$verbose" ] && echo "expecting flist file $fl_file"
	if [ ! -f $fl_file ] ; then
		echo "package $package is not installed (moved the flist file?)"
		exit 1
	fi

	if [ "$simulate" != "1" ] ; then

	# care about info/dir file
	rm -f /tmp/.hdw-get/dir* ; modify_info_dir=""
	if [ -n "`grep usr/share/info/dir $fl_file`" ] ; then
	modify_info_dir="1"
	cp $inst_root/usr/share/info/dir /tmp/.hdw-get/
	for info_obj in `grep 'usr/share/info/.*.info' $fl_file | \
			 awk -F/ '{ print $4 }'`; do
		info_o=`echo $info_obj | sed 's/.info//'`
		[ "$verbose" ] && echo "deleting $info_o from info/dir"
		grep -v \($info_o\)\. /tmp/.hdw-get/dir | \
		grep -v ^$package\ info\ -\ added\ by\ hdw > \
		/tmp/.hdw-get/dir~
		mv /tmp/.hdw-get/dir~ /tmp/.hdw-get/dir
	done
	fi

	# remove the files/dirs
	((supposed = 0))
	((counter = 0))
	cat $fl_file | while read pkg file; do
		((supposed += 1))
		if [ -e $inst_root/$file ] ; then
			if [ "$file" != "tmp" ] ; then
			if [ -d $inst_root/$file ] ; then
				rmdir -p $inst_root/$file > /dev/null 2>&1
			else
				rm -f $inst_root/$file > /dev/null 2>&1
				rmdir -p `dirname $file` > /dev/null 2>&1
			fi
			((counter += 1))
			fi
		fi
		echo -en "\rremoved $counter from $supposed supposed files"
	done
	echo -en "\ndone.\n"

	# reinstall info/dir file
	if [ "$modify_info_dir" ] ; then
		[ "$verbose" ] && echo "restoring modified info/dir file"
		cp /tmp/.hdw-get/dir $inst_root/usr/share/info/dir
		rm -f /tmp/.hdw-get/dir*
	fi

	fi

# install
elif [ "$install" = "1" ] ; then
	# does inst_root exist
	if [ ! -d $inst_root ] ; then
		echo
		echo "the install directory doesnt exist, create it first"
		echo
		exit 1
	fi

	# does package exist?
	pkg=`basename $package | sed 's/.tar.bz2//'`
	if [ -f $inst_root/$FL_DIR/${pkg}-[0-9]* ] ; then
		echo "$pkg seems to be installed, candidates:"
		for i in $inst_root/$FL_DIR/${pkg}-[0-9]*; do
			echo "-> $i"
		done
		exit 1
	fi

	# package in database?
	if [ "`grep "^$pkg\ -" $DBF`" = "" ] ; then
		echo "sorry, package $package not found ..."
		echo "(use 'hdw-get update' to update database)"
		exit 1
	fi

	# check for runtime deps
	if [ -z $no_deps ] ; then
		deps="`grep "^$pkg\ -" $DBF | sed 's/.* -//'`"
		udeps=""
		tudeps=""
		for dep in $deps; do
			[ ! -f $inst_root/$FL_DIR/$dep-* ] && \
				udeps="$dep $udeps"
		done
		while [ "$udeps" != "$tudeps" ] ; do
			tudeps="$udeps"
			for tdep in $tudeps; do
				deps="`grep "^$tdep\ -" $DBF | sed 's/.* -//'`"
				for dep in $deps; do
					exists=0
					[ -f $inst_root/$FL_DIR/$dep-* ] && \
						break
					for i in $udeps; do
						[ "$i" = "$dep" ] && exists=1
					done
					[ "$exists" = "0" ] && \
						udeps="$udeps $dep"
				done
			done
		done
		echo "$pkg depends on the following uninstalled packages:"
		echo "$udeps"
		echo
		echo "continue? [y,n] (default 'n')"
		if [ -z $auto_resolve_deps ] ; then
			read answer
			if [ "$answer" = "y" ] ; then
				for i in $udeps; do
					[ "$simulate" != "1" ] && \
					$0 -no-dep install $inst_root $i
				done
			fi
		fi
	fi
	
	if [ "$simulate" != "1" ] ; then

	# store info/dir file
	rm -f /tmp/.hdw-get/dir*
	if [ -f $inst_root/usr/share/info/dir ] ; then
		[ "$verbose" ] && echo "backing up info/dir file"
		cp $inst_root/usr/share/info/dir /tmp/.hdw-get/
	fi
	
	# check/install package
	if [ -f $package ] ; then
		echo "extracting `basename $package` to $inst_root ..."
		tar --use-compress-program=bzip2 -xf $package \
			-C $inst_root
		echo "done"
	elif [ ! -z "$bin_dir" -a -f $bin_dir/${hdw_arch}-${hdw_arch_opt}/$package/$package.tar.bz2 ] ; then
		echo "extracting $package to $inst_root ..."
		tar --use-compress-program=bzip2 \
		-xf $bin_dir/$package/$package.tar.bz2 -C $inst_root
	elif [ ! -z "$net_addr" ] ; then
		echo "getting package $package via net ..."
		wget $net_addr/hdw-linux/hdw-linux-$hdw_version/binaries/${hdw_arch}-${hdw_arch_opt}/$package/${package}.tar.bz2 > /dev/null 2>&1
		retval=$?
		if [ "$retval" != "0" ] ; then
			echo "package not found at $net_addr, aborting ..."
			exit
		else
			echo "extracting $package to $inst_root ..."	
			tar --use-compress-program=bzip2 \
				-xf ${package}.tar.bz2 -C $inst_root
			rm ${package}.tar.bz2
			echo "done"
		fi
	else
		echo "sorry, package $package not found"
		exit 1
	fi

	# merging info/dir file
	if [ "$pkg" != "00-dirtree" ] ; then
	fl_file=`ls -A $inst_root/$FL_DIR/$pkg-[0-9]* | tail -1`
	if [ -n "`grep usr/share/info/dir $fl_file`" ] ; then
		echo "" >> /tmp/.hdw-get/dir
		echo "$pkg info - added by hdw `date`" >> /tmp/.hdw-get/dir
		for info_obj in `grep 'usr/share/info/.*.info' $fl_file | \
					awk -F/ '{ print $4 }'`; do
			info_o=`echo $info_obj | sed 's/.info//'`
			if [ -z "`grep \($info_o\)\. /tmp/.hdw-get/dir`" ]
			then
				[ "$verbose" ] && \
					echo "merging $info_o to info/dir"
				grep \($info_o\)\. \
					$inst_root/usr/share/info/dir >> \
					/tmp/.hdw-get/dir
			fi
		done
	fi
	fi
	[ -f /tmp/.hdw-get/dir ] && \
		cp /tmp/.hdw-get/dir $inst_root/usr/share/info/dir
	rm -f /tmp/.hdw-get/dir

	fi

# dist-install
elif [ "$d_install" = "1" ] ; then
	# check dist file
	if [ ! -f $dist_file ] ; then
		echo "$dist_file not found, aborting"
	else
		# install all distribution related packages
		for package in `cat $dist_file`; do
			# execute myelf
			$0 install $inst_root $package
		done
	fi
fi

# remove temp directory
rm -rf /tmp/.hdw-get