#!/bin/bash
# Author:  yeho <lj2007331 AT gmail.com>
# BLOG:  https://blog.linuxeye.cn
#
# Notes: OneinStack for CentOS/RedHat 6+ Debian 7+ and Ubuntu 12+
#
# Project home page:
#       https://oneinstack.com
#       https://github.com/lj2007331/oneinstack

export PATH=/sbin:/bin:/usr/sbin:/usr/bin:/usr/local/sbin:/usr/local/bin
clear
printf "
#######################################################################
#       OneinStack for CentOS/RedHat 6+ Debian 7+ and Ubuntu 12+      #
#                 FTP virtual user account management                 #
#       For more information please visit https://oneinstack.com      #
#######################################################################
"
# Check if user is root
[ $(id -u) != "0" ] && { echo "${CFAILURE}Error: You must be root to run this script${CEND}"; exit 1; }

oneinstack_dir=$(dirname "`readlink -f $0`")
pushd ${oneinstack_dir} > /dev/null
. ./options.conf
. ./include/color.sh

[ ! -d "${pureftpd_install_dir}" ] && { echo "${CFAILURE}FTP server does not exist! ${CEND}"; exit 1; }

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
[ -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
  Directory=${Directory:-${wwwroot_dir}}
  if [ ! -d "${Directory}" ]; then
    echo "${CWARNING}The directory does not exist${CEND}"
  else
    break
  fi
done
}

while :; do
  printf "
What Are You Doing?
\t${CMSG}1${CEND}. UserAdd
\t${CMSG}2${CEND}. UserMod
\t${CMSG}3${CEND}. UserPasswd
\t${CMSG}4${CEND}. UserDel
\t${CMSG}5${CEND}. ListAllUser
\t${CMSG}6${CEND}. ShowUser
\t${CMSG}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