#!/bin/sh

# 確認
zenity --question --no-wrap --title="ブートセクタの保存ツール" --text="これからブートセクタ(MBR,PBR)のバックアップを行います。\n今、Ubuntuで起動している場合には、OKを押してください。"
if [ $? != "0" ]; then
    zenity --info  --title="ブートセクタの保存ツール" --text="キャンセルされました。\nブートセクタの保存ツールを終了します。"
    exit 1
fi

# 保存先チェック
if [ -e $HOME/bootsect ]; then
    zenity --question --no-wrap --title="ブートセクタの保存ツール" --text="$HOME/bootsect ディレクトリが既に存在します。\n\nこのディレクトリの中の、以前のブートセクタのバックアップファイルやログファイル（list.txt)を\n削除（上書き）しても良いですか？\n\n良ければ、OKを押してください。"
    if [ $? != "0" ]; then
        zenity --info --no-wrap --title="ブートセクタの保存ツール" --text="$HOME/bootsect ディレクトリの\n名前を変更してから再度このツールを実行してください。\n\nブートセクタの保存ツールを終了します。"
        exit 2
    fi
else
    mkdir $HOME/bootsect
fi

# 確認
zenity --info --no-wrap --title="ブートセクタの保存ツール" --text="$HOME/bootsect ディレクトリの中にブートセクタを保存します。"

# 管理者権限取得
gksudo -m "ディスクの情報を集めるのに管理者権限が必要となります。ご自身のパスワードを入力して下さい。" echo > /dev/null

# MBR取得
DEVICELIST=$(sudo fdisk -l | grep /dev/sd | grep ディスク | cut -d" " -f2 | cut -d: -f1)
for a in $DEVICELIST
do
    DISKID1=$(sudo fdisk -l | grep -A 3 "ディスク $a" | tail -n 1 | tr -d " ")
    NAME1=$(echo $a | cut -d/ -f3)
    sudo dd if=$a of=$HOME/bootsect/${NAME1}mbr bs=512 count=1 2> /dev/null && zenity --info --no-wrap --title="ブートセクタの保存ツール" --text="\nディスク ${a}:${DISKID1} のMBRが\n${NAME1}mbr にバックアップされました。" && echo "${DISKID1} ${NAME1}mbr" > $HOME/bootsect/list.txt
done

# 管理者権限取得
gksudo -m "ディスクの情報を集めるのに管理者権限が必要となります。ご自身のパスワードを入力して下さい。" echo > /dev/null

# PBR取得
DEVICELIST=$(sudo fdisk -l | grep /dev/sd | grep -v ディスク | cut -d" " -f1)
for a in $DEVICELIST
do
    NAME1=$(echo $a | cut -d\/ -f3)
    FILESYSTEMTYPE=$(sudo fdisk -l | grep $a | grep -v ディスク | tr -d \* | tr -s " " | cut -d" " -f5)
    if [ $FILESYSTEMTYPE != 82 ] && [ $FILESYSTEMTYPE != 5 ];
    then
        UUID1=$(sudo vol_id --export $a | grep ID_FS_UUID= | cut -c12-)
        sudo dd if=$a of=$HOME/bootsect/${NAME1}pbr bs=512 count=1 2>/dev/null && zenity --info --no-wrap --title="ブートセクタの保存ツール" --text="\nパーティション ${a}:UUID=${UUID1} のPBRが\n${NAME1}pbr にバックアップされました。"; echo "$UUID1 ${NAME1}pbr" >> $HOME/bootsect/list.txt
    fi
done

# 終了
zenity --info --no-wrap --title="ブートセクタの保存ツール" --text="作業記録が bootsect ディレクトリ（フォルダ）の list.txt に保存されました。\n\nCDからの起動の場合、システムを終了すると bootsect ディレクトリごと、\nその中に保存されたブートセクタのバックアップは無くなってしまい、危険です。\n\n$HOME ディレクトリを表示しますので\nbootsect ディレクトリを内蔵ハードディスクやUSBメディアなどにコピーしておいて下さい。\n\nOKを押すと、ブートセクタの保存ツールを終了します。お疲れさまでした。"

# ホームを開く
nautilus $HOME

exit 0