mkimg.sh

#!/bin/bash

export WORKDIR=$(dirname $(readlink -f "$0"))
cd ${WORKDIR}

if [ $# -lt 3 ];then
	echo "Usage: $0 soc machine dist [custom]"
	echo "Example: $0 rk3568 h68k focal"
	exit 1
fi

soc=$1
machine=$2
dist=$3
custom=$4

# check parameters
if [ -f "./env/soc/${soc}.env" ];then
	source ./env/soc/${soc}.env
else
	echo "The soc env file is not exists: ./env/soc/${soc}.env"
	exit 1
fi

if [ -f "./env/machine/${machine}.env" ];then
	source ./env/machine/${machine}.env
else
	echo "The machine env file is not exists: ./env/machine/${machine}.env"
	exit 1
fi

if [ -f "./env/linux/${dist}.env" ];then
	source ./env/linux/${dist}.env
else
	echo "The dist env file is not exists: ./env/linux/${dist}.env"
	if [ -f "./env/linux/private/${dist}.env" ];then
		echo "The private environment file ./env/linux/private/${dist}.env has been found."
		source ./env/linux/private/${dist}.env
	else
		echo "The dist env file is not exists: ./env/linux/private/${dist}.env"
		exit 1
	fi
fi

# The custom env file
# The variable values in it can override the variable values of the same name in the previous three files
if [ -f "./env/custom/${custom}.env" ];then
	source ./env/linux/${custom}.env
fi
# end of check parameters

BC=$(which bc)
if [ "$BC" == "" ];then
	echo "The bc program is not installed, please install bc."
	echo "Example: sudo apt-get install -y bc"
	exit 1
fi

if [ -n "$DIST_ALIAS" ];then
	os_release=${DIST_ALIAS}
else
	os_release=$dist
fi

if [ -n "$DIST_ALIAS" ];then
	rootfs_source=${DIST_ALIAS}
else
	rootfs_source=${dist}
fi

if [ ! -d "${WORKDIR}/build/${rootfs_source}" ];then
	echo "The rootfs of dist ${rootfs_source} is not exists, please make rootfs first!"
	exit 1
fi

case $OS_RELEASE in
	bionic|focal|jammy) 		os_name='ubuntu';;
	buster|bullseys|bookworm)	os_name='debian';;
	*)				os_name='unknown';;
esac

if [ -n "${DEFAULT_FSTYPE}" ];then
	case ${DEFAULT_FSTYPE} in
		btrfs|xfs|ext4) rootfs_fstype=${DEFAULT_FSTYPE}
				;;
			*)	rootfs_fstype=btrfs
	esac
else
	rootfs_fstype=btrfs
fi

bootloader_mb=16
if [ -n "${DEFAULT_BOOTFS_MB}" ];then
	bootfs_mb=${DEFAULT_BOOTFS_MB}
else
	bootfs_mb=256
fi

rootfs_source_mb=$(du -m -d1 "${WORKDIR}/build/${rootfs_source}" | tail -n1 | awk '{print $1}')
echo "The rootfs source size is ${rootfs_source_mb} MB"

# modules size (estimated value)
modules_mb=150

if [ "$rootfs_fstype" == "btrfs" ];then
	# the btrfs compress rate (estimated value)
	compress_rate=0.618
	target_img_mb=$(echo -e "(($rootfs_source_mb + $modules_mb) * $compress_rate + $bootloader_mb + $bootfs_mb) / 1\nquit\n" | ${BC} -q)
else
	# reserved size for xfs or ext4
	reserved_mb=320
	target_img_mb=$(( $rootfs_source_mb + $modules_mb + $bootloader_mb + $bootfs_mb + $reserved_mb))
fi
echo "The target image size is ${target_img_mb} MB"

output_img=${WORKDIR}/build/${machine_name}_${os_name}_${os_release}_v$(date +%Y%m%d).img
echo "Create a blank disk image: $output_img ... "
echo ./scripts/diskinit.sh "${output_img}" "${target_img_mb}" "${bootloader_mb}" "${bootfs_mb}" "${rootfs_fstype}"
./scripts/diskinit.sh "${output_img}" "${target_img_mb}" "${bootloader_mb}" "${bootfs_mb}" "${rootfs_fstype}"
if [ $? -eq 0 ];then
	echo "succeeded"
	echo
else
	echo "failed"
	exit 1
fi

rm -rf build/temp_root && mkdir -p build/temp_root

umount -f ${WORKDIR}/build/${rootfs_source}/dev/pts 2>/dev/null
umount -f ${WORKDIR}/build/${rootfs_source}/dev 2>/dev/null
umount -f ${WORKDIR}/build/${rootfs_source}/sys 2>/dev/null
umount -f ${WORKDIR}/build/${rootfs_source}/run 2>/dev/null
umount -f ${WORKDIR}/build/${rootfs_source}/proc 2>/dev/null
echo "Make the target image ... "
echo scripts/write_target_img.sh "${WORKDIR}/build/${rootfs_source}" "${WORKDIR}/build/temp_root" "${output_img}" "${rootfs_fstype}" "${FIRSTBOOT}"
scripts/write_target_img.sh "${WORKDIR}/build/${rootfs_source}" "${WORKDIR}/build/temp_root" "${output_img}" "${rootfs_fstype}" "${FIRSTBOOT}"
if [ $? -eq 0 ];then
	echo "The target image [${output_img}] has been created successfully"
	echo
else
	echo "failed"
	exit 1
fi

rm -rf ${WORKDIR}/build/temp_root