#!/bin/bash
  
# 定义脚本所在目录的路径  
script_dir=$(cd "$(dirname "$0")" && pwd)  


# 定义默认的exclude_file和include_file  
default_username="root"  
default_server="192.168.12.52"  
default_sources=$(ls $script_dir)  
default_destination="$script_dir/"  
default_exclude_file="$script_dir/exclude.txt"  
default_include_file="$script_dir/include.txt"  

  
#################################################################
####脚本变量定义#################################################
#################################################################

# 移动脚本  
move_script() {  
  read -p "请输入移动路径: " dist_dir  
  mv rsync_*.sh include.txt exclude.txt $dist_dir
}

# 定义主函数  
rsync_files() {  
  read -p "请输入同步的文件(逗号分隔): " key  
  server="$default_server"  
  username="$default_username"
  sources="$default_sources"
  destination="$default_destination"  
  
    
  # 检查必要参数是否提供，否则使用默认值  
  if [[ -z "$server" ]]; then  
    server="$default_server"  
  fi 

  if [[ -z "$username" ]]; then  
    username="$default_username"  
  fi 
  
  # 检查目标目录是否为空，如果是，则设置为当前脚本所在的目录  
  if [[ -z "$sources" ]]; then  
    sources= $(ls $default_sources) 
  fi  
  
  # 检查目标目录是否为空，如果是，则设置为当前脚本所在的目录  
  if [[ -z "$destination" ]]; then  
    destination="$default_destination"  
  fi  
  
  if [[ -z "$exclude_file" ]]; then  
    exclude_file="$default_exclude_file"  
  fi   
    
  if [[ -z "$include_file" ]]; then  
    include_file="$default_include_file"  
  fi  
  
  # 输出信息  
  echo "---------------------------------------------"  
  echo "服务器：$server"  
  echo "用户名：$username"   
  #echo "所有资源：$sources"  
  echo "同步文件：$key"  
  echo "目标目录：$destination"  
  echo "使用 exclude_file 文件: $exclude_file"  
  echo "使用 include_file 文件: $include_file"  
  echo "开始同步至目录..."  
  echo "---------------------------------------------"  
  
  # 检查文件是否存在  
  if [[ ! -f "$exclude_file" ]]; then  
    echo "exclude_file 文件不存在: $exclude_file"  
    exit 1  
  fi  
    
  if [[ ! -f "$include_file" ]]; then  
    echo "include_file 文件不存在: $include_file"  
    exit 1  
  fi  
  
  for source in $sources; do 
	if [ "$source" != "$key" ]; then  
	# 	echo "当前匹配 $script_dir/$key, 当前文件: $source"
		continue  
	fi  

	if [ "$source" == "exclude.txt" ]; then  
		continue  
	fi  
	if [ "$source" == "include.txt" ]; then  
		continue  
	fi  
	if [[ $source =~ rsync_.*.sh ]]; then 
		continue  
	fi   	
	
    echo "当前同步资源文件: $script_dir/$source"  
	# 判断是否为文件  
	if [ -f "$script_dir/$source" ]; then  
		echo "文件：rsync -a --exclude-from=$exclude_file "$script_dir/$source" $username@$server:$destination$source"
		
	#	rsync -a --exclude-from="$exclude_file" \  
	#	"$script_dir/$source" \  
	#	"$username@$server:$destination$source"
	fi  
	  
	# 判断是否为目录  
	if [ -d "$script_dir/$source" ]; then  
		echo "目录：rsync -a --exclude-from=$exclude_file "$script_dir/$source/" $username@$server:$destination$source/"
		
	#	rsync -a --exclude-from="$exclude_file" \  
	#	"$script_dir/$source/" \  
	#	"$username@$server:$destination$source/"
	fi


  done 
}


# 定义主函数  
rsync_all_files() {  
  server="$default_server"  
  username="$default_username"
  sources="$default_sources"
  destination="$default_destination"  
  
    
  # 检查必要参数是否提供，否则使用默认值  
  if [[ -z "$server" ]]; then  
    server="$default_server"  
  fi 

  if [[ -z "$username" ]]; then  
    username="$default_username"  
  fi 
  
  # 检查目标目录是否为空，如果是，则设置为当前脚本所在的目录  
  if [[ -z "$sources" ]]; then  
    sources= $(ls $default_sources) 
  fi  
  
  # 检查目标目录是否为空，如果是，则设置为当前脚本所在的目录  
  if [[ -z "$destination" ]]; then  
    destination="$default_destination"  
  fi  
  
  if [[ -z "$exclude_file" ]]; then  
    exclude_file="$default_exclude_file"  
  fi   
    
  if [[ -z "$include_file" ]]; then  
    include_file="$default_include_file"  
  fi  
    
  
  # 输出信息  
  echo "---------------------------------------------"  
  echo "服务器：$server"  
  echo "用户名：$username"   
  # echo "所有资源：$sources"  
  echo "目标目录：$destination"  
  echo "使用 exclude_file 文件: $exclude_file"  
  echo "使用 include_file 文件: $include_file"  
  echo "开始同步至目录..."  
  echo "---------------------------------------------"  
  
  
  # 检查文件是否存在  
  if [[ ! -f "$exclude_file" ]]; then  
    echo "exclude 文件不存在: $exclude_file"  
    exit 1  
  fi  
    
  if [[ ! -f "$include_file" ]]; then  
    echo "include 文件不存在: $include_file"  
    exit 1  
  fi  
  
  
  for source in $sources; do 	
	if [ "$source" == "exclude.txt" ]; then  
		continue  
	fi  
	if [ "$source" == "include.txt" ]; then  
		continue  
	fi  
	if [[ $source =~ rsync_.*.sh ]]; then 
		continue  
	fi   
	
	
    # echo "当前同步资源文件: $script_dir/$source"  
	# 判断是否为文件  
	if [ -f "$script_dir/$source" ]; then  
		echo "rsync -a --exclude-from=$exclude_file "$script_dir/$source" $username@$server:$destination$source"
		
	#	rsync -a --exclude-from="$exclude_file" \  
	#	"$script_dir/$source" \  
	#	"$username@$server:$destination$source"
	fi  
	  
	# 判断是否为目录  
	if [ -d "$script_dir/$source" ]; then  
		echo "rsync -a --exclude-from=$exclude_file "$script_dir/$source/" $username@$server:$destination$source/"
		
	#	rsync -a --exclude-from="$exclude_file" \  
	#	--include-from="$include_file" \  
	#	"$script_dir/$source/" \  
	#	"$username@$server:$destination$source/"
	fi
	
  done 
}



#################################################################
#################################################################

show(){
	echo -e "\n"
	echo "----------------------------------"
	echo "please enter your choise:"
	echo "(1) 异步迁移指定文件|目录"
	echo "(2) 异步迁移所有文件"
	echo "(3) 移动脚本"
	echo "(6) 退出菜单"
	echo "----------------------------------"
}

menu(){
	input=0
	while [ $input -ne 6 ]
	do
	show
	read input
	case $input in
		1) rsync_files;;  #异步迁移指定
		2) rsync_all_files;;  #异步迁移所有文件
		3) move_script;;  #移动脚本
		6) exit;;
	esac
	done
}

menu