#!/bin/bash
export LANG=en_US.UTF-8
export LC_ALL=en_US.UTF-8
# ==DIS/AUTOEXEC==
# === Create Img ===
# create_image.sh — создание ограниченного образа с выбранного устройства Raspberry Pi
# с возможностью сохранить на внешний диск или сетевой ресурс (NAS)
 
set -euo pipefail
 
# --- Определение источника ---
echo "🗂️ Доступные источники (разделы с rootfs):"
mapfile -t SOURCES < <(
  lsblk -nrpo NAME,SIZE,FSTYPE,MOUNTPOINT | awk '/ext4/ {
    mount=$4; label=""
    if (mount == "/") label=" (текущая система)"
    printf("%s:%s:%s:%s%s\n", $1, $2, $3, mount, label)
  }'
)
 
if [[ ${#SOURCES[@]} -eq 0 ]]; then
  echo "❌ Нет доступных источников."
  exit 1
fi
 
i=1
echo ""
for src in "${SOURCES[@]}"; do
  IFS=":" read -r dev size fs mp label <<< "$src"
  echo "$i - $dev ($size, $fs, $mp)$label"
  ((i++))
done
 
echo "0 - Отмена"
read -rp "🔍 Выберите устройство-источник: " choice
[[ "$choice" == "0" || -z "$choice" ]] && echo "❎ Отменено." && exit 0
(( choice < 1 || choice > ${#SOURCES[@]} )) && echo "❌ Неверный выбор." && exit 1
 
SEL_SRC="${SOURCES[$((choice-1))]}"
IFS=":" read -r SRC_DEV SRC_SIZE SRC_FS SRC_MP SRC_LABEL <<< "$SEL_SRC"
 
if [[ "$SRC_DEV" =~ ^(/dev/.+p[0-9]+)$ ]]; then
  SRC_PARENT=$(lsblk -no PKNAME "$SRC_DEV" | head -n1)
  SRC_PARENT="/dev/$SRC_PARENT"
else
  SRC_PARENT="$SRC_DEV"
fi
 
# --- Выбор цели ( диск/каталог ) ---
NAS_PATH="/mnt/backup_nas"
if mountpoint -q "$NAS_PATH"; then
  mapfile -t DEVICES < <(lsblk -nrpo NAME,SIZE,MOUNTPOINT | grep -E "/dev/sd|/dev/nvme" | awk '{print $1":"$2":"$3}')
  DEVICES+=("$NAS_PATH::NAS")
else
  mapfile -t DEVICES < <(lsblk -nrpo NAME,SIZE,MOUNTPOINT | grep -E "/dev/sd|/dev/nvme" | awk '{print $1":"$2":"$3}')
fi
 
if [[ ${#DEVICES[@]} -eq 0 ]]; then
  echo "❌ Нет доступных устройств."
  exit 1
fi
 
echo "💿 Выберите место для сохранения образа:"
i=1
for dev in "${DEVICES[@]}"; do
  IFS=":" read -r name size mnt <<< "$dev"
  label=""
  [[ "$mnt" == "NAS" ]] && label="🛧 сетевой диск"
  [[ -z "$mnt" && -z "$label" ]] && label="🔌 не примонтировано"
  echo "$i - $name $size $mnt $label"
  ((i++))
done
 
echo "0 - Отмена"
read -rp "🔍 Ваш выбор: " dchoice
[[ "$dchoice" == "0" || -z "$dchoice" ]] && echo "❎ Отменено." && exit 0
(( dchoice < 1 || dchoice > ${#DEVICES[@]} )) && echo "❌ Неверный выбор." && exit 1
 
SEL_DEV="${DEVICES[$((dchoice-1))]}"
IFS=":" read -r DEV_NAME DEV_SIZE DEV_MNT <<< "$SEL_DEV"
TEMP_MOUNTED=false
 
if [[ "$DEV_NAME" == "$NAS_PATH" ]]; then
  TARGET_DIR="$NAS_PATH"
else
  MOUNTED_PATH=$(lsblk -nrpo NAME,MOUNTPOINT "$DEV_NAME" | awk '$2!="" {print $2; exit}')
  if [[ -n "$MOUNTED_PATH" ]]; then
    TARGET_DIR="$MOUNTED_PATH"
  else
    i=0
    while true; do
      MNT_PATH="/mnt/rpi_target$i"
      if ! mountpoint -q "$MNT_PATH"; then
        break
      fi
      ((i++))
    done
    sudo mkdir -p "$MNT_PATH"
    FIRST_PART=$(lsblk -nrpo NAME "$DEV_NAME" | grep -v "^$DEV_NAME" | head -n1)
    if [[ -z "$FIRST_PART" ]]; then
      echo "❌ Не найден раздел."
      exit 1
    fi
    if sudo mount "$FIRST_PART" "$MNT_PATH"; then
      TARGET_DIR="$MNT_PATH"
      TEMP_MOUNTED=true
    else
      echo "❌ Не удалось примонтировать."
      exit 1
    fi
  fi
fi
 
# --- Выбор имени файла ---
DATE=$(date +%F_%H-%M-%S)
echo ""
echo "📁 Выберите способ задания имени файла:"
echo "1 - 📅 Использовать стандартное имя ($DATE.img)"
echo "2 - ✏️  Ввести своё имя вручную (по умолчанию: $DATE)"
echo "0 - ❌ Отмена"
read -rp "👉 Ваш выбор: " name_choice
case "$name_choice" in
  1)
    IMG_NAME="$DATE.img"
    ;;
  2)
    echo -n "📝 Имя для создаваемого образа [по умолчанию: $DATE]: "
    read -r CUSTOM_NAME
    IMG_NAME="${CUSTOM_NAME:-$DATE}"
    [[ ! "$IMG_NAME" =~ \.img$ ]] && IMG_NAME+=".img"
    ;;
  0|"")
    echo "❎ Отменено."
    exit 0
    ;;
  *)
    echo "❌ Неверный выбор."
    exit 1
    ;;
esac
 
FINAL_IMG="$TARGET_DIR/$IMG_NAME"
 
END_SECTOR=$(sudo fdisk -l "$SRC_PARENT" | awk -v dev="$SRC_DEV" '$1 ~ dev { end=$3 } END { print end }')
SECTOR_SIZE=$(sudo blockdev --getss "$SRC_PARENT")
TOTAL_BYTES=$(( (END_SECTOR + 1) * SECTOR_SIZE ))
TOTAL_MB=$(( (TOTAL_BYTES + 1024*1024 - 1) / (1024*1024) + 16 ))
 
echo "[*] Снимаю образ: $FINAL_IMG ($TOTAL_MB МБ)..."
sudo dd if="$SRC_PARENT" of="$FINAL_IMG" bs=1M count="$TOTAL_MB" status=progress
sync
 
if sudo fdisk -l "$FINAL_IMG" >/dev/null 2>&1; then
  echo "✅ Образ сохранён: $FINAL_IMG"
else
  echo "❌ Образ создан, но таблица разделов не читается."
  exit 1
fi
 
$TEMP_MOUNTED && sudo umount "$TARGET_DIR"
exit 0