#!/bin/sh
#
# Copyright (c) 1995-2001,2004 Silicon Graphics, Inc.  All Rights Reserved.
# 
# 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 2 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, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
#

# Get standard environment
. $PCP_DIR/etc/pcp.env
# ... and _is_archive()
. $PCP_SHARE_DIR/lib/utilproc.sh

_usage()
{
    echo "Usage: mkaf [findopts] filename ..."
}

# strip any known compression suffix from the filename(s) on stdin
#
_strip_compress_suffix()
{
    sed \
	-e 's/\.xz$//' \
	-e 's/\.lzma$//' \
	-e 's/\.bz2$//' \
	-e 's/\.bz$//' \
	-e 's/\.gz$//' \
	-e 's/\.Z$//' \
	-e 's/\.z$//' \
	-e 's/\.zst$//' \
    # end
}

if [ $# -eq 0 ]
then
    _usage
    exit 1
fi

tmp=`mktemp -d "$PCP_TMPFILE_DIR/pm_mkaf.XXXXXXXXX"` || exit 1
status=0
trap "rm -rf $tmp; exit \$status" 0 1 2 3 15
findopts=""

while [ $# -gt 0 ]
do
    case $1
    in
	-?)
	    _usage
	    exit 1
	    ;;
	-*)
	    findopts="$findopts $1"
	    ;;

	*)
	    if [ -d $1 ]
	    then
		[ -z "$findopts" ] && findopts="-follow"
		$PCP_ECHO_PROG >&2 $PCP_ECHO_N "Searching \"find $1 $findopts ...\" $PCP_ECHO_C"
		find $1 $findopts -type f -print \
		| while read file
		do
		    if _is_archive $file
		    then
			echo $file | _strip_compress_suffix >>$tmp/base
		    fi
		done
		$PCP_ECHO_PROG >&2 " done"
	    elif [ ! -f $1 ]
	    then
		echo >&2 "mkaf: $1: No such file"
	    elif _is_archive $1
	    then
		echo $1 | _strip_compress_suffix >>$tmp/base
	    else
		echo >&2 "mkaf: $1: Not a PCP archive file"
	    fi
	    ;;

    esac

    shift
done


if [ ! -s $tmp/base ]
then
    echo >&2 "mkaf: Warning: no PCP archives found, so no folio created"
    status=1
    exit
fi

host=somehost
which hostname >/dev/null 2>&1 && host=`hostname`

cat <<End-of-File
PCPFolio
Version: 1
# use pmafm(1) to process this PCP archive folio
#
Created: on $host at `date`
Creator: pmchart
# 		Host			Basename
#
End-of-File

sed <$tmp/base \
    -e 's/\.[0-9][0-9]*$//' \
    -e 's/\.meta$//' \
    -e 's/\.index$//' \
| sort -u \
| while read base
do
    host=`pmdumplog -l $base 2>&1 | sed -n -e '/^Performance metrics/s/.* host //p'`
    if [ -z "$host" ]
    then
	echo >&2 "mkaf: Warning: cannot extract hostname from archive \"$base\" ... skipped"
    else
	printf "%-15s %-23s %s\n" "Archive:" "$host" "$base"
    fi
done

exit