#!/bin/bash

YEAR=`date +2003-%Y`

HGROOT=`hg root`

# file enumaration modes

function all_files() {

    hg status -a -m -c |

    cut -d ' ' -f 2 | grep -E '(\.(cc|h|dox)$|Makefile\.am$)' |

    while read file; do echo $HGROOT/$file; done

}

function modified_files() {

    hg status -a -m |

    cut -d ' ' -f 2 | grep -E  '(\.(cc|h|dox)$|Makefile\.am$)' |

    while read file; do echo $HGROOT/$file; done

}

function changed_files() {

    {

        if [ -n "$HG_PARENT1" ]

        then

            hg status --rev $HG_PARENT1:$HG_NODE -a -m

        fi

        if [ -n "$HG_PARENT2" ]

        then

            hg status --rev $HG_PARENT2:$HG_NODE -a -m

        fi

    } | cut -d ' ' -f 2 | grep -E '(\.(cc|h|dox)$|Makefile\.am$)' | 

    sort | uniq |

    while read file; do echo $HGROOT/$file; done

}

function given_files() {

    for file in $GIVEN_FILES

    do

	echo $file

    done

}

# actions

function update_action() {

    if ! diff -q $1 $2 >/dev/null

    then

	echo -n " [$3 updated]"

	rm $2

	mv $1 $2

	CHANGED=YES

    fi

}

function update_warning() {

    echo -n " [$2 warning]"

    WARNED=YES

}

function update_init() {

    echo Update source files...

    TOTAL_FILES=0

    CHANGED_FILES=0

    WARNED_FILES=0

}

function update_done() {

    echo $CHANGED_FILES out of $TOTAL_FILES files has been changed.

    echo $WARNED_FILES out of $TOTAL_FILES files has been warned.

}

function update_begin() {

    ((TOTAL_FILES++))

    CHANGED=NO

    WARNED=NO

}

function update_end() {

    if [ $CHANGED == YES ]

    then

	((++CHANGED_FILES))

    fi

    if [ $WARNED == YES ]

    then

	((++WARNED_FILES))

    fi

}

function check_action() {

    if ! diff -q $1 $2 >/dev/null

    then

	echo

	echo -n "      $3 failed at line(s): "

	echo -n $(diff $1 $2 | grep '^[0-9]' | sed "s/^\(.*\)c.*$/ \1/g" | 

	          sed "s/,/-/g" | paste -s -d',')

	FAILED=YES

    fi

}

function check_warning() {

    echo

    if [ "$2" == 'long lines' ]

    then

        echo -n "      $2 warning at line(s): "

        echo -n $(grep -n -E '.{81,}' $1 | sed "s/^\([0-9]*\)/ \1\t/g" | 

                  cut -f 1 | paste -s -d',')

    else

        echo -n "      $2 warning"

    fi

    WARNED=YES

}

function check_init() {

    echo Check source files...

    FAILED_FILES=0

    WARNED_FILES=0

    TOTAL_FILES=0

}

function check_done() {

    echo $FAILED_FILES out of $TOTAL_FILES files has been failed.

    echo $WARNED_FILES out of $TOTAL_FILES files has been warned.

    if [ $FAILED_FILES -gt 0 ]

    then

	return 1

    elif [ $WARNED_FILES -gt 0 ]

    then

	if [ "$WARNING" == 'INTERACTIVE' ]

	then

	    echo -n "Assume as normal behaviour? (yes/no) "

	    while read answer

	    do

		if [ "$answer" == 'yes' ]

		then

		    return 0

		elif [ "$answer" == 'no' ]

		then

		    return 1

		fi

		echo -n "Assume as normal behaviour? (yes/no) "		    

	    done

	elif [ "$WARNING" == 'WERROR' ]

	then

	    return 1

	fi

    fi

}

function check_begin() {

    ((TOTAL_FILES++))

    FAILED=NO

    WARNED=NO

}

function check_end() {

    if [ $FAILED == YES ]

    then

	((++FAILED_FILES))

    fi

    if [ $WARNED == YES ]

    then

	((++WARNED_FILES))

    fi

}

# checks

function header_check() {

    if echo $1 | grep -q -E 'Makefile\.am$'

    then

	return

    fi

    TMP_FILE=`mktemp`

    (echo "/* -*- mode: C++; indent-tabs-mode: nil; -*-

 *

 * This file is a part of LEMON, a generic C++ optimization library.

 *

 * Copyright (C) "$YEAR"

 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

 * (Egervary Research Group on Combinatorial Optimization, EGRES).

 *

 * Permission to use, modify and distribute this software is granted

 * provided that this copyright notice appears in all copies. For

 * precise terms see the accompanying LICENSE file.

 *

 * This software is provided \"AS IS\" with no warranty of any kind,

 * express or implied, and with no claim as to its suitability for any

 * purpose.

 *

 */

"

    awk 'BEGIN { pm=0; }

     pm==3 { print }

     /\/\* / && pm==0 { pm=1;}

     /[^:blank:]/ && (pm==0 || pm==2) { pm=3; print;}

     /\*\// && pm==1 { pm=2;}

    ' $1

    ) >$TMP_FILE

    "$ACTION"_action "$TMP_FILE" "$1" header

}

function tabs_check() {

    if echo $1 | grep -q -v -E 'Makefile\.am$'

    then

        OLD_PATTERN=$(echo -e '\t')

        NEW_PATTERN='        '

    else

        OLD_PATTERN='        '

        NEW_PATTERN=$(echo -e '\t')

    fi

    TMP_FILE=`mktemp`

    cat $1 | sed -e "s/$OLD_PATTERN/$NEW_PATTERN/g" >$TMP_FILE

    "$ACTION"_action "$TMP_FILE" "$1" 'tabs'

}

function spaces_check() {

    TMP_FILE=`mktemp`

    cat $1 | sed -e 's/ \+$//g' >$TMP_FILE

    "$ACTION"_action "$TMP_FILE" "$1" 'trailing spaces'

}

function long_lines_check() {

    if cat $1 | grep -q -E '.{81,}'

    then

	"$ACTION"_warning $1 'long lines'

    fi

}

# process the file

function process_file() {

    echo -n "    $ACTION $1..."

    CHECKING="header tabs spaces long_lines"

    "$ACTION"_begin $1

    for check in $CHECKING

    do

	"$check"_check $1

    done

    "$ACTION"_end $1

    echo

}

function process_all {

    "$ACTION"_init

    while read file

    do

	process_file $file

    done < <($FILES)

    "$ACTION"_done

}

while [ $# -gt 0 ]

do

    if [ "$1" == '--help' ] || [ "$1" == '-h' ]

    then

	echo -n \

"Usage:

  $0 [OPTIONS] [files]

Options:

  --dry-run|-n

     Check the given files, but do not modify them.

  --interactive|-i

     If --dry-run is specified and files are warned, then a message is

     prompted whether the warnings should be turned to errors.

  --werror|-w

     If --dry-run is specified, the warnings are turned to errors.

  --all|-a

     Check all source files in the repository.

  --modified|-m

     Check only the modified source files. This option is proper to

     use before a commit. E.g. all files which are modified or added

     into the repository will be updated.

  --changed|-c

     Check only the changed source files compared to the parent(s) of

     the current hg node.  This option is proper to use as hg hook

     script. E.g. to check all your commited source files with this

     script add the following section to the appropriate .hg/hgrc

     file.

       [hooks]

       pretxncommit.checksources = scripts/unify-sources.sh -c -n -i

  --help|-h

     Print this help message.

  files

     The files to check/unify. If no file names are given, the modified

     source files will be checked/unified (just like using the

     --modified|-m option).

"

        exit 0

    elif [ "$1" == '--dry-run' ] || [ "$1" == '-n' ]

    then

	[ -n "$ACTION" ] && echo "Conflicting action options" >&2 && exit 1

	ACTION=check

    elif [ "$1" == "--all" ] || [ "$1" == '-a' ]

    then

	[ -n "$FILES" ] && echo "Conflicting target options" >&2 && exit 1

	FILES=all_files

    elif [ "$1" == "--changed" ] || [ "$1" == '-c' ]

    then

	[ -n "$FILES" ] && echo "Conflicting target options" >&2 && exit 1

	FILES=changed_files

    elif [ "$1" == "--modified" ] || [ "$1" == '-m' ]

    then

	[ -n "$FILES" ] && echo "Conflicting target options" >&2 && exit 1

	FILES=modified_files

    elif [ "$1" == "--interactive" ] || [ "$1" == "-i" ]

    then

	[ -n "$WARNING" ] && echo "Conflicting warning options" >&2 && exit 1

	WARNING='INTERACTIVE'

    elif [ "$1" == "--werror" ] || [ "$1" == "-w" ]

    then

	[ -n "$WARNING" ] && echo "Conflicting warning options" >&2 && exit 1

	WARNING='WERROR'

    elif [ $(echo x$1 | cut -c 2) == '-' ]

    then

	echo "Invalid option $1" >&2 && exit 1

    else

	[ -n "$FILES" ] && echo "Invalid option $1" >&2 && exit 1

	GIVEN_FILES=$@

	FILES=given_files

	break

    fi

    shift

done

if [ -z $FILES ]

then

    FILES=modified_files

fi

if [ -z $ACTION ]

then

    ACTION=update

fi

process_all