#!/bin/bash

RED='\e[0;31m'      #Red
GREEN='\e[0;32m'    #Green
YELLOW='\e[0;33m'   #Yellow
CYAN='\e[0;36m'     #Cyan
CRESET='\e[0m'		#Reset color

INDEX_DIR=$ROOT/var/lib/scratchpkg/index
LOCK_FILE=$ROOT/var/lib/scratchpkg/spkg.lock
TMP_PKGINSTALL=$ROOT/var/lib/scratchpkg/spkg.install
HOOK_DIR=/etc/hooks
SYSUSERS_DIR=/etc/sysusers

nocolor() {
	RED=
	GREEN=
	YELLOW=
	CYAN=
	CRESET=
}

msg() {
	echo -e "${GREEN}==>${CRESET} $1"
}

msg2() {
	echo -e " ${CYAN}*${CRESET} $1"
}

msgerr() {
	echo -e "${RED}==> ERROR:${CRESET} $1"
}

msgwarn() {
	echo -e "${YELLOW}==> WARNING:${CRESET} $1"
}

runhooks() {
	
	if [ "$UPGRADE_PKG" ]; then
		opr=upgrade
	else
		opr=install
	fi
	
	for hook in $(ls $HOOK_DIR/*.hook 2>/dev/null); do
		description=$(grep "^# description[[:blank:]]*:" $hook | sed 's/^# description[[:blank:]]*:[[:blank:]]*//')
		operation=$(grep "^# operation[[:blank:]]*:" $hook | sed 's/^# operation[[:blank:]]*:[[:blank:]]*//')
		target=$(grep "^# target[[:blank:]]*:" $hook | sed 's/^# target[[:blank:]]*:[[:blank:]]*//')
		if [ -n "$description" ] && [ -n "$operation" ] && [ -n "$target" ]; then
			if [ "$(echo $operation | grep -w "$opr" )" ]; then
				if [ "$(grep -E $target $INDEX_DIR/$name/.files)" ]; then
					msg2 "$description"
					. $hook
					if [ "`type -t exechook`" = "function" ]; then
						exechook
					fi
				fi
			fi
		fi
		unset description operation target
		unset -f exechook
	done
	
}

sysusers() {
	
	if [ "$(ls $SYSUSERS_DIR/*.conf 2>/dev/null)" ]; then
		for file in $(ls $SYSUSERS_DIR/*.conf); do
			while read uid gid mode dir junk; do
				# Ignore comments and blank lines
				case "${uid}" in
					""|\#*) continue ;;
				esac
				if [ -e "$dir" ]; then
					if [ "$uid" != '-' ]; then
						getent passwd $uid >/dev/null
						if [[ "$?" = 0 ]]; then
							chown "$uid" "$dir"
						fi
					fi
					if [ "$gid" != '-' ]; then
						getent group $gid >/dev/null
						if [[ "$?" = 0 ]]; then
							chgrp "$gid" "$dir"
						fi
					fi
					if [ "$mode" != '-' ]; then
						chmod "$mode" "$dir"
					fi
				fi
			done < "$file"
		done
	fi

}

help() {
	
	cat << EOF
Usage:
  $(basename $0) package.spkg.txz <options>

Options:
  -u, --upgrade              upgrade package
  -r, --reinstall            reinstall package
  -d, --no-dep               skip dependency check
  -c, --ignore-conflict      ignore conflict when installing package
  -v, --verbose              print files installed
  -h, --help                 show this help message
      --no-preinstall        skip preinstall script before build/install package
      --no-postinstall       skip postinstall script after install package
      --no-preupgrade        skip preupgrade script before upgrade package
      --no-postupgrade       skip postupgrade script after upgrade package
      --no-backup            skip backup when upgrading package
      --no-color             disable colour for output

Example:
  installpkg foobar-1.0-1.spkg.txz -uc --no-backup       upgrade package foobar-1.0-1 without backup its 
                                                         old configuration files and skip conflict check
EOF
	
}

extract_opt() {
	for opt in $@; do
		case $opt in
			  --*) OPTS+=($opt) ;;
			   -*) for (( i=1; i<${#opt}; i++ )); do OPTS+=(-${opt:$i:1}); done ;;
			    *) OPTS+=($opt) ;;
		esac
	done
	echo ${OPTS[@]}
}

parse_opts() {
	
	if [ -z "$1" ]; then
		SHOWHELP=yes
	else
		while [ "$1" ]; do
			case $1 in
			-u |           --upgrade) UPGRADE_PKG=yes ;;
			-r |         --reinstall) REINSTALL_PKG=yes ;;
			-d |            --no-dep) IGNORE_DEP=yes ;;
			-c |   --ignore-conflict) IGNORE_CONFLICT=yes ;;
			-v |           --verbose) VERBOSE_INSTALL=yes ;;
			-h |              --help) SHOWHELP=yes ;;
			         --no-preinstall) NO_PREINSTALL=yes ;;
			        --no-postinstall) NO_POSTINSTALL=yes ;;
			  	     --no-preupgrade) NO_PREUPGRADE=yes ;;
				    --no-postupgrade) NO_POSTUPGRADE=yes ;;
				         --no-backup) NO_BACKUP=yes ;;
			   		      --no-color) NOCOLOR=yes ;;
		                    --root=*) ROOT="${1#*=}" ;;
					      *.spkg.txz) PKGNAME="$1" ;;
					               *) msg "Invalid option! ($1)"; exit 1 ;;
			esac
			shift
		done
	fi
	
}

ret() {
	# remove lock file on exit
	rm -f $LOCK_FILE
	exit $1
}

parse_opts $(extract_opt $@)

# show help page
if [ "$SHOWHELP" ] || [ -z "$PKGNAME" ]; then
	help
	ret 0
fi
	
# disable color for output
if [ "$NOCOLOR" ]; then
	nocolor
fi

mkdir -p $INDEX_DIR

# check for lock file
if [ -f $LOCK_FILE ]; then
	msgerr "Cant install/remove package simultaneously."
	msgerr "remove '$LOCK_FILE' if no install/remove package process running."
	exit 1
else
	touch $LOCK_FILE
fi

if [ -n "$PKGNAME" ]; then
	BASEPKGNAME=$(basename $PKGNAME)
fi

# check existence of package file
if [ ! -f "$PKGNAME" ]; then
	msgerr "Package '$1' not exist!"
	ret 1
fi

echo -ne "Loading $BASEPKGNAME...\033[0K\r"

# check integrity of package
if ! tar -tf $PKGNAME &>/dev/null; then
	msgerr "Package '$1' is corrupted!"
	ret 1
fi
echo -ne "\033[0K"

# check for root access
if [ "$UID" != "0" ]; then
	msgerr "Installing package need root access!"
	ret 1
fi

name=$(echo $BASEPKGNAME | sed 's/.spkg.txz//' | rev | cut -d - -f 3- | rev)
version=$(echo $BASEPKGNAME | sed 's/.spkg.txz//' | rev | cut -d - -f 2 | rev)
release=$(echo $BASEPKGNAME | sed 's/.spkg.txz//' | rev | cut -d - -f 1 | rev)
		
# get package information if installed
if [ -e $INDEX_DIR/$name/.pkginfo ]; then
	iname=$(cat $INDEX_DIR/$name/.pkginfo | grep ^name | cut -d " " -f3-)
	iversion=$(cat $INDEX_DIR/$name/.pkginfo | grep ^version | cut -d " " -f3-)
	irelease=$(cat $INDEX_DIR/$name/.pkginfo | grep ^release | cut -d " " -f3-)
	ALREADYINSTALLED=yes
fi

if [ "$ALREADYINSTALLED" = "yes" ] && [ ! "$UPGRADE_PKG" ] && [ ! "$REINSTALL_PKG" ]; then
	msg "Package '$name' already installed. ($iversion-$irelease)"
	ret 0
fi

# cant reinstall if not installed and cant upgrade if up-to-date
if [ "$UPGRADE_PKG" ] || [ "$REINSTALL_PKG" ]; then	
	if [ "$ALREADYINSTALLED" != "yes" ]; then
		msgerr "Package '$name' not installed."
		ret 1
	fi
	if [ "$UPGRADE_PKG" ]; then	
		if [ "$version-$release" = "$iversion-$irelease" ]; then
			msg "Package '$name' is up-to-date. ($iversion-$irelease)"
			ret 0
		fi
	fi
fi

### INSTALL PACKAGE INTO SYSTEM ###
opr=Installing; oprdone=installed
[ "$UPGRADE_PKG" ] && { opr=Upgrading; oprdone=upgraded; }
[ "$REINSTALL_PKG" ] && { opr=Reinstalling; oprdone=reinstalled; }
msg "$opr '$name-$version-$release'..."

#ignore conflict
if [ ! "$IGNORE_CONFLICT" ]; then
	msg2 "Checking file conflict..."
	while IFS=' ' read -r line; do
			if [ "$line" = "${line%.*}.spkgnew" ]; then
				line=${line%.*}
			fi
			if [ -e "$ROOT/$line" ]; then
				if [ "$UPGRADE_PKG" ] || [ "$REINSTALL_PKG" ]; then
					if [ ! "$(grep -Fx "$line" "$INDEX_DIR/$name/.files")" ]; then
						fileconflict+=(${line})
					fi
				else
					fileconflict+=(${line})
				fi
			fi
	done < <(tar -tf "$PKGNAME" --exclude=.pkginfo --exclude=.pkginstall --exclude=.pkgreadme ${excludefile[@]} | grep -v '/$')
		
	if [ "${#fileconflict[@]}" -gt 0 ]; then
		msgerr "File conflict found:"
		for fc in ${fileconflict[@]}; do
			msg2 "$fc"
		done
		ret 1
	fi
fi

if [ "$ROOT" = "" ]; then
	if [ ! "$REINSTALL_PKG" ]; then
		if [ $(tar -tf "$PKGNAME" | grep -x ".pkginstall") ]; then
			source <(tar -xf "$PKGNAME" .pkginstall -O)
		fi
	fi
fi

# run preinstall script if no --no-preinstall flag and not upgrade package
if [ ! "$NO_PREINSTALL" ] && [ ! "$UPGRADE_PKG" ]; then
	if [ "`type -t pre_install`" = "function" ]; then
		msg2 "Running preinstall script..."
		pre_install "$version"
	fi
fi

# run preupgrade script if package upgrade
if [ "$UPGRADE_PKG" ] && [ ! "$NO_PREUPGRADE" ]; then
	if [ "`type -t pre_upgrade`" = "function" ]; then
		msg2 "Running preupgrade script..."
		pre_upgrade "$version" "$iversion"
	fi
fi	

#installing package into system
msg2 "Extracting package..."
installcmd="tar --keep-directory-symlink -p -x -v -f $PKGNAME -C ${ROOT:-/} --exclude=.pkginfo --exclude=.pkginstall --exclude=.pkgreadme"

rm -f $TMP_PKGINSTALL
$installcmd | while IFS=' ' read line; do 
	if [ "$line" = "${line%.*}.spkgnew" ]; then
		line=${line%.*}
		if [ "$UPGRADE_PKG" ] || [ "$REINSTALL_PKG" ]; then
			if [ ! -e "$ROOT/$line" ]; then
				mv "$ROOT/$line".spkgnew "$ROOT/$line"
			fi
		else
			mv "$ROOT/$line".spkgnew "$ROOT/$line"
		fi
	fi
	echo "$line" >> $TMP_PKGINSTALL
done

# remove old files from old package that not exist in new package
if [ "$UPGRADE_PKG" ] || [ "$REINSTALL_PKG" ]; then
	msg2 "Removing old files..."
	grep -v '/$' $INDEX_DIR/$name/.files | while IFS=' ' read line; do
		if [ ! "$(grep -Fx "$line" $TMP_PKGINSTALL)" ]; then
			rm "$ROOT/$line" &>/dev/null
		fi
	done

	while IFS=' ' read -r line; do
		if [ ! "$(tac $TMP_PKGINSTALL | grep -x "$line")" ] && [ ! "$(grep -Rx --exclude-dir="$name" -w "$line" "$INDEX_DIR")" ]; then
			rmdir "$ROOT/$line" &>/dev/null
		fi
	done < <(tac $INDEX_DIR/$name/.files | grep '/$')
fi	

# register package into database
rm -fr $INDEX_DIR/$name
mkdir $INDEX_DIR/$name
echo "name = $name" > $INDEX_DIR/$name/.pkginfo
echo "version = $version" >> $INDEX_DIR/$name/.pkginfo
echo "release = $release" >> $INDEX_DIR/$name/.pkginfo
mv $TMP_PKGINSTALL $INDEX_DIR/$name/.files
tar -x -f $PKGNAME -C $INDEX_DIR/$name .pkginstall .pkgreadme >/dev/null 2>&1

if [ ! "$NO_POSTINSTALL" ] && [ ! "$UPGRADE_PKG" ]; then
	if [ "`type -t post_install`" = "function" ]; then
		msg2 "Running postinstall script..."
		post_install "$version"
	fi
fi

if [ "$UPGRADE_PKG" ] && [ ! "$NO_POSTUPGRADE" ]; then
	if [ "`type -t post_upgrade`" = "function" ]; then
		msg2 "Running postupgrade script..."
		post_upgrade "$version" "$iversion"
	fi
fi

if [ "$ROOT" = "" ]; then
	runhooks
	sysusers
fi

msg "Package '$name-$version-$release' $oprdone."

# running ldconfig
if [ "$ROOT" = "" ] && [ -x /sbin/ldconfig ]; then
	/sbin/ldconfig
fi

ret 0