RhodeCode

#!/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.

}

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 -n " [$3 failed]"

	FAILED=YES

    fi

}

function check_warning() {

    echo -n " [$2 warning]"

    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.

    if [ $FAILED_FILES -gt 0 ]

    then

	return 1

    elif [ $WARNED_FILES -gt 0 ]

    then

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

	then

	    while read answer

	    do

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

		then

		    return 0

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

		then

		    return 1

		fi

	    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" '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

  --interactive|-i

  --werror|-w

  --all|-a

     All files in the repository will be checked.

  --modified|-m

  --changed|-c

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

       [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 will be checked/unified

"

        exit 0

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

    then

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

	ACTION=check

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

    then

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

	FILES=all_files

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

    then

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

	FILES=changed_files

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

    then

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

	FILES=modified_files

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

    then

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

	WARNING='INTERACTIVE'

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

    then

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

	WARNING='WERROR'

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

    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