Subversion Repositories programming

#!/bin/bash

############################### CONFIG OPTIONS #################################

DLDIR="/home/irasnyd/downloads/usenet"              # "global" download dir  ###

GLOGDIR="/home/irasnyd/downloads/usenet/logs"       # "global" logdir        ###

UNEX_LOG="${GLOGDIR}/unextractable"                 # "global" unextractable ###

UNREP_LOG="${GLOGDIR}/unrepairable"                 # "global" unrepairable  ###

NFODIR="/data/NFO"                                  # "global" nfo storage   ###

SFVDIR="/data/NFO/sfv"                              # "global" sfv storage   ###

################################################################################

################################################################################

############################### FUNCTIONS ######################################

### Clean up temp files ###

cleanup_temp() {

    rm -f parfiles;

    rm -f goodfiles;

    rm -f completefiles;

    rm -f currentset;

    rm -f unrepairable;

    rm -f parjoinfiles;

}

### Clean up global logs ###

cleanup_logs() {

    rm -f "${UNREP_LOG}";

    rm -f "${UNEX_LOG}";

}

### Function which prints error logs ###

print_errors() {

   # Print Unrepairable

   echo

   if [ -f "${UNREP_LOG}" ]

   then

      echo '****************************************'

      echo '*** THE FOLLOWING WERE UNREPAIRABLE ****'

      echo '****************************************'

      echo

      cat "${UNREP_LOG}"

   else

      echo 'No Unrepairable Files!'

   fi

   echo

   # Print Unextractable

   if [ -f "${UNEX_LOG}" ]

   then

      echo '****************************************'

      echo '*** THE FOLLOWING WERE UNEXTRACTABLE ***'

      echo '****************************************'

      echo

      cat "${UNEX_LOG}"

   else

      echo 'No Unextractable Files!'

   fi

}

### delete the currentset ###

del_currentset() {

    while read CSNAME;

        do rm "${CSNAME}"

    done < currentset

}

### does everything that needs to be done in the dir passed to it ###

rarslave_dir() {

    OPERATING_DIR="$1"

    ORIG_DIR="$(pwd)"

    # move into the working dir

    cd "${OPERATING_DIR}"

    # print starting message

    print_working_message

    # Set/Clean up the Environment

    cleanup_temp

    # Repair everything, and sort it into "goodfiles" and "badfiles"

    find ./ -maxdepth 1 -iregex '.*\.par2$' \! -iregex  '.*\.vol[0-9]\++[0-9]\+\.par2$' -print0 \

        | xargs -0 -n1 -i rarslave "{}" "parsort"

    # When using <=findutils-4.2.20 (or thereabouts), use the following. The "+" operator

    # has changed in later versions.

    #find ./ -maxdepth 1 -iregex '.*\.par2$' \! -iregex  '.*\.vol[0-9]+\+[0-9]+\.par2$' -print0 \

    #    | xargs -0 -n1 -i rarslave "{}" "parsort"

    # Extract files, and delete the set if it extracted correctly

    sed 's/.par2$//' goodfiles | sed 's/.PAR2$//' > completefiles

    # for each FILENAME in completefiles, extract them

    while read FILENAME;

        # we are out of filenames

        if [ -z "${FILENAME}" ]; then break; fi;

        # get the set of files associated with this par2

        do ls "${FILENAME}"* | grep -i "rar$\|r..$\|par2$" > currentset;

        # remove pars from currentset, then get the first thing that

        # has "rar" or "r00" at the END only

        # store above result in EXFILE (file to extract)

        EXFILE="$(grep -vi "par2$" currentset | grep -m1 "rar$\|r00$")";

        # if there are files to extract

        if [ ! -z "${EXFILE}" ]; then

            # (extract the file AND remove the current set) OR save the filename to "unextractable"

            ( rar e -o+ "${EXFILE}" && del_currentset ) || echo "${FILENAME}" >> "${UNEX_LOG}";

        else

            echo "No File to extract, removing currentset";

            del_currentset;

        fi;

    done < completefiles

    # part for parjoin

    sed 's/.par2$//' unrepairable | sed 's/.PAR2$//' > parjoinfiles

    while read FILENAME;

        # no more filenames

        if [ -z "${FILENAME}" ]; then break; fi

        # no appropriate file attached to this par2

        if [ ! -f "${FILENAME}".001 ]; then continue; fi;

        do rarslave "${FILENAME}" "parjoin";

    done < parjoinfiles

    # Clean up temp files used

    cleanup_temp

    # print finishing message

    print_finished_message

    # go back to the starting dir

    cd "${ORIG_DIR}"

}

### the "parjoin" command ###

rarslave_parjoin() {

    PJNAME="$1"

    lxsplit -j "${PJNAME}".001                  ### Attempt to join the file

    par2repair "${PJNAME}".*???2                ### Attempt to repair the joined file

    RETURN_CODE=$?                              ### Get the return code of the repair

    if [ ${RETURN_CODE} -eq 0 ]; then           ### If the repair was successful

            rm "${PJNAME}".???*                 ### then remove all the source files

    else                                        ### If the repair was not successful

            rm "${PJNAME}"                      ### then remove the attempted file

    fi

}

### the "parsort" command. attempts to repair and sort into ###

### "goodfiles" and "unrepairable" ###

rarslave_parsort() {

    PARNAME="$1"

    par2repair "${PARNAME}" \

    && echo "${PARNAME}" >> goodfiles \

    || echo "${PARNAME}" >> unrepairable

}

### the "nfosort" command. moves all NFO's to ${NFODIR} ###

nfosort() {

    find "${DLDIR}" -iregex '.*\.nfo$' -exec mv '{}' "${NFODIR}" \;

}

### the "remove extraneous" command. removes all *.1 files (repair leftovers)

### as well as moving all sfv's to ${SFVDIR}

remove_extraneous() {

    find "${DLDIR}" -iregex '^.*\.1$' -exec rm '{}' \;

    find "${DLDIR}" -iregex '^.*\.sfv$' -exec mv '{}' "${SFVDIR}" \;

}

### prints the beginning message ###

print_working_message() {

    echo '================================================================================'

    echo "Working in $(pwd)"

    echo '================================================================================'

}

### prints the ending message ###

print_finished_message() {

    echo '================================================================================'

    echo "Done in $(pwd)"

    echo '================================================================================'

}

### print a welcome message ###

print_welcome_message() {

    echo '================================================================================'

    echo '== This is rarslave, a free program designed to assist you in repairing and   =='

    echo '== extracting files that you have downloaded from USENET. It will work on any =='

    echo '== file set with a par2 and rar. It also works for sets with just par2        =='

    echo '== associated. In that case, it repairs then removes the par2 (if successful) =='

    echo '==                                                                            =='

    echo '== Written By ---- Ira Snyder                                                 =='

    echo '== Start Date ---- 06-08-2005 (ported from zsh)                               =='

    echo '== Last Revised -- 06-08-2005 (appears not to have major bugs now)            =='

    echo '== License ------- GNU General Public License v2                              =='

    echo '== License ------- http://www.gnu.org/licenses/gpl.txt                        =='

    echo '==                                                                            =='

    echo '== Thanks for using this program!                                             =='

    echo '================================================================================'

    echo

}

############################### END FUNCTIONS ##################################

################################################################################

GLOBAL_ORIG_DIR="$(pwd)"

# if we got no args, run in $DLDIR

if [ -z "$@" ]; then

    print_welcome_message

    # switch to the dir from the config and set up the environment

    cd "${DLDIR}"

    cleanup_logs

    # run rarslave recursively on every directory . and below

    find ./ -type d -print0 | xargs -0 -n1 | xargs -I{} rarslave "{}"

    nfosort

    remove_extraneous

    print_errors

    # Print Parting Message

    echo

    echo "All Finished. Good Bye!"

# we have some args...

else

    # if there is no 2nd arg, run in the directory given by $1

    if [ -z "$2" ]; then

        rarslave_dir "$@"

    # there is a second arg...

    else

        # run the built-in "parsort" command if we got it

        if [ "$2" = "parsort" ]; then

            rarslave_parsort "$1"

        fi

        # run the built-in "parjoin" command if we got it

        if [ "$2" = "parjoin" ]; then

            rarslave_parjoin "$1"

        fi

    fi

fi

cd "${GLOBAL_ORIG_DIR}"