X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=gopher.sh;h=58175702600c07be27fe6f97fddb896273c14163;hb=fcda6ae3df75d214b74022323d421283764f27ff;hp=f91be7a8c6f27fa1a30be4086280f1fab2a0538b;hpb=0eb21b2178dbb2cc1faea64447b861a91e442b90;p=gofetch.git

diff --git a/gopher.sh b/gopher.sh
index f91be7a..5817570 100755
--- a/gopher.sh
+++ b/gopher.sh
@@ -1,97 +1,222 @@
 #!/bin/sh
 
+# $0 [server] ([selector]) ([port]) ([mode])
+# 	server: the gopher server to dig to
+#	selector: the gopher selector (default is empty)
+#	port: the port to use (default is 70)
+#	mode: the filetype mode (default depends upon selector)
+#		0: plain text
+#		1: menu (dir-like)
+#		...
+#		download: fake mode to download the result without changes
+
+# Manual:
+# 	When asked for a number (non-dialog uses only), you can also
+#	enter 'n' or 'p' to get the next or previous item (+1 or -1)
+
+# ENV variables:
+# 	USE_DIALOG: force the usage of the command 'dialog'
+#		  : true if dialog is found
+#		0 : do not
+#		1 : force the use of dialog
+#	LINK_COLOR: escape sequences colour (def: 2)
+#		- : means no escape sequence
+#		1 : means colour 1
+#		2 : means colour 2
+#		[...]
+# 	INVERT    : invert the output for image viewing (for white backgrounds)
+# 		0 : do not invert (default)
+#		1 : invert 
+
+# EXIT Codes:
+#	  0: ok
+#	  1: syntax error
+#	  2: cannot contact server
+#	  3: unknown selector mode
+#	255: special exit more 'q'
+
 SERVER="$1"
 SELECTOR="$2"
 PORT="$3"
 MODE="$4"
 
+PREFIX="[0-9hIg+]"
+
 # Defaults:
 [ "$PORT" = "" ] && PORT=70
 if [ "$MODE" = "" ]; then
 	# "" or dir-like selector? -> 1 ; if not -> 0 
 	echo "$SELECTOR" | grep "/$" >/dev/null && MODE=1 || MODE=0
 	[ "$SELECTOR" = "" ] && MODE=1
+	
+	# check explicit modes:
+	if echo "$SELECTOR" | grep "^/\($PREFIX\|download\)/" >/dev/null; then
+		MODE="`echo "$SELECTOR" | cut -f2 -d/`"
+		SELECTOR="`echo "$SELECTOR" | sed 's:^/[^/]*/::'`"
+	fi
 fi
 
 if [ "$SERVER" = "" ]; then
 	echo "Syntax error: $0 [SERVER] ([SELECTOR]) ([PORT]) ([MODE])" >&2
-	exit 2
+	exit 1
+fi
+
+# can be "-" for no escape sequences
+[ "$LINK_COLOR" = ""  ] && LINK_COLOR=2
+
+# Start and end link tags
+SL=
+EL=
+if [ "$LINK_COLOR" != "-" ]; then
+	SL="`tput setf $LINK_COLOR``tput setaf $LINK_COLOR`"
+	EL="`tput init`"
+	export LESS="${LESS}-R"
+fi
+
+# 'dialog' or text
+if [ "$USE_DIALOG" = "" ]; then
+	if dialog --help >/dev/null 2>&1; then
+		USE_DIALOG=1
+	else
+		USE_DIALOG=0
+	fi
+fi
+
+# Invert image viewer
+if [ "$INVERT" = 1 ]; then
+	INVERT="--invert"
+else
+	INVERT=
 fi
 
-# $0 [FILE]
+# $0 [FILE] (dialog)
+# Display a gopher menu for the given resource
 cat_menu() {
 	i=0
-	cat "$1" | grep '^[i0-9]' | while read ln; do
-		ln="`echo "$ln" | cut -f1`"
+	cat "$1" | grep "^i\|^$PREFIX" | while read ln; do
 		if echo "$ln" | grep "^i" >/dev/null 2>&1; then
-			echo "$ln" | sed "s:^.:	:g"
-		elif echo "$ln" | grep "^[0-9]" >/dev/null 2>&1; then
+			if [ "$2" != dialog ]; then
+				echo "$ln" | sed "s:^.\([^\t]*\).*$:	\1:g"
+			else
+				echo "$ln" | sed 's:":'"''"':g;s:^.\([^\t]*\).*$:"      " "\1":g'
+			fi
+		elif echo "$ln" | grep "^$PREFIX" >/dev/null 2>&1; then
 			i=`expr $i + 1`
-			i=`printf %2.f $i`
+			[ $i -le 9 ] && i=0$i 
 			field="`echo "$ln" | cut -c1`"
 			case "$field" in
 				0) typ='TXT';;
-				1) typ='DIR';;
-				7) typ='SRH';;
-				8) typ='TEL';;
-				*) typ='???';;
+				1) typ='DIR';; # menu, actually
+				7) typ='(?)';; # query
+				8) typ='TEL';; # TELnet (not TELephone)
+				h) typ='WEB';; # HTML
+				I) typ='IMG';;
+				g) typ='GIF';;
+				+) typ='SVR';; # redundant server
+				*) typ='!!!';;
 			esac
-			echo "$ln" | sed "s:^.:$typ $i	:g"
-		#else
-			# Bad line
+			if [ "$2" != dialog ]; then
+				echo "$ln" | sed "s:^.\\([^\t]*\\).*:$typ $i	$SL\\1$EL:g"
+			else
+				echo "$ln" | sed "s:"'"'":'':g;s:^.\\([^\t]*\\).*:"'"'"$typ $i"'"'" "'"\1"'":g"
+			fi
 		fi
 	done
 }
 
-# $0 [FILE]
-choices() {
-	i=0
-	cat "$1" | grep '^[0-9]' | while read choice; do
-		i=`expr $i + 1`
-		i=`printf %2.f $i`
-		
-		echo "$i --> $choice"
-	done
-}
-
 # $0 [FILE] [INDEX] [FIELD]
+# Get a field from the given-by-index link in FILE
+#
 # Fields:
 # 	1 = type/name
 # 	2 = selector
 # 	3 = server
 # 	4 = port
 getsel() {
-	cat "$1" | grep '^[0-9]' | tail -n+"$2" | head -n 1 | cut -f"$3"
+	cat "$1" | grep "^$PREFIX" | tail -n+"$2" | head -n 1 | cut -f"$3"
 }
 
-tmp="`mktemp gofetch.current_page.XXXXX`"
+# Save page content to 'tmp' file
+tmp="`mktemp -t gofetch.current_page.XXXXXX`"
 finish() {
-  rm -rf "$tmp"
+  rm -rf "$tmp" "$tmp.jpg" "$tmp.menu" "$tmp.choice"
 }
 trap finish EXIT
 
+echo "$SELECTOR" | nc "$SERVER" "$PORT" > "$tmp"
+if [ $? != 0 ]; then
+	echo Cannot contact gopher server "[$SERVER]" >&2
+	exit 2
+fi
+
 if [ $MODE = 1 ]; then
-	echo "$SELECTOR" | nc "$SERVER" "$PORT" | sed 's:\r::g' > "$tmp"
-else
-	echo "$SELECTOR" | nc "$SERVER" "$PORT" > "$tmp"
+	sed --in-place 's:\r::g' "$tmp"
 fi
 
+# Process page content
 case "$MODE" in
+download)
+	# Special, fake mode, only from top-level
+	cat "$tmp"
+;;
 0)
 	cat "$tmp" | less
 ;;
-1)
-	CHOICE=start
+1|+)
+	CHOICE=0
 	while [ "$CHOICE" != "" ]; do
-		cat_menu "$tmp" | less
-		read -p "[$SELECTOR]: " CHOICE
+		if [ "$USE_DIALOG" = 1 ]; then
+			> "$tmp.menu"
+			cat_menu "$tmp" dialog | while read ln; do
+				echo -n " $ln" >> "$tmp.menu"
+			done
+			[ "$LINES"   = "" ] && LINES=`tput lines`
+			[ "$COLUMNS" = "" ] && COLUMNS=`tput cols`
+			title="$SERVER: $SELECTOR"
+			dialog	--extra-button --extra-label Back \
+				--cancel-label Exit \
+				--no-shadow \
+				--menu "$title" \
+				"$LINES" "$COLUMNS" "$LINES" \
+				--file "$tmp.menu" 2>"$tmp.choice"
+			val=$?
+			clear
+			
+			if [ $val = 3 ]; then
+				CHOICE=""
+			elif [ $val = 1 ]; then
+				CHOICE="q"
+			else
+				CHOICE="`cat "$tmp.choice" | cut -c5-`"
+			fi
+			
+			rm "$tmp.menu"
+			rm "$tmp.choice"
+		else
+			cat_menu "$tmp" | less
+			read -p "[$SELECTOR]: " NEW_CHOICE
+			if [ "$NEW_CHOICE" = p ]; then
+				CHOICE=`expr "$CHOICE" - 1 2>/dev/null`
+				if [ "$CHOICE" -lt 0 ]; then
+					CHOICE=`cat "$tmp" | grep "^$PREFIX" | wc --lines`
+				fi
+			elif [ "$NEW_CHOICE" = n ]; then
+				CHOICE=`expr "$CHOICE" + 1 2>/dev/null`
+			else
+				CHOICE="$NEW_CHOICE"
+			fi
+		fi
+		
+		[ "$CHOICE" = q ] && exit 255 # force-quit
 		index="`expr 1 \* "$CHOICE" 2>/dev/null`"
 		if [ "$index" != "" ]; then
 			goto_server="`getsel "$tmp" $index 3`"
 			goto_sel="`getsel "$tmp" $index 2`"
 			goto_port="`getsel "$tmp" $index 4`"
 			goto_mode="`getsel "$tmp" $index 1 | cut -c1`"
+			echo "Digging to [$goto_server:$goto_port] $index [$goto_sel]..."
 			sh "$0" "$goto_server" "$goto_sel" "$goto_port" "$goto_mode"
+			[ $? = 255 ] && exit 255 # force-quit
 		fi
 	done
 ;;
@@ -108,8 +233,29 @@ case "$MODE" in
 9)
 	echo "<BINARY FILE>" | less
 ;;
+g|I)
+	if img2aa --help >/dev/null 2>&1; then
+		img2aa --mode=DITHERING \
+			$INVERT \
+			--width=74 "$tmp" | less
+	elif jp2a -h >/dev/null 2>&1; then
+		if convert -h >/dev/null 2>&1; then
+			convert "$tmp" "$tmp.jpg"
+			jp2a --border --chars=" .-+=o8#"\
+				$INVERT \
+				--width=74 "$tmp.jpg" | less
+		else
+			echo "required program not found to view images: convert" \
+				| less
+		fi
+	else
+		echo "required program not found to view images:" \
+			jp2a or img2aa | less
+	fi
+;;
 *)
 	echo "unknwon selector mode: <$MODE>" | less
 	exit 3
 ;;
 esac
+