#!/bin/bash
# based on https://gist.github.com/deanet/3427090
#
# useful $HOME/.gdrive.conf options:
#    curl_args="--limit-rate 500K"


browser="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/35.0.1916.153 Safari/537.36"

if [ -e $HOME/.gdrive.conf ]
then
    . $HOME/.gdrive.conf
fi

# destination_folder_id might be set in ~/.gdrive.conf
# but can be overridden as last parameter on command line
lastparm=${@: -1}
if [ ! -z "$lastparm" ]
then
    if expr "$lastparm" : '^[A-Za-z0-9]\{28\}$' > /dev/null
    then
        destination_folder_id=$lastparm
        # all but last word
        set -- "${@:0:$#}"
    fi
fi

old_umask=`umask`
umask 0077

if [ -z "$username" ]
then
    read -p "username: " username
    unset token
    echo "username=$username" >> $HOME/.gdrive.conf
fi

if [ -z "$account_type" ]
then
    if expr "$username" : '^[^@]*$' > /dev/null || expr "$username" : '.*@gmail.com$' > /dev/null
    then
        account_type=GOOGLE
    else
        account_type=HOSTED
    fi
fi

if [ -z "$password$token" ]
then
    read -s -p "password: " password
    unset token
    echo
fi

if [ -z "$token" ]
then
    echo "Obtaining token..."
    token=`curl --silent --data-urlencode Email=$username --data-urlencode Passwd="$password" --data accountType=$account_type --data service=writely --data source=cURL "https://www.google.com/accounts/ClientLogin" | sed -ne s/Auth=//p`
    sed -ie '/^token=/d' $HOME/.gdrive.conf
    if [ -z "$token" ]; then
        echo "Failed to get token. Most likely your username/password was wrong."
        exit
    else
        echo "Got token"
        echo "token=$token" >> $HOME/.gdrive.conf
    fi
fi
umask $old_umask

echo "Starting upload..."
for file in "$@"
do
    if [ "$file" != "$0" ]; then
        slug=`basename "$file"`
        mime_type=`file --brief --mime-type "$file"`
        echo "Getting upload link..."
        upload_link=`curl --silent --show-error --insecure --request POST --header "Content-Length: 0" --header "Authorization: GoogleLogin auth=${token}" --header "GData-Version: 3.0" --header "Content-Type: $mime_type" --header "Slug: $slug" "https://docs.google.com/feeds/upload/create-session/default/private/full${destination_folder_id+/folder:$destination_folder_id/contents}?convert=false" --dump-header - | sed -ne s/"Location: "//p`
        echo "Starting upload of $file:"
        curl --progress-bar --request POST --output /dev/null --data-binary "@$file" --header "Authorization: GoogleLogin auth=${token}" --header "GData-Version: 3.0" --header "Content-Type: $mime_type" --header "Slug: $slug" "$upload_link" $curl_args
    fi
done