Contents of /bti/trunk/bti-shrink-urls

#!/bin/bash
# Copyright (C) 2009 Bart Trojanowski <bart@jukie.net>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation version 2 of the License.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

needs_escape=true

shrink_host=2tu.us
test -f ~/.bti && . ~/.bti

while test -n "$1" ; do
        word="$1"
        shift
        case "$word" in
            --escaped)
                needs_escape=
                ;;
            --help|-h)
                cat <<END
bti-shrink-urls - convert URLs to a shorter form using a web service

    $0 [--escaped] [<url>]

Currently supported: 2tu.us (default), bit.ly, j.mp.
END
                exit 0
                ;;
            *)
                URL=$word
                ;;
        esac
done

function convert_url() {
        local url=$1
        test -n "$url" || return 1
        test "${url%%:*}" = 'http' || return 1

        local urllen="${#url}"

        # http://en.wikipedia.org/wiki/Percent-encoding
        if test -n "$needs_escape" ; then
                url=$(echo "$url" | sed -e 's/\%/%25/g' \
                                        -e 's/!/%21/g' \
                                        -e 's/*/%2A/g' \
                                        -e "s/'/%27/g" \
                                        -e 's/(/%28/g' \
                                        -e 's/)/%29/g' \
                                        -e 's/;/%3B/g' \
                                        -e 's/:/%3A/g' \
                                        -e 's/@/%40/g' \
                                        -e 's/&/%26/g' \
                                        -e 's/=/%3D/g' \
                                        -e 's/+/%2B/g' \
                                        -e 's/\$/%24/g' \
                                        -e 's/,/%2C/g' \
                                        -e 's,/,%2F,g' \
                                        -e 's/?/%3F/g' \
                                        -e 's/#/%23/g' \
                                        -e 's/\[/%5B/g' \
                                        -e 's/]/%5D/g')
        fi

        case $shrink_host in
        2tu.us)
            local submit="http://2tu.us/?save=y&url=$url"
            local res=$(wget -q -O - "$submit" | awk -F"'" '/Your tight URL is:/ { print $2 }')
            ;;
        bit.ly|j.mp)
            if [ -z "$shrink_bitly_login" -o -z "$shrink_bitly_key" ]; then
                echo "To use $shrink_host you must set 'shrink_bitly_login' and 'shrink_bitly_key' in ~/.bti" >&2
                exit 1
            fi
            local submit="http://api.bit.ly/v3/shorten?format=txt&login=$shrink_bitly_login&apiKey=$shrink_bitly_key&domain=$shrink_host&longUrl=$url"
            local res=$(wget -q -O - "$submit")
            ;;
        *)
            echo "Shrinking with $shrink_host is not supported." >&2
            exit 1
            ;;
        esac

        if test "${res%%:*}" = 'http' -a "${#res}" -lt "$urllen" ; then
                echo $res
                return 0
        fi
        return 1
}

function die() {
        echo >&2 $@
        exit 1
}

if test -n "$URL" ; then
        convert_url "$URL" || die "Failed to shrink '$URL'"
        exit $?
fi

test -t 0 && echo >&2 "Type in some urls and I'll try to shrink them for you..."
while read line ; do
        convert_url "$line" || echo $line
done
1	#!/bin/bash
2	# Copyright (C) 2009 Bart Trojanowski <bart@jukie.net>
3	#
4	# This program is free software; you can redistribute it and/or modify it
5	# under the terms of the GNU General Public License as published by the
6	# Free Software Foundation version 2 of the License.
7	#
8	# This program is distributed in the hope that it will be useful, but
9	# WITHOUT ANY WARRANTY; without even the implied warranty of
10	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11	# General Public License for more details.
12	#
13	# You should have received a copy of the GNU General Public License along
14	# with this program; if not, write to the Free Software Foundation, Inc.,
15	# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16
17	needs_escape=true
18
19	shrink_host=2tu.us
20	test -f ~/.bti && . ~/.bti
21
22	while test -n "$1" ; do
23	word="$1"
24	shift
25	case "$word" in
26	--escaped)
27	needs_escape=
28	;;
29	--help\|-h)
30	cat <<END
31	bti-shrink-urls - convert URLs to a shorter form using a web service
32
33	$0 [--escaped] [<url>]
34
35	Currently supported: 2tu.us (default), bit.ly, j.mp.
36	END
37	exit 0
38	;;
39	*)
40	URL=$word
41	;;
42	esac
43	done
44
45	function convert_url() {
46	local url=$1
47	test -n "$url" \|\| return 1
48	test "${url%%:*}" = 'http' \|\| return 1
49
50	local urllen="${#url}"
51
52	# http://en.wikipedia.org/wiki/Percent-encoding
53	if test -n "$needs_escape" ; then
54	url=$(echo "$url" \| sed -e 's/\%/%25/g' \
55	-e 's/!/%21/g' \
56	-e 's/*/%2A/g' \
57	-e "s/'/%27/g" \
58	-e 's/(/%28/g' \
59	-e 's/)/%29/g' \
60	-e 's/;/%3B/g' \
61	-e 's/:/%3A/g' \
62	-e 's/@/%40/g' \
63	-e 's/&/%26/g' \
64	-e 's/=/%3D/g' \
65	-e 's/+/%2B/g' \
66	-e 's/\$/%24/g' \
67	-e 's/,/%2C/g' \
68	-e 's,/,%2F,g' \
69	-e 's/?/%3F/g' \
70	-e 's/#/%23/g' \
71	-e 's/\[/%5B/g' \
72	-e 's/]/%5D/g')
73	fi
74
75	case $shrink_host in
76	2tu.us)
77	local submit="http://2tu.us/?save=y&url=$url"
78	local res=$(wget -q -O - "$submit" \| awk -F"'" '/Your tight URL is:/ { print $2 }')
79	;;
80	bit.ly\|j.mp)
81	if [ -z "$shrink_bitly_login" -o -z "$shrink_bitly_key" ]; then
82	echo "To use $shrink_host you must set 'shrink_bitly_login' and 'shrink_bitly_key' in ~/.bti" >&2
83	exit 1
84	fi
85	local submit="http://api.bit.ly/v3/shorten?format=txt&login=$shrink_bitly_login&apiKey=$shrink_bitly_key&domain=$shrink_host&longUrl=$url"
86	local res=$(wget -q -O - "$submit")
87	;;
88	*)
89	echo "Shrinking with $shrink_host is not supported." >&2
90	exit 1
91	;;
92	esac
93
94	if test "${res%%:*}" = 'http' -a "${#res}" -lt "$urllen" ; then
95	echo $res
96	return 0
97	fi
98	return 1
99	}
100
101	function die() {
102	echo >&2 $@
103	exit 1
104	}
105
106	if test -n "$URL" ; then
107	convert_url "$URL" \|\| die "Failed to shrink '$URL'"
108	exit $?
109	fi
110
111	test -t 0 && echo >&2 "Type in some urls and I'll try to shrink them for you..."
112	while read line ; do
113	convert_url "$line" \|\| echo $line
114	done