#=----------------------------------------------------------------------------=#
#                                 OS on Kaleid                                 #
#                                                                              #
#   Desc:       Grub installation script for the OS/K image                    #
#                                                                              #
#                                                                              #
#   Copyright © 2018-2021 The OS/K Team                                        #
#                                                                              #
#   This file is part of OS/K.                                                 #
#                                                                              #
#   OS/K is free software: you can redistribute it and/or modify               #
#   it under the terms of the GNU General Public License as published by       #
#   the Free Software Foundation, either version 3 of the License, or          #
#   any later version.                                                         #
#                                                                              #
#   OS/K is distributed in the hope that it will be useful,                    #
#   but WITHOUT ANY WARRANTY# without even the implied warranty of             #
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              #
#   GNU General Public License for more details.                               #
#                                                                              #
#   You should have received a copy of the GNU General Public License          #
#   along with OS/K.  If not, see <https://www.gnu.org/licenses/>.             #
#=----------------------------------------------------------------------------=#

#Color codes
CL='\033[0;32m'
CL2='\033[1;36m'
CL3='\033[0m'
NC='\033[1;37m'

LOSETUP=$(which losetup)

set -e #exit if error

echo ${CL2}[install-os-k.sh]${NC} Mount image... 		\(udisksctl\)${CL3}
lpblock=$(udisksctl loop-setup --offset 1048576 --no-user-interaction --file $1 | cut -d ' ' -f 5 | sed 's/\.//g')

echo ${CL2}[install-os-k.sh]${NC} Sync image... 		\(sync\)${CL3}
sync

if [ -z $lpblock ];
then
	echo Impossible to find an available loop device. Fatal error !
	exit 13
fi

mountpoint=$(udisksctl info -b $lpblock | grep MountPoint | cut -d ':' -f 2 | sed 's/ //g')

if [ -z $mountpoint ];
then
	mountpoint=$(udisksctl mount -b $lpblock | cut -d ' ' -f 4 | sed 's/\.//g')
fi

if [ -z $mountpoint ];
then
	echo Impossible to mount the image at $lpblock. Fatal error !
	exit 13
fi

echo ${CL2}[install-os-k.sh]${NC} Copying grub.cfg to $lpblock as $mountpoint
cp $2 $mountpoint/boot/grub/grub.cfg
echo ${CL2}[install-os-k.sh]${NC} Copying kernel to $lpblock as $mountpoint
cp $3 $mountpoint/boot/kaleid

echo ${CL2}[install-os-k.sh]${NC} Sync image... 		\(sync\)${CL3}
sync

echo ${CL2}[install-os-k.sh]${NC} Unmount image... 		\(udisksctl\)${CL3}
udisksctl unmount --no-user-interaction -b $lpblock

lpblock=$($LOSETUP -j $1 | cut -d ':' -f 1)
if [ -z $lpblock ];
then
	exit 0
fi

udisksctl loop-delete -b $lpblock