xref: /titanic_41/usr/src/lib/libshell/common/scripts/shtinyurl.sh (revision bb294766735789e8f3edfcbac1edb83dfb9ee53b) 
1#!/usr/bin/ksh93
2
3#
4# CDDL HEADER START
5#
6# The contents of this file are subject to the terms of the
7# Common Development and Distribution License (the "License").
8# You may not use this file except in compliance with the License.
9#
10# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
11# or http://www.opensolaris.org/os/licensing.
12# See the License for the specific language governing permissions
13# and limitations under the License.
14#
15# When distributing Covered Code, include this CDDL HEADER in each
16# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
17# If applicable, add the following below this CDDL HEADER, with the
18# fields enclosed by brackets "[]" replaced with your own identifying
19# information: Portions Copyright [yyyy] [name of copyright owner]
20#
21# CDDL HEADER END
22#
23
24#
25# Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
26# Use is subject to license terms.
27#
28
29# Solaris needs /usr/xpg6/bin:/usr/xpg4/bin because the tools in /usr/bin are not POSIX-conformant
30export PATH=/usr/xpg6/bin:/usr/xpg4/bin:/bin:/usr/bin
31
32# Make sure all math stuff runs in the "C" locale to avoid problems
33# with alternative # radix point representations (e.g. ',' instead of
34# '.' in de_DE.*-locales). This needs to be set _before_ any
35# floating-point constants are defined in this script).
36if [[ "${LC_ALL}" != "" ]] ; then
37    export \
38        LC_MONETARY="${LC_ALL}" \
39        LC_MESSAGES="${LC_ALL}" \
40        LC_COLLATE="${LC_ALL}" \
41        LC_CTYPE="${LC_ALL}"
42        unset LC_ALL
43fi
44export LC_NUMERIC=C
45
46function fatal_error
47{
48	print -u2 "${progname}: $*"
49	exit 1
50}
51
52# parse HTTP return code, cookies etc.
53function parse_http_response
54{
55	nameref response="$1"
56	typeset h statuscode statusmsg i
57
58	# we use '\r' as additional IFS to filter the final '\r'
59	IFS=$' \t\r' read -r h statuscode statusmsg  # read HTTP/1.[01] <code>
60	[[ "$h" != ~(Eil)HTTP/.* ]]         && { print -u2 -f $"%s: HTTP/ header missing\n" "$0" ; return 1 ; }
61	[[ "$statuscode" != ~(Elr)[0-9]* ]] && { print -u2 -f $"%s: invalid status code\n"  "$0" ; return 1 ; }
62	response.statuscode="$statuscode"
63	response.statusmsg="$statusmsg"
64
65	# skip remaining headers
66	while IFS='' read -r i ; do
67		[[ "$i" == $'\r' ]] && break
68
69		# strip '\r' at the end
70		i="${i/~(Er)$'\r'/}"
71
72		case "$i" in
73			~(Eli)Content-Type:.*)
74				response.content_type="${i/~(El).*:[[:blank:]]*/}"
75				;;
76			~(Eli)Content-Length:[[:blank:]]*[0-9]*)
77				integer response.content_length="${i/~(El).*:[[:blank:]]*/}"
78				;;
79			~(Eli)Transfer-Encoding:.*)
80				response.transfer_encoding="${i/~(El).*:[[:blank:]]*/}"
81				;;
82		esac
83	done
84
85	return 0
86}
87
88function cat_http_body
89{
90	typeset emode="$1"
91	typeset hexchunksize="0"
92	integer chunksize=0
93
94	if [[ "${emode}" == "chunked" ]] ; then
95		while IFS=$'\r' read hexchunksize &&
96			[[ "${hexchunksize}" == ~(Elri)[0-9abcdef]* ]] &&
97			(( chunksize=16#${hexchunksize} )) && (( chunksize > 0 )) ; do
98			dd bs=1 count="${chunksize}" 2>/dev/null
99		done
100	else
101		cat
102	fi
103
104	return 0
105}
106
107function request_tinyurl
108{
109	# site setup
110	typeset url_host="tinyurl.com"
111	typeset url_path="/api-create.php"
112	typeset url="http://${url_host}${url_path}"
113	integer netfd # http stream number
114	typeset inputurl="$1"
115	compound httpresponse # http response
116	typeset request=""
117
118	# we assume "inputurl" is a correctly encoded URL which doesn't
119	# require any further mangling
120	url_path+="?url=${inputurl}"
121
122	request="GET ${url_path} HTTP/1.1\r\n"
123	request+="Host: ${url_host}\r\n"
124	request+="User-Agent: ${http_user_agent}\r\n"
125	request+="Connection: close\r\n"
126
127	redirect {netfd}<> "/dev/tcp/${url_host}/80"
128	(( $? != 0 )) && { print -u2 -f $"%s: Could not open connection to %s.\n" "$0" "${url_host}" ;  return 1 ; }
129
130	# send http post
131	{
132		print -n -- "${request}\r\n"
133	}  >&${netfd}
134
135	# process reply
136	parse_http_response httpresponse <&${netfd}
137	response="${ cat_http_body "${httpresponse.transfer_encoding}" <&${netfd} ; }"
138
139	# close connection
140	redirect {netfd}<&-
141
142	if (( httpresponse.statuscode >= 200 && httpresponse.statuscode <= 299 )) ; then
143		print -r -- "${response}"
144		return 0
145	else
146		print -u2 -f $"tinyurl response was (%s,%s):\n%s\n" "${httpresponse.statuscode}" "${httpresponse.statusmsg}" "${response}"
147		return 1
148	fi
149
150	# not reached
151}
152
153function request_trimurl
154{
155	# site setup
156	typeset url_host="api.tr.im"
157	typeset url_path="/api/trim_url.xml"
158	typeset url="http://${url_host}${url_path}"
159	integer netfd # http stream number
160	typeset inputurl="$1"
161	compound httpresponse # http response
162	typeset request=""
163
164	# we assume "inputurl" is a correctly encoded URL which doesn't
165	# require any further mangling
166	url_path+="?url=${inputurl}"
167
168	request="GET ${url_path} HTTP/1.1\r\n"
169	request+="Host: ${url_host}\r\n"
170	request+="User-Agent: ${http_user_agent}\r\n"
171	request+="Connection: close\r\n"
172
173	redirect {netfd}<> "/dev/tcp/${url_host}/80"
174	(( $? != 0 )) && { print -u2 -f $"%s: Could not open connection to %s.\n" "$0" "${url_host}" ;  return 1 ; }
175
176	# send http post
177	{
178		print -n -- "${request}\r\n"
179	}  >&${netfd}
180
181	# process reply
182	parse_http_response httpresponse <&${netfd}
183	response="${ cat_http_body "${httpresponse.transfer_encoding}" <&${netfd} ; }"
184
185	# close connection
186	redirect {netfd}<&-
187
188	if (( httpresponse.statuscode >= 200 && httpresponse.statuscode <= 299 )) ; then
189		# the statement below should really parse the XML...
190		print -r -- "${response/~(Elr).*(\<url\>)(.*)(\<\/url\>).*/\2}"
191		return 0
192	else
193		print -u2 -f $"tr.im response was (%s,%s):\n%s\n" "${httpresponse.statuscode}" "${httpresponse.statusmsg}" "${response}"
194		return 1
195	fi
196
197	# not reached
198}
199
200function usage
201{
202	OPTIND=0
203	getopts -a "${progname}" "${shtinyurl_usage}" OPT '-?'
204	exit 2
205}
206
207# program start
208builtin basename
209builtin cat
210builtin date
211builtin uname
212
213typeset progname="${ basename "${0}" ; }"
214
215# HTTP protocol client identifer
216typeset -r http_user_agent="shtinyurl/ksh93 (2009-08-12; ${ uname -s -r -p ; })"
217
218typeset -r shtinyurl_usage=$'+
219[-?\n@(#)\$Id: shtinyurl (Roland Mainz) 2009-08-12 \$\n]
220[-author?Roland Mainz <roland.mainz@nrubsig.org>]
221[+NAME?shtinyurl - create short alias URL from long URL]
222[+DESCRIPTION?\bshtinyurl\b is a small utility which passes a given URL
223	to internet service which creates short aliases in the
224	form of http://<servicename>/XXXXXXXX to redirect long URLs.]
225[+?The first arg \burl\b describes a long URL which is transformed into
226	a tinyurl.com short alias.]
227[P:provider?Service provider (either \'tinyurl.com\' or \'tr.im\').]:[mode]
228
229url
230
231[+SEE ALSO?\bksh93\b(1), \brssread\b(1), \bshtwitter\b(1), http://www.tinyurl.com, http://tr.im]
232'
233
234typeset service_provider="tr.im"
235
236while getopts -a "${progname}" "${shtinyurl_usage}" OPT ; do
237#	printmsg "## OPT=|${OPT}|, OPTARG=|${OPTARG}|"
238	case ${OPT} in
239		P)	service_provider="${OPTARG}" ;;
240		*)	usage ;;
241	esac
242done
243shift $((OPTIND-1))
244
245# expecting at least one more argument
246(( $# >= 1 )) || usage
247
248typeset url="$1"
249shift
250
251case "${service_provider}" in
252	"tinyurl.com")
253		request_tinyurl "${url}"
254		exit $?
255		;;
256	"tr.im")
257		request_trimurl "${url}"
258		exit $?
259		;;
260	*)
261		fatal_error "Unsupported service provider."
262esac
263
264# not reached
265
266# EOF.
267