DEPENDS += "mdm-image"

OTA_TARGET_BASENAME_UBI     = "${MACHINE}-ota-target-image-ubi"
OTA_TARGET_IMAGE_ROOTFS_UBI = "${IMAGE_ROOTFS}/../${OTA_TARGET_BASENAME_UBI}"
OUTPUT_FILE_UBI             = "${DEPLOY_DIR_IMAGE}/${OTA_TARGET_BASENAME_UBI}.zip"
BASE_IMAGE_ROOTFS_UBI       = "$(echo ${IMAGE_ROOTFS} | sed 's#-recovery##')"
#OUTPUT_FILE_UBI = "${OTA_TARGET_BASENAME_UBI}.zip"

python do_rootfs_append(){
    bb.build.exec_func('recovery_ubi', d)
}

recovery_ubi[cleandirs] = "${OTA_TARGET_IMAGE_ROOTFS_UBI}"
recovery_ubi[dirs]      = "${OTA_TARGET_IMAGE_ROOTFS_UBI}"

recovery_ubi() {
#rm -rf ${OTA_TARGET_IMAGE_ROOTFS_UBI}
#mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}

#Create directory structure for targetfiles.zip
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/BOOTABLE_IMAGES
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/DATA
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/OTA
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/SYSTEM
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RADIO

echo "base image rootfs: ${BASE_IMAGE_ROOTFS_UBI}"
# copy radio directory content into RADIO directory
cp -r ${WORKSPACE}/oe-core/meta-msm/recipes/base-files-recovery/${MACHINE}/radio/. ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RADIO/.

# copy the boot\recovery images
cp ${DEPLOY_DIR_IMAGE}/${MACHINE}-boot.img ${OTA_TARGET_IMAGE_ROOTFS_UBI}/BOOTABLE_IMAGES/boot.img

cp ${DEPLOY_DIR_IMAGE}/${MACHINE}-boot.img ${OTA_TARGET_IMAGE_ROOTFS_UBI}/BOOTABLE_IMAGES/recovery.img


# copy the contents of system rootfs
cp -r ${BASE_IMAGE_ROOTFS_UBI}/. ${OTA_TARGET_IMAGE_ROOTFS_UBI}/SYSTEM/.

#copy contents of recovery rootfs
cp -r ${IMAGE_ROOTFS}/. ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/.

#generate recovery.fstab which is used by the updater-script
#echo #mount point fstype device [device2] >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/recovery.fstab
echo /boot     mtd     boot >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/recovery.fstab
echo /cache    ubifs  cache >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/recovery.fstab
echo /data     ubifs  userdata >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/recovery.fstab
echo /recovery mtd     recovery >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/recovery.fstab
echo /system   ubifs  system >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/recovery.fstab

#Copy contents of userdata rootfs
if [ -d ${TMPDIR}/rootfs/${MACHINE}-usrfs/ ]; then
   cp -r ${TMPDIR}/rootfs/${MACHINE}-usrfs/. ${OTA_TARGET_IMAGE_ROOTFS_UBI}/DATA/.
fi

#Getting content for OTA folder
mkdir -p ${OTA_TARGET_IMAGE_ROOTFS_UBI}/OTA/bin
cp   ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/usr/bin/applypatch ${OTA_TARGET_IMAGE_ROOTFS_UBI}/OTA/bin/.

cp   ${OTA_TARGET_IMAGE_ROOTFS_UBI}/RECOVERY/usr/bin/updater ${OTA_TARGET_IMAGE_ROOTFS_UBI}/OTA/bin/.


# copy contents of META folder
#recovery_api_version is from recovery module
echo recovery_api_version=3 >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#blocksize = BOARD_FLASH_BLOCK_SIZE
echo blocksize=131072 >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#boot_size: Size of boot partition from partition.xml
echo boot_size=0x00A00000 >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#recovery_size : Size of recovery partition from partition.xml
echo recovery_size=0x00A00000 >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#system_size : Size of system partition from partition.xml
echo system_size=0x00A00000 >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#userdate_size : Size of data partition from partition.xml
echo userdata_size=0x00A00000 >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#mkyaffs2_extra_flags : -c $(BOARD_KERNEL_PAGESIZE) -s $(BOARD_KERNEL_SPARESIZE)
echo mkyaffs2_extra_flags=-c 4096 -s 16 >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#extfs_sparse_flag : definition in build
echo extfs_sparse_flags=-s >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#default_system_dev_certificate : Dummy location
echo default_system_dev_certificate=build/abcd >> ${OTA_TARGET_IMAGE_ROOTFS_UBI}/META/misc_info.txt

#delete old files
rm -rf ${DEPLOY_DIR_IMAGE}/${OTA_TARGET_BASENAME_UBI}
# Pack the files into the deploy dir
cp -r ${OTA_TARGET_IMAGE_ROOTFS_UBI} ${DEPLOY_DIR_IMAGE}

# Zip ${MACHINE}-ota-target-image.zip
rm -f ${OUTPUT_FILE_UBI}
cd ${DEPLOY_DIR_IMAGE}/${OTA_TARGET_BASENAME_UBI}/
rm -rf ./SYSTEM/dev
rm -rf ./RECOVERY/dev
# Remove config and cp device_table-minimal.txt
rm -rf ./SYSTEM/etc/config
cp ${TOPDIR}/../meta/files/device_table-minimal.txt ./device_table.txt
zip -qry ${OUTPUT_FILE_UBI} ./
# Run fs_config on all the system files in the zip, and save the output
zipinfo -1 ${OUTPUT_FILE_UBI} | awk -F/ 'BEGIN { OFS="/" } /^SYSTEM\// {$1 = "system"; print}' | ${STAGING_BINDIR_NATIVE}/fs_config > ./META/filesystem_config.txt
zipinfo -1 ${OUTPUT_FILE_UBI} | awk -F/ 'BEGIN { OFS="/" } /^DATA\// {$1 = "data"; print}' | ${STAGING_BINDIR_NATIVE}/fs_config >> ./META/filesystem_config.txt
zip -q ${OUTPUT_FILE_UBI} ./META/filesystem_config.txt
}