#!/bin/bash

DISK="/dev/sda"
VAR_MOUNT="/var"
HOME_MOUNT="/home"

VAR_SIZE_GB="4096" # 4TB in GB
HOME_SIZE_GB="10240" # 10TB in GB
 

VAR_SECTORS=$((VAR_SIZE_GB * 1024 * 1024 * 1024 / 512))
#HOME_SECTORS=$((HOME_SIZE_GB * 1024 * 1024 * 1024 / 512))

HOME_START_SECTOR=$((FIRST_SECTOR + VAR_SECTORS))

# Use fdisk to create the partitions
# echo -e "g\nn\n1\n\n\n+$VAR_SECTORS\nt\n20\nn\n2\n$HOME_START_SECTOR\n+$HOME_SECTORS\nt\n20\nw\n" | fdisk "$DISK"
echo -e "g\nn\n1\n\n\n+$VAR_SECTORS\nt\n20\nn\n2\n\n\n\n\nt\n20\nw\n" | fdisk "$DISK"

VAR_PART=/dev/sda1
HOME_PART=/dev/sda2

# Format the partitions
mkfs.ext4 -F "$VAR_PART" -L var
mkfs.ext4 -F "$HOME_PART" -L home

# Create mount points if they don't exist
mkdir -p "$VAR_MOUNT"
mkdir -p "$HOME_MOUNT"

systemctl daemon-reload

# Mount the partitions
mount "$VAR_PART" "$VAR_MOUNT"
mount "$HOME_PART" "$HOME_MOUNT"

# Add entries to /etc/fstab for persistent mounting
echo "#$VAR_PART" >> /etc/fstab
echo "UUID=$(blkid -s UUID -o value "$VAR_PART") $VAR_MOUNT ext4 defaults 0 2" >> /etc/fstab
echo "#$HOME_PART" >> /etc/fstab
echo "UUID=$(blkid -s UUID -o value "$HOME_PART") $HOME_MOUNT ext4 defaults 0 2" >> /etc/fstab

echo "Partitions created and mounted successfully."
echo "Added entries to /etc/fstab for persistent mounting."