#!/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)
# Copyright (C) 2023 JELOS (https://github.com/JustEnoughLinuxOS)

if [ -e /storage/.please_resize_me ] ; then
  . /usr/lib/jelos/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=":"} /'$(cat /proc/cmdline | sed -e 's#^.*disk=UUID=##g' -e 's#\ .*$##g')'/ {print $1}')
  if [ "$(echo ${PART} | awk '/mmcblk/ || /nvme/ {print $1}')" ]
  then
    DISK=$(echo ${PART} | sed s/p[0-9]//g)
    PARTNUM=$(echo ${PART} | sed "s#^.*[0-9]p##g")
  else
    DISK=$(echo ${PART} | sed s/[0-9]//g)
    PARTNUM=$(echo ${PART} | sed "s#^.*sd[a-z]##g")
  fi
  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...      " "parted -fs ${DISK} print &>/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 spinner "Checking storage file system... " "e2fsck -f -p ${PART} &>/dev/null"

    StartProgress countdown "Rebooting in 5s...     " 5 "NOW"
  fi
fi
sync &>/dev/null
sync &>/dev/null
sync &>/dev/null
reboot &>/dev/null
