#!/bin/bash

################################################################################
#      This file is part of OpenELEC - http://www.openelec.tv
#      Copyright (C) 2009-2014 Stephan Raue (stephan@openelec.tv)
#
#  OpenELEC 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 2 of the License, or
#  (at your option) any later version.
#
#  OpenELEC 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 OpenELEC.  If not, see <http://www.gnu.org/licenses/>.
################################################################################

################################################################################
# variables such as $ROOT $PATH etc... that are required for this
# script to work must be passed via env ... in scripts/image
################################################################################

# set variables
  OE_TMP=$(mktemp -d)
  LOOP=$(losetup -f)

  SYSTEM_SIZE=512
  STORAGE_SIZE=32 # STORAGE_SIZE must be >= 32 !

  DISK_SIZE=$(( $SYSTEM_SIZE + $STORAGE_SIZE + 4 ))
  DISK="$TARGET_IMG/$IMAGE_NAME.img"

# functions
  cleanup() {
    echo "image: cleanup..."
    umount "$OE_TMP" &>/dev/null || :
    losetup -d "$LOOP"
    [ -f "$OE_TMP/ldlinux.sys" ] && chattr -i "$OE_TMP/ldlinux.sys" || :
    rm -rf "$OE_TMP"
    exit
  }

trap cleanup SIGINT

# generate volume id for fat partition
if [ "$UEFI" = "yes" ] ; then
  UUID_1=$(date '+%d%m')
  UUID_2=$(date '+%M%S')
  FAT_VOL_ID="${UUID_1}${UUID_2}"
  UUID_SYSTEM="${UUID_1}-${UUID_2}"
  DISK="$TARGET_IMG/$IMAGE_NAME-efi.img"
fi

# ensure loopX not in use
  umount "$OE_TMP" &>/dev/null || :
  umount "$LOOP" &>/dev/null >/dev/null || :
  losetup -d "$LOOP" &>/dev/null >/dev/null || :

# create an image
  echo "image: creating image: $DISK..."
  dd if=/dev/zero of="$DISK" bs=1M count="$DISK_SIZE"
  sync

# write a disklabel
  echo "image: creating partition table on $DISK..."
  losetup "$LOOP" "$DISK"
  if [ "$UEFI" = "yes" ] ; then
    parted -s "$LOOP" mklabel gpt
  else
    parted -s "$LOOP" mklabel msdos
  fi
  sync

# create part1
  echo "image: creating part1 on $DISK..."
  SYSTEM_PART_END=$(( $SYSTEM_SIZE * 1024 * 1024 / 512 + 2048 ))
  if [ "$BOOTLOADER" = "syslinux" ]; then
    if [ "$UEFI" = "yes" ] ; then
      parted -s "$LOOP" -a min unit s mkpart primary fat32 2048 $SYSTEM_PART_END
    else
      parted -s "$LOOP" -a min unit s mkpart primary ext4 2048 $SYSTEM_PART_END
    fi
  elif [ "$BOOTLOADER" = "bcm2835-bootloader" -o "$BOOTLOADER" = "u-boot" ]; then
    parted -s "$LOOP" -a min unit s mkpart primary fat32 2048 $SYSTEM_PART_END
  fi
  if [ "$UEFI" = "yes" ] ; then
    parted -s "$LOOP" set 1 legacy_boot on
  else
    parted -s "$LOOP" set 1 boot on
  fi

# create part2
  echo "image: creating part2 on $DISK..."
  STORAGE_PART_START=$(( $SYSTEM_PART_END + 2048 ))
  STORAGE_PART_END=$(( $STORAGE_PART_START + (( $STORAGE_SIZE * 1024 * 1024 / 512 )) ))
  parted -s "$LOOP" -a min unit s mkpart primary ext4 $STORAGE_PART_START $STORAGE_PART_END
  sync

if [ "$BOOTLOADER" = "syslinux" ]; then
  # write mbr
    echo "image: writing mbr..."
    if [ "$UEFI" = "yes" ] ; then
      MBR="$ROOT/$TOOLCHAIN/share/syslinux/gptmbr.bin"
    else
      MBR="$ROOT/$TOOLCHAIN/share/syslinux/mbr.bin"
    fi
    if [ -n "$MBR" ]; then
      dd bs=440 count=1 conv=notrunc if="$MBR" of="$LOOP"
    fi
    sync
fi

# create filesystem on part1
  losetup -d "$LOOP"
  echo "image: creating filesystem on part1..."
  OFFSET=$(( 2048 * 512 ))
  SIZELIMIT=$(( $SYSTEM_SIZE * 1024 * 1024 ))
  losetup -o $OFFSET --sizelimit $SIZELIMIT "$LOOP" "$DISK"
  if [ "$BOOTLOADER" = "syslinux" ]; then
    if [ "$UEFI" = "yes" ] ; then
      mkfs.vfat -i "$FAT_VOL_ID" "$LOOP"
    else
      mke2fs -q -t ext4 -m 0 "$LOOP"
      tune2fs -U $UUID_SYSTEM "$LOOP"
      e2fsck -n "$LOOP"
    fi
  elif [ "$BOOTLOADER" = "bcm2835-bootloader" -o "$BOOTLOADER" = "u-boot" ]; then
    mkfs.vfat "$LOOP"
  fi
  sync

# mount partition
  echo "image: mounting part1 on $OE_TMP..."
  mount "$LOOP" "$OE_TMP"

if [ "$BOOTLOADER" = "syslinux" ]; then
  # create bootloader configuration
    echo "image: creating bootloader configuration..."
    cat << EOF > "$OE_TMP"/syslinux.cfg
UI vesamenu.c32
PROMPT 0
MENU TITLE Boot Menu
MENU BACKGROUND splash.png
TIMEOUT 50
DEFAULT live

MENU WIDTH 70
MENU MARGIN 15
MENU ROWS 2
MENU HSHIFT 4
MENU VSHIFT 13
MENU TIMEOUTROW 10
MENU TABMSGROW 8
MENU CMDLINEROW 8
MENU HELPMSGROW 13
MENU HELPMSGENDROW 26
MENU CLEAR

MENU COLOR border       30;44   #40ffffff #00000000 std
MENU COLOR title        1;36;44 #ff8bbfe3 #00000000 std
MENU COLOR sel          7;37;40 #80f0f0f0 #ff606060 all
MENU COLOR unsel        37;44   #50ffffff #00000000 std
MENU COLOR help         37;40   #c0ffffff #a0000000 std
MENU COLOR timeout_msg  37;40   #80ffffff #00000000 std
MENU COLOR timeout      1;37;40 #c0ffffff #00000000 std
MENU COLOR msg07        37;40   #90ffffff #a0000000 std
MENU COLOR tabmsg       31;40   #ff868787 #00000000 std

LABEL installer
  MENU LABEL Run $DISTRONAME Installer
  KERNEL /KERNEL
  APPEND boot=UUID=$UUID_SYSTEM installer quiet tty vga=current vt.global_cursor_default=0 loglevel=2

LABEL live
  MENU LABEL Run $DISTRONAME Live
  KERNEL /KERNEL
  APPEND boot=UUID=$UUID_SYSTEM disk=UUID=$UUID_STORAGE quiet ssh vga=current vt.global_cursor_default=0 loglevel=2
EOF

  # install extlinux
    echo "image: installing extlinux to part1..."
    if [ "$UEFI" = "yes" ] ; then
      syslinux --heads=4 --sector=32 -i "$LOOP"
    else
      extlinux --heads=4 --sector=32 -i "$OE_TMP"
    fi

  # copy files
    echo "image: copying files to part1..."
    cp $TARGET_IMG/$IMAGE_NAME.kernel "$OE_TMP/$KERNEL_NAME"
    cp $TARGET_IMG/$IMAGE_NAME.system "$OE_TMP/SYSTEM"
    cp $RELEASE_DIR/splash.png "$OE_TMP"
    cp $ROOT/$TOOLCHAIN/share/syslinux/vesamenu.c32 "$OE_TMP"
    cp $ROOT/$TOOLCHAIN/share/syslinux/libcom32.c32 "$OE_TMP"
    cp $ROOT/$TOOLCHAIN/share/syslinux/libutil.c32 "$OE_TMP"
    if [ "$UEFI" = "yes" ] ; then
      mkdir -p "$OE_TMP/EFI/BOOT"
      cp $ROOT/$TOOLCHAIN/share/syslinux/bootx64.efi "$OE_TMP/EFI/BOOT"
      cp $ROOT/$TOOLCHAIN/share/syslinux/ldlinux.e64 "$OE_TMP/EFI/BOOT"
      cp $ROOT/$TOOLCHAIN/share/syslinux/bootia32.efi "$OE_TMP/EFI/BOOT"
      cp $ROOT/$TOOLCHAIN/share/syslinux/ldlinux.e32 "$OE_TMP/EFI/BOOT"  
      cat << EOF > "$OE_TMP"/EFI/BOOT/syslinux.cfg
DEFAULT installer

LABEL installer
  KERNEL /KERNEL
  APPEND boot=UUID=$UUID_SYSTEM installer quiet tty vga=current vt.global_cursor_default=0 loglevel=2
EOF
    fi
elif [ "$BOOTLOADER" = "bcm2835-bootloader" ]; then
  # create bootloader configuration
    echo "image: creating bootloader configuration..."
    cat << EOF > "$OE_TMP"/cmdline.txt
boot=/dev/mmcblk0p1 disk=/dev/mmcblk0p2 quiet ssh vt.global_cursor_default=0 loglevel=2
EOF

  # copy files
    echo "image: copying files to part1..."
    cp $TARGET_IMG/$IMAGE_NAME.kernel "$OE_TMP/$KERNEL_NAME"
    cp $TARGET_IMG/$IMAGE_NAME.system "$OE_TMP/SYSTEM"
    cp $RELEASE_DIR/3rdparty/bootloader/bootcode.bin "$OE_TMP"
    cp $RELEASE_DIR/3rdparty/bootloader/fixup.dat "$OE_TMP"
    cp $RELEASE_DIR/3rdparty/bootloader/start.elf "$OE_TMP"
    cp $RELEASE_DIR/3rdparty/bootloader/config.txt "$OE_TMP"

    for dtb in $RELEASE_DIR/3rdparty/bootloader/*.dtb ; do
      if [ -f $dtb ] ; then
        cp "$dtb" "$OE_TMP"
      fi
    done

    if [ -d $RELEASE_DIR/3rdparty/bootloader/overlays ]; then
      cp -r $RELEASE_DIR/3rdparty/bootloader/overlays "$OE_TMP"
    fi

elif [ "$BOOTLOADER" = "u-boot" ]; then
  # create bootloader configuration
    echo "image: creating bootloader configuration..."
    if [ -n "$UBOOT_SYSTEM" -a -f "$RELEASE_DIR/3rdparty/bootloader/uEnv-$UBOOT_SYSTEM.txt" ]; then
      cp "$RELEASE_DIR/3rdparty/bootloader/uEnv-$UBOOT_SYSTEM.txt" "$OE_TMP/uEnv.txt"
    elif [ -f "$RELEASE_DIR/3rdparty/bootloader/uEnv.txt" ]; then
      cp $RELEASE_DIR/3rdparty/bootloader/uEnv.txt "$OE_TMP"
    elif [ -f "$RELEASE_DIR/3rdparty/bootloader/boot.scr" ]; then
      cp $RELEASE_DIR/3rdparty/bootloader/boot.scr "$OE_TMP"
    elif [ -f "$RELEASE_DIR/3rdparty/bootloader/boot.ini" ]; then
      cp $RELEASE_DIR/3rdparty/bootloader/boot.ini "$OE_TMP"
    fi

    echo "image: installing u-boot bootloader..."
    if [ -n "$UBOOT_SYSTEM" -a -f "$RELEASE_DIR/3rdparty/bootloader/SPL-$UBOOT_SYSTEM" ]; then
      dd if="$RELEASE_DIR/3rdparty/bootloader/SPL-$UBOOT_SYSTEM" of="$DISK" bs=512 seek=2 conv=fsync,notrunc
    elif [ -e "$RELEASE_DIR/3rdparty/bootloader/SPL" ]; then
      dd if="$RELEASE_DIR/3rdparty/bootloader/SPL" of="$DISK" bs=512 seek=2 conv=notrunc
    elif [ -n "$UBOOT_SYSTEM" -a -f "$RELEASE_DIR/3rdparty/bootloader/u-boot-$UBOOT_SYSTEM.imx" ]; then
      dd if="$RELEASE_DIR/3rdparty/bootloader/u-boot-$UBOOT_SYSTEM.imx" of="$DISK" bs=512 seek=2 conv=fsync,notrunc
    elif [ -e "$RELEASE_DIR/3rdparty/bootloader/u-boot.imx" ]; then
      dd if="$RELEASE_DIR/3rdparty/bootloader/u-boot.imx" of="$DISK" bs=512 seek=2 conv=notrunc
    elif [ -n "$UBOOT_SYSTEM" -a -f "$RELEASE_DIR/3rdparty/bootloader/uboot-sunxi-$UBOOT_SYSTEM.bin" ]; then
      dd if="$RELEASE_DIR/3rdparty/bootloader/uboot-sunxi-$UBOOT_SYSTEM.bin" of="$DISK" bs=1k seek=8 conv=fsync,notrunc
    elif [ -e "$RELEASE_DIR/3rdparty/bootloader/u-boot-sunxi-with-spl.bin" ]; then
      dd if=$RELEASE_DIR/3rdparty/bootloader/u-boot-sunxi-with-spl.bin of="$DISK" bs=1024 seek=8 conv=notrunc
    elif [ -d "$RELEASE_DIR/3rdparty/bootloader/hardkernel" ]; then
      dd if="$RELEASE_DIR/3rdparty/bootloader/hardkernel/bl1.bin.hardkernel" of="$DISK" bs=512 seek=1 conv=notrunc
      dd if="$RELEASE_DIR/3rdparty/bootloader/hardkernel/bl2.bin.hardkernel" of="$DISK" bs=512 seek=31 conv=notrunc
      dd if="$RELEASE_DIR/3rdparty/bootloader/u-boot.bin" of="$DISK" bs=512 seek=63 conv=notrunc
      dd if="$RELEASE_DIR/3rdparty/bootloader/hardkernel/tzsw.bin.hardkernel" of="$DISK" bs=512 seek=719 conv=notrunc
    elif [ -e "$RELEASE_DIR/3rdparty/bootloader/bl1.bin.hardkernel" ]; then
      dd if="$RELEASE_DIR/3rdparty/bootloader/bl1.bin.hardkernel" of="$DISK" bs=1 count=442 conv=notrunc
      dd if="$RELEASE_DIR/3rdparty/bootloader/bl1.bin.hardkernel" of="$DISK" bs=512 skip=1 seek=1 conv=notrunc
      dd if="$RELEASE_DIR/3rdparty/bootloader/u-boot.bin" of=$DISK bs=512 seek=64 conv=notrunc
    fi

    echo "image: copying files to part1..."
    cp $TARGET_IMG/$IMAGE_NAME.kernel "$OE_TMP/$KERNEL_NAME"
    cp $TARGET_IMG/$IMAGE_NAME.system "$OE_TMP/SYSTEM"

    if [ -n "$UBOOT_SYSTEM" -a -f "$RELEASE_DIR/3rdparty/bootloader/u-boot-$UBOOT_SYSTEM.img" ]; then
      cp "$RELEASE_DIR/3rdparty/bootloader/u-boot-$UBOOT_SYSTEM.img" "$OE_TMP/u-boot.img"
    elif [ -f $RELEASE_DIR/3rdparty/bootloader/u-boot.img ] ; then
      cp $RELEASE_DIR/3rdparty/bootloader/u-boot.img "$OE_TMP"
    fi

    if [ -f "$RELEASE_DIR/3rdparty/bootloader/$UBOOT_SYSTEM.fex" ]; then
      cp $RELEASE_DIR/3rdparty/bootloader/$UBOOT_SYSTEM.fex $RELEASE_DIR/3rdparty/bootloader/script.bin
    fi

    if [ -f "$RELEASE_DIR/3rdparty/bootloader/script.bin" ]; then
      cp $RELEASE_DIR/3rdparty/bootloader/script.bin "$OE_TMP"
    fi

    for dtb in $RELEASE_DIR/3rdparty/bootloader/*.dtb ; do
      if [ -f $dtb ] ; then
        cp "$dtb" "$OE_TMP"
      fi
    done

    if [ -f "$RELEASE_DIR/3rdparty/bootloader/cubieboard2.bin" ]; then
      cp $RELEASE_DIR/3rdparty/bootloader/* "$OE_TMP"
    fi
fi # bootloader

if [ -f "$INSTALL_SRC_DIR/aml_autoscript.src" ]; then
  cat "$INSTALL_SRC_DIR/aml_autoscript.src" > "$OE_TMP/aml_autoscript.src"
  echo "run storeargs" >> "$OE_TMP/aml_autoscript.src"
  echo "run bootcmd" >> "$OE_TMP/aml_autoscript.src"
  $MKIMAGE_TOOL_CMD "$OE_TMP/aml_autoscript.src" "$OE_TMP/aml_autoscript"
fi

# unmount part1
  echo "image: unmounting part1..."
  sync
  umount "$LOOP"

# create filesystem on part2
  losetup -d "$LOOP"
  echo "image: creating filesystem on part2..."
  OFFSET=$(( $STORAGE_PART_START * 512 ))
  SIZELIMIT=$(( $STORAGE_SIZE * 1024 * 1024 ))
  losetup -o $OFFSET --sizelimit $SIZELIMIT "$LOOP" "$DISK"
  mke2fs -q -t ext4 -m 0 "$LOOP"
  tune2fs -U $UUID_STORAGE "$LOOP"
  e2fsck -n "$LOOP"
  sync

# mount part2
  echo "image: mounting part2 on $OE_TMP..."
  mount "$LOOP" "$OE_TMP"

# add resize mark
  if [ "$UEFI" != "yes" ] ; then
    touch "$OE_TMP/.please_resize_me"
    sync
  fi

# unmount part2
  echo "image: unmounting part2..."
  umount "$LOOP"
  sync

# gzip
  echo "image: compressing..."
  gzip $DISK

# cleanup
  cleanup
