#!/bin/sh

BOLD="\033[1;33m"
NORM="\033[0m"
INFO="$BOLD Info:$NORM"
ERROR="$BOLD Error:$NORM"
WARNING="$BOLD Warning:$NORM"
INPUT="$BOLD => $NORM"
if [ -z $1 ]
then
	chrootDir=debian
else
	chrootDir=$1
fi
iSystem=jessie
chrootBaseDir=/DataVolume/$chrootDir
debootstrapPkgName=debootstrap_1.0.89~bpo8+1_all.deb
projectURL=https://github.com/MyBookLive/chroot/raw/master
isServicesInstalled=no
WGET="wget --no-check-certificate -q -O"
echo -e $INFO This script will guide you through the chroot-based services
echo -e $INFO installation on Western Digital My Book Live \(Duo\) and My Cloud NAS.
echo -e $INFO The goal is to install Debian $iSystem environment with no interference
echo -e $INFO with firmware. You will be asked later about which services to install
echo -en $INPUT Do you wish to continue [y/n]?
read userAnswer
if [ "$userAnswer" != "y" ]
then
	echo -e $INFO Ok then. Exiting.
	exit 0
fi

if [ -e /etc/init.d/chroot_$chrootDir.sh ]
then
	echo -e $ERROR Chroot\'ed services start/stop script detected! Please, remove
	echo -e $ERROR previous installation or specify destination folder name
	echo -e $ERROR and run script again with folder name parameter, for example:
	echo -e $ERROR ./install.sh my_debian
	exit 1
fi
if [ -d $chrootBaseDir ]
then
	echo -e $WARNING Previous chroot environment will be moved to $chrootBaseDir.old
	[ -d $chrootBaseDir.old ] || mkdir $chrootBaseDir.old
	mv -f $chrootBaseDir/* $chrootBaseDir.old
else
	mkdir $chrootBaseDir
fi
echo -e $INFO Deploying a debootstrap package...
$WGET /tmp/$debootstrapPkgName $projectURL/$debootstrapPkgName
dpkg -i /tmp/$debootstrapPkgName
rm -f /tmp/$debootstrapPkgName
echo -e $INFO Preparing a new Debian $iSystem chroot file base. Please, be patient,
echo -e $INFO may takes a long time on low speed connection...
#debootstrap --no-check-gpg --no-check-certificate --variant=minbase --exclude=yaboot,udev,dbus,systemd --include=locales,mc,aptitude,wget,dialog,apt-utils,sysvinit-core,sysvinit,sysvinit-utils $iSystem $chrootBaseDir http://archive.debian.org/debian/
debootstrap --no-check-gpg --no-check-certificate --variant=minbase --exclude=yaboot,udev,dbus,systemd --include=locales,mc,aptitude,wget,dialog,apt-utils,sysvinit,sysvinit-utils $iSystem $chrootBaseDir http://archive.debian.org/debian/
echo "share:x:1000:root,www-data,daapd" >> $chrootBaseDir/etc/group
cat > $chrootBaseDir/usr/sbin/policy-rc.d <<EOF
#!/bin/sh
exit 101
EOF
chmod a+x $chrootBaseDir/usr/sbin/policy-rc.d
sed -e "s/# ru_RU.UTF-8 UTF-8/ru_RU.UTF-8 UTF-8/" -e "s/# en_US.UTF-8 UTF-8/en_US.UTF-8 UTF-8/" $chrootBaseDir/etc/locale.gen >$chrootBaseDir/etc/locale.gen.sed
mv -f $chrootBaseDir/etc/locale.gen.sed $chrootBaseDir/etc/locale.gen
chroot $chrootBaseDir locale-gen
chroot $chrootBaseDir apt-get update > /dev/null 2>&1
echo -e $INFO A Debian $iSystem chroot environment  installed.
echo -e $INFO Now deploying services start script...
$WGET $chrootBaseDir/chroot_$chrootDir.sh $projectURL/wedro_chroot.sh
eval sed -i 's,__CHROOT_DIR_PLACEHOLDER__,$chrootBaseDir,g' $chrootBaseDir/chroot_$chrootDir.sh
chmod +x $chrootBaseDir/chroot_$chrootDir.sh
touch $chrootBaseDir/chroot-services.list
$chrootBaseDir/chroot_$chrootDir.sh install
echo >> $chrootBaseDir/root/.bashrc
echo PS1=\'\(chroot-$chrootDir\)\\w\# \' >> $chrootBaseDir/root/.bashrc
$chrootBaseDir/chroot_$chrootDir.sh start
echo -e $INFO ...finished.

echo -en $INPUT Do you wish to install miniDLNA UPnP/DLNA server [y/n]?
read userAnswer
if [ "$userAnswer" == "y" ]
then
	isServicesInstalled=yes
	echo -e $INFO UPnP/DLNA content will be taken from \"Public/Shared Music\",
	echo -e $INFO \"Public/Shared Pictures\" and\"Public/Shared Videos\" shares.
	chroot $chrootBaseDir apt-get --force-yes -qqy install minidlna
	killall minidlna > /dev/null 2>&1
	[ -d "/DataVolume/shares/Public/Shared Music" ] || mkdir "/DataVolume/shares/Public/Shared Music"
	[ -d "/DataVolume/shares/Public/Shared Pictures" ] || mkdir "/DataVolume/shares/Public/Shared Pictures"
	[ -d "/DataVolume/shares/Public/Shared Videos" ] || mkdir "/DataVolume/shares/Public/Shared Videos"
	sed -i 's|^media_dir=/var/lib/minidlna|media_dir=A,/mnt/Public/Shared Music\nmedia_dir=P,/mnt/Public/Shared Pictures\nmedia_dir=V,/mnt/Public/Shared Videos|g' $chrootBaseDir/etc/minidlna.conf
	rm -f $chrootBaseDir/var/lib/minidlna/files.db
	echo minidlna >> $chrootBaseDir/chroot-services.list
	echo -e $INFO MiniDLNA is installed.
fi

echo -en $INPUT Do you wish to install Transmission BitTorrent client [y/n]?
read userAnswer
if [ "$userAnswer" == "y" ]
then
	isServicesInstalled=yes
	[ -d /DataVolume/shares/Public/Torrents ] || mkdir /DataVolume/shares/Public/Torrents
	echo -e $INFO Torrents content will be downloaded to \"Public/Torrents\" share. Installing...
	chroot $chrootBaseDir apt-get --force-yes -qqy install transmission-daemon
	$WGET $chrootBaseDir/etc/transmission-daemon/settings.json $projectURL/settings.json
	chmod +rw $chrootBaseDir/etc/transmission-daemon/settings.json
	echo transmission-daemon >> $chrootBaseDir/chroot-services.list
	echo -e $INFO Transmission is installed.
fi

if [ "$isServicesInstalled" == "yes" ]
then
	echo -en $INPUT Do you wish to start chroot\'ed services right now [y/n]?
	read userAnswer
	if [ "$userAnswer" == "y" ]
	then
		/etc/init.d/chroot_$chrootDir.sh stop
		sleep 5
		/etc/init.d/chroot_$chrootDir.sh start
	fi
fi
echo -e $INFO Congratulation! Installation finished. You\'ve got a working
echo -e $INFO Debian $iSystem environment onboard.  You may install any services
echo -e $INFO you wish, but don\'t forget to add it\'s names to
echo -e $INFO $chrootBaseDir/chroot-services.list
echo -e $INFO /etc/init.d/chroot_$chrootDir.sh script is used
echo -e $INFO to start or stop chroot\'ed services.
echo -e $INFO Found bug? Please, report us!
echo -e $INFO https://github.com/MyBookLive/chroot/issues