ddm

#!/bin/bash
#
### LICENSE ###
# ddm - dresden-departure-monitor:
# getting departures for stops in Dresden
#
# Copyright (C) 2011 don_philipe - don_philipe@gmx.de
#
# 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, either version 3 of the License, or
# (at your option) any later version.
#
# 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, see <http://www.gnu.org/licenses/>.
###
#


OPTIONS="-q --keep-session-cookies --save-cookies cookies --load-cookies cookies -T 100"
URL="www.dvb.de"
VERSION="Version: 1.1"

# default:
displayed=15		# number of displayed departures
loop=false
loop_sec=30		# repeat getting departuretimes every x seconds
city="Dresden"
stop=""
headlinelength=50		# length of the headline without stop


helptext()
{
  cat <<HELP

Syntax: ddm [options] stop
Shows the departuretimes, routes, vehicles and directions for a given stop in Dresden.
Passing the stop becomes obsolete if you add one in the script.

Options:	
	-d num		changes the number of displayed departures (default: 15)
	-h, --help	shows this helptext
	-l [sec]	get departuretimes repeatedly every 30 second or add an 
			own interval in seconds (greater than 0)
	-v, --version	returns the scriptversion

HELP
}


	
getSessionId()
{
	grep /de/Fahrplan/Abfahrtsmonitor/haltestellenauswahl.do?uk_id abfahrt > sessionid
	sessionid="$(sed 's/  <form action="\/de\/Fahrplan\/Abfahrtsmonitor\/haltestellenauswahl.do?uk_id=//;s/#result" id="am_form" method="post" >//' sessionid)"
}


clearDir()
{
	rm abfahrt
	rm cookies
	rm sessionid
}


getDepartures()
{
  	wget $OPTIONS -O abfahrt --post-data "amform%5Bdatum%5D=${cur_date}&amform%5Bzeit%5D=${cur_time}&amform%5Bstartort%5D=${city}&amform%5Bstartname%5D=${stop}&amform%5Bstarttype%5D=stop" "${URL}/de/Fahrplan/Abfahrtsmonitor/haltestellenauswahl.do?uk_id=${sessionid}" #%20
	# test if stop is valid:
	if [ -n "$(grep Unbekannte\ Haltestelle\. abfahrt)" ]
	then
		echo -e "\nUnknown stop!\n"
		clearDir
		exit 0
	fi

	# get correct stopname:
	stop="$(grep \>$city, abfahrt)"
	stop="${stop##*,}"
	
	# replacing umlauts
	stop="$(echo $stop | sed 's/\&#228;/ä/g;s/\&#246;/ö/g;s/&#252;/ü/g;s/\&#223;/\ß/g')"
	
	#s//Ä/g
	#s//Ü/g
	#s//Ö/g
	
	
	# adds the length of the stopname to the headlinelength:
	let headlinelength=$headlinelength+${#stop}	
	for (( i=1 ; i<= $headlinelength ; i++ ))
	do
		underline="${underline}="
	done

	getSessionId

	# getting departuretimes again with only two vehicles
	wget $OPTIONS -O abfahrt --post-data 'amform%5Bverkehrsmittelcheckboxen%5D=strassenbahn&amform%5Bverkehrsmittelcheckboxen%5D=stadtbus' "${URL}/de/Fahrplan/Abfahrtsmonitor/abfahrten.do?uk_id=${sessionid}"
}


clearTime()	# remove the leading 0 if present
{
	tmp="${1}"
	if [ "$(echo ${tmp} | cut -c 1)" == "0" ]
	then
		tmp=$(echo ${tmp} | cut -c 2)
	fi
	echo ${tmp}
}


calcRemainingTime()	# doesnt work correct around 0 o'clock
{
	dep_time_h=$(clearTime ${timearray[$i]%%:*})
	cur_time_h=$(clearTime ${cur_time%%:*})
	dep_time_m=$(clearTime ${timearray[$i]##*:})
	cur_time_m=$(clearTime ${cur_time##*:})

	if [ $dep_time_h -lt $cur_time_h ]
	then
		timeshift=24
	else
		timeshift=0
	fi

	if [ $(( $dep_time_h - $cur_time_h )) -eq 0 ]	# same hour
	then
		remaining_time[$i]="$(( $dep_time_m - $cur_time_m ))"
	else							# 1,2,... hour later
		if [ $(( $dep_time_m - $cur_time_m )) -lt 0 ]
		then	#remaining_time not greater than 59 min
			minutes="$(( 60 + $(( $dep_time_m - $cur_time_m )) ))"
			if [ $(( $dep_time_h - $cur_time_h )) -lt 0 ]
			then	# for daychange
				hours="$(( 23 + $(( $dep_time_h - $cur_time_h )) ))"
				remaining_time[$i]="$hours:$minutes"
			else
				remaining_time[$i]=$minutes
			fi
		else	# remaining_time greater than 59 min
			remaining_time[$i]="$(( $timeshift + $dep_time_h - $cur_time_h )):$(( $dep_time_m - $cur_time_m ))"
		fi
	fi
}


processHtml()
{
	startline="$(grep -n \"abfahrten_container\" abfahrt)"
	startline="${startline%%:*}"
	endline="$(grep -n /de/Fahrplan/Abfahrtsmonitor/spaeter.do abfahrt)"
	endline="${endline%%:*}"

	dep_times="$(sed -n ''${startline}','${endline}'p' abfahrt | sed '/[012][0-9]\:[0-5][0-9]/!d;s/[^0-9\:]//g')"
	vehicle="$(sed -n ''${startline}','${endline}'p' abfahrt | sed '/^.*pikto_[a-z]-\?[a-z]*\.gif.*$/!d;s/^.*pikto_//g;s/\.gif.*$/ /g;s/\b./\U&/g')"
	route="$(sed -n ''${startline}','${endline}'p' abfahrt | sed '/<td>[1-9][0-9]\?[0-9]\?<\/td>/!d;s/[^0-9]//g;')" #s/[1-9][^0-9]/&\ /g')"
	direction="$(sed -n ''${startline}','${endline}'p' abfahrt | sed '/          \t.*[A-ZÄÖÜ][a-zäöüß]*.*[A-ZÄÖÜ]\?[a-zäöüß]*\t/!d;s/          \t//g;s/\ /+/g;s/\t/ /g')"
	
	for (( i=1 ; i <= $displayed ; i++ ))	# loop starts with 1 for the calculation two lines below
	do
		if [ "$dep_times" == "" ]
		then
			break
		fi
		
		timearray[$i]="$(echo ${dep_times:0:6})"	# ${String:Position:Length}
		dep_times="$(echo $dep_times | sed 's/[0-9][0-9]\:[0-9][0-9]//')"	# cut the currently used time
		
		vehiclearray[$i]="$(echo `expr "$vehicle" : '\([A-ZÄÖÜ]-\?B\?[a-zäöüß]*\ \)'`)"	# -\?B\? for S-Bahn
		vehicle="$(echo $vehicle | sed 's/[A-ZÄÖÜ]-\?B\?[a-zäöüß]*\ //')"
		routearray[$i]="$(echo `expr "$route" : '\([1-9][0-9][0-9]\|[1-9][0-9]\|[1-9]\)'`)" 	#\|S\ [1-9] - add for S-Bahn at the end
		route="$(echo $route | sed 's/[1-9][0-9][0-9]\ \|[1-9][0-9]\ \|[1-9]\ //')" 	#\|S\ [1-9]\ - add for S-Bahn at the end
		directionarray[$i]="$(echo `expr "$direction" : '\([A-ZÄÖÜ][A-ZÄÖÜa-zäöüß+\.,]*\)'`)"
		directionarray[$i]=$(echo ${directionarray[$i]} | sed 's/+/\ /g')
		direction="$(echo $direction | sed 's/[A-ZÄÖÜ][A-ZÄÖÜa-zäöüß+\.,]*\ //')"
		calcRemainingTime
	done
}


output()
{
	echo -e "\nDeparturemonitor for ${stop}, ${city}, ${cur_date}, ${cur_time}:\n${underline}\n"
	echo -e "Time\tRoute\tVehicle\t\tDirection\t\tRemaining time"	# tableheader
	# output lines:
	for (( i=1 ; i <= $displayed ; i++ ))
	do
		if [ "${timearray[$i]}" == "" ]	# break loop if not enough departures
		then
			break
		fi
		
		tabs="\t\t"
		if [ ${#directionarray[$i]} -lt 8 ]
		then
			tabs="\t\t\t"
		fi
		if [ ${#directionarray[$i]} -gt 15 ]
		then
			tabs="\t"
		fi
		echo -e "${timearray[$i]}\t${routearray[$i]}\t${vehiclearray[$i]}\t\t${directionarray[$i]}$tabs${remaining_time[$i]}"
	done
	echo -e "\n"
}



######################
# Script starts here #
######################

while [ "${1}" != ''  ]
do
	[ "${1}" == "-d" ] && shift && displayed="${1}" && shift
	[ "${1}" == "-h" ] || [ "${1}" == "--help" ] && helptext && exit 0
	[ "${1}" == "-l" ] && shift && loop=true && loop_sec="${1}" && shift
	[ "${1}" == "-v" ] || [ "${1}" == "--version" ] && echo $VERSION && exit 0
	[ -n "${1}" ] && stop="${1} ${2}" && shift && shift	# 2x shift for stops containing two words

if [ $(( $displayed + 0 )) -eq 0 ] ; then
	echo "Syntax error! Parameter not found or 0."
	exit 0
fi
if [ "$(echo $loop_sec | sed '/[0-9]/!d')" == "" ] ; then
	stop="${loop_sec} ${stop}"
	loop_sec=30
fi

done

# get website for initial sessionid:
wget $OPTIONS -O abfahrt "${URL}/de/Fahrplan/Abfahrtsmonitor"

# set date and time:
cur_date="$(date +%d.%m.%Y)"
cur_time="$(date +%H:%M)"
	
if [ "$(cat abfahrt)" == "" ]
then
	echo -e "\nCan't reach the server!\n"
	clearDir
	exit 0
fi

getSessionId
clear
echo "Download departuretimes..."
getDepartures
echo "Process information..."
processHtml
output
clearDir
	

while $loop
do
	sleep $loop_sec
	
	# get website for initial sessionid:
	wget $OPTIONS -O abfahrt "${URL}/de/Fahrplan/Abfahrtsmonitor"

	# set date and time:
	cur_date="$(date +%d.%m.%Y)"
	cur_time="$(date +%H:%M)"
	
	if [ "$(cat abfahrt)" == "" ]
	then
		echo -e "\nCan't reach the server!\n"
		clearDir
		exit 0
	fi
	
	headlinelength=50
	underline=""

	getSessionId
	getDepartures
	processHtml
	clear
	output
	clearDir	
done

exit 0