#!/bin/bash

set -euo pipefail

# Check if the the user has invoked the image with flags.
# eg. "metricbeat -c metricbeat.yml"
if [[ -z $1 ]] || [[ ${1:0:1} == '-' ]] ; then
  exec metricbeat "$@"
else
  # They may be looking for a Beat subcommand, like "metricbeat setup".
  subcommands=$(metricbeat help \
                  | awk 'BEGIN {RS=""; FS="\n"} /Available Commands:/' \
                  | awk '/^\s+/ {print $1}')

  # If we _did_ get a subcommand, pass it to metricbeat.
  for subcommand in $subcommands; do
      if [[ $1 == $subcommand ]]; then
        exec metricbeat "$@"
      fi
  done
fi

# If neither of those worked, then they have specified the binary they want, so
# just do exactly as they say.
exec "$@"
