#!/usr/bin/bash
VERSION=23

PROJECT=$1  # Save the PowerVC Project name
USER=$2     # Save the PowerVC Admin user name
PASSWD=$3   # Save the PowerVC Admin user password
if [[ "$PROJECT" == "" ]]
then 
	echo Hint: ./meter Project Admin-User Password
	echo The project names are listed with: openstack project list 
	exit 42
fi
# Change these to your local costs
export CPUUNITCOST=12.50
export RAMUNITCOST=0.01
export DISKUNITCOST=0.50

# earliest timestamp as PowerVC did not support metering before 2016 June (YYY-MM-DD) this is a good start time
export TIMESTAMP="2016-06-07T00:00:00"

# Remind the user of the details we are fetching, the project and user name and cost numbers used
echo Metering for project:$PROJECT and user:$USER 
echo Costs: $CPUUNITCOST per CPU per day \(CPU Entitlement\)
echo Costs: $RAMUNITCOST per MB of memory per day
echo Costs: $DISKUNITCOST per GB of disk space per day

# We need the Project Id from the Project name supplied so look it up
set  OS_USERNAME=$USER
set  OS_PASSWORD=$PASSWORD
PROJECTID=$( openstack project list -f value | grep " "$PROJECT"$" | awk '{ printf "%s", $1 }' )
if [[ "$PROJECTID" == "" ]]
then 
	echo Project not found.  PROJECT = $PROJECT has PROJECTID = $PROJECTID
	exit 44
fi

# Get a PowerVC REST API authorisation token for later on getting the metering data
cat >out.auth.json <<EOF
{"auth": 
    {"scope": {"project": {"name": "$PROJECT", "domain": {"name": "Default"} } }, 
     "identity": {"methods": ["password"], "password": {"user": {"domain": {"name": "Default"}, "name": "$USER", "password": "$PASSWD" } } }
    }
}
EOF

# curl to Keystone security to logon and get the access token for later stats access.
curl -s -k -1 -i -X POST https://localhost:5000/v3/auth/tokens \
-H "Accept: application/json" \
-H "Content-Type: application/json" \
-d @out.auth.json 1>token 

TOKEN=$(grep ^X-Subject-Token token | awk '{ print $2 }' )
if [[ "$TOKEN" == "" ]]
then
echo No security token found for this user and password combination in the token file
exit 43
fi
# Debug echo Authorisation Token is $TOKEN

# everything ready  = TOKEN by the right project and user plus the PROJECTID
# get the metering / billing information from Ceilometer once each for CPU, memory and disk space
# note 86400 is the seconds in a day
# note http://localhost:8777/ could be replaced with the full https://localhost/powervc/openstack/metering/

curl -s -1 -k -H "X-Auth-Token:$TOKEN" \
-X GET "http://localhost:8777/v2/meters/total_vcpu/statistics?q.field=timestamp&q.op=gt&q.value=$TIMESTAMP&aggregate.func=max&period=86400&q.field=project_id&q.value=$PROJECTID" 1>out.vcpu

curl -s -1 -k -H "X-Auth-Token:$TOKEN" \
-X GET "http://localhost:8777/v2/meters/total_memory/statistics?q.field=timestamp&q.op=gt&q.value=$TIMESTAMP&aggregate.func=max&period=86400&q.field=project_id&q.value=$PROJECTID" 1>out.memory

curl -s -1 -k -H "X-Auth-Token:$TOKEN" \
-X GET "http://localhost:8777/v2/meters/total_volumes/statistics?q.field=timestamp&q.op=gt&q.value=$TIMESTAMP&aggregate.func=max&period=86400&q.field=project_id&q.value=$PROJECTID" 1>out.volumes


# output the spreadsheet comma separated values format file - heading information - 1st line creates the file then the rest append to it
echo PowerVC Ceilometer Metering stats for Billing - meter Script Version $VERSION >$PROJECT.$USER.bill.csv
echo Project,$PROJECT >$PROJECT.$USER.bill.csv
echo User,$USER >>$PROJECT.$USER.bill.csv
echo Date,`date` >>$PROJECT.$USER.bill.csv
echo  >>$PROJECT.$USER.bill.csv
echo CPU Costs,$CPUUNITCOST, per CPU per day \(CPU Entitlement\) >>$PROJECT.$USER.bill.csv
echo Memory Costs,$RAMUNITCOST, per MB of memory per day >>$PROJECT.$USER.bill.csv
echo Disk Costs,$DISKUNITCOST, per GB of disk space per day >>$PROJECT.$USER.bill.csv

# next process the ghastly json format for vcpu, memory and volumes
echo "" >>$PROJECT.$USER.bill.csv
echo "VCPU Costs (actually Entitled Capacity)" >>$PROJECT.$USER.bill.csv
cat out.vcpu | grep "^\[" | sed 's/, {\"/\n/g' | sed 's/"//g' | sed 's/://g' | sed 's/,//g'  | \
awk 'BEGIN { printf "Date,vCPU,UnitPrice,Price\n" } { printf "%s,%.2f,%.2f,%.2f\n", $12, $4, COST, $4 * COST }' COST=$CPUUNITCOST | sed -e 's/T000000//' >>$PROJECT.$USER.bill.csv

echo "" >>$PROJECT.$USER.bill.csv
echo "Memory RAM Costs in Megabytes" >>$PROJECT.$USER.bill.csv
cat out.memory | grep "^\[" | sed 's/, {\"/\n/g' | sed 's/"//g' | sed 's/://g' | sed 's/,//g'  | \
awk 'BEGIN { printf "Date,MB,UnitPrice,Price\n" } { printf "%s,%.2f,%.2f,%.2f\n", $12, $4, COST, $4 * COST }' COST=$RAMUNITCOST | sed -e 's/T000000//' >>$PROJECT.$USER.bill.csv

echo "" >>$PROJECT.$USER.bill.csv
echo "DISK Costs in Gigabytes" >>$PROJECT.$USER.bill.csv
cat out.volumes | grep "^\[" | sed 's/, {\"/\n/g' | sed 's/"//g' | sed 's/://g' | sed 's/,//g'  | \
awk 'BEGIN { printf "Date,GB,UnitPrice,Price\n" } { printf "%s,%.2f,%.2f,%.2f\n", $12, $4, COST, $4 * COST }' COST=$DISKUNITCOST | sed -e 's/T000000//' >>$PROJECT.$USER.bill.csv

echo "" >>$PROJECT.$USER.bill.csv
echo "End of file" >>$PROJECT.$USER.bill.csv

# comment out the rm for debugging
rm out.* token
echo Metering and Billing for project:$PROJECT and user:$USER output-file: $PROJECT.$USER.bill.csv