#!/bin/bash

function install_rtorrent {

# Install rTorrent
aptitude update && aptitude -y install rtorrent

} # end function install_rtorrent


function configure_user {

# Simple check to see if rtorrent is installed
if [ ! -d "/usr/share/doc/rtorrent" ]; then 
    echo -e "\033[31;1mERROR: Please ensure that rTorrent is installed on this system. \033[0m"
    exit 1
fi
	
# Create a "watch" directory to allow automatic torrent downloads
# Also create sessions directory to store torrent session info
mkdir /home/$torrent_user/{watch,rtorrent_session}

# Optimise the config file
cat > /home/$torrent_user/.rtorrent.rc <<EOF

# Maximum and minimum number of peers to connect to per torrent
#min_peers = 40
max_peers = 100

# Same as above but for seeding completed torrents (-1 = same as downloading)
#min_peers_seed = 10
#max_peers_seed = 50

# Maximum number of simultanious uploads per torrent
max_uploads = 25

# Global upload and download rate in KiB. "0" for unlimited
download_rate = 2000
upload_rate = 1000

# Default directory to save the downloaded torrents
#directory = ./

# Default session directory.
session = ./rtorrent_session

# Watch a directory for new torrents, and stop those that have been deleted
schedule = watch_directory,5,5,load_start=./watch/*.torrent
schedule = untied_directory,5,5,stop_untied=

# Close torrents when diskspace is low.
schedule = low_diskspace,5,60,close_low_diskspace=100M

# Seed to a ratio of 2. If less than 200MB is uploaded, seed until ratio is 5
ratio.enable=
ratio.min.set=200;
ratio.max.set=500;
ratio.upload.set=200M;

# Port range to use for listening.
port_range = 40000-45000

# Start opening ports at a random position within the port range.
#port_random = no

# Check hash for finished torrents.
#check_hash = no

# Set whetever the client should try to connect to UDP trackers.
#use_udp_trackers = yes

# Enable DHT support for trackerless torrents or when all trackers are down.
# May be set to "disable" (completely disable DHT), "off" (do not start DHT),
# "auto" (start and stop DHT as needed), or "on" (start DHT immediately).
# The default is "off". For DHT to work, a session directory must be defined.
#dht = auto
#dht_port = 6881

# Enable peer exchange (for torrents not marked private)
#peer_exchange = yes

EOF

# Set proper ownership for the config file and the watch folder
chown $torrent_user:$torrent_user /home/$torrent_user/.rtorrent.rc
chown $torrent_user:$torrent_user /home/$torrent_user/watch
chown $torrent_user:$torrent_user /home/$torrent_user/rtorrent_session

# Output msg to console
echo -e "\033[35;1m A 'watch' folder has been created in the /home/$torrent_user/watch. \033[0m"
echo -e "\033[35;1m Torrent files uploaded there will be downloaded automatically. \033[0m"
echo -e "\033[35;1m Client configured to seed to a ratio of 2.0. Settings are stored in /home/$torrent_user/.rtorrent.rc \033[0m"

} # end function configure_user


#### Main program begins ####

if [ ! -n "$1" ]; then
    echo ""
    echo -e "\033[35;1mSelect from the options below to use this script:- \033[0m"
    echo -n  "$0"
    echo -ne "\033[36m install\033[0m"
    echo     " - Installs rTorrent."

    echo -n  "$0"
    echo -ne "\033[36m configure User\033[0m"
    echo     " - Configure rTorrent for \"User\"."

    echo ""
    exit
fi


case $1 in

install)
    install_rtorrent
;;

configure)
    # Check whether a linux username is entered.
	if [ $# -ne 2 ]; then
	    echo -e "\033[31;1mERROR: Please enter the required parameters.\033[0m"
		exit 1
	fi

    torrent_user=$2
	
    # Check if the torrent user exists on system. 
    if [ -d "/home/$torrent_user" ]; then
        configure_user
    else
        # If user doesn't exist, exit.
        echo -e "\033[31;1mERROR: User $torrent_user doesn't exist! \033[0m"
        exit 1
    fi
;;
esac