DiskUtils.sh

#!/bin/bash

# Define colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'

log() {
  echo -e "$(date '+%Y-%m-%d %H:%M:%S') - $1"
}

# Check for sudo permissions
if [ "$EUID" -ne 0 ]; then
  log "${RED}Please run this script as root!${NC}"
  exit 1
fi

display_disk_info() {
  local disk="$1"
  log "${YELLOW}Disk Information for $disk:${NC}"
  fdisk -l "$disk"
}

image_disk() {
  fdisk -l
  echo -e -n "Enter the path of the disk to image (e.g., ${YELLOW}/dev/sda1${NC}): "
  read -r disk

  if [ ! -e "$disk" ]; then
    log "${RED}Error: Disk '$disk' not found.${NC}"
    return 1
  fi

  display_disk_info "$disk"

  echo -e -n "${YELLOW}Proceed to image the disk? [y/N] ${NC}"
  read -r response
  if [[ ! "$response" =~ ^([yY][eE][sS]|[yY])$ ]]; then
    log "${YELLOW}Disk imaging aborted.${NC}"
    return 1
  fi

  echo -e -n "Enter the path to save the image (e.g., ${YELLOW}Downloads/disk.img${NC}): "
  read -r path

  if [ ! -d "$(dirname "$path")" ]; then
    log "${RED}Error: Directory '$(dirname "$path")' does not exist.${NC}"
    return 1
  fi

  dd if="$disk" of="$path" bs=4M status=progress
  log "${GREEN}Disk imaging completed successfully.${NC}"
}

securely_erase_disk() {
  fdisk -l
  echo -e -n "Enter the disk to erase (e.g., ${YELLOW}/dev/sda1${NC}): "
  read -r disk

  if [ ! -e "$disk" ]; then
    log "${RED}Error: Disk '$disk' not found.${NC}"
    return 1
  fi

  display_disk_info "$disk"

  echo -e -n "${YELLOW}Proceed to erase the disk? [y/N] ${NC}"
  read -r response
  if [[ ! "$response" =~ ^([yY][eE][sS]|[yY])$ ]]; then
    log "${YELLOW}Disk erasure aborted.${NC}"
    return 1
  fi

  dd if=/dev/urandom of="$disk" bs=4k status=progress
  log "${GREEN}Disk securely erased successfully.${NC} To use the disk again, format it."
}

format_disk(){
    fdisk -l
    echo -e -n "Enter the disk to format (e.g., ${YELLOW}/dev/sda1${NC}): "
    read -r disk

    if [ ! -e "$disk" ]; then
      log "${RED}Error: Disk '$disk' not found.${NC}"
      return 1
    fi

    display_disk_info "$disk"

    echo -e "${YELLOW}Choose the format option:"
    echo -e "1. Format as ext4 filesystem (used for Linux)"
    echo -e "2. Format as NTFS filesystem (used for Windows)"
    echo -e "3. Format as FAT32 filesystem (used for USB drives)"
    echo -e "4. Format as exFAT filesystem (used for USB drives)"
    read -r format_choice

    case $format_choice in
      1)
        mkfs.ext4 "$disk"
        log "${GREEN}Disk formatted as ext4 filesystem successfully.${NC}"
        ;;
      2)
        mkfs.ntfs "$disk"
        log "${GREEN}Disk formatted as NTFS filesystem successfully.${NC}"
        ;;
      3)
        mkfs.fat -F32 "$disk"
        log "${GREEN}Disk formatted as FAT32 filesystem successfully.${NC}"
        ;;
      4)
        mkfs.exfat "$disk"
        log "${GREEN}Disk formatted as exFAT filesystem successfully.${NC}"
        ;;
      *)
        log "${RED}Invalid option!${NC}"
        return 1
        ;;
    esac

}

PS3='Enter your choice: '
select choice in "Image Disk" "Securely Erase Disk" "Format Disk" "Exit"
do
  case $choice in
    "Image Disk")
      image_disk
      ;;
    "Securely Erase Disk")
      securely_erase_disk
      ;;
    "Format Disk")
      format_disk
      ;;
    "Exit")
      log "${YELLOW}Exiting the script.${NC}"
      break
      ;;
    *)
      log "${RED}Invalid option!${NC}"
      break
      ;;
  esac
done