#!/bin/ksh
#
# $Log: make-html-index,v $
# Revision 1.4  2008/09/05 03:19:18  mbaker
# changed recursion method at high level
#
# Revision 1.3  2008/09/01 21:00:22  mbaker
# changed recursion algorithm
#
# Revision 1.2  2008/09/01 13:45:06  mbaker
# added flagfile
#
# Revision 1.1  2008/06/26 03:09:37  mbaker
# Initial revision
#
#########################################################################
#DOCBEGIN
#
#Create index.html for all listed files in a directory.
#
#ASSUMPTIONS: 
#
# OVERWRITE any/all existing index.html 
#        UNLESS 
#           set from command line arguements.
#        OR 
#           if a flag files exists. 
# 
#
#  Matt Baker - mbaker@computeranddata.com
#  Latest copy available at www.computeranddata.com/scripts
#  Version 1.2  09/01/2008
#
#DOCEND
#########################################################################

#########################################################################
#usage
#
usage ()
{
cat <<EOF
Usage: $0: [-debug] [-v] [-u www.mysite.dom] [-r /www] [-R] [-n] [-f filename] -d directory
       $0: -h

      -d:       Directory name to: work in OR to work in AND traverse from.
      -f:       Filename to use instead of the default index.html.
      -n:       No overwrite of any existing index.html files (or Filename 
                from the -f option).  
				The default IS to overwrite.
				If the dontreindex file exists in the directory, 
				no overwrite is done either.
      -r:       Docroot.  i.e. /www
      -R:       Recursively (traverse) from the DIRECTORY downward creating
                an index.html (or -f filename) file in each directory.
      -u:       URL of this site.  i.e.  www.computeranddata.com

      -v:       Verbose mode.  This also is a "report" mode in the
                output file.  The output can then be parsed later.

      -debug:   Debug mode, does a 'set -x' 
   -info|doc:   Print out header of program (document)
      -h:       Help - synopsis

EOF
}

#########################################################################
#
#VARIABLES AREA
#
set -o allexport

DEBUG=0
VERBOSE=0
RECURSION=0
DIRECTORY=""
FLAGFILE=dontreindex
MYDOCROOT=""
MYURL=""
OVERWRITE=1
OUTPUTFILE=index.html

set +o allexport

###########
#read in options & arguments
###########

while (( $# ))
do
	case $1 in

		-d)     shift
			DIRECTORY=$1
			shift

			if [[ ! -d $DIRECTORY ]]
			then
				print "ERROR:3: Direcory $DIRECTORY not found or is a file."
				exit 4
			fi
			;;

		-f)     shift
			OUTPUTFILE=$1

			if [[ $(print $OUTPUTFILE | cut -c1) == '-' ]]
			then
				print "ERROR:2:Must give filename to option -f"
				usage
				exit 2
			fi

			shift
			;;

		-r)     shift
			MYDOCROOT=$1

			if [[ $(print $MYDOCROOT | cut -c1) == '-' ]]
			then
				print "ERROR:2:Must give docroot (i.e. /www) to -r"
				usage
				exit 2
			fi

			shift
			;;

		-u)     shift
			MYURL=$1

			if [[ $(print $MYURL | cut -c1) == '-' ]]
			then
				print "ERROR:2:Must give url (i.e. www.computeranddata.com) to -u"
				usage
				exit 2
			fi

			shift
			;;

		-R)     RECURSION=1
			shift
			;;

		-n)     OVERWRITE=0
			shift
			;;

		-debug) DEBUG=1
			shift
			;;

		-v) VERBOSE=1
			shift
			;;

		-info | -doc) 
			awk '/DOCBEGIN/, /DOCEND/' $0 \
						| sed -e 's/^#//' \
						| sed -e '/DOCBEGIN/d' \
						| sed -e '/DOCEND/d' \
						|  more
			exit 0
			;;


		-h|*)   if [[ $1 != -h ]]
			then
				print "\n ERROR:1: OPTION: $1 is invalid.\n"
			else
				print ""
			fi
			usage
			exit 1
			;;
	esac
done

if (( DEBUG ))
then
	print "INFO: Command line option settings"
	print "INFO: ...RECURSION: $RECURSION" 
	print "INFO: ...DIRECTORY: $DIRECTORY" 
	print "INFO: ...OUTPUTFILE: $OUTPUTFILE" 
	print "INFO: ...OVERWRITE: $OVERWRITE" 
	print "INFO: ...VERBOSE: $VERBOSE" 

	set -x
fi

#########################################################################
#
#FUNCTIONS AREA
#
###############
# makeindex
makeindex () {
	if (( DEBUG ))
	then
		set -x
	fi

	#
	#IF 
	#  FLAGFILE doesn't exist
	#  IF 
	#     overwrite is true
	#    OR 
	#     overwrite is false AND the file doesn't exist
	#  THEN overwrite
	#ELSE 
	#  DONT overwrite
	#FI
	#
	if [[ ! -f $FLAGFILE ]]
	then
		if [[ ($OVERWRITE == 1) || ($OVERWRITE == 0 && ! -f $OUTPUTFILE) ]]
		then
			if (( VERBOSE ))
			then
				print "INFO: Creating $OUTPUTFILE for $DIRECTORY..."
			fi
	
			MYDIRCWD=$(pwd)
			MYDIRHOME=$(echo "${MYDIRCWD##$MYDOCROOT}")
			cat > $OUTPUTFILE <<-EOF
				<HTML>
				<HEAD> <TITLE>Index file for $MYDIRHOME</TITLE> </HEAD>
				<BODY>
				EOF
	
			for FILE in $(ls | grep -v $OUTPUTFILE)
			do
				if (( VERBOSE ))
				then
					print "INFO: Working on file $FILE ..."
				fi
	
				if [[ -n $MYDOCROOT || -n $MYURL ]]
				then
					MYCWD=$(pwd)
					MYHOME=$(echo "${MYCWD##$MYDOCROOT}")
					if [[ -d $FILE ]]
					then
						print "<A HREF="http://${MYURL}$MYHOME/$FILE/">$FILE</A><BR>" \
									>> $OUTPUTFILE
					else
						print "<A HREF="http://${MYURL}$MYHOME/$FILE">$FILE</A><BR>" \
									>> $OUTPUTFILE
					fi
				else
					if [[ -d $FILE ]]
					then
						print "<A HREF="$FILE"/>$FILE</A><BR>" >> $OUTPUTFILE
					else
						print "<A HREF="$FILE">$FILE</A><BR>" >> $OUTPUTFILE
					fi
				fi
			done
		
			cat >> $OUTPUTFILE <<-EOF
				</BODY>
				</HTML>
				EOF
		else 
			if (( VERBOSE ))
			then
				print "INFO: Skipping $OUTPUTFILE for $DIRECTORY as it already exists."
			fi
		fi
	#
	#Don't overwrite if file exists
	#
	else
		if [[ -f $FLAGFILE ]]
		then
			if (( VERBOSE ))
			then
				print "INFO: Skipping $OUTPUTFILE for $DIRECTORY as flagfile exists."
			fi
		fi
	fi
}

###############
# nextdir
nextdir () {
	if (( DEBUG ))
	then
		set -x
	fi

	DIRECTORY=$1
	cd $DIRECTORY > /dev/null 2>&1
	if (( VERBOSE ))
	then
		print "INFO: Working in Directory $(pwd) ..."
	fi

	makeindex
	for DIR in $(ls -l | grep ^d | awk '{print $NF}')
	do
		################
		# if directory, recursion, if not set
		################
		if [[ $RECURSION == 1 ]]
		then
			if (( VERBOSE ))
			then
				print "INFO: Going into Directory $DIR ..."
			fi

			nextdir "$DIR"
			cd .. > /dev/null 2>&1
		fi
	done
}

#########################################################################
#main
#########################################################################
if [[ -z $DIRECTORY ]]
then
	print "ERROR:4:DIRECTORY not set."
	usage
	exit 1
fi


nextdir $DIRECTORY


if (( VERBOSE ))
then
	print "INFO: Done with $0"
fi