#!/bin/sh

# SPDX-License-Identifier: GPL-2.0-or-later
# Copyright (C) 2009-2014 Stephan Raue (stephan@openelec.tv)
# Copyright (C) 2018-present Team CoreELEC (https://coreelec.org)

if [ -e /storage/.please_resize_me ] ; then
  . /usr/lib/coreelec/functions

  hidecursor

  # this sh** was never intended to be used
  # on already installed and runing system
  if [ -d /storage/.kodi -o -d /storage/.config -o -d /storage/.cache ] ; then
    rm -f /storage/.please_resize_me
    sync
    echo "Resizing is not permitted - the system has already been initialised."
    StartProgress countdown "Rebooting in 15s... " 15 "NOW"
    reboot -f
  fi

  # get the disk. /storage on 2nd partition
  PART=$(grep "/storage " /proc/mounts | cut -d" " -f1)
  DISK=$(echo ${PART} | sed s/p2$//g)

  rm -f /storage/.please_resize_me
  sync

  # just in case
  if [ ! -z "${DISK}" -a ! -z "${PART}" ] ; then
    umount ${PART}

    clear >/dev/console
    echo "PARTITION RESIZING IN PROGRESS"
    echo ""
    echo "Please do not reboot or turn off your @DISTRONAME@ device!"
    echo ""

    # identify the partition scheme, and if gpt fix minor issues such as gpt header not at end of disk
    SCHEME=$(blkid -s PTTYPE -o value ${DISK})
    if [ "$SCHEME" = "gpt" ]; then
      StartProgress spinner "Checking layout...      " "sgdisk -e ${DISK} &>/dev/null"
    fi

    StartProgress spinner "Resizing partition...   " "parted -s -a optimal -m ${DISK} resizepart 2 4GiB &>/dev/null"
    StartProgress spinner "Checking file system... " "e2fsck -f -p ${PART} &>/dev/null"
    StartProgress spinner "Resizing file system... " "resize2fs ${PART} &>/dev/null"
    GAMESSTART="$(echo $(parted -m ${DISK} print all 2>/dev/null | awk 'BEGIN {FS=":"} /^2/ {gsub ("[A-Z]",""); print $3}') + 17 2>/dev/null | bc 2>/dev/null)"
    StartProgress spinner "Creating exFAT partition..." "parted -s -a optimal -m ${DISK} mkpart primary ntfs ${GAMESSTART}MB 100% &>/dev/null"
    partprobe &>/dev/null
    StartProgress spinner "Formatting exFAT partition..." "mkfs.exfat -n GAMES ${DISK}p3 &>/dev/null"
    StartProgress countdown "Rebooting in 5s...     " 5 "NOW"
  fi
fi
reboot -f &>/dev/null