#!/bin/bash
# Author: Steven Shiau <steven _at_ nchc org tw>
# License: GPL

# Since at the time this script is executed, there is no clonezilla files in the system. Therefore we can not use functions from Clonezilla. hardcoded here.
# Decide the path for live media.
# casper 1.77+debian-7 uses /live_media, 1.110 uses /cdrom
# live-initramfs uses /live/image.
possible_path="/live_media /cdrom /live/image"
# Possible kernel/initrd paths are /casper (created by casper) or /live (created by live-initramfs)
live_sys_files_dir_list="casper live isolinux"

LIVE_MEDIA=""
# It's too slow to use find to search the whole system... Therefore we use this method.
for i in $possible_path; do
  for j in $live_sys_files_dir_list; do
    if [ -f "$i/$j/filesystem.squashfs" ]; then
      LIVE_MEDIA="$i"
      break
    fi
  done
  [ -n "$LIVE_MEDIA" ] && break
done

if [ -z "$LIVE_MEDIA" ]; then
  echo "///WARNING/// filesystem.squashfs not found! No idea where is LIVE_MEDIA!!!"
fi

# Prepare drbl/clonezilla runtime programs and dir
if [ -e /$LIVE_MEDIA/pkg/custom-ocs ]; then
  # We do not put custom-ocs in /opt/drbl/sbin/, we put it in /opt/local/sbin/ so that when we use clonezilla live to create a custom clonezilla live, it won't be put in the custom clonezilla live. It should be put by ocs-iso, not in the tarball.
  echo "Preparing customized Clonezilla program /opt/local/sbin/custom-ocs..."
  mkdir -p /opt/local/sbin/
  cp -f /$LIVE_MEDIA/pkg/custom-ocs /opt/local/sbin/
  chmod 755 /opt/local/sbin/custom-ocs
fi
