#!/bin/bash
# Apr 2008, Dominik Klein
# Display scores of Linux-HA resources
#
# Aug 2008, Ronny Becker
# Added "reset failcount" function
#
# Known issues:
# * cannot get resource[_failure]_stickiness values for master/slave and clone resources
#   if those values are configured as meta attributes of the master/slave or clone resource
#   instead of as meta attributes of the encapsulated primitive

tmpfile=/tmp/dkshowscorestmpfiledk
tmpfile2=/tmp/dkshowscorestmpfile2dk
> $tmpfile
> $tmpfile2

i=1

if [ `crmadmin -D | cut -d' ' -f4` != `uname -n|tr "[:upper:]" "[:lower:]"` ] 
  then echo "Warning: Script is not running on DC. This will be slow."
fi

sortby=1
[ -n "$1" ] && [ "$1" = "node" ] && sortby=3

export default_stickiness=`cibadmin -Q -o crm_config 2>/dev/null|grep "default[_-]resource[_-]stickiness"|grep -o -E 'value ?= ?"[^ ]*"'|cut -d '"' -f 2|grep -v "^$"`
export default_failurestickiness=`cibadmin -Q -o crm_config 2>/dev/null|grep "resource[_-]failure[_-]stickiness"|grep -o -E 'value ?= ?"[^ ]*"'|cut -d '"' -f 2|grep -v "^$"`


2>&1 ptest -LVs | grep -v group_color | sed 's/dump_node_scores\:\ //' > $tmpfile

parseline() {
	line="$1"
        node=`echo $line|cut -d " " -f 9|sed 's/://'`
        res=`echo $line|cut -d " " -f 5`
        score=`echo $line|cut -d " " -f 10|sed 's/1000000/INFINITY/'`
}

get_stickiness() {
	res="$1"
	# get meta attribute resource_stickiness
	if ! stickiness=`crm_resource -g resource_stickiness -r $res --meta 2>/dev/null`
	then
		# if that doesnt exist, get syntax like <primitive resource-stickiness="100"
		if ! stickiness=`crm_resource -x -r $res 2>/dev/null | grep -E "<master|<primitive|<clone" | grep -o "resource[_-]stickiness=\"[0-9]*\"" | cut -d '"' -f 2 | grep -v "^$"`
		then 
			# if no resource-specific stickiness is confiugured, use the default value
			stickiness="$default_stickiness"
		fi	
	fi

	# get meta attribute resource_failure_stickiness
	if ! failurestickiness=`crm_resource -g resource_failure_stickiness -r $res --meta 2>/dev/null`
	then
		# if that doesnt exist, use the default value
		failurestickiness="$default_failurestickiness"
	fi	
}

get_failcount() {
	res="$1"
	node="$2"
        failcount=`crm_failcount -G -r $res -U $node 2>/dev/null|grep -o -E 'value ?= ?INFINITY|value ?= ?[0-9]*'|cut -d '=' -f 2|grep -v "^$"`
}

# display allocation scores
grep -v master_color $tmpfile | grep -v clone_color | while read line
do
	unset node res score stickiness failcount failurestickiness
	parseline "$line"
	get_stickiness $res
	get_failcount $res $node
	printf "%-4s%-32s%-16s%-32s%-11s%-9s%-16s\n" $i $res $score $node $stickiness $failcount $failurestickiness
        i=$(($i + 1))
done >> $tmpfile2

# display promotion scores
grep master_color $tmpfile | while read line
do
	unset node res score stickiness failcount failurestickiness
	parseline "$line"
	inflines=`grep master_color $tmpfile | grep $res | grep 1000000 | wc -l`
	if [ $inflines -eq 1 ]
	then
		# [10:24] <beekhof> the non INFINITY values are the true ones
		# [10:25] <kleind> except for when the actually resulting score is [-]INFINITY
		# [10:25] <beekhof> yeah
		actualline=`grep master_color $tmpfile | grep $res | grep -v 1000000`
		parseline "$actualline"
	fi
	get_stickiness $res
	get_failcount $res $node
	res=$res"_(master)"
	printf "%-4s%-32s%-16s%-32s%-11s%-9s%-16s\n" $i $res $score $node $stickiness $failcount $failurestickiness
        i=$(($i + 1))
done | sort | uniq >> $tmpfile2

# Heading
printf "%-4s%-32s%-16s%-32s%-11s%-9s%-16s\n" "ID" "Resource" "Score" "Node" "Stickiness" "#Fail" "Fail-Stickiness"

sort -k $sortby $tmpfile2

# reset failcount if wanted
echo ""
while [ 0 -lt 1 ]
do
 echo -n "If you want to reset a resources failcount, please type the ID (otherwise CTRL-C):"
 read userselect
 echo " resetting failcount for ID: $userselect ..."
 resource=`grep "^${userselect} " /tmp/dkshowscorestmpfile2dk | awk '{print $2}'`
 node=`grep "^${userselect} " /tmp/dkshowscorestmpfile2dk | awk '{print $4}'`
 crm_failcount -D -r $resource -U $node
 echo "done"
 echo ""
 echo ""
done

rm $tmpfile $tmpfile2