#!/bin/bash

function runQuery(){
    mysql -s -s -h $host -P $port -u $user -p$password -e "$1"
    if [ $? -ne 0 ]
    then
        echo "Failed to execute query: $1"
        exit
    fi
}

# Transform grants to from external hosts to MaxScale's host
function getGrants(){
    result=$(runQuery "show grants for $1"|sed -e "s/@[^ ]*/@'$maxscalehost'/" -e "s/ *IDENTIFIED BY.*//" -e "s/$/;/")
    echo "$result"
}

user=$(whoami)
host=$(hostname)
port=3306
include_root="and user <> 'root'"

if [ "$1" == "--help" ] || [ $# -eq 0 ]
then
    echo "$0 -u USER -p PASSWORD -h HOST -P PORT [-r]"
    exit
fi

while getopts "u:p:h:P:r" var
do
    case $var in
        u)
            user=$OPTARG
            ;;

        p)
            password=$OPTARG
            ;;

        h)
            host=$OPTARG
            ;;

        P)
            port=$OPTARG
            ;;
        r)
            include_root=""
            ;;
    esac
done

# Get the MaxScale hostname from the backend server
maxscalehost=$(runQuery "select user()")
maxscalehost=${maxscalehost#*@}

# List all the users
runQuery "select concat(\"'\", user, \"'\", '@', \"'\", host, \"'\") from mysql.user where user <> '' and host <> '%' $include_root"|while read i
do
    getGrants "$i"
done