#!/bin/bash
#/**
#         restoreBaseImage
#@brief Script de ejemplo para restaurar una imagen 
#@brief (puede usarse como base para el programa de restauración de imágenes usado por OpenGNSys Admin).
#@param 1 REPO|CACHE
#@param 2 imagen
#@param 3 disco 
#@param 4 particion 
#@param 5 protocolo 
#@param 6 opciones protocolo
#@return 
#@exception OG_ERR_FORMAT  # 1 formato incorrecto.
#@exception updateCache    # 15 No hay cache 
#@exception updateCache    # 16 no hay espacio sufiente 
#@exception "Error al montar la imagen" INTERNACIONALIZAR
#@todo: Probar rsync desde repo sin samba
#@todo: Se puede dar la opcion de que mantenga los ficheros nuevos.
#@todo: Se puede dar la opcion que la trasferencia se manden los datos comprimidos.
#@note  se toma como punto de partida el script restoreImage
#@version 1.0 - restauración de imagen con sincronizacion.
#@author  
#@date   2012-12-04
#*/ ##
trap "ogUnlock $3 $4; umount $DIRMOUNT; hose $REPOIP 2009 --out sh -c \"echo -ne UMOUNT_IMAGE $2 $IMGEXT\" "  1 2 3 6 9 14 15

# Limpiamos los archivo de log
echo "" >$OGLOGCOMMAND

TIME1=$SECONDS
PROG="$(basename $0)"
if [ $# -lt 4 ]; then
    ogRaiseError $OG_ERR_FORMAT "$MSG_FORMAT: $PROG REPO|CACHE imagen ndisco nparticion [ UNICAST|MULTICAST|TORRENT ] [opciones protocolo]"
    exit $?
fi
echo "[1] $MSG_SCRIPTS_START  $0 $*" | tee -a $OGLOGFILE

IMGEXT="img"

# Comprobamos si la imagen o la particion estan bloqueada:
ogIsImageLocked "$1" "$2.$IMGEXT" && exit $(ogRaiseError $OG_ERR_LOCKED "$1 $2.$IMGEXT"; echo $?)
ogIsLocked "$3" "$4" && exit $(ogRaiseError $OG_ERR_LOCKED  "$3 $4"; echo $?)

# Carga del configurador del engine
[ -z $OGENGINECONFIGURATE ] && source /opt/opengnsys/etc/engine.cfg

# Procesar parámetros de entrada
REPOFILE=$(ogGetPath "REPO" "$2.$IMGEXT") || exit $(ogRaiseError $OG_ERR_NOTFOUND "REPO, $2.$IMGEXT"; echo $?)

# Comprobar que es sincronizable (con REPO)
# TODO: crear mensaje de error
file $REPOFILE | grep " BTRFS Filesystem" 2>&1 > /dev/null
[ $? == 0 ]  ||  exit  $(ogRaiseError  "$1 $2"; echo $?)

PROTO=${5:-"UNICAST"}
if [ "$1" == "CACHE" -o "$1" == "cache" ]; then
   # Las imagenes sincronizables aunque sean iguales no tienen el mismo sum. 
   # Sincronizamos si existe el fichero y si no usamos updateCache.
   echo "[10] Realizando un updateCache REPO $2.img $PROTO $6" | tee -a $OGLOGFILE
   echo "      updateCache REPO" "/$2.img" "$PROTO" "$6"   | tee -a $OGLOGFILE
   updateCache REPO "/$2.img" "$PROTO" "$6"     &>> $OGLOGCOMMAND
   RETVAL=$?
   if [ "$RETVAL" != "0" ] 
        then 
        echo "[49] Fin del updateCache REPO $2.img $5 $6 con error $RETVAL" | tee -a $OGLOGFILE 
        # RC=15 No hay cache 
        # RC=16 no hay espacio sufiente 
        exit $RETVAL 
   fi 
fi 

TIME2=$SECONDS

PART=$(ogDiskToDev "$3" "$4") || exit $?

# Restaurar la imagen.
echo "[40] Restaurar imagen en $PART"  | tee -a $OGLOGFILE
DEST=$(ogMount "$3" "$4") 
# Formateamos si es necesario
if [  $? -ne 0 ] ; then
	# TODO ogGetImageInfo
	DIRMOUNT="/tmp/mount$$"
	mkdir $DIRMOUNT
	mount -o compress=lzo $REPOFILE $DIRMOUNT
	FSTYPE=$(head -1 $DIRMOUNT/ogimg.info |cut -d: -f3)	
	umount $DIRMOUNT
	echo "      Es necesario formatear la partición en $FSTYPE" | tee -a $OGLOGFILE
	ogFormat "$3" "$4" "$FSTYPE"
	ogMount "$3" "$4"  || exit $?
fi

ogLock $3 $4

# Montamos la imagen basica.
ogMountImage "$1" "$2" "img" 
#  Comprobar que la imagen completa se ha montado.
ogWaitMountImage "$1" "$2" "img" 
# Sincronizamos la partición con la imagen.
echo  "[60] Sincronizando la partición con la imagen"  |tee -a $OGLOGFILE
ogSyncRestore "$1" "$2" "img" $3 $4 

TIMEAUX2=$[SECONDS-TIME2]
echo "      Fin Restauracion: tiempo parcial : $[TIMEAUX2/60]m $[TIMEAUX2%60]s" | tee -a $OGLOGFILE

# Desmontamos imagen basica.
ogUnmountImage "$1" "$2" "img"

# Restableciendo acl
ogUnlock $3 $4
echo "[70] Restableciendo lista de control de acceso." |tee -a $OGLOGFILE
ogRestoreInfoImage $3 $4  &>>$OGLOGCOMMAND


# Llamar al script de post-configuración del sistema operativo.
echo "[90] Post-configuracion de arranque del sistema." |tee -a $OGLOGFILE
configureOs $3 $4  &>>$OGLOGCOMMAND

TIME=$[SECONDS-TIME1]
echo "[100] Duracion de la operacion $[TIME/60]m $[TIME%60]s" |tee -a $OGLOGFILE