apt-get update
apt-get install -y \
  grub-pc grub-pc-bin \
  grub-efi-amd64 grub-efi-amd64-bin shim-signed os-prober

grub-install --target=i386-pc --boot-directory=/boot /dev/sdX

grub-install --target=x86_64-efi \
  --efi-directory=/boot/efi --boot-directory=/boot \
  --recheck --removable

update-grub  # 或 grub-mkconfig -o /boot/grub/grub.cfg

#!/usr/bin/env bash
# portable_boot.sh
# 目的：把一顆 Linux 磁碟（Debian/Ubuntu/Pop!_OS）做成「可攜、獨立、雙開（UEFI+BIOS）」。
# 功能：
#  1) 自動偵測磁碟與分割（ESP/bios_grub/root）
#  2) 若「缺 bios_grub」，在 GPT 磁碟的「開頭或結尾」自動建立 2MiB 的 BIOS Boot 分割（flags=bios_grub）
#  3) 安裝 GRUB（i386-pc + x86_64-efi），並停用 os-prober 讓選單僅保留本機系統（獨立）
#  4) 備份開機資產（MBR/GPT前置、分割表、ESP、/boot、grub、kernelstub）
#  5) 切換獨立/非獨立（關/開 os-prober）
# 使用：
#   sudo ./portable_boot.sh           # 正式執行
#   sudo ./portable_boot.sh --dry-run # 預演，不寫入

set -Eeuo pipefail
shopt -s extglob

# --- 參數：--dry-run 會預演所有步驟，不改動系統 ---
DRY=false; [[ "${1-}" == "--dry-run" ]] && DRY=true && echo "🟡 DRY-RUN：預演模式（不會寫入磁碟）"

# --- 小工具：統一跑指令並顯示 ---
RUN(){ echo "👉 $*"; $DRY || eval "$@"; }

# --- 收尾：確保掛載都卸掉，避免殘留 ---
cleanup(){
  echo "🧹 收尾卸載..."
  for d in dev proc sys run; do mountpoint -q "/mnt/$d" && umount "/mnt/$d" || true; done
  mountpoint -q /mnt/boot/efi && umount /mnt/boot/efi || true
  mountpoint -q /mnt && umount /mnt || true
}
trap cleanup EXIT

err(){ echo "❌ $*" >&2; exit 1; }

# ========== 共用輔助 ==========
list_disks(){ echo "📀 可用磁碟（僅列出 TYPE=disk）："; lsblk -d -o NAME,SIZE,MODEL,TYPE | awk '$4=="disk"{printf("  %s  %8s  %s\n",$1,$2,$3)}'; }
pick_disk(){
  list_disks
  read -rp "請輸入『目標磁碟』（例：sda 或 nvme0n1）： " X
  [[ -z "${X:-}" ]] && err "未輸入磁碟"
  DISK="/dev/$X"; [[ -b "$DISK" ]] || err "找不到磁碟 $DISK"
  echo "🧭 目標磁碟：$DISK"
}
scan_parts(){
  echo "🔎 掃描分割：$DISK"
  lsblk -o NAME,SIZE,FSTYPE,PARTLABEL,PARTFLAGS,MOUNTPOINT "$DISK"
  # 偵測 ESP（vfat，且 flags 含 boot/esp）
  ESP_PART="$(lsblk -prno NAME,FSTYPE,PARTFLAGS "$DISK" | awk '/ vfat / && /boot|esp/ {print $1; exit}')"
  # 偵測 bios_grub（flags 含 bios_grub）
  BIOS_PART="$(lsblk -prno NAME,PARTFLAGS "$DISK" | awk '/bios_grub/ {print $1; exit}')"
  # 偵測 root 候選（取同顆磁碟最大的 ext4）
  ROOT_PART="$(lsblk -prbo NAME,SIZE,FSTYPE "$DISK" | awk '$3=="ext4"{print $1" "$2}' | sort -k2 -nr | head -1 | awk '{print $1}')"
  echo "   ➤ ESP(UEFI)：${ESP_PART:-<未找到>}"
  echo "   ➤ BIOS區   ：${BIOS_PART:-<未找到>}"
  echo "   ➤ ROOT候選 ：${ROOT_PART:-<未找到>}"
  # 允許手動覆蓋
  read -rp "覆蓋 ESP（直接輸入 /dev/XXX；留空=使用偵測）： " a; [[ -n "${a:-}" ]] && ESP_PART="$a"
  read -rp "覆蓋 BIOS（/dev/XXX；留空=使用偵測）： " a; [[ -n "${a:-}" ]] && BIOS_PART="$a"
  read -rp "覆蓋 ROOT（/dev/XXX；留空=使用偵測）： " a; [[ -n "${a:-}" ]] && ROOT_PART="$a"
  [[ -n "${ROOT_PART:-}" ]] || err "找不到 root 分割（ext4）。"
  [[ -n "${ESP_PART:-}" ]] || err "找不到 ESP（FAT32＋boot,esp）。請先用 GParted 建立。"
}
ensure_mounts(){
  echo "🗂 掛載 ROOT 與 EFI（準備 chroot）"
  RUN "umount -R /mnt 2>/dev/null || true"
  RUN "mount $ROOT_PART /mnt"
  RUN "mkdir -p /mnt/boot/efi"
  RUN "mount $ESP_PART /mnt/boot/efi"
  for d in dev proc sys run; do RUN "mount --bind /$d /mnt/$d"; done
}
stop_apt_locks(){
  echo "⏸️ 停用自動更新守護程式（避免 apt/dpkg 鎖）"
  RUN "systemctl stop packagekit 2>/dev/null || true"
  RUN "systemctl stop apt-daily.service apt-daily-upgrade.service 2>/dev/null || true"
}
fix_dpkg(){
  echo "🔧 修復 dpkg/apt 狀態並更新索引"
  RUN "chroot /mnt /bin/bash -c 'dpkg --configure -a || true; apt-get -y -f install || true; apt-get update'"
}
install_grub_tools(){
  echo "📦 安裝 GRUB 工具（盡量同時準備 BIOS+UEFI；失敗則退而求其次）"
  if ! $DRY; then
    chroot /mnt /bin/bash -c "DEBIAN_FRONTEND=noninteractive apt-get install -y grub-pc grub-pc-bin grub-efi-amd64 grub-efi-amd64-bin shim-signed" || \
    chroot /mnt /bin/bash -c "DEBIAN_FRONTEND=noninteractive apt-get install -y grub-pc grub-pc-bin || true"
  else
    echo "（DRY）略過 apt 安裝"
  fi
}
disable_os_prober(){
  echo "🔒 停用 os-prober（讓 GRUB 『只顯示本機系統』，完全獨立）"
  RUN "chroot /mnt /bin/bash -c \"sed -i 's/^#\\?GRUB_DISABLE_OS_PROBER=.*/GRUB_DISABLE_OS_PROBER=true/' /etc/default/grub; grep -q GRUB_DISABLE_OS_PROBER /etc/default/grub || echo GRUB_DISABLE_OS_PROBER=true >> /etc/default/grub\""
}
enable_os_prober(){
  echo "🔓 開啟 os-prober（GRUB 會列出其他硬碟上的系統）"
  RUN "chroot /mnt /bin/bash -c \"sed -i 's/^#\\?GRUB_DISABLE_OS_PROBER=.*/GRUB_DISABLE_OS_PROBER=false/' /etc/default/grub; grep -q GRUB_DISABLE_OS_PROBER /etc/default/grub || echo GRUB_DISABLE_OS_PROBER=false >> /etc/default/grub\""
}
write_grub(){
  echo "💾 安裝 GRUB：BIOS + UEFI"
  local disk="/dev/$(lsblk -no pkname "$ROOT_PART")"
  [[ -b "$disk" ]] || err "無法推得磁碟於 $ROOT_PART"
  # BIOS（需要 bios_grub）
  if [[ -n "${BIOS_PART:-}" ]]; then
    RUN "chroot /mnt grub-install --target=i386-pc --boot-directory=/boot $disk"
  else
    echo "⚠️ 未找到 bios_grub；跳過 BIOS 安裝（僅做 UEFI）"
  fi
  # UEFI
  RUN "chroot /mnt grub-install --target=x86_64-efi --efi-directory=/boot/efi --boot-directory=/boot --recheck --removable || true"
}
run_kernelstub_if_exists(){
  echo "🧰 Pop!_OS：若有 kernelstub，強制同步 kernel/initrd 到 EFI"
  RUN "chroot /mnt /bin/bash -c 'if command -v kernelstub >/dev/null 2>&1; then kernelstub --force || true; fi'"
}
gen_grub_cfg(){
  echo "⚙️ 生成 grub.cfg"
  RUN "chroot /mnt /bin/bash -c 'update-grub || grub-mkconfig -o /boot/grub/grub.cfg'"
}

# ========== 自動建立 bios_grub（若缺） ==========
# 說明：僅在「磁碟為 GPT」且「有足夠未配置空間」時嘗試建立 2MiB 分割並設 flags=bios_grub
ensure_bios_grub(){
  # 先檢查是否已有
  if [[ -n "${BIOS_PART:-}" ]]; then
    echo "✅ 已存在 bios_grub 分割：$BIOS_PART（略過建立）"
    return 0
  fi
  echo "🧱 未偵測到 bios_grub，嘗試自動建立 2MiB BIOS Boot 分割（僅限 GPT）"

  # 檢查分割表型別
  local label; label="$(parted -s "$DISK" print | awk -F: '/Partition Table/ {gsub(/^[ \t]+/,"",$2); print $2}')"
  [[ "$label" == "gpt" ]] || err "此磁碟非 GPT（目前僅支援 GPT 自動建立 bios_grub）。請改用 GParted 建立後再跑。"

  # 取得 free 區資訊（MiB 單位）
  local free_head="" free_tail=""
  # 開頭 free：查詢 print free 第一個 free 區塊
  free_head="$(parted -m "$DISK" unit MiB print free | awk -F: 'NR>2 && $1!~/^[0-9]+$/ && /free/ {gsub(/MiB/,"",$2); gsub(/MiB/,"",$3); print $2,$3; exit}')"
  # 結尾 free：取最後一個 free 區塊
  free_tail="$(parted -m "$DISK" unit MiB print free | awk -F: 'NR>2 && /free/ {last=$0} END{if(last!=""){split(last,a,":"); gsub(/MiB/,"",a[2]); gsub(/MiB/,"",a[3]); print a[2],a[3]}}')"

  create_bios_grub_at(){
    local start="$1" end="$2"
    echo "🛠️ 嘗試建立 bios_grub：${start}MiB → ${end}MiB"
    RUN "parted -s $DISK unit MiB mkpart primary $start $end"
    # 取最新分割編號（最後一個分割）
    local lastp; lastp="$(lsblk -prno NAME "$DISK" | tail -1)"
    RUN "parted -s $DISK set ${lastp##*[^0-9]} bios_grub on"
    BIOS_PART="$lastp"
    echo "✅ 建立完成：$BIOS_PART（flags=bios_grub）"
  }

  # 嘗試優先在開頭預留 1→3 MiB；若不足，再在結尾預留 2 MiB
  if [[ -n "$free_head" ]]; then
    read s e <<<"$free_head"
    # 需要至少 3MiB（保留 1→3 的小窗），避免覆蓋 GPT 保護區
    if (( $(echo "$e - $s >= 3" | bc -l) )); then
      create_bios_grub_at "1" "3"
      return 0
    fi
  fi
  if [[ -n "$free_tail" ]]; then
    read s e <<<"$free_tail"
    # 需要至少 2MiB
    if (( $(echo "$e - $s >= 2" | bc -l) )); then
      # 從尾端往回 2MiB
      local start end
      end="$(printf "%.0f" "$e")"
      start="$(( end - 2 ))"
      create_bios_grub_at "$start" "$end"
      return 0
    fi
  fi

  err "找不到可用的未分配空間（≥2MiB）可建立 bios_grub；請手動用 GParted 挪出 2MiB 再跑。"
}

# ========== 備份（含說明檔） ==========
backup_boot_assets(){
  echo "📦 建立開機資產備份（MBR/GPT前置、分割表、ESP、/boot、grub、kernelstub）"
  read -rp "備份輸出目錄（例：/mnt/usbbackup）： " DEST
  [[ -n "${DEST:-}" ]] || err "未輸入輸出目錄"
  local ts base work
  ts="$(date +%Y%m%d-%H%M%S)"
  base="boot-backup-$(basename "$DISK")-$(hostname)-$ts"
  work="/tmp/$base"
  $DRY || mkdir -p "$work" "$DEST"

  echo "🧲 備份磁碟前置區（4MiB：MBR/GPT header + 空間）"
  RUN "dd if=$DISK of=$work/${base}.first4MiB.img bs=1M count=4 status=none"

  echo "📃 匯出分割表（sfdisk）"
  RUN "sfdisk -d $DISK > $work/${base}.sfdisk.txt"

  echo "📁 備份 ESP 內容"
  $DRY || mkdir -p /mnt/esp
  RUN "mount $ESP_PART /mnt/esp"
  RUN "rsync -aHAX --delete /mnt/esp/ $work/esp/"
  RUN "umount /mnt/esp"

  echo "🧷 備份 /boot 與 grub 設定"
  $DRY || mkdir -p "$work/boot"
  [[ -d /boot ]] && RUN "rsync -aHAX /boot/ $work/boot/" || true
  [[ -f /etc/default/grub ]] && RUN "cp -a /etc/default/grub $work/" || true
  [[ -d /etc/default/grub.d ]] && RUN "rsync -aHAX /etc/default/grub.d $work/" || true

  echo "🧰 備份 kernelstub（Pop!_OS）"
  RUN "chroot /mnt /bin/bash -c '[ -f /etc/kernelstub/configuration ] && cp -a /etc/kernelstub/configuration /tmp/kernelstub.conf || true'"
  $DRY || { [ -f /mnt/tmp/kernelstub.conf ] && cp -a /mnt/tmp/kernelstub.conf "$work/kernelstub.configuration" || true; }

  echo "🗜 打包壓縮（tar.gz）"
  RUN "tar -C $work -czf $DEST/${base}.tar.gz ."
  if ! $DRY; then
    echo "✅ 備份完成：$DEST/${base}.tar.gz"
    cat > "$DEST/${base}-RESTORE.txt" <<'DOC'
還原指南：
1) 還原分割表（如需重建整顆碟）：
   sudo sfdisk /dev/sdX < boot-backup-*.sfdisk.txt
2) 還原 ESP 內容：
   sudo mount /dev/sdXn_efi /mnt/esp
   sudo rsync -aHAX --delete ./esp/ /mnt/esp/
   sudo umount /mnt/esp
3) 還原 /boot 與 grub 設定（已 chroot 到該系統）：
   sudo rsync -aHAX ./boot/ /boot/
   sudo cp -a ./grub /etc/default/grub 2>/dev/null || true
   sudo rsync -aHAX ./grub.d/ /etc/default/grub.d/ 2>/dev/null || true
4) 重裝 GRUB（雙開）：
   grub-install --target=i386-pc --boot-directory=/boot /dev/sdX
   grub-install --target=x86_64-efi --efi-directory=/boot/efi --boot-directory=/boot --recheck --removable
   update-grub || grub-mkconfig -o /boot/grub/grub.cfg
DOC
  fi
}

# ========== 動作集合 ==========
action_make_independent_dualboot(){
  pick_disk
  scan_parts
  # 自動補齊 bios_grub（若缺）
  ensure_bios_grub
  # 按偵測結果掛載並 chroot
  ensure_mounts
  stop_apt_locks
  fix_dpkg
  install_grub_tools
  disable_os_prober            # → 讓它「完全獨立」
  write_grub                   # → 安裝 BIOS + UEFI
  run_kernelstub_if_exists     # → Pop!_OS 同步 kernel 到 EFI
  gen_grub_cfg                 # → 產生 grub.cfg（只剩本機）
  echo "✅ 完成：此系統已成為『可攜、獨立、雙開（UEFI+BIOS）』"
}

action_toggle_isolation(){
  pick_disk
  scan_parts
  ensure_mounts
  read -rp "輸入 1=關閉 os-prober(獨立) / 2=開啟 os-prober(列出他系統)：[1/2] " m
  case "$m" in
    1) disable_os_prober ;;
    2) enable_os_prober ;;
    *) err "輸入無效" ;;
  esac
  gen_grub_cfg
  echo "✅ 已更新 GRUB 選單"
}

action_backup(){
  pick_disk
  scan_parts
  ensure_mounts
  backup_boot_assets
}

action_create_bios_grub_only(){
  pick_disk
  scan_parts
  ensure_bios_grub
  echo "✅ bios_grub 已就緒：$BIOS_PART"
}

# ========== 主選單 ==========
if [[ "${1-}" != "--dry-run" && -n "${1-}" ]]; then
  echo "用法：sudo $0 [--dry-run]"
  exit 1
fi

echo "=============================="
echo " 可攜獨立雙開工具 v2（Debian/Ubuntu/Pop!_OS）"
echo "=============================="
echo "1) 讓此系統成為『可攜、獨立、雙開（UEFI+BIOS）』"
echo "2) 切換『獨立/非獨立』（關/開 os-prober）"
echo "3) 備份開機資產（MBR/GPT前置、分割表、ESP、/boot、grub…）"
echo "4) 自動建立 BIOS Boot 分割（若缺，GPT 且有空間）"
echo "q) 離開"
read -rp "請選擇： " choice

case "$choice" in
  1) action_make_independent_dualboot ;;
  2) action_toggle_isolation ;;
  3) action_backup ;;
  4) action_create_bios_grub_only ;;
  q|Q) echo "bye"; exit 0 ;;
  *) err "選項無效" ;;
esac