#/bin/bash
#
# Script para monitoração de serviço
#
# Procedimentos:
# Este script foi desenvolvido com intuito de facilitar a administração
# de processos do servidor, implementando o recurso de "kill" para os
# processos de um determinado usuário, por sessão (tty) e/ou processo
# específico.
#

# Variáveis do script
KILL="/bin/kill"
PS="/bin/ps"
GREP="/bin/grep"
AWK="/bin/awk"
MORE="/bin/more"

if [ "$1" ];
then
  opc1=$(echo "$1" | $AWK -F'=' '{print $1}')
  opcPar1=$(echo "$1" | $AWK -F'=' '{print $2}')
fi
if [ "$2" ];
then
  opc2=$(echo "$2" | $AWK -F'=' '{print $1}')
  opcPar2=$(echo "$2" | $AWK -F'=' '{print $2}')
fi
if [ "$3" ];
then
  opc3=$(echo "$3" | $AWK -F'=' '{print $1}')   opcPar3=$(echo "$3" | $AWK -F'=' '{print $2}') fi

varOpc=$1
parUsuario=$2
parOpc=$3

# Fazer verificação dos parâmetros recebidos
if [ ! "$1" ];
then
  # Mostrar mensagem de utilização
  echo "kill_user.sh - Utilitário para Gerenciamento de Processos"
  echo -e "\n kill_user.sh é uma ferramenta utilizada para matar os"
  echo "processos de um determinado usuário, em uma determinada sessão, "
  echo "tendo como opcional a seleção de um processo específico."
  echo -e "\nUtilização: kill_user.sh -user=usuario [-tty=tty] [-prog=programa] \n"
  echo -e "Opções: \n"
  echo -e " -user \t - Especifica o usuário para busca de processos (obrigatório)"
  echo -e " -prog \t - Especifica programa da sessão do usuário (opcional)"
  echo -e " -tty \t - Especifica terminal da sessão do usuário (opcional)"
  echo -e "Exemplo:"
  echo -e " $ $0 -user=usuario -tty=tty1 -prog=programa \n"
  exit 1
fi

# Usuário
# Parâmetros recebidos pelo script
if [ "$opc1" -a "$opc1" == "-user" -a "$opcPar1" ];
then
  processos=$($PS aux | $AWK -F' ' '{print $1,$2,$7,$11}' | $GREP ^$opcPar1 | $GREP -v grep)
  parUsuario=$opcPar1
fi

# Programa

if [ "$opc2" -a "$opc2" == "-prog" -a "$opcPar2" ];
then
  if [ "$processos" ];
  then
    processos=$(echo "$processos" | $AWK -F' ' '{print $1,$2,$3,$4}' | $GREP $opcPar2 | $GREP -v grep)
  else
    processos=$($PS aux | $AWK -F' ' '{print $1,$2,$7,$11}' | $GREP $opcPar2 | $GREP -v grep)
  fi
elif [ "$opc2" -a "$opc2" == "-tty" -a "$opcPar2" ];
then
  if [ "$processos" ];
  then
  processos=$(echo "$processos" | $AWK -F' ' '{print $1,$2,$3,$4}' | $GREP $opcPar2 | $GREP -v grep)
  else
  processos=$($PS aux | $AWK -F' ' '{print $1,$2,$7,$11}' | $GREP $opcPar2 | $GREP -v grep)
  fi
fi

# Seção (tty)
if [ "$opc3" -a "$opc3" == "-tty" -a "$opcPar3" ];
then
  if [ "$processos" ];
  then
    processos=$(echo "$processos" | $AWK -F' ' '{print $1,$2,$3,$4}' | $GREP $opcPar3 | $GREP -v grep)
  else
    processos=$($PS aux | $AWK -F' ' '{print $1,$2,$7,$11}' | $GREP $opcPar3 | $GREP -v grep)
  fi
elif [ "$opc3" -a "$opc3" == "-prog" -a "$opcPar3" ];
then
  if [ "$processos" ];
  then
    processos=$(echo "$processos" | $AWK -F' ' '{print $1,$2,$3,$4}' | $GREP $opcPar3 | $GREP -v grep)
  else
  processos=$($PS aux | $AWK -F' ' '{print $1,$2,$7,$11}' | $GREP $opcPar3 | $GREP -v grep)
  fi
fi

# Mostrar listagem de processos encontrados
echo -e "\n--------------------------------------------------------------------------------"
echo -e " Kill User - Gerenciador Simples de processos para sistemas UNIX - Shell Script"
echo -e "--------------------------------------------------------------------------------"
echo -e " Processos encontrados para usuário [$parUsuario]: \n"
echo "$processos" | $AWK -F' ' '{print $2,$3,$4}' | $MORE

echo -e "\n\nConfirme execução ? [s/N]: \c";
read varConfirma

if [ "$varConfirma" == "s" -o "$varConfirma" == "S" ];
then
  # Executar comando
  echo -e "\nProcessos do usuário [$parUsuario] foram cancelados com sucesso!\n"
  pidProcessos=$(echo "$processos" | $AWK -F' ' '{print $2}')
  $KILL -9 $pidProcessos 2> /dev/null
else
  # Mostrar mensagem de aviso
  echo -e "\nProcesso do usuário [$parUsuario] não foram cancelados..."
  echo -e "Você não confirmou a execução!\n"
fi