#!/bin/bash

#author liding@zlhy.com

function helpfunc(){

         echo -e "\033[36m  Please check your parameters, -D for db name, -M main library IP, -I for authorized IP address

           options:

           -D   Need to authorize the db name

           -M   Execute on this IP

           -I   Authorized IP

           -T   Authorization type 1 is the program account, 2 read-only account, 3 administrator account ; you can specify multiple to ',' separation

           eg : mysqlha_winreplication.sh -D  -M -I -T

           Generate 3 accounts:

                  1st: program link account. permissions: SELECT, INSERT, UPDATE, DELETE, EXECUTE, no DROP, CREATE;

                  2nd: read-only account. permissions: Select

                  3rd: administrator account. permissions: ALL \033[0m"

}

############################################

type=2

if [ $# -lt 1 ] ;

    then

        helpfunc

         exit 1

else

       while getopts "M:D:I:T:" Option

       do

               case $Option in

                       D) db=$OPTARG;;

                       M) master_ip=$OPTARG;;

                       I) grant_ip=$OPTARG;;

                       T) type=$OPTARG;;

                        *) helpfunc; exit 1; ;;

               esac

       done

fi

type=$(echo $type |tr ',' ' ')

function GetPwd(){

status=$1

if [ $status -eq 0 ];then

        A=`head -c 500 /dev/urandom | tr -dc A-Z |head -c 1`

        B=`head -c 500 /dev/urandom | tr -dc [:alnum:]| head -c 6`

        C=`echo $RANDOM$RANDOM|cut -c 2`

        echo $A$B$C

else

        A=`head -c 500 /dev/urandom | tr -dc A-Z |head -c 1`

        B=`head -c 500 /dev/urandom | tr -dc [:alnum:]| head -c 12`

        C=`echo $RANDOM$RANDOM|cut -c 2`

        echo $A$B$C

fi

}

        #       echo -e "\033[42;37m     \033[0m"

function GetProgram(){

pro_pwd=$(GetPwd 0)

#ssh administrator@${master_ip}  "cd /cygdrive/d/data/mysqld_master && cmd /c sql.bat \"$sql\"

##读写

pro_sql="GRANT SELECT,INSERT,UPDATE,DELETE,EXECUTE ON ${db}.* TO ${db}@'"localhost"' IDENTIFIED BY '"$pro_pwd"'; "

pro_sql1="GRANT SELECT,INSERT,UPDATE,DELETE,EXECUTE ON ${db}.* TO ${db}@'"${grant_ip}"' IDENTIFIED BY '"$pro_pwd"'; "

for sql in "$pro_sql" "$pro_sql1"

do

        if ssh administrator@${master_ip}  "cd /cygdrive/d/data/mysqld_master && cmd /c sql.bat \"$sql\" ";then

                :

        else

                echo -e "\033[31m  To grant authorization{IP:${master_ip} 权限:读写 } faild \033[0m"

        fi

done

echo -e "\033[32m   To grant authorization{IP:${master_ip} 权限:读写 } sucess   \033[0m"

echo -e "\033[32m   读写账户:${db}                                                      \033[0m"

echo -e "\033[32m   读写密码:${pro_pwd}                                                 \033[0m"

}

function GetSelect(){

select_pwd=$(GetPwd 1)

##只读

select_sql="GRANT SELECT ON ${db}.* TO ${db}_r@'"localhost"' IDENTIFIED BY '"$select_pwd"';"

select_sql1="GRANT SELECT ON ${db}.* TO ${db}_r@'"${grant_ip}"' IDENTIFIED BY '"$select_pwd"';"

for sql in "$select_sql" "$select_sql1"

do

        if ssh administrator@${master_ip}  "cd /cygdrive/d/data/mysqld_master && cmd /c sql.bat \"$sql\" ";then

                :

        else

                echo -e "\033[31m  To grant authorization{IP:${master_ip} 权限:只读 } faild \033[0m"

        fi

        else

                echo -e "\033[31m  To grant authorization{IP:${master_ip} 权限:只读 } faild \033[0m"

        fi

done

echo -e "\033[32m   To grant authorization{IP:${master_ip} 权限:只读 } sucess   \033[0m"

echo -e "\033[32m   只读账户:${db}_r                                            \033[0m"

echo -e "\033[32m   只读密码:${select_pwd}                                              \033[0m"

}

function GetAdmin(){

admin_pwd=$(GetPwd 2)

#管理员

admin_sql="grant select,insert,update,delete,create,drop,alter,index,lock tables ,CREATE TEMPORARY TABLES  ON ${db}.* TO ${db}_admin@'"localhost"' IDENTIFIED BY  '"$admin_pwd"';"

admin_sql="grant select,insert,update,delete,create,drop,alter,index,lock tables ,CREATE TEMPORARY TABLES  ON ${db}.* TO ${db}_admin@'"127.0.0.1"' IDENTIFIED BY  '"$admin_pwd"';"

for sql in "$admin_sql" "$admin_sql1"

do

        if ssh administrator@${master_ip}  "cd /cygdrive/d/data/mysqld_master && cmd /c sql.bat \"$sql\" ";then

                :

        else

                echo -e "\033[31m  To grant authorization{IP:${master_ip} 权限:管理员 } faild \033[0m"

        fi

done

echo -e "\033[32m   To grant authorization{IP:localhost 权限:管理员 } sucess   \033[0m"

echo -e "\033[32m   管理员账户:${db}_admin                                              \033[0m"

echo -e "\033[32m   管理员密码:${admin_pwd}                                             \033[0m"

}

function main(){

for stu in $type

do

        if [ $stu -eq 1 ];then

                GetProgram

        elif [ $stu -eq 2 ];then

                GetSelect

        elif [ $stu -eq 3 ];then

                GetAdmin

        else

                echo -e "\033[31m  Please check if your type parameter is in range \033[0m"

        fi

done

}

main

@echo off

set allparam=%1

D:\data\mysqld_master\bin\mysql.exe -uroot -e %allparam%

@echo off

set allparam=%2

set port=%1

echo %1

echo %2

D:\data\mysqld_slave\bin\mysql.exe -uroot -P%port% -e %allparam%