#!/bin/bash
#
# v1.0.3 nwgat
#
# Credits: A heavily modified version of this idea and script http://www.torrent-invites.com/showthread.php?t=132965 towards a simplified end user experience.
# Authors: Lordhades - Adamaze - userdocs
# Script URL: https://git.io/v6Mza
# wget -qO ~/lftpsync.sh https://git.io/v6Mza
#
#### Editing option 0 is only required if you need to specify the tmp directory. This is the location the lock file, PID file and log file are created and checked by this script.
### Editing options 1 is only required if you have set have a private key you wish to use
## Editing options 2 - 5 is required.
# Editing options 6 - 10 is optional.
#
# 0: Optional - This variable specifies the location of the tmp folder to use for the lock, PID and log file. This default should be working on both linux and windows at the same time. On Linux by using the /tmp folder and Windows by using the included tmp folder in the lftp directory. You should not need to change this unless you want to specify the tmp directory for debugging.
tmpdir="/tmp"
# 1: Optional - This variable will specify the path to the key folder where the script will look for your ssh private keyfiles. You will probably need to change this on linux if you are using private keys to something relative like ~/.ssh
keydirectory="/keys"
# If you place a private key in the key folder you can give the script the exact name of this file here including the file extension if applicable.
keyname=""
# 2: Your sftp/ftp username goes here replacing username inside the double quotes.
username="USERNAME"
# 3: Your sftp/ftp password. If you have set up a private key file then you can ignore this variable and leave it as it is.
password="PASSWORD"
# 4: Your seedbox server URL/hostname
hostname="SERVER.SEEDBOX.COM"
# 5: The remote directory on the seedbox you wish to mirror from. Can now be passed to the script directly using "$1". It must exist on the remote server.
remote_dir='/REMOTE/DIR'
# 6: Optional - The local directory your files will be mirrored to. It is relative to the portable folder and will be created if it does not exist so the default setting will work.
local_dir="/LOCAL/DIR"
# 7: Optional - Set the SSH port if yours is not the default.
port="22"
# 8: Optional - The number of parallel files to download. It is set to download 1 file at a time.
parallel="1"
# 9: Optional - set maximum number of connections lftp can open
default_pget="20"
# 10: Optional - Set the number of connections per file lftp can open
pget_mirror="20"
# 11: Optional - Add custom arguments or flags here.
#args="-c -e --only-newer --ignore-time"
#
[[ ! -z "$1" ]] && remote_dir="$1"
#
base_name="$(basename "$0")"
lock_file="$tmpdir/$base_name.lock"
#
# This checks to see if LFTP is actually running and if the lock file exists. It LFTP is not running and there is a lock file it will be automatically cleared allowing the script to run.
[[ -z $(ps -p $(sed -rn 's/\[(.*)\](.*)/\1/p;1q' $tmpdir/$base_name.PID 2> /dev/null) 2> /dev/null | awk 'FNR==2{print $1}') ]] && rm -f "$tmpdir/$base_name.PID" "$lock_file" "$tmpdir/$base_name.log"
#
trap "rm -f $lock_file" SIGINT SIGTERM
#
if [[ -f "$lock_file" ]]
#
then
    echo "$base_name is running already."
    exit
else
    touch "$lock_file"
    lftp -e "debug -Tpo $tmpdir/$base_name.PID 0;set sftp:connect-program ssh -a -x -i $keydirectory/$keyname" -p "$port" -u "$username,$password" "sftp://$hostname" <<-EOF
    set sftp:auto-confirm yes
    set mirror:parallel-transfer-count "$parallel"
    set pget:default-n $default_pget
    set mirror:use-pget-n $pget_mirror
    mirror $args --log="$tmpdir/$base_name.log" "$remote_dir/Games" "$local_dir/Games"
    mirror $args --log="$tmpdir/$base_name.log" "$remote_dir/Misc" "$local_dir/Misc"
    mirror $args --log="$tmpdir/$base_name.log" "$remote_dir/Movies" "$local_dir/Movies"
    mirror $args --log="$tmpdir/$base_name.log" "$remote_dir/Music" "$local_dir/Music"
    mirror $args --log="$tmpdir/$base_name.log" "$remote_dir/Software" "$local_dir/Software"
    mirror $args --log="$tmpdir/$base_name.log" "$remote_dir/TV Shows" "$local_dir/TV Shows"
    quit
    EOF
    #
    rm -f "$tmpdir/$base_name.PID" "$lock_file" "$tmpdir/$base_name.log"
    #
    trap - SIGINT SIGTERM
    exit
fi