#!/bin/bash

#####################################################
# Podcatcher Script
#
# you must first make a text file in the main directory
# and name the text file podlist.txt
# 
# Made By : Ibrahim Riad (Heema)
# http://sites.google.com/site/heematux/
#
# Dependencies : wget , awk , sed , xmlstarlet or xsltproc
#
# Some code was inspired from bashpodder
# http://linc.homeunix.org:8080/scripts/bashpodder/
#
# last update : 23-02-2010
VER=1.6.3
#
#####################################################

#####################################################
# Declaring variables
#####################################################

main_directory="$HOME/Podcast"
temp_directory="$main_directory/temp"
download_directory="$main_directory/shows"
archive_directory="/media/win_c/Documents and Settings/Ibrahim Riad/Desktop/shows"

#!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
#NOTE: Anything in the temp_directory will be deleted
#!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

history="$main_directory/downloaded_history.txt"
podcast_number="4"
#alarm="/usr/lib/openoffice/share/gallery/sounds/applause.wav"
multimedia_player="xmms"
stream_player="vlc"
bittorrent_client="/home/heema/Programs/Transgui/transgui"

#<--------------------------------------------------->

Progressbar ()
{

SOME="="
EMPTY=" "
COLUMNS=50

LEFT=$(($COLUMNS*100/20))
BAR_LENGTH=$(($(($COLUMNS*100))-$((2*$LEFT))-200))
DOT=$(($BAR_LENGTH/100))
PERCENT=$(($1*$DOT))

NUMCHARS=0

for ((j=0;$j<$LEFT;j=$j+100))
do
	echo -n " "
done
echo -n "["

for ((j=0;$j<$PERCENT;j=$j+100))
do
	echo -n "$SOME"
	NUMCHARS=$NUMCHARS+1
done

for ((j=$PERCENT;j<$BAR_LENGTH && $NUMCHARS<$((BAR_LENGTH/100));j=$j+100))
do
	echo -n "$EMPTY"
	NUMCHARS=$NUMCHARS+1
done

echo "]"
}

#<--------------------------------------------------->

# Creates a lock file to stop a second occurrence of the script
if [ -f "$main_directory/lock" ];then
	echo ""
	echo -e "\033[0;31mThe script is already runing , if it isn't then delete the lock file ( "$main_directory/lock" )\033[0m"
	echo ""
	exit
else
	touch "$main_directory/lock"
fi

# Checks for xmlstarlet or xsltproc if there installed
has_xmlstarlet=`whereis xmlstarlet`
has_xsltproc=`whereis xsltproc`

if [[ $has_xmlstarlet != "xmlstarlet:" ]];then
	PARSE_FEED="xmlstarlet tr"
elif [[ $has_xsltproc != "xsltproc:" ]];then
	PARSE_FEED="xsltproc"
else
        echo ""
        echo "Sorry, but this script requires either xmlstarlet or xsltproc and they could not be located."
        echo ""
        exit 0
fi

function clean_up {

	# Perform program exit housekeeping
        echo ""
	echo ""
	echo -e "\033[0;31mThe script was exited unexpectatly\033[0m"
	echo ""
	echo -e "\033[0;31mTemp files removed\033[0m"
	echo ""
	sleep 1
	rm "$main_directory/lock"
	rm "$temp_directory"/* 2>/dev/null
	exit
}

trap clean_up SIGHUP SIGINT SIGTERM

#<--------------------------------------------------->

#####################################################
# Specifying options
#####################################################

while [ "$1" != "" ];do
   case $1 in
             -n|--podcast_number ) podcast_number="$2"
	                 ;;
             -m|--manage_feeds ) cat -n "$main_directory/podlist.txt"
	   echo ""
	   echo "1) Add feed"
	   echo "2) Delete feed"
	   echo "3) Check status of feeds"
	   echo ""
	   echo "0) Close"
	   echo ""
	   echo -n "Enter choise: "
	   read manage_choise
	   case $manage_choise in
	       1 )
	       echo -n "Enter Feed: "
	       read feed_entered
	       echo "$feed_entered" >> "$main_directory/podlist.txt"
	       ;;
	       2 )
	            echo -n "Enter Number you want to delete: "
		    read number_delete
		    echo ""
		    cat "$main_directory/podlist.txt" | sed -e "$number_delete"d | sed -e '/^$/d' > "$main_directory/podlist.txt"
	       ;;
	       3 ) clear
		   for x in `cat "$main_directory/podlist.txt"`;do
		       echo ""
                       echo -n "$x          "
		       wget -t 2 -o "$temp_directory/wget_temp.txt" --spider "$x" && tail -n 2 "$temp_directory/wget_temp.txt" | head -n 1
                   done
                   rm "$main_directory/lock" 2>/dev/null
		  exit
		  ;;
	       * ) rm "$main_directory/lock" 2>/dev/null
		   exit
		   ;;
	       
	       esac
	   rm "$main_directory/lock" 2>/dev/null
	   exit
	      ;;
             -c|--clean ) echo -n "Are you sure you want to clean your download directory ? (y/n) "
			read CLEAN_CHOISE
			if [ "CLEAN_CHOISE" == "y" ];then
			    rm "$download_directory"/* 2>/dev/null
			    rm "$temp_directory"/* 2>/dev/null
			    echo -e "\033[0;31mDirectory Cleaned\033[0m"
			else
			    exit
			fi
			exit
                         ;;
		 -a|--archive ) if [ ! -d "$archive_directory" ];then
				echo -e "\033[0;32mMaking archive directory\033[0m"
				echo ""
				mkdir "$archive_directory"
			fi
			# Remove Bittorent files
			for br in `ls "$download_directory"`
			do
				Bittrm=$(file -b "$download_directory"/"$br" | awk '{print $1}')
				    if [ "$Bittrm" == "BitTorrent" ];then
					rm "$download_directory"/"$br"
				    fi
			done
			rm "$download_directory"/latest.m3u 2>/dev/null
			mv -v "$download_directory"/* "$archive_directory" 2>/dev/null
			rm "$main_directory/lock" 2>/dev/null
			echo -e "\033[0;31mPodcasts archived\033[0m"
			
			
			exit
                         ;;
	     -p|--play ) $multimedia_player "$download_directory/latest.m3u" &
			rm "$main_directory/lock" 2>/dev/null
			exit
			;;
	     -d|--download ) download="y"
                        ;;
	     -f|--full-catalogue ) full="y"
                        ;;
	     -h|--help ) echo "
Podster $VER
Developed by Ibrahim Riad (Heema) - http://sites.google.com/site/heematux/

Usage: $0 [OPTIONS]

Options are:
        -a, --archive                   Copies the podcasts to a specified location
        -n, --podcast_number            Specify the number of podcast per each stream to be downloaded
        -m, --manage_feeds              Lets you add , delete and check the status of the feeds
        -c, --clean                     Delete the podcasts and playlist
        -p, --play                      Play the playlist
        -d, --download                  Downloads automatically new podcasts
	-f, --full-catalogue            Adds the previous podcasts to your history without downloading them
        -h, --help                      Display this text and exit

Examples:
        $0 -n 1                         Will download the latest podcasts
        $0 -n 1 -d                      Will download the latest podcasts without confirmation
"
rm "$main_directory/lock" 2>/dev/null
                        exit
                        ;;
   esac
   shift
done

#<--------------------------------------------------->

clear

#####################################################
# Creating directories
#####################################################

mkdir -p "$main_directory"
mkdir -p "$temp_directory"
mkdir -p "$download_directory"
touch "$history"

#<--------------------------------------------------->

UPDATE () {

#####################################################
# Downloading feeds
#####################################################

echo "#################################"
echo "# Downloading feeds"
echo "#################################"
echo ""

if [[ ! -f "$main_directory/podlist.txt" || ! -s "$main_directory/podlist.txt" ]]
then
    touch "$main_directory/podlist.txt"
    echo -e "\033[0;31mFile "$main_directory/podlist.txt" is empty\033[0m"
    echo ""
    exit
fi

echo ""

Total=$(cat -n "$main_directory/podlist.txt" | awk '{print $1}' | tail -n 1)

for i in `cat "$main_directory/podlist.txt"`
do
Current=$(ls "$HOME/Podcast/temp" | wc -l)
Sub=$(($Current*100))
Percent=$(($Sub/$Total))

clear

echo "#################################"
echo "# Downloading feeds"
echo "#################################"
echo ""
echo ""
echo -e -n "\033[0;32m $Percent%\033[0m" 
Progressbar $Percent
echo ""
echo ""
wget -t 1 -T 20 -P "$temp_directory" "$i"
done

#<--------------------------------------------------->

#####################################################
# Filter the feeds from the enclosure
#####################################################

if [ ! -f "$main_directory/parse_enclosure.xsl" ]
then
	echo ""
	echo '<?xml version="1.0"?>
	<stylesheet version="1.0"
		xmlns="http://www.w3.org/1999/XSL/Transform">
		<output method="text"/>
		<template match="/">
		<apply-templates select="/rss/channel/item/enclosure"/>
		</template>
		<template match="enclosure">
		<value-of select="@url"/><text>&#10;</text>
	</template>
	</stylesheet>' > "$main_directory/parse_enclosure.xsl"
fi

if [ ! -f "$main_directory/parse_all.xsl" ]
then
	echo ""
	echo '<?xml version="1.0"?>
	<stylesheet version="1.0"
		xmlns="http://www.w3.org/1999/XSL/Transform">
		<output method="text"/>
		<template match="/">
		<apply-templates select="/rss/channel/item"/>
		</template>
		<template match="item">
		<value-of select="title"/><text>&#10;</text>
		<value-of select="enclosure/@url"/><text>&#10;</text>
	</template>
	</stylesheet>' > "$main_directory/parse_all.xsl"
fi

cd "$temp_directory"

echo ""
echo "Please wait... Parsing the feeds"
echo ""

for p in `ls`
do
	$PARSE_FEED "$main_directory/parse_enclosure.xsl" $p >> $p.log 2>/dev/null
	$PARSE_FEED "$main_directory/parse_all.xsl" $p >> title.txt 2>/dev/null
	#xmlstarlet tr "$main_directory/parse_enclosure.xsl" $p >> $p.log 2>/dev/null
	#xmlstarlet tr "$main_directory/parse_all.xsl" $p >> title.txt 2>/dev/null
done

#<--------------------------------------------------->

#####################################################
# Clean the output of the filter
#####################################################

for x in `ls *.log`
do
    cat "$x" | head -n $podcast_number >> final.txt
done

#<--------------------------------------------------->

}

clear

DOWNLOAD () {

#####################################################
# Download the podcast and keep a history of it
#####################################################

echo "#################################"
echo "# Download or stream the podcast"
echo "#################################"
echo ""

echo "Options:"
echo "--------"
echo ""
echo "Press Ctrl+c to quit at any time"
echo ""
echo "! Resume is supported !"
echo ""
echo "d = Download , s = Stream , c = Catalogue , n = Skip"
echo ""
echo ""

echo "New shows:"
echo "----------"
echo ""

NEW_SHOWS=0

for t in `cat final.txt`
do
  test=$(grep -F "$t" "$history") 2>/dev/null
  status=$(echo $?)
  
     if [ $status -eq 1 ];then
	 clean=$(basename "$t")
	 TITLE=$(grep -F -B 1 "$clean" title.txt)
	 echo "$TITLE"
	 echo ""
   	NEW_SHOWS=$((NEW_SHOWS+1))
     fi
done

echo ""
echo "( $NEW_SHOWS new shows )"
echo ""

echo ""
echo "Actions:"
echo "--------"
echo ""

for d in `cat final.txt`
do
    temp=$(grep -F "$d" "$history") 2>/dev/null
    exists=$(echo $?)
    clean_title=$(basename "$d")
    
    if [ $exists -eq 1 ];then
	if [ "$download" == "y" ];then
           choise="d"
	elif [ "$full" == "y" ];then
	    echo "$d" >> "$history"
	    echo ""
	    echo "$clean_title has been added"
	continue
	else
	    read -p "Download , Stream or catalogue : $clean_title (d/s/c/n) ? " choise
	    echo "$choise*$d" >> choises.txt
	fi
	
	echo ""
	
    fi
done

cat choises.txt | sort > choises-sorted.txt

for d in `cat choises-sorted.txt`
do

	Option=$(echo "$d" | cut -c 1)
	Filepath=$(echo "$d" | awk -F* '{print $2}')
	clean_tag=$(basename "$d")

	### Downloading ###
	# you can prioritize the downloads by adding a number after d : d1, d2, d3,....

	if [ "$Option" == "d" ] || [[ "$Option" == "d[0-9]*" ]] || [ "$Option" == "D" ] || [ "$Option" == "y" ];then
	    echo ""
	    echo -e "\033[0;32mDownloading : \033[0m" "$clean_tag"
	    echo ""
	
	# wget -c -P "$download_directory" "$Filepath" && echo "$Filepath" >> "$history"
	    wget -c -O "$download_directory/$clean_tag" "$Filepath" && echo "$Filepath" >> "$history"
	    
	    # This line for converting the %5B and %5D from the filename and removing question marks at end
	    clean_tag_bit=$(echo "$clean_tag" | sed -e s/%5B/[/ | sed -e s/%5D/]/ | sed -e s/%28/\(/ | sed -e s/%29/\)/ | sed -e 's/?.*$//')
	    
	    mv "$download_directory/$clean_tag" "$download_directory/$clean_tag_bit"
	    
	    ### Check if its a torrent ###
	    
	    Bitt=$(file -b "$download_directory/$clean_tag_bit" | awk '{print $1}')
	    if [ "$Bitt" == "BitTorrent" ];then
		nohup "$bittorrent_client" "$download_directory/$clean_tag_bit" &
		sleep 3
	    fi

	### Streaming ###

	elif [ "$Option" == "s" ] || [ "$Option" == "S" ];then
	    echo ""
	    echo -e "\033[0;32mStreaming : \033[0m" "$clean_tag"
	    echo ""
	    echo "$Filepath" >> "$history"
	    $stream_player "$Filepath" &

	### Catalogue ###

	elif [ "$Option" == "c" ];then
	    echo ""
	    echo "Added $clean_tag"
	    echo ""
	    echo "$Filepath" >> "$history"
	
	### Skipping ###

	else
	    echo ""
	    echo -e "\033[0;31mSkipping : \033[0m" "$clean_tag"
	    echo ""
	fi
done

#<--------------------------------------------------->

#####################################################
# Create an m3u playlist
#####################################################

# remove added junk from filename
#for n in `ls "$download_directory"`
#do
#   echo "$n" | grep "?" 2>/dev/null
#   if [ $? -eq 0 ];then
#   	NEWURL=$(echo "$n" | awk -F "?" '{print $1}')
#   	mv "$download_directory/$n" "$download_directory/$NEWURL" 2>/dev/null
#   fi
#done

ls -1rc "$download_directory" | grep -v m3u > "$download_directory/latest.m3u"

#<--------------------------------------------------->

}

CLEANUP () {

#####################################################
# Cleanup
#####################################################

rm "$main_directory/lock"
rm "$main_directory"/temp/* 2>/dev/null

}

UPDATE
DOWNLOAD
CLEANUP

echo "#################################"
echo -e "#""\033[0;32m Done \033[0m"
echo "#################################"
echo ""