#!/bin/bash
set -ex

# KHẮC PHỤC LỖI: Sử dụng /run thay vì /tmp vì /tmp có thể chưa sẵn sàng lúc boot sớm
export TMPDIR=/run/growpart_tmp
mkdir -p $TMPDIR

# Tìm phân vùng đang mount vào /
ROOT_PART=$(findmnt -n -o SOURCE / | cut -d'[' -f1)

# Lấy tên ổ đĩa (ví dụ sda)
DISK=$(lsblk -no PKNAME $ROOT_PART)

# Lấy số thứ tự phân vùng (ví dụ 1)
PART_NUM=$(echo $ROOT_PART | grep -o '[0-9]*$')

echo "Đang thực hiện growpart cho /dev/$DISK phân vùng $PART_NUM..."

# Thực hiện mở rộng bảng phân vùng
growpart /dev/$DISK $PART_NUM || true

# Kiểm tra loại FS để dùng lệnh tương ứng
FS_TYPE=$(findmnt -n -o FSTYPE /)

if [ "$FS_TYPE" == "xfs" ]; then
    xfs_growfs /
elif [ "$FS_TYPE" == "btrfs" ]; then
    # Fedora mặc định dùng Btrfs, phải dùng lệnh này
    btrfs filesystem resize max /
else
    # Dành cho EXT4
    resize2fs $ROOT_PART
fi

# Dọn dẹp thư mục tạm
rm -rf $TMPDIR