#根据情况设置为50M(对于isoa服务开发来说足够了)
maxmemory=3145728

#设置回收站所在的目录
trash=$HOME/.trash
#设置日志文件所在的目录
mvlog=$trash/trash.log

from1=$1
from2=$2

var_pwd=
var_father=

#回收站若不存在，则新建之
if [ ! -e $trash ];then
 mkdir -p $trash
 chmod 755 $trash
fi

#产生7位的随机数
function rand()
{
 a=(0 1 2 3 4 5 6 7 8 9 a b c d e A B C D E F)
 for ((i=0;i<7;i++))
 do
  echo -n ${a[$RANDOM%${#a[*]}]}
 done
}

random=$(rand)

#文件不存在时的提示信息
function file_null()
{
 local file=$1
 echo "rm: cannot remove '$file': No such file or directory"
}

#打印参数出错后的提示信息
function echo_msg()
{
 echo -n "rm: missing operand
Try 'rm --help' for more information.
"
}

function echo_msg2()
{
 echo -n "rm: invalid option  '$1'
Try 'rm --help' for more information.
"
}

#回收站管理函数
function deal()
{
 local tmp=$(mktemp /tmp/tfile.XXXXXX)
 local num=$(($(cat $mvlog| wc -l)/2))
 #awk -F: -v nu=$num  -v trash=$trash '{if (NR<=nu) system("rm -rf "trash"'/'"$2"':'"$3""); \
#else print $0}' $mvlog | sort -o $mvlog
 awk -F: -v nu=$num  -v trash=$trash '{if (NR<=nu) system("rm -rf "trash"'/'"$2"':'"$3""); \
else print $0}' $mvlog >> $tmp
 mv $tmp $mvlog
}

JUG=
#目录处理函数
function jug_cur()
{

 local tmp=
 local dirname=$1
 local jug=${dirname/\/*/}
 if [ "$jug" == "." ];then
  var_pwd=${dirname/./$(pwd)}
JUG=0
 elif [ "$jug" == ".." ];then
  tm=$(pwd)
  tmp=${tm%/*}  
  var_father=${dirname/../$tmp}  
  JUG=1
#elif [ "$jug" == "~" ];then
#return 2
 else
JUG=2
 fi 
}

#命令不带参数时的普通文件删除函数
function rm1
{
 local filename=$(basename $from1)
 local dirname=$(dirname $from1)

 jug_cur $dirname
 if [ "$JUG" -eq 0 ];then
  dirname=$var_pwd
 elif [ $JUG -eq 1 ];then
  dirname=$var_father
 fi 

 if [ -d "$from1" ];then
  echo "rm: cannot remove '$from1': Is a directory"
 else
  if [ ! -e $from1 ];then
   file_null $from1
  else
   echo "$dirname:$filename:$random:$(date +%Y-%m-%d.%T)" >> $mvlog
   mv "$from1" "$trash/$filename:$random"
  fi
 fi
}

#rm -i
function rmi()
{
 local filename=$(basename $from2)
 local dirname=$(dirname $from2)

 jug_cur $dirname
 if [ $JUG -eq 0 ];then
  dirname=$var_pwd
 elif [ $JUG -eq 1 ];then
  dirname=$var_father
 fi 

 if [ -f "$from2" ];then
  echo -n "rm: remove regular file '$from2'?"
  read answer
  if [ "$answer" = 'y' -o "$answer" = 'Y' ];then
   echo "$dirname:$filename:$random:$(date +%Y-%m-%d.%T)" >> $mvlog
   mv "$from2" "$trash/$filename:$random"
  fi
 else
  if [ ! -e $from2 ];then
   file_null $from2
  else
   echo "rm: cannot remove '$from2': Is a directory"
  fi
 fi
}

#rm -f
function rmf()
{
 local filename=$(basename $from2)
 local dirname=$(dirname $from2)

 jug_cur $dirname
 if [ $JUG -eq 0 ];then
  dirname=$var_pwd
 elif [ $JUG -eq 1 ];then
  dirname=$var_father
 fi 

 if [ -f "$from2" ];then
  echo "$dirname:$filename:$random:$(date +%Y-%m-%d.%T)" >> $mvlog
  mv "$from2" "$trash/$filename:$random"
 else
  if [ ! -e $from2 ];then
   :
  else
   echo "rm: cannot remove '$from2': Is a directory"
  fi
 fi
}

#rm -r
function rmr()
{
 local filename=$(basename $from2)
 local dirname=$(dirname $from2)

 jug_cur $dirname
 if [ $JUG -eq 0 ];then
  dirname=$var_pwd
 elif [ $JUG -eq 1 ];then
  dirname=$var_father
 fi 

 if [ "$from2" = "." -o "$from2" = ".." ];then
  echo "rm: cannot remove directory: '$from2'"
 elif [ -e "$from2" ];then
  echo "$dirname:$filename:$random:$(date +%Y-%m-%d.%T)" >> $mvlog
  mv "$from2" "$trash/$filename:$random"
 else
  file_null $from2
 fi
}

#rm -rf
function rmrf()
{
 local filename=$(basename $from2)
 local dirname=$(dirname $from2)

 jug_cur $dirname
 if [ $JUG -eq 0 ];then
  dirname=$var_pwd
 elif [ $JUG -eq 1 ];then
  dirname=$var_father
 fi 

 if [ "$from2" = "." -o "$from2" = ".." ];then
  echo "rm: cannot remove directory: '$from2'"
 elif [ -e "$from2" ];then
  echo "$dirname:$filename:$random:$(date +%Y-%m-%d.%T)" >> $mvlog
  mv "$from2" "$trash/$filename:$random"
 else
  :
 fi
}

#rm -ir
function rmir()
{
 local filename=$(basename $from2)
 local dirname=$(dirname $from2)

 jug_cur $dirname
 if [ $JUG -eq 0 ];then
  dirname=$var_pwd
 elif [ $JUG -eq 1 ];then
  dirname=$var_father
 fi 

 if [ -e "$from2" ];then
  if [ -d "$from2" ];then
   echo -n "rm: remove directory '$from2'?"
  else
   echo -n "rm: remove regular file '$from2'?"
  fi

  read answer
  if [ "$answer" = 'y' -o "$answer" = 'Y' ];then
   echo "$dirname:$filename:$random:$(date +%Y-%m-%d.%T)" >> $mvlog
   mv "$from2" "$trash/$filename:$random"
  fi
 else
  if [ ! -e $from2 ];then
   file_null $from2
  fi
 fi

}

#清空回收站
function rmc()
{
 /bin/rm -rf $trash
}

function rml()
{
 local tmp=$(mktemp /tmp/tfile.XXXXXX)
 clear

 if [ ! -d "$trash" ];then
  mkdir $trash
 fi

 if [ ! -f "$mvlog" ];then
  touch $mvlog
 fi
 line=$(cat -n $mvlog | awk -F: '{print $1, "FileName:"$2, "Time: "$4":"$5":"$6}')
 linecount=$(cat $mvlog | wc -l)
 echo -e "$line"
 echo
 echo
 echo "[$linecount] Please enter the file you want to restore (replaced with the line number)"
 printf "RowNumber: "
 read answer
 if [ "$answer" = 'q' -o "$answer" = 'Q' -o "$answer" = "" ];then
  :
 else
  printf "Please confirm (Y/N): "
  read answer1
  if [ "$answer1" = 'y' -o "$answer1" = 'Y' ];then
   address=$(sed -n "$answer""p" $mvlog | awk -F: '{print $1}')
   filename=$(sed -n "$answer""p" $mvlog | awk -F: '{print $2}')
   filerand=$(sed -n "$answer""p" $mvlog | awk -F: '{print $3}')
   fullname=$address/$filename
   if [ -e "$fullname" ];then
    echo "The file exist!"
    sleep 0.5
   else
    old="$trash/$filename:$filerand"
    new="$address/$filename"
    mv "$old" "$new"
    #deline=$(cat $mvlog|sed "$answer""d" | sort -o $mvlog)
    deline=$(cat $mvlog|sed "$answer""d" >> $tmp)
    mv $tmp $mvlog
    echo "restore success!"
    sleep 0.5
   fi
  fi
 fi
}

function help()
{
 cat << 'EOF'
Usage: rm [OPTION]... FILE...
Remove (unlink) the FILE(s).

-f, --force   ignore nonexistent files, never prompt
-i, --interactive prompt before any removal
--no-preserve-root do not treat `/' specially (the default)
--preserve-root   fail to operate recursively on `/'
-r, -R, --recursive   remove directories and their contents recursively
--help display this help and exit

By default, rm does not remove directories.  Use the --recursive (-r or -R)
option to remove each listed directory, too, along with all of its contents.

To remove a file whose name starts with a `-', for example `-foo',
use one of these commands:
rm -- -foo

rm ./-foo

Note that if you use rm to remove a file, it is usually possible to recover
the contents of that file.  If you want more assurance that the contents are
truly unrecoverable, consider using shred.

Report bugs to <[email protected]>.
EOF
}

#脚本开始

#检测回收站已用存储空间，如果已经达到最大值，则删除日志文件中位于前面的一半的文件
mem=$(du -s $trash|awk '{print $1}')
if [ "$mem" -gt $maxmemory ];then
 deal
fi

if [ "$#" -eq 0 ];then
 echo_msg
fi

if [ "$#" -eq 1 ];then
 case "$from1" in
  -i)
  echo_msg

  -f)
  echo_msg

  -r | -R)
  echo_msg

  -ir|-ri|-iR|-Ri|-if|-fi|-rf|-fr|-Rf|-fR)
  echo_msg

  -l)
  rml

  -c)
  rmc

  --help)
  help

  -*)
  echo_msg2 $from1

  *)
  rm1

 esac
fi

if [ "$#" -ge 2 ];then
 until  [ "$2" = "" ]
 do
  from2=$2
  case "$from1" in
   -i)
   rmi

   -f)
   rmf

   -r|-R)
   rmr

   -l)
   rml

   -rf|-Rf|-fr|-fR)
   rmrf

   -ir|-ri|-iR|-Ri)
   rmir

   -if|-fi)
   rmf

   --help)
   help
   exit 1

   -*)
   echo_msg2 $from1
   exit 1

   *)
   {
    until [ "$1" = "" ]
    do
     from1=$1
     rm1
     shift
    done
   }

  esac
  shift
 done
fi

exit