#!/bin/bash

# Gettorrent Script by alxrt

# Синтаксис:
#      ./gettorrent TOPIC_ID <-h HOST> <-u USERNAME> <-p PASSWORD> <-o filepath> 
#   TOPIC_ID: Идентификатор топика, к которому относится торент.
# -h HOST: Хост трекера. Обычно rutracker.org или pornolab.net ;-)
#     -u USERNAME: Имя пользователя
#       -p PASSWORD: Пароль пользователя
#    -o filepath: Имя сохраняемого файла

# Настройки

TR_HOST="rutracker.org"
TR_USER=""
TR_PASSWORD=""

SC_COOKIE="/tmp/gt-$TR_HOST-$TR_USER.ck"
SC_UA="Mozilla/5.0 (X11; U; Linux x86_64; ru; rv:1.9.2.15) Gecko/20110303 Ubuntu/10.10 (maverick) Firefox/3.6.15"
SC_FILEPATH=""
#SC_PROXY="" # hostname:port
#SC_PROXY_AUTH="" #user:password

if < -n "$SC_PROXY" >; then
       curl_proxy="-x $SC_PROXY -U $SC_PROXY_AUTH"
   echo $curl_proxy
fi

#Functions

die() {
echo "Синтаксис:"
    echo    "./gettorrent TOPIC_ID <-h HOST> <-u USERNAME> <-p PASSWORD> <-o filepath>"
 echo    "TOPIC_ID: Идентификатор топика, к которому относится торент."
      echo    "-h HOST: Хост трекера. Обычно rutracker.org или pornolab.net ;-)"
  echo    "-u USERNAME: Имя пользователя"
     echo    "-p PASSWORD: Пароль пользователя"
  echo    "-o filepath: Имя сохраняемого файла"
       exit
}

waitf() {
    PERIOD="1."`head -n 1 /dev/urandom |md5sum |cut -d " " -f1|sed -e "s/<a-za-z>.//g"`
    sleep 0.$period
}

check_auth() {
    echo "Cheking auth"
    if < -w $SC_COOKIE >; then
     result=`curl --trace-ascii - -b $SC_COOKIE $curl_proxy -c $SC_COOKIE -A "$SC_UA" "http://$TR_HOST/forum/index.php"|egrep -o 'action=".*login\.php"'|egrep -o "<a-za-z0-9>*login\.php"`
  if < -z "$result" >; then 
              if < -z `echo $result|grep "http:"` >; then
                     autpath="http://$TR_HOST/forum/$result"
               else
                 autpath="$result"
             fi
           return `true`
        else return `false`; fi
    else
    result=`curl --trace-ascii - -c $SC_COOKIE $curl_proxy -A "$SC_UA" "http://$TR_HOST/forum/index.php"|egrep -o 'action=".*login\.php"'|egrep -o "</a-za-z0-9><a-za-z0-9>*login\.php"`
      if < -z `echo $result|grep "http:"` >; then
             autpath="http://$TR_HOST/forum/$result"
       else
         autpath="$result"
     fi
   echo $result
 return `false`
    fi


}

make_auth(){
    echo "Makin auth"
    if < -w $SC_COOKIE >; then
        cookie_data=`cat $SC_COOKIE`
  curl --trace-ascii - -b $SC_COOKIE  $curl_proxy -c $SC_COOKIE -A "$SC_UA" -d "login_username=$TR_USER" -d "login_password=$TR_PASSWORD" --data-binary "login=%C2%F5%EE%E4"  "$autpath">/dev/null
    else
      curl --trace-ascii - -c $SC_COOKIE $curl_proxy -A "$SC_UA" -d "login_username=$TR_USER" -d "login_password=$TR_PASSWORD" --data-binary "login=%C2%F5%EE%E4"  "$autpath">/dev/null
    fi
}

parse_topic(){
       echo "Parsing topic"
        result=`curl --trace-ascii - -b $SC_COOKIE $curl_proxy -c $SC_COOKIE -A "$SC_UA" "http://$TR_HOST/forum/viewtopic.php?t=$SC_TOPICID"|egrep -o "</a-za-z0-9><a-za-z0-9>*dl\.php"|head -n1`
        if < -z `echo $result|grep "http:"` >; then
                     dlpath="http://$TR_HOST/forum/$result"'?t='"$SC_TOPICID"
      else
                 dlpath="$result"'?t='"$SC_TOPICID"
    fi
   echo $dlpath
}

get_torrent(){
    echo "Getting torrent"
    if < -z "$SC_FILEPATH" >; then
        curl --trace-ascii - -d "" $curl_proxy  -b $SC_COOKIE -c $SC_COOKIE -A "$SC_UA" -o "<$TR_HOST>.t$SC_TOPICID.torrent"  -e "http://$TR_HOST/forum/viewtopic.php?t=$SC_TOPICID" "$dlpath"
    else
     curl --trace-ascii - -d "" $curl_proxy -b $SC_COOKIE -c $SC_COOKIE -A "$SC_UA" -o $SC_FILEPATH -e "http://$TR_HOST/forum/viewtopic.php?t=$SC_TOPICID" "$dlpath"
    fi
}

check_arg() {
 if < -z `echo $1|egrep '^-'` >; then 
          return `true`;
      else
                return `false`;
     fi
}

# Body of script
tst=`printf $1|egrep '^<0-9>+$'`
if < -z $tst >; then 
    die
else 
        SC_TOPICID=$1
fi

while shift; do
    case $1 in
 "-h")
           shift&&check;_arg&&TR;_HOST="$1"||die
     ;;
 "-u")
           shift&&check;_arg&&TR;_USER=$1||die
       ;;
 "-p")
           shift&&check;_arg&&TR;_PASSWORD=$1||die
   ;;
 "-o")
           shift&&check;_arg&&SC;_FILEPATH=$1||die
   ;;
    esac
done

SC_COOKIE="/tmp/gt-$TR_HOST-$TR_USER.ck"


if check_auth; then
    waitf
    parse_topic
    waitf
    get_torrent
else
    waitf
    make_auth
    waitf
    parse_topic
    waitf    
    get_torrent
fi
</a-za-z0-9></a-za-z>