#!/bin/bash
SUDO_BINARY="/usr/bin/sudo"
 
if [[ $# -eq 0 ]] ; then
  echo 'usage: ambari-sudo.sh [sudo_arg1, sudo_arg2 ...] command [arg1, arg2 ...]'
  exit 1
fi
 
# if user is non-root
if [ "$EUID" -ne 0 ] ; then
  $SUDO_BINARY "$@"
else
  ENV=()
  SUDO_ARGS=()
 
  for i ; do
    if [[ "$i" == *"="* ]] ; then
      ENV+=("$i")
      shift
    elif [[ "$i" == "-"* ]] ; then
      SUDO_ARGS+=("$i")
      shift
    else
      break
    fi
  done
   
  #echo "sudo arguments: ${SUDO_ARGS[@]}"
  #echo "env: ${ENV[@]}"
  #echo "args: $@"
 
  if [ "$ENV" ] ; then
    export "${ENV[@]}"
  fi
 
  "$@"
fi

       for name [ [ in [ word ... ] ] ; ] do list ; done
              The list of words following in is expanded, generating a list of
              items.  The variable name is set to each element of this list in
              turn, and list is executed each time.  If the in word  is  omit‐
              ted,  the  for  command  executes  list once for each positional
              parameter that is set (see PARAMETERS below).  The return status
              is  the  exit  status of the last command that executes.  If the
              expansion of the items following in results in an empty list, no
              commands are executed, and the return status is 0.