#=----------------------------------------------------------------------------=#
#                            GNU GPL OS/K                                      #
#                                                                              #
#   Desc:       OS/K image maker script                                        #
#                                                                              #
#                                                                              #
#   Copyright © 2018-2019 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'

set -e #exit if error

## Create the image
echo ${CL2}[create_disk.sh]${NC} Creating image... 	\(dd\)${CL3}
dd if=/dev/zero of=$1 bs=512 count=131072 status=progress

## Partition the image
echo ${CL2}[create_disk.sh]${NC} Partitionning image... 	\(parted\)${CL3}
sudo parted --script $1 mktable msdos mkpart primary 2048s 100% set 1 boot on

## Mount
echo ${CL2}[create_disk.sh]${NC} Mount image... 		\(losetup\)${CL3}
lpblock=$(sudo losetup -f)
sudo losetup $lpblock $1 -o 1048576
echo ${CL2}[create_disk.sh]${NC} Sync image... 		\(sync\)${CL3}
sync

if [ -z $lpblock ];
then
	echo No loop device partition to target. Fatal error !
	return 13
fi

## Format
echo ${CL2}[create_disk.sh]${NC} Formatting image... 	\(mkdosfs\)${CL3}
sudo mkdosfs -F32 -f 2 $lpblock
echo ${CL2}[create_disk.sh]${NC} Sync image... 		\(sync\)${CL3}
sync

## Detach
echo ${CL2}[create_disk.sh]${NC} Unmount image... 		\(udisksctl\)${CL3}
sudo losetup -d $lpblock

## Mount
echo ${CL2}[create_disk.sh]${NC} Mount image... 		\(losetup\)${CL3}
lpblock=$(sudo losetup -f)
sudo losetup $lpblock $1 -o 1048576
lpblockbase=$(sudo losetup -f)
sudo losetup $lpblockbase $1

if [ -z $lpblockbase ];
then
	echo No loop device to target. Fatal error !
	return 13
fi

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

sudo udisksctl mount -b $lpblock --no-user-interaction

## Sync
echo ${CL2}[create_disk.sh]${NC} Sync image... 		\(sync\)${CL3}
sync

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

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

## Install Grub
echo ${CL2}[create_disk.sh]${NC} Install grub into $lpblockbase as $mountpoint... 		\(grub-install\)${CL3}

sudo grub-install --force --target=i386-pc --root-directory=$mountpoint --boot-directory=$mountpoint/boot  --no-floppy --modules="part_msdos biosdisk fat multiboot configfile" $lpblockbase

## Sync
echo ${CL2}[create_disk.sh]${NC} Sync image... 		\(sync\)${CL3}
sync

## Unmount
echo ${CL2}[create_disk.sh]${NC} Unmount image... 		\(udisksctl\)${CL3}
sudo udisksctl unmount --no-user-interaction -b $lpblock
sudo losetup -d $lpblock
sudo losetup -d $lpblockbase

echo ${CL2}[create_disk.sh]${CL} Terminated without error.${CL3}