#!/bin/sh
#
# This file if part of nzbget
#
# Example postprocessing script for NZBGet
#
# Copyright (C) 2008 Peter Roubos <[email protected]>
# Copyright (C) 2008 Otmar Werner
# Copyright (C) 2008-2012 Andrey Prygunkov <[email protected]>
# Copyright (C) 2012 Antoine Bertin <[email protected]>
#
# 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 2 of the License, or
# (at your option) any later version.
#
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
#
#

#######################    Usage instructions     #######################
# o  Script will unrar downloaded rar files, join ts-files and rename img-files
#    to iso.
#
# o  To use this script with nzbget set the option "PostProcess" in
#    nzbget configuration file to point to this script file. E.g.:
#        PostProcess=/home/user/nzbget/nzbget-postprocess.sh
#
# o  The script needs a configuration file. An example configuration file
#    is provided in file "postprocess-example.conf". Put the configuration file
#    into the directory where nzbget's configuration file (nzbget.conf) or where
#    this script itself is located. Then edit the configuration file in any
#    text editor to adjust the settings.
#
# o  You can also edit the script's configuration via web-interface (requires
#    NZBGetWeb 1.4 or later). Set the options "PostProcessConfigFile" and
#    "PostProcessConfigTemplate" to point to "postprocess-example.conf"
#    (including full path). The both options are under the section
#    "CONFIGURATION OF POSTPROCESSING-SCRIPT" in NZBGetWeb.
#
# o  There are few options, which can be ajdusted for each nzb-file
#    individually. To view/edit them in web-interface click on a spanner icon
#    near the name of nzb-file.
#
# o  The script supports the feature called "delayed par-check".
#    That means it can try to unpack downloaded files without par-checking
#    them fisrt. Only if unpack fails, the script schedules par-check,
#    then unpacks again.
#    To use delayed par-check set following options in nzbget configuration file:
#        ParCheck=no
#        ParRepair=yes
#        LoadPars=one (or) LoadPars=all
#
# o  If you want to par-check/repair all files before trying to unpack them,
#    set option "ParCheck=yes".
#
####################### End of Usage instructions #######################


# NZBGet passes following arguments to postprocess-programm as environment
# variables:
#  NZBPP_DIRECTORY    - path to destination dir for downloaded files;
#  NZBPP_NZBFILENAME  - name of processed nzb-file;
#  NZBPP_PARFILENAME  - name of par-file or empty string (if no collections were
#                       found);
#  NZBPP_PARSTATUS    - result of par-check:
#                       0 = not checked: par-check disabled or nzb-file does
#                           not contain any par-files;
#                       1 = checked and failed to repair;
#                       2 = checked and successfully repaired;
#                       3 = checked and can be repaired but repair is disabled;
#  NZBPP_NZBCOMPLETED - state of nzb-job:
#                       0 = there are more collections in this nzb-file queued;
#                       1 = this was the last collection in nzb-file;
#  NZBPP_PARFAILED    - indication of failed par-jobs for current nzb-file:
#                       0 = no failed par-jobs;
#                       1 = current par-job or any of the previous par-jobs for
#                           the same nzb-files failed;
#  NZBPP_CATEGORY     - category assigned to nzb-file (can be empty string).


# Name of script's configuration file
SCRIPT_CONFIG_FILE="nzbget-postprocess.conf"

# Exit codes
POSTPROCESS_PARCHECK_CURRENT=91
POSTPROCESS_PARCHECK_ALL=92
POSTPROCESS_SUCCESS=93
POSTPROCESS_ERROR=94
POSTPROCESS_NONE=95

# Check if the script is called from nzbget
if [ "$NZBPP_DIRECTORY" = "" -o "$NZBOP_CONFIGFILE" = "" ]; then
    echo "*** NZBGet post-process script ***"
    echo "This script is supposed to be called from nzbget (0.7.0 or later)."
    exit $POSTPROCESS_ERROR
fi

# Check if postprocessing was disabled in postprocessing parameters
# (for current nzb-file) via web-interface or via command line with
# "nzbget -E G O PostProcess=no <ID>"
if [ "$NZBPR_PostProcess" = "no" ]; then
    echo "[WARNING] Post-Process: Postprocessing disabled for this nzb-file, exiting"
    exit $POSTPROCESS_NONE
fi

echo "[INFO] Post-Process: Post-process script successfully started"

# Determine the location of configuration file (it must be stored in
# the directory with nzbget.conf or in this script's directory).
ConfigDir="${NZBOP_CONFIGFILE%/*}"
ScriptConfigFile="$ConfigDir/$SCRIPT_CONFIG_FILE"
if [ ! -f "$ScriptConfigFile" ]; then
    ConfigDir="${0%/*}"
    ScriptConfigFile="$ConfigDir/$SCRIPT_CONFIG_FILE"
fi
if [ ! -f "$ScriptConfigFile" ]; then
    echo "[ERROR] Post-Process: Configuration file $ScriptConfigFile not found, exiting"
    exit $POSTPROCESS_ERROR
fi

# Readg configuration file
while read line; do eval "$line"; done < $ScriptConfigFile

# Check nzbget.conf options
BadConfig=0

if [ "$NZBOP_ALLOWREPROCESS" = "yes" ]; then
    echo "[ERROR] Post-Process: Please disable option \"AllowReProcess\" in nzbget configuration file"
    BadConfig=1
fi

if [ "$NZBOP_LOADPARS" = "none" ]; then
    echo "[ERROR] Post-Process: Please set option \"LoadPars\" to \"One\" or \"All\" in nzbget configuration file"
    BadConfig=1
fi

if [ "$NZBOP_PARREPAIR" = "no" ]; then
    echo "[ERROR] Post-Process: Please set option \"ParRepair\" to \"Yes\" in nzbget configuration file"
    BadConfig=1
fi

if [ "$BadConfig" -eq 1 ]; then
    echo "[ERROR] Post-Process: Existing because of not compatible nzbget configuration"
    exit $POSTPROCESS_ERROR
fi

# Check if all collections in nzb-file were downloaded
if [ ! "$NZBPP_NZBCOMPLETED" -eq 1 ]; then
    echo "[INFO] Post-Process: Not the last collection in nzb-file, exiting"
    exit $POSTPROCESS_SUCCESS
fi

# Check par status
if [ "$NZBPP_PARSTATUS" -eq 1 -o "$NZBPP_PARSTATUS" -eq 3 -o "$NZBPP_PARFAILED" -eq 1 ]; then
    if [ "$NZBPP_PARSTATUS" -eq 3 ]; then
        echo "[WARNING] Post-Process: Par-check successful, but Par-repair disabled, exiting"
    else
        echo "[WARNING] Post-Process: Par-check failed, exiting"
    fi
    exit $POSTPROCESS_ERROR
fi

# Check if destination directory exists (important for reprocessing of history items)
if [ ! -d "$NZBPP_DIRECTORY" ]; then
    echo "[ERROR] Post-Process: Nothing to post-process: destination directory $NZBPP_DIRECTORY doesn't exist"
    exit $POSTPROCESS_ERROR
fi

cd "$NZBPP_DIRECTORY"

# If not just repaired and file "_brokenlog.txt" exists, the collection is damaged
# exiting with returning code $POSTPROCESS_PARCHECK_ALL to request par-repair
if [ ! "$NZBPP_PARSTATUS" -eq 2 ]; then
    if [ -f "_brokenlog.txt" ]; then
        if (ls *.[pP][aA][rR]2 >/dev/null 2>&1); then
            echo "[INFO] Post-Process: Brokenlog found, requesting par-repair"
            exit $POSTPROCESS_PARCHECK_ALL
        fi
    fi
fi

# All checks done, now processing the files

# Flag indicates that something was unrared
Unrared=0

# Unrar the files (if any) to the temporary directory, if there are no rar files this will do nothing
if (ls *.rar >/dev/null 2>&1); then

    # Check if unrar exists
    $UnrarCmd >/dev/null 2>&1
    if [ "$?" -eq 127 ]; then
        echo "[ERROR] Post-Process: Unrar not found. Set the path to unrar in script's configuration"
        exit $POSTPROCESS_ERROR
    fi

    # Make a temporary directory to store the unrarred files
    ExtractedDirExists=0
    if [ -d $ExtractedDir ]; then
        ExtractedDirExists=1
    else
        mkdir $ExtractedDir
    fi

    echo "[INFO] Post-Process: Unraring"
    rarpasswordparam=""
    if [ "$NZBPR_Password" != "" ]; then
        rarpasswordparam="-p$NZBPR_Password"
    fi

    $UnrarCmd x -y -p- "$rarpasswordparam" -o+ "*.rar"  ./$ExtractedDir/
    if [ "$?" -eq 3 ]; then
        echo "[ERROR] Post-Process: Unrar failed"
        if [ "$ExtractedDirExists" -eq 0 ]; then
            rm -R $ExtractedDir
        fi
        # for delayed par-check/-repair at least one par-file must be already downloaded
        if (ls *.[pP][aA][rR]2 >/dev/null 2>&1); then
            echo "[INFO] Post-Process: Requesting par-repair"
            exit $POSTPROCESS_PARCHECK_ALL
        fi
        exit $POSTPROCESS_ERROR
    fi
    Unrared=1

    # Remove the rar files
    if [ "$DeleteRarFiles" = "yes" ]; then
        echo "[INFO] Post-Process: Deleting rar-files"
        rm *.r[0-9][0-9] >/dev/null 2>&1
        rm *.rar >/dev/null 2>&1
        rm *.s[0-9][0-9] >/dev/null 2>&1
    fi

    # Go to the temp directory and try to unrar again.
    # If there are any rars inside the extracted rars then these will no also be unrarred
    cd $ExtractedDir
    if (ls *.rar >/dev/null 2>&1); then
        echo "[INFO] Post-Process: Unraring (second pass)"
        $UnrarCmd x -y -p- -o+ "*.rar"

        if [ "$?" -eq 3 ]; then
            echo "[INFO] Post-Process: Unrar (second pass) failed"
            exit $POSTPROCESS_ERROR
        fi

        # Delete the Rar files
        if [ "$DeleteRarFiles" = "yes" ]; then
            echo "[INFO] Post-Process: Deleting rar-files (second pass)"
            rm *.r[0-9][0-9] >/dev/null 2>&1
            rm *.rar >/dev/null 2>&1
            rm *.s[0-9][0-9] >/dev/null 2>&1
        fi
    fi

    # Move everything back to the Download folder
    mv * ..
    cd ..
    rmdir $ExtractedDir
fi

# If download contains only nzb-files move them into nzb-directory
# for further download
# Check if command "wc" exists
wc -l . >/dev/null 2>&1
if [ "$?" -ne 127 ]; then
    AllFilesCount=`ls -1 2>/dev/null | wc -l`
    NZBFilesCount=`ls -1 *.nzb 2>/dev/null | wc -l`
    if [ "$AllFilesCount" -eq "$NZBFilesCount" ]; then
        echo "[INFO] Moving downloaded nzb-files into incoming nzb-directory for further download"
        mv *.nzb $NZBOP_NZBDIR
    fi
fi

# Clean up
echo "[INFO] Post-Process: Cleaning up"
chmod -R a+rw .
rm *.nzb >/dev/null 2>&1
rm *.sfv >/dev/null 2>&1
rm *.1 >/dev/null 2>&1
rm _brokenlog.txt >/dev/null 2>&1
if [ "$Unrared" -eq 1 ]; then
    # Delete par2-file only if there were files for unpacking.
    rm *.[pP][aA][rR]2 >/dev/null 2>&1
fi

if [ "$JoinTS" = "yes" ]; then
    # Join any split .ts files if they are named xxxx.0000.ts xxxx.0001.ts
    # They will be joined together to a file called xxxx.0001.ts
    if (ls *.ts >/dev/null 2>&1); then
        echo "[INFO] Post-Process: Joining ts-files"
        tsname=`find . -name "*0001.ts" |awk -F/ '{print $NF}'`
        cat *0???.ts > ./$tsname
    fi

    # Remove all the split .ts files
    echo "[INFO] Post-Process: Deleting source ts-files"
    rm *0???.ts >/dev/null 2>&1
fi

if [ "$RenameIMG" = "yes" ]; then
    # Rename img file to iso
    # It will be renamed to .img.iso so you can see that it has been renamed
    if (ls *.img >/dev/null 2>&1); then
        echo "[INFO] Post-Process: Renaming img-files to iso"
        imgname=`find . -name "*.img" |awk -F/ '{print $NF}'`
        mv $imgname $imgname.iso
    fi
fi

############################
### BEGIN CUSTOMIZATIONS ###
############################

# Move categories to /share/yourdirectory and remove download destination directory
if [ "$NZBPP_CATEGORY" = "Series" ]; then # Change here.
        echo "[INFO] Post-Process: Moving TV shows to /volume1/series" # Change here.
        cp -R "$NZBPP_DIRECTORY" "/volume1/series" >/dev/null 2>&1 # Change here.
        if [ "$?" -ne 0 ]; then
           echo "[ERROR] Post-Process: Moving to /volume1/series failed" # Change here.
           exit $POSTPROCESS_ERROR
        else
           rm -fr *
           cd ..
           rmdir "$NZBPP_DIRECTORY"
           NZBPP_DIRECTORY="/volume1/series"
        fi
fi

if [ "$NZBPP_CATEGORY" = "Movies" ]; then # Change here.
        echo "[INFO] Post-Process: Moving Movies to /volume2/downloaded" # Change here.
        cp -R "$NZBPP_DIRECTORY" "/volume2/downloaded" >/dev/null 2>&1 # Change here.
        if [ "$?" -ne 0 ]; then
           echo "[ERROR] Post-Process: Moving to /volume2/downloaded" # Change here.
           exit $POSTPROCESS_ERROR
        else
           rm -fr *
           cd ..
           rmdir "$NZBPP_DIRECTORY"
           NZBPP_DIRECTORY="/volume2/downloaded"
        fi
fi

##########################
### END CUSTOMIZATIONS ###
##########################

if [ "$SickBeard" = "yes" -a "$NZBPP_CATEGORY" = "$SickBeardCategory" -a -e "$SabToSickBeard" ]; then
    # Call SickBeard's postprocessing script
    echo "[INFO] Post-Process: Running SickBeard's postprocessing script"
    $PythonCmd $SabToSickBeard "$NZBPP_DIRECTORY" "$NZBPP_NZBFILENAME" >/dev/null 2>&1
fi

if [ "$CouchPotato" = "yes" -a "$NZBPP_CATEGORY" = "$CouchPotatoCategory" -a -e "$SabToCouchPotato" ]; then
    # Call CouchPotato's postprocessing script
    echo "[INFO] Post-Process: Running CouchPotato's postprocessing script"
    $PythonCmd $SabToCouchPotato "$NZBPP_DIRECTORY" "$NZBPP_NZBFILENAME" >/dev/null 2>&1
fi

# Check if destination directory was set in postprocessing parameters
# (for current nzb-file) via web-interface or via command line with
# "nzbget -E G O DestDir=/new/path <ID>"
if [ "$NZBPR_DestDir" != "" ]; then
    mkdir $NZBPR_DestDir
    mv * $NZBPR_DestDir >/dev/null 2>&1
    cd ..
    rmdir $NZBPP_DIRECTORY
fi

# All OK, requesting cleaning up of download queue
exit $POSTPROCESS_SUCCESS