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

# Load DRBL setting and functions
if [ ! -f "/opt/drbl/sbin/drbl-conf-functions" ]; then
  echo "Unable to find /opt/drbl/sbin/drbl-conf-functions! Program terminated!" 
  exit 1
fi
. /opt/drbl/sbin/drbl-conf-functions

# Load some necessary setting from drbl-ocs.conf
. /opt/drbl/conf/drbl-ocs.conf

# Load basic functions of ocs
. /opt/drbl/sbin/ocs-functions

# Settings
DEBIAN_ISO_URL="http://free.nchc.org.tw/drbl-core/iso/debian-live-for-ocs.iso"
DEBIAN_ISO_PATH="$(dirname $DEBIAN_ISO_URL)"
DEBIAN_ISO="$(basename $DEBIAN_ISO_URL)"
# http://free.nchc.org.tw/drbl-core/iso/MD5SUMS
md5_file="MD5SUMS"
md5_file_url="$DEBIAN_ISO_PATH/$md5_file"
FILE_LIMIT="4500"  # unit: MB

# other default settings
# If insert_mode=prog_only, only copy DRBL/Clonezilla programs only, no ocs images.
insert_mode="prog_and_img"
img_size_sum=0

#
prog="$(basename $0)"

# functions
USAGE() {
    echo "Usage:"
    echo "To put clonezilla image into live CD:"
    echo "$prog [OPTION] CLONEZILLA_IMAGE_NAME"
    echo "OPTION:"
    language_help_prompt_by_idx_no
    echo "-p, --image-path   Assign the clonezilla image path where CLONEZILLA_IMAGE_NAME exists. Default = $ocsroot"
    echo "-s, --skip-image   Do not include any clonezilla image. The is used to created a live CD with DRBL/Clonezilla programs only."
    echo "$prog will download a template Debian live CD for clonezilla iso file from $DEBIAN_ISO_URL if ncecessary."
    echo "NOTE! Due to the limitation in mkisofs, this program does not work in clonezilla image file larger than $FILE_LIMIT MB."
    echo "Ex:"
    echo "To put clonezilla image sarge-ocs (located in /home/partimag in clonezilla server) to Live CD, you can run:"
    echo "$prog sarge-ocs"
    echo "To put more images, just append them, such as:"
    echo "$prog sarge-ocs etch-ocs"
}
#
check_file_size_in_ocs_image() {
  local img_path="$1"
  local size_t
  for ifile in $img_path/*; do
    size_t="$(du -ms $ifile | awk -F" " '{print $1}')"
    if [ "$size_t" -gt "$FILE_LIMIT" ]; then
      [ "$BOOTUP" = "color" ] && $SETCOLOR_FAILURE
      echo "$ifile size ($size_t MB) is too large! mkisofs is not able to process that!"
      echo "Unable to create this clonezilla image live CD!"
      [ "$BOOTUP" = "color" ] && $SETCOLOR_NORMAL
      exit 1
    fi
  done
}

#
check_if_root

# Parse command-line options
while [ $# -gt 0 ]; do
  case "$1" in
    -l|--language)
            shift
            if [ -z "$(echo $1 |grep ^-.)" ]; then
              # skip the -xx option, in case 
              specified_lang="$1"
              shift
            fi
	    [ -z "$specified_lang" ] && USAGE && exit 1
            ;;
    -p|--image-path)
            shift
            if [ -z "$(echo $1 |grep ^-.)" ]; then
              # skip the -xx option, in case 
              image_path="$1"
              shift
            fi
	    [ -z "$image_path" ] && USAGE && exit 1
            ;;
    -s|--skip-image)
            insert_mode="prog_only"
            shift ;;
    -*)     echo "${0}: ${1}: invalid option" >&2
            USAGE >& 2
            exit 2 ;;
    *)      break ;;
  esac
done
ocs_image="$*"
# strip the / to avoid problem
ocs_image="$(echo $ocs_image | sed -e "s|/||g")"
      
[ -z "$ocs_image" -a "$insert_mode" = "prog_and_img" ] && USAGE && exit 1
[ -z "$image_path" ] && image_path=$ocsroot

if ! type mkisofs 2>/dev/null; then
  echo "mkisofs not found!"
  echo "Program terminated!"
  exit 1
fi
#
ask_and_load_lang_set $specified_lang

iso_file="$(basename $DEBIAN_ISO_URL)"
if [ ! -f "$iso_file" ]; then
  echo "We need Debian live CD for Clonnezilla to create clonezilla cd."
  echo "Downloadling the iso file from $DEBIAN_ISO_URL..."
  wget $DEBIAN_ISO_URL
  get_iso_rc=$?
  # validate it
  if [ "$get_iso_rc" -eq 0 ]; then
    echo -n "Validating $DEBIAN_ISO... "
    md5_tmp="$(mktemp -d /tmp/isomd5.XXXXX)"
    #  wget http://opensource.nchc.org.tw/drbl-core/iso/MD5SUMS
    echo -n "Downloading $md5_file_url... "
    wget -r -l1 --no-parent --passive-ftp -e robots=off -q -nd --retr-symlinks -P "$md5_tmp" $md5_file_url
    if ! grep -w "$DEBIAN_ISO" $md5_tmp/MD5SUMS | md5sum -c; then
      [ "$BOOTUP" = "color" ] && $SETCOLOR_FAILURE
      echo "$DEBIAN_ISO is broken! Something went wrong! Try to remove $DEBIAN_ISO and run this program again."
      [ "$BOOTUP" = "color" ] && $SETCOLOR_NORMAL
      echo "Program terminated!"
      exit 1
    fi
  else
    [ "$BOOTUP" = "color" ] && $SETCOLOR_FAILURE
    echo "Unable to download $DEBIAN_ISO_URL!"
    [ "$BOOTUP" = "color" ] && $SETCOLOR_NORMAL
    echo "Program terminated!"
    exit 1
  fi
fi

if [ "$insert_mode" = "prog_and_img" ]; then
  echo "Creating clonezilla ISO with image(s) $ocs_image from $image_path..."
  ocs_live_prog="/opt/drbl/setup/files/ocs/ocs-live-restore"
  ocs_imgs_with_abs_path=""
  for im in $ocs_image; do
    if [ ! -d "$image_path/$im" ]; then
      [ "$BOOTUP" = "color" ] && $SETCOLOR_FAILURE
      echo "clonezilla image path $image_path/$im is NOT found!"
      [ "$BOOTUP" = "color" ] && $SETCOLOR_NORMAL
      echo "Program terminated!"
      exit 1
    fi
    ocs_imgs_with_abs_path="$ocs_imgs_with_abs_path $image_path/$im=$image_path/$im"
    convert_ocs_format_from_1.5_to_2.0_or_newer $image_path/$im/
    check_file_size_in_ocs_image $image_path/$im/
    img_size="$(du -ms $image_path/$im/ | awk -F" " '{print $1}')" # unit: MB
    img_size_sum="$((img_size_sum+img_size))"
  done
else
  echo "Creating clonezilla Live CD ISO file..."
  ocs_live_prog="/opt/drbl/setup/files/ocs/ocs-live-save"
fi
# use the 1st image name as iso filename
iso_label_tag="$(echo $ocs_image | awk -F" " '{print $1}')"
[ -z "$iso_label_tag" ] && iso_label_tag="drbl"
# mkisofs only allow 32 characters for volume ID
iso_label_tag="${iso_label_tag:0:31}"

target_iso="clonezilla-${iso_label_tag}.iso"

DEBIAN_ISO_TMP="$(mktemp -d /tmp/ocs-iso.XXXXXX)"
trap "[ -d "$DEBIAN_ISO_TMP" ] && umount $DEBIAN_ISO_TMP &>/dev/null && rm -rf $DEBIAN_ISO_TMP" HUP INT QUIT TERM EXIT
ISOLNX_TMP="$(mktemp -d /tmp/isolnx.XXXXXX)"
mount -o loop $DEBIAN_ISO $DEBIAN_ISO_TMP

template_iso_size="$(du -ms $DEBIAN_ISO | awk -F" " '{print $1}')"
drbl_prog_size="$(du -ms /opt/drbl/ | awk -F" " '{print $1}')" 
target_iso_size="$((template_iso_size+drbl_prog_size+img_size_sum))"

echo "Estimated target ISO file \"$target_iso\" size: $target_iso_size MB"
if [ "$target_iso_size" -gt "$FILE_LIMIT" ]; then
  [ "$BOOTUP" = "color" ] && $SETCOLOR_WARNING
  echo "The target ISO file is too large to fit in a DVD disk."
  [ "$BOOTUP" = "color" ] && $SETCOLOR_NORMAL
  echo "Do you want to continue ?"
  echo -n "[y/N] "
  read cont_ans
  case "$cont_ans" in
    y|Y|[yY][eE][sS])
       echo $msg_ok_let_do_it
       ;;
    *)
       echo "Abort!"
       exit 2
  esac
fi

# Create the iso image file
# isolinux should be rw, so we have to copy it, and exclude the one in iso image.
cp -a $DEBIAN_ISO_TMP/isolinux $ISOLNX_TMP/
mkisofs -o $target_iso \
 -A "Clonezilla live CD" \
 -V "$iso_label_tag" \
 -publisher "DRBL/Clonezilla http://drbl.sf.net http://clonezilla.sf.net" \
 -r -J -l \
 -b isolinux/isolinux.bin -c isolinux/boot.catalog \
 -no-emul-boot -boot-load-size 4 -boot-info-table \
 -x $DEBIAN_ISO_TMP/isolinux \
 -graft-points $DEBIAN_ISO_TMP \
  /isolinux/=$ISOLNX_TMP/isolinux \
  /opt/drbl/=/opt/drbl \
  /ocs-live=$ocs_live_prog \
  $ocs_imgs_with_abs_path

rc=$?
if [ "$rc" -gt 0 ]; then
  [ "$BOOTUP" = "color" ] && $SETCOLOR_FAILURE
  echo "Something went wrong!"
  [ "$BOOTUP" = "color" ] && $SETCOLOR_NORMAL
  umount $DEBIAN_ISO_TMP &>/dev/null
  [ ! -z "$DEBIAN_ISO_TMP" ] && rm -rf $DEBIAN_ISO_TMP
  [ ! -z "$ISOLNX_TMP" ] && rm -rf $ISOLNX_TMP
  [ -f "$target_iso" ] && rm -fv $target_iso
  echo "Program terminated!"
  exit 1
fi

# unmount all iso file
umount $DEBIAN_ISO_TMP &>/dev/null

# Clean the tmp working directory
echo "Cleaning tmp dirs..."
[ ! -z "$DEBIAN_ISO_TMP" ] && rm -rf $DEBIAN_ISO_TMP
[ ! -z "$ISOLNX_TMP" ] && rm -rf $ISOLNX_TMP

# 
[ "$BOOTUP" = "color" ] && $SETCOLOR_WARNING
echo "Now you can burn the iso file \"$target_iso\" into CD/DVD, then use it to boot to use clonezilla."
[ "$BOOTUP" = "color" ] && $SETCOLOR_NORMAL
