drk.sh

#!/bin/bash

#init variables
submission_type="u"
submission_chosen=false

needs_help=false

authentication=""
password=""

# get last argument
for i in $@; do :; done
last_arg=$i

# validation functions
function validate_submission_choice {
    if [[ $submission_chosen = true ]]; then
        echo "Only one of the folowing can be chosen:"
        echo "-s (snippet) -f (file) -u (url)"
        exit
    else
        submission_chosen=true
    fi
}

# soruce: http://stackoverflow.com/a/10660730
rawurlencode() {
    local string="${1}"
    local strlen=${#string}
    local encoded=""

    for (( pos=0 ; pos<strlen ; pos++ )); do
        c=${string:$pos:1}
        case "$c" in
            [-_.~a-zA-Z0-9] )
                o="${c}" 
                ;;
            * )
                printf -v o '%%%02x' "'$c"
                ;;
        esac
        encoded+="${o}"
    done

    REPLY="${encoded}"   #+or echo the result (EASIER)... or both... :p
}

# check arguments
while getopts hsfup:a: option
do
    case "${option}"
    in
        s) 
            validate_submission_choice
            submission_type="s";
            ;;
        f)
            validate_submission_choice
            submission_type="f";
            ;;
        u)
            validate_submission_choice
            submission_type="u";
            ;;
        h)
            needs_help=true
            ;;
        p)
            password=$OPTARG
            ;;
        a)
            authentication=$OPTARG
            ;;
    esac
done

if [[ $needs_help == true ]] || [[ "$#" -eq 0 ]]; then
    echo '''
    Usage:
    Display a resource by passing darkshare URL:
    `drk.sh http://drk.sh/aBz`
    
    Create a resource by passing one of the following options:
    -s (snippet) -f (file) -u (url)
    
    Examples:
    Snippet:
    `drk.sh -s ~/path/to/file.txt`
    File:
    `drk.sh -s ~/path/to/file.jpg`
    URL:
    `drk.sh -s http://duckduckgo.com`
    '''
    exit
fi

if [[ $submission_chosen == false ]]; then
    if [[ $last_arg != *drk.sh* ]]; then
        echo "Please pass a valid darkshare url"
        exit;
    fi

    curl_cmd="curl -s -X GET "

    if [[ -n $authentication ]]; then
        curl_cmd="$curl_cmd -u $authentication"
    fi

    if [[ -n $password ]]; then
        curl_cmd="$curl_cmd $last_arg?password=$password"
    else
        curl_cmd="$curl_cmd $last_arg"
    fi

    response=$($curl_cmd)

    echo "$response"
    exit
fi

## Snippets
if [[ $submission_type = "s" ]]; then

    # needs file as last agrument
    if [[ $last_arg == -* ]]; then
        echo "Please supply a last argument."
        exit
    fi

    # if file exists
    if [[ -f $last_arg ]]; then

        curl_cmd="curl -s -X POST "

        if [[ -n $authentication ]]; then
            curl_cmd="$curl_cmd -u $authentication"
        fi
        
        cat_output=`cat $last_arg`

        rawurlencode "$cat_output"; cat_output=${REPLY}

        curl_cmd="$curl_cmd -d body=$cat_output https://drk.sh/s"

        response=$($curl_cmd)

        echo $response

    else
        echo "File $last_arg not found."
        exit
    fi
    
fi

# Files
if [[ $submission_type = "f" ]]; then

    if [[ $last_arg == -* ]]; then
        echo "Please supply a last argument."
        exit
    fi

    # if file exists
    if [[ -f $last_arg ]]; then
        curl_cmd="curl -s -X POST "

        if [[ -n $authentication ]]; then
            curl_cmd="$curl_cmd -u $authentication"
        fi
        
        curl_cmd="$curl_cmd -F path=@$last_arg https://drk.sh/f/"

        response=$($curl_cmd)

        echo $response
    else
        echo "File $last_arg not found."
        exit
    fi
    
fi

# URL's
if [[ $submission_type = "u" ]]; then

    if [[ $last_arg == -* ]]; then
        echo "Please supply a last argument."
        exit
    fi

    curl_cmd="curl -s -X POST "

    if [[ -n $authentication ]]; then
        curl_cmd="$curl_cmd -u $authentication"
    fi
    
    curl_cmd="$curl_cmd -d destination=$last_arg https://drk.sh/"

    response=$($curl_cmd)

    echo $response
    
fi