#!/bin/bash

set -euo pipefail

cp target/debug/examples/libloadrust.so /usr/lib/collectd/loadrust.so
cp target/debug/examples/libwrite_logrs.so /usr/lib/collectd/write_logrs.so
cp target/debug/examples/libmyerror.so /usr/lib/collectd/myerror.so

cat <<EOF | tee /etc/collectd/collectd.conf
Hostname "localhost"
LoadPlugin loadrust
LoadPlugin write_logrs
LoadPlugin csv
LoadPlugin logfile
LoadPlugin myerror

<Plugin logfile>
    LogLevel info
    File "/var/lib/collectd/log"
</Plugin>
<Plugin write_logrs>
    LogTimings "INFO"
</Plugin>
<Plugin csv>
  DataDir "/var/lib/collectd/csv"
  StoreRates false
</Plugin>
<Plugin loadrust>
</Plugin>
EOF

systemctl restart collectd
sleep 25
systemctl status collectd
systemctl stop collectd

grep_test() {
    echo grep "$1" "$2"
    EXIT_CODE=0

    # With set -e, we don't want to exit immediately, but instead add context
    # of what grep failed, so we force the command to succeed while capturing
    # the failing command's exit status:
    # https://stackoverflow.com/a/45729843/433785
    grep "$1" "$2" || EXIT_CODE=$? && true
    if [[ $EXIT_CODE -ne 0 ]]; then
        echo "Not found: $1 in $2"
        echo "contents of $2:"
        cat "$2"
    fi
    return $EXIT_CODE
}

grep_test 'epoch,shortterm,midterm,longterm' /var/lib/collectd/csv/localhost/loadrust/load*
grep_test 'A raw log with argument: 10' /var/lib/collectd/log
grep_test 'collectd logging configuration: Some' /var/lib/collectd/log
grep_test 'write_logrs: write_logrs: rust logging configuration: Some' /var/lib/collectd/log
grep_test 'write_logrs: write_logrs: flushing: timeout: no timeout, identifier: no identifier' /var/lib/collectd/log
grep_test 'write_logrs: write_logrs: yes drop is called' /var/lib/collectd/log
grep_test 'myerror: collectd_plugin::api::logger: read error: plugin encountered an error; bailing' /var/lib/collectd/log
grep_test 'read-function of plugin `myerror'"'"' failed.' /var/lib/collectd/log
grep_test 'myerror: collectd_plugin::api::logger: read error: plugin panicked' /var/lib/collectd/log
grep_test 'myerror: collectd_plugin::api::logger: panic hook error: plugin panicked: (examples/myerror.rs: 45): Oh dear what is wrong!?' /var/lib/collectd/log

exit $?