#!/bin/bash
#===============================================================================
#
# DIRECTORY:
#   /home/*/.local/share/nautilus/scripts/04-RAW/
# OR
#   /home/*/.gnome2/nautilus-sctipts/04-RAW/ (deprecated)
#
# FILE:
#   01-testdisk
#
# USAGE:
#   Right klick on a RAW image (.dd) and
#   choose this nautilus script from the context menu.
#
# OPTIONS:
#   none
#
# DESCRIPTION:
#   Recovers partition in a writable RAW image (if present/possible).
#
#
# REQUIREMENTS:
#   bash, zenity, sudo and testdisk
#
# BUGS:
#   ---
#
# NOTES:
#   Tested on
#   - Debian 8+
#   - Arch Linux
#
# AUTHOR:
#   Patrick Neumann, patrick@neumannsland.de
#
# COMPANY:
#   (privately)
#
# VERSION:
#   0.9 (beta)
#
# LINK TO THE MOST CURRENT VERSIONS:
#   https://
#
# CREATED:
#   24.03.2016
#
# COPYRIGHT (C):
#   2015-2020 - Patrick Neumann
#
# LICENSE:
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
# WARRANTY:
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# TODO:
#   - make a copy of the logfile (history)?
#
# HISTORY:
#   0.9 - Patrick Neumann - Initial (public) release
#
#===============================================================================

#-------------------------------------------------------------------------------
# Additional supported Distribution(s) (add before Library!).
#-------------------------------------------------------------------------------
SUPPORTED_OSR="arch"

#-------------------------------------------------------------------------------
# Check for library (casualscripter_nautilus-scripts_functions.sh).
#-------------------------------------------------------------------------------
readonly LIBRARY="${0%/*/*}/.casualscripter_nautilus-scripts_functions.sh"
if [ ! -f "${LIBRARY}" ] ; then
  zenity --error \
         --text \
         "ERROR: casualscripter_nautilus-scripts_functions.sh MISSING!"
  exit 1
fi

source "${LIBRARY}"

#-------------------------------------------------------------------------------
# Checks (see library "casualscripter_nautilus-scripts_functions.sh").
#-------------------------------------------------------------------------------
check_dep "${DATE_BIN}" "date"
check_dep "${RM_BIN}" "coreutils"
check_dep "${GTERMINAL_BIN}" "gnome-terminal"
check_dep "${SUDO_BIN}" "sudo"
check_dep "${TESTDISK_BIN}" "testdisk"
check_dep "${SLEEP_BIN}" "coreutils"
check_dep "${PGREP_BIN}" "procps"
check_dep "${MV_BIN}" "coreutils"

check_ext "${SOURCE}" "dd"

#-------------------------------------------------------------------------------
# Avoid unintended changes to the Image.
#-------------------------------------------------------------------------------
check_if_pwd_is_used_as_mountpoint

#-------------------------------------------------------------------------------
# ... testdisk...
#-------------------------------------------------------------------------------
readonly TESTDISK_LOG="${DIRNAME}/testdisk-$( ${DATE_BIN} "+%Y%m%dT%H%M%S" ).log"

if [ -f "${TESTDISK_LOG}" ] ; then
  ${RM_BIN} "${TESTDISK_LOG}"
fi

cd "${DIRNAME}"
${GTERMINAL_BIN} --execute ${SUDO_BIN} ${TESTDISK_BIN} /log "${SOURCE}"

${SLEEP_BIN} 3

# We have to wait until multi_sigfind has finished...
while ${PGREP_BIN} --full "${TESTDISK_BIN}" > /dev/null 2>&1; do
  ${SLEEP_BIN} 1
done

${MV_BIN} "testdisk.log" "${TESTDISK_LOG}"

#-------------------------------------------------------------------------------
# Display content of the resultfile "testdisk.log".
#-------------------------------------------------------------------------------
display_resultfile "${TESTDISK_LOG}"

exit 0