#!/usr/bin/env bash

############################################################
#                                                          #
#  EvoAdmin : gestion des comptes FTP virtuels de ProFTPd  #
#                                                          #
#  Copyright (c) 2009 Evolix - Tous droits reserves        #
#                                                          #
#  @author Sebastien Palma <spalma@evolix.fr>              #
#  @version 1.0                                            #
#                                                          #
############################################################

# vim: expandtab softtabstop=4 tabstop=4 shiftwidth=4 showtabline=2

set -o errexit
set -o pipefail
set -o nounset
#set -x

VPASSWD_PATH="/etc/proftpd/vpasswd"
FTPLOG_PATH="/var/log/evolix-ftp.log"

usage() {

cat <<EOT >&2

Usage: $0 [ACTION UID [PARAMS]]

    Manage ProFTPd virtual accounts.

    -a ACTION
       l (list), a (add), m (modify), d (delete) ProFTPd virtual account(s)

    -u UID
       UID of the real account the virtual FTP accounts will be related to

    Available PARAMS are :

    -n ACCOUNTNAME
       Name of the ProFTPd virtual account you want to add/edit/delete.
       Mandatory in all actions.

    -f FOLDER
       Directory that the virtual account will have as home directory.
       Mandatory for add and modify action.

    -p PASSWD
       Password for the virtual account.
       Mandatory for add and modify action.

EOT
}

log_msg() {
    curdate="$(date +"%Y/%m/%d %H:%M:%S")"
    echo "$curdate $1" >> "$FTPLOG_PATH"
}

get_user_login_by_UID() {
    uid=$1
    grep "$uid" /etc/passwd | awk -F : "{if (\$3==$uid) print \$1}"
}

list_accounts_by_UID() {
  uid=$1

  while IFS=$'\n' read -r line;
  do
     line_uid="$(echo "$line" | cut -d":" -f3)"

     if [[ ! "$uid" ]] ||  [[ "$line_uid" == "$uid" ]]; then
       username="$(get_user_login_by_UID "$line_uid")"
       account="$(echo "$line" | cut -d":" -f1)"
       path="$(echo "$line" | cut -d":" -f6)"
       size="$(du -s "$path" | cut -f 1)"
       #modif="$(cat $path/.lastmodified)"
       # Passage en minuscule ?
       #account="$(echo $account | tr '[A-Z]' '[a-z]')"
       #path="$(echo $path | tr '[A-Z]' '[a-z]')"

       echo "$username:$account:$path:$size${modif:+:$modif}"
       
     fi
  done < "$VPASSWD_PATH"
}

add_account() {
  user_id=$1
  account_name=$2
  path=$3
  passwd=$4

  cmd="{if (\$3==$user_id) print \$4}"
  user_gid="$(awk -F : "$cmd" /etc/passwd)"

  # Si le répertoire de travail du compte FTP n'existe pas, on le crée
  if [[ ! -d "$path" ]]; then
    mkdir -p "$path"
    chown "$user_id":"$user_gid" "$path"
    # fix by tmartin : s/655/755/
    chmod 755 "$path"
  fi

  echo "$passwd" | ftpasswd --passwd --file=$VPASSWD_PATH --name="$account_name" --uid="$user_id" --gid="$user_gid" --home="$path" --shell=/bin/false --stdin
  log_msg "Creation du compte $account_name (uid=$user_id, gid=$user_gid, home=$path)"
}

edit_password() {
  account_name=$1
  passwd=$2

  echo "$passwd" | ftpasswd --passwd --file="$VPASSWD_PATH" --name="$account_name" --uid=9999 --gid=9999 --home=/dev/null --shell=/dev/null --change-password --stdin

}


delete_account() {

  account_name=$1

  ftpasswd --passwd --file=$VPASSWD_PATH --name="$account_name" --uid=9999 --gid=9999 --home=/dev/null --shell=/dev/null --delete-user
  log_msg "Suppression du compte $account_name"
}

while getopts ha:u:n:f:p: opt; do
   case "$opt" in
   a)
    in_action=$OPTARG
    ;;
   u)
    in_userid=$OPTARG
    ;;
   n)
    in_accountname=$OPTARG
    ;;
   f)
    in_workpath=$OPTARG
    ;;
   p)
    in_password=$OPTARG
    ;;
   h)
    usage
    exit 1
    ;;
   *)
    usage
    exit 1
    ;;
   esac
done

case "${in_action-}" in
   l)
    echo -e "$(list_accounts_by_UID "${in_userid-}")"
    exit 1
    ;;
   a)
    if [[ -z "${in_userid-}" ]]; then
        echo "User ID not specified"
    elif [[ $in_userid = *[!0-9]* ]]; then
        echo "User ID must be a non negative integer"
    elif [[ -z "${in_accountname-}" ]]; then
        echo "Account name not specified"
    elif [[ -z "${in_workpath-}" ]]; then
        echo "A directory was not specified"
    elif [[ -z "${in_password-}" ]]; then
        echo "A password was not specified"
    else
        echo -e -n \
            "$(add_account \
                "$in_userid" \
                "$in_accountname" \
                "$in_workpath" \
                "$in_password")"
    fi
    exit 1
    ;;
   m)
    if [[ -z "${in_accountname-}" ]]; then
        echo "Account name not specified"    
    elif [[ -z "${in_password-}" ]]; then
        echo "A password was not specified"
    else
        echo -e -n \
            "$(edit_password \
                "$in_accountname" \
                "$in_password")"
    fi
    exit 1;
    ;;
   d)
    if [[ -z "${in_accountname-}" ]]; then
        echo "Account name not specified"
    else
        echo -e -n \
            "$(delete_account "$in_accountname")"
    fi
    exit 1;
    ;;
   *)
    usage
    exit 1
    ;;
esac