#!/bin/sh
#####################
# Name: yt-down
# Lisc: GPLv2
# Auth: iluaster
# Main: jadedctrl
# Desc: Fetches YT videos.
#####################

# Based on https://github.com/iluaster/Youtube_downloader

# Usage: yt-down -[sSD] $video_id
# -s for stream; -S for stream and skip prompt; 
# -D for download and skip prompt; No arg for download.


if [ "$1" = "-s" ]
then
    stream_mode=1
    id=$2
elif [ "$1" = "-S" ]
then
    stream_mode=1
    skip_prompt=1
    id=$2
elif [ "$1" = "-D" ]
then
    stream_mode=0
    skip_prompt=1
    id=$2
else
    stream_mode=0
    id=$1
fi

if [ -e "./lib/yt-meta" ]
then
    alias yt-channel="./lib/yt-channel"
    alias yt-search="./lib/yt-down"
    alias yt-meta="./lib/yt-meta"
    alias yt-search="./lib/yt-search"
    alias yt-down="./lib/yt-channel"
    alias st-download="./lib/st-download"
    alias st-video="./lib/st-video"
fi


if echo "$id" | grep "youtube.com"
then 
    id="$(echo "$id" | sed 's/.*video_id=//')"
elif [ -z "$id" ]
then
    echo "No video specified."
    exit 1
else
    id="$id"
fi
    
name="http://www.youtube.com/get_video_info?video_id=$id"

declare -i line=0 2> /dev/null

if $? 2> /dev/null
then
    echo > /dev/null
else
    line=0
fi

select_option ()
{
    echo "Formats:"
    cat /tmp/video_type_option.txt | while IFS='' read -r i
    do
        line=$((line+1)) 
        # Replace this pipe method so $line isn't lost
        echo "${line}. $i"
    done
    if [ $skip_prompt -eq 0 ]
    then
        printf '\033[0;32m>>>\033[0m '
        IFS=''
        read -r n
    else
        n=1
    fi

    if [ $n -le 5 ];
    then
        head -n "$n" /tmp/tmp3.txt | tail -n 1 > /tmp/tmp4.txt
    else
        echo "$line $n"
        echo "Input Error!"
        line=0
        select_option
    fi
}


st-download "$name" "/tmp/${id}_url.txt"

# Cut and filter mp4 url

cp "/tmp/${id}_url.txt" /tmp/tmp2.txt
sed -e 's/&/\n/g' /tmp/tmp2.txt| grep 'url_encoded_fmt_stream_map'> /tmp/tmp3.txt
sed -i -e 's/%2C/,/g' /tmp/tmp3.txt 
sed -i -e 's/,/\n/g' /tmp/tmp3.txt

# Print out total video format name and quality
perl -ne 'print "$2,$1\n" if /quality%3D(.*?)%.*video%252F(.*?)(%|\n)/' /tmp/tmp3.txt > /tmp/video_type_option.txt

# If video format name is prior to quality
perl -ne 'print "$1,$2\n" if /video%252F(.*?)%.*quality%3D(.*?)(%|\n)/' /tmp/tmp3.txt >> /tmp/video_type_option.txt
sed -i -e 's/x-flv/flv/g' /tmp/video_type_option.txt

select_option
  
# Set file extension name variable and video quality variable
extension_name=$(head -n "$n" /tmp/video_type_option.txt | tail -n 1 | cut -d "," -f 1)
quality_name=$(head -n "$n" /tmp/video_type_option.txt | tail -n 1 | cut -d "," -f 2)

sed -i -e 's/%26/\&/g' /tmp/tmp4.txt
sed -i -e 's/&/\n/g' /tmp/tmp4.txt
grep 'http' /tmp/tmp4.txt > /tmp/tmp5.txt
grep 'sig%3D' /tmp/tmp4.txt >> /tmp/tmp5.txt
perl -pe 's/\n//g' /tmp/tmp5.txt | sed -e 's/sig%3D/\&signature%3D/g' > /tmp/tmp6.txt
sed -i -e 's/url%3D//g' /tmp/tmp6.txt
# url decoding
cat /tmp/tmp6.txt | sed -e 's/%25/%/g' -e 's/%25/%/g' -e 's/%3A/:/g' -e 's/%2F/\//g' -e 's/%3F/\?/g' -e 's/%3D/=/g' -e 's/%26/\&/g' > /tmp/tmp7.txt


if [ $stream_mode -eq 1 ]
then
    st-video "$(cat /tmp/tmp7.txt)"
else
    st-download "$(cat /tmp/tmp7.txt)" "${id}_${quality_name}.${extension_name}"
fi
  
 
rm -f /tmp/tmp[2-7].txt "/tmp/${id}_url.txt" /tmp/video_type_option.txt