#!/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
  fi

  # get the disk. /storage on 2nd partition
  PART=$(blkid | awk  'BEGIN {FS=":"} /STORAGE/ {print $1}')
  DISK=$(echo ${PART} | sed s/p[0-9]//g)
  PARTNUM=$(echo ${PART} | sed "s#^.*[0-9]p##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 storage partition...   " "parted -s -a optimal -m ${DISK} resizepart ${PARTNUM} 100% &>/dev/null"
    StartProgress spinner "Checking storage file system... " "e2fsck -f -p ${PART} &>/dev/null"
    StartProgress spinner "Resizing storage file system... " "resize2fs ${PART} &>/dev/null"
    StartProgress countdown "Rebooting in 5s...     " 5 "NOW"
  fi
fi
reboot &>/dev/null