#!/bin/bash
# Date:2018/11/29
# Author:  linyaohong
# BLOG:  ************
# Qq:122123498
# Version:1.0
# User: Centos 7

export PATH=/sbin:/bin:/usr/sbin:/usr/bin:/usr/local/sbin:/usr/local/bin

# set color
echo=echo
for cmd in echo /bin/echo; do
  $cmd >/dev/null 2>&1 || continue
  if ! $cmd -e "" | grep -qE '^-e'; then
    echo=$cmd
    break
  fi
done
CSI=$($echo -e "\033[")
CEND="${CSI}0m"
CDGREEN="${CSI}32m"
CRED="${CSI}1;31m"
CGREEN="${CSI}1;32m"
CYELLOW="${CSI}1;33m"
CBLUE="${CSI}1;34m"
CMAGENTA="${CSI}1;35m"
CCYAN="${CSI}1;36m"
CSUCCESS="$CDGREEN"
CFAILURE="$CRED"
CQUESTION="$CMAGENTA"
CWARNING="$CYELLOW"
CMSG="$CCYAN"

# Set pureftpd variable
pureftpd_install_dir=/usr/local/pureftpd
run_user=www
FTP_conf=${pureftpd_install_dir}/etc/pure-ftpd.conf
FTP_tmp_passfile=${pureftpd_install_dir}/etc/pureftpd_psss.tmp
Puredbfile=${pureftpd_install_dir}/etc/pureftpd.pdb
Passwdfile=${pureftpd_install_dir}/etc/pureftpd.passwd
FTP_bin=${pureftpd_install_dir}/bin/pure-pw

# Check if user is root
[ $(id -u) != "0" ] && { echo "${CFAILURE}Error: You must be root to run this script${CEND}"; exit 1; }

# Check pureftpd install or no
[ ! -d "${pureftpd_install_dir}" ] && { echo "${CFAILURE}FTP server does not exist! ${CEND}"; exit 1; }
[ -z "`grep ^PureDB $FTP_conf`" ] && { echo "${CFAILURE}pure-ftpd is not own password database${CEND}" ; exit 1; }

USER() {
while :; do echo
  read -e -p "Please input a username: " User
  if [ -z "$User" ]; then
    echo "${CWARNING}username can't be NULL! ${CEND}"
  else
    break
  fi
done
}

PASSWORD() {
while :; do echo
  read -e -p "Please input the password: " Password
  [ -n "`echo $Password | grep '[+|&]'`" ] && { echo "${CWARNING}input error,not contain a plus sign (+) and &${CEND}"; continue; }
  if (( ${#Password} >= 5 ));then
    echo -e "${Password}\n$Password" > $FTP_tmp_passfile
    break
  else
    echo "${CWARNING}Ftp password least 5 characters! ${CEND}"
  fi
done
}

DIRECTORY() {
while :; do echo
  read -e -p "Please input the directory(Default directory: ${wwwroot_dir}): " Directory
  if [ -z "$Directory" ]; then
    Directory="${wwwroot_dir}"
  fi
  if [ ! -d "$Directory" ]; then
    echo "${CWARNING}The directory does not exist${CEND}"
  else
    break
  fi
done
}

while :; do
  printf "
What Are You Doing?
\t${CGREEN}1${CEND}. UserAdd
\t${CGREEN}2${CEND}. UserMod
\t${CGREEN}3${CEND}. UserPasswd
\t${CGREEN}4${CEND}. UserDel
\t${CGREEN}5${CEND}. ListAllUser
\t${CGREEN}6${CEND}. ShowUser
\t${CGREEN}q${CEND}. Exit
"
  read -e -p "Please input the correct option: " Number
  if [[ ! $Number =~ ^[1-6,q]$ ]]; then
    echo "${CFAILURE}input error! Please only input 1~6 and q${CEND}"
  else
    case "$Number" in
    1)
      USER
      [ -e "$Passwdfile" ] && [ -n "`grep ^${User}: $Passwdfile`" ] && { echo "${CQUESTION}[$User] is already existed! ${CEND}"; continue; }
      PASSWORD;DIRECTORY
      $FTP_bin useradd $User -f $Passwdfile -u $run_user -g $run_user -d $Directory -m < $FTP_tmp_passfile
      $FTP_bin mkdb $Puredbfile -f $Passwdfile > /dev/null 2>&1
      echo "#####################################"
      echo
      echo "[$User] create successful! "
      echo
      echo "You user name is : ${CMSG}$User${CEND}"
      echo "You Password is : ${CMSG}$Password${CEND}"
      echo "You directory is : ${CMSG}$Directory${CEND}"
      echo
      ;;
    2)
      USER
      [ -e "$Passwdfile" ] && [ -z "`grep ^${User}: $Passwdfile`" ] && { echo "${CQUESTION}[$User] was not existed! ${CEND}"; continue; }
      DIRECTORY
      $FTP_bin usermod $User -f $Passwdfile -d $Directory -m
      $FTP_bin mkdb $Puredbfile -f $Passwdfile > /dev/null 2>&1
      echo "#####################################"
      echo
      echo "[$User] modify a successful! "
      echo
      echo "You user name is : ${CMSG}$User${CEND}"
      echo "You new directory is : ${CMSG}$Directory${CEND}"
      echo
      ;;
    3)
      USER
      [ -e "$Passwdfile" ] && [ -z "`grep ^${User}: $Passwdfile`" ] && { echo "${CQUESTION}[$User] was not existed! ${CEND}"; continue; }
      PASSWORD
      $FTP_bin passwd $User -f $Passwdfile -m < $FTP_tmp_passfile
      $FTP_bin mkdb $Puredbfile -f $Passwdfile > /dev/null 2>&1
      echo "#####################################"
      echo
      echo "[$User] Password changed successfully! "
      echo
      echo "You user name is : ${CMSG}$User${CEND}"
      echo "You new password is : ${CMSG}$Password${CEND}"
      echo
      ;;
    4)
      if [ ! -e "$Passwdfile" ]; then
        echo "${CQUESTION}User was not existed! ${CEND}"
      else
        $FTP_bin list
      fi

      USER
      [ -e "$Passwdfile" ] && [ -z "`grep ^${User}: $Passwdfile`" ] && { echo "${CQUESTION}[$User] was not existed! ${CEND}"; continue; }
      $FTP_bin userdel $User -f $Passwdfile -m
      $FTP_bin mkdb $Puredbfile -f $Passwdfile > /dev/null 2>&1
      echo
      echo "[$User] have been deleted! "
      ;;
    5)
      if [ ! -e "$Passwdfile" ]; then
        echo "${CQUESTION}User was not existed! ${CEND}"
      else
        $FTP_bin list
      fi
      ;;
    6)
      USER
      [ -e "$Passwdfile" ] && [ -z "`grep ^${User}: $Passwdfile`" ] && { echo "${CQUESTION}[$User] was not existed! ${CEND}"; continue; }
      $FTP_bin show $User
      ;;
    q)
      exit
      ;;
    esac
  fi
done