HyprSpike

#!/usr/bin/env bash
set -euo pipefail

# Ensure script is run as root
if [[ $EUID -ne 0 ]]; then
  echo "This script must be run as root."
  exit 1
fi

BASE_DEV="/dev/nvme0n1"
# Detect partitions
parts=( $(ls ${BASE_DEV}p* 2>/dev/null) )
if [[ ${#parts[@]} -eq 0 ]]; then
  echo "No partitions detected on ${BASE_DEV}."
  exit 1
fi

echo "Available partitions on ${BASE_DEV}:"
for p in "${parts[@]}"; do
  num=${p##*p}
  echo "  $num -> $p"
done

# Prompt for root btrfs partition
read -rp "Enter partition number to format as Btrfs (e.g., 5 for nvme0n1p5): " sel
TARGET="${BASE_DEV}p${sel}"
if [[ ! -b $TARGET ]]; then
  echo "Partition $TARGET does not exist."
  exit 1
fi

echo "Formatting $TARGET as Btrfs..."
mkfs.btrfs -f "$TARGET"

echo "Mounting $TARGET to /mnt..."
mount "$TARGET" /mnt

# Create subvolumes
echo "Creating subvolumes..."
btrfs subvolume create /mnt/@
btrfs subvolume create /mnt/@home
btrfs subvolume create /mnt/@log
btrfs subvolume create /mnt/@cache

# Unmount and remount with options
umount /mnt

MOUNT_OPTS="ssd,noatime,compress=zstd,space_cache=v2,discard=async"
echo "Mounting @ subvolume..."
mount -o ${MOUNT_OPTS},subvol=@ "$TARGET" /mnt

# Create mount points
echo "Creating directories for subvolumes and EFI mount..."
mkdir -p /mnt/{boot/efi,home,var/log,var/cache}

# Mount other subvolumes
echo "Mounting @home subvolume..."
mount -o ${MOUNT_OPTS},subvol=@home "$TARGET" /mnt/home

echo "Mounting @log subvolume..."
mount -o ${MOUNT_OPTS},subvol=@log "$TARGET" /mnt/var/log

echo "Mounting @cache subvolume..."
mount -o ${MOUNT_OPTS},subvol=@cache "$TARGET" /mnt/var/cache

# Mount EFI partition
echo
read -rp "Enter EFI partition (e.g., /dev/nvme0n1p1): " EFI_PART
if [[ ! -b $EFI_PART ]]; then
  echo "EFI partition $EFI_PART does not exist."
  exit 1
fi

echo "Mounting EFI partition to /mnt/boot/efi..."
mount "$EFI_PART" /mnt/boot/efi

echo "All done! Subvolumes created and mounted under /mnt."