#!/bin/ksh
# Purpose                      : Transfer files from Local to Remote/Remote to 
#Name of the file that is executed
SCRIPT_NAME=`basename ${0} .ksh`    
#Lock file/Temporary file to ensure that the files are not run multiple times                                         
LCK_FILE="/tmp/${SCRIPT_NAME}_${2}.run"  
#File used for build of SFTP tasks                                    
TMP_SFTP_SCRPT="/tmp/sftp_script_${2}_$$.ftp"  
#Denotes the current directory                              
DIR_NAME=`dirname ${0}`    
#File that defines the source and target specifications                                                  
FILE_CONFIG="${DIR_NAME}/${1}.config"                                        
FOUND=false
TIMESTAMP=`date '+%d-%b-%Y_%R'`
FILES_PATH="/tmp/Files2Transfer_${2}"

#
# Copy files from local system to remote system
#
Local2Remote() {

	cd ${FILES_PATH} 2>/dev/null
	if [[ ${?} -gt 0 ]];then
          print -u2 "LocalTarget path not found";
          rm ${LCK_FILE}
          exit -1;
        fi

	echo cd ${FILE_TARGET_PATH} > ${TMP_SFTP_SCRPT}
	echo mput ${FILE_MASK} >> ${TMP_SFTP_SCRPT}
	echo bye >> ${TMP_SFTP_SCRPT}

	sftp -oPort=${FILE_TARGET_PORT} -b ${TMP_SFTP_SCRPT} ${FILE_TARGET_ACCOUNT} 2>/dev/null 1>&2

	rm ${TMP_SFTP_SCRPT}

}

#
# Copy files from remote system to local system
#
Remote2Local() {

	cd ${FILES_PATH} 2>/dev/null
	if [[ ${?} -gt 0 ]];then
          print -u2 "LocalSource path not found";
          rm ${LCK_FILE}
          exit 1;
        fi

	echo cd ${FILE_SOURCE_PATH} > ${TMP_SFTP_SCRPT}
	echo mget ${FILE_MASK} >> ${TMP_SFTP_SCRPT}
	
# The below line needs to be modified to enable archive functionality. 	
  echo rm ${FILE_MASK} >> ${TMP_SFTP_SCRPT}    

# Added below line to copy and remove the files to the desired folder 
# echo mv ${file_mask} >> ${tmp_sftp_scrpt}                                  	
		          
# End of change 
	echo bye >> ${TMP_SFTP_SCRPT}

	sftp -oPort=${FILE_SOURCE_PORT} -b ${TMP_SFTP_SCRPT} ${FILE_SOURCE_ACCOUNT} 2>/dev/null 1>&2

	rm ${TMP_SFTP_SCRPT}
			
}

#
# Main function
#

if [ ! -f ${FILE_CONFIG} ]; then
	print -u2 "Configuration file not found"
	rm ${LCK_FILE}
        exit 1
fi

if [ -f ${LCK_FILE} ]; then
	print -u2 "Script with parameter ${CL_NAME} already running, exiting..."
	exit 1
else
	touch ${LCK_FILE}
fi

exec < ${FILE_CONFIG}
	while IFS=\| read -r name source_account source_port source_path target_account target_port target_path mask; do
	case ${name} in
		\#* | "")
			continue
			;;
	esac
		
			eval FILE_SOURCE_ACCOUNT=${source_account}
                        eval FILE_SOURCE_PORT=${source_port}
			eval FILE_SOURCE_PATH=${source_path}
			eval FILE_TARGET_ACCOUNT=${target_account}
                        eval FILE_TARGET_PORT=${target_port}
			eval FILE_TARGET_PATH=${target_path}
			eval FILE_MASK=${mask} 
			FOUND=true
			break

done

if [ ${FOUND} = false ]; then
	print -u2 "File to transfer not configured"
    rm ${LCK_FILE}
	exit 2
fi

mkdir ${FILES_PATH}

set -f
Remote2Local
Local2Remote
set +f

rm -Rf ${FILES_PATH}

rm ${LCK_FILE}

exit 0

CL_NAME=${1}
SCRIPT_NAME=`basename ${0} .ksh`                                   # Name of the shell script which is executed
LCK_FILE="/tmp/${SCRIPT_NAME}_${2}.run"                            # Lock file/Temporary file to ensure that the files are not run multiple times
TMP_SFTP_SCRPT="/tmp/sftp_script_${2}_$$.ftp"                      # File used for build of SFTP tasks
DIR_NAME=`dirname ${0}`                                            # Directory from which shell script is executed
FILE_CONFIG="${DIR_NAME}/${1}.config"                              # File that defines the source and target specifications
FOUND=false
TIMESTAMP=`date '+%d-%b-%Y_%R'`
FILES_PATH="/tmp/Files2Transfer_${2}"  
ARCHIVE_PATH="/tmp/iBuyFilesTransfered/archive"

#
# Copy local files to remote file system
#
Local2Remote() {

	cd ${FILES_PATH} 2>/dev/null 
	if [[ ${?} -gt 0 ]];then 
          print -u2 "LocalTarget path not found";
          rm ${LCK_FILE}
          exit -1;
        fi

	echo cd ${FILE_TARGET_PATH} > ${TMP_SFTP_SCRPT} 
	echo mput ${FILE_MASK} >> ${TMP_SFTP_SCRPT}  
	echo bye >> ${TMP_SFTP_SCRPT}

	sftp -oPort=${FILE_TARGET_PORT} -b ${TMP_SFTP_SCRPT} ${FILE_TARGET_ACCOUNT} 2>/dev/null 1>&2 

	rm ${TMP_SFTP_SCRPT}

}


#
# Copy remote files to local file system
#
Remote2Local() {

	cd ${FILES_PATH} 2>/dev/null
	if [[ ${?} -gt 0 ]];then
          print -u2 "LocalSource path not found";
          rm ${LCK_FILE}
          exit 1;
        fi

	echo cd ${FILE_SOURCE_PATH} > ${TMP_SFTP_SCRPT}
	echo mget ${FILE_MASK} >> ${TMP_SFTP_SCRPT}
	# Changes added to archive the files
	# echo rm ${FILE_MASK} >> ${TMP_SFTP_SCRPT} Commented this line as the below zip command moves the files into archive and deletes the files from the system
	echo zip -m archive.zip ${FILE_MASK} >> ${TMP_SFTP_SCRPT} 
    # End of change
	echo bye >> ${TMP_SFTP_SCRPT}

	sftp -oPort=${FILE_SOURCE_PORT} -b ${TMP_SFTP_SCRPT} ${FILE_SOURCE_ACCOUNT} 2>/dev/null 1>&2 

	rm ${TMP_SFTP_SCRPT}
			
}

#
# Main function
#

if [ ! -f ${FILE_CONFIG} ]; then
	print -u2 "Configuration file not found"
	rm ${LCK_FILE}
        exit 1
fi

if [ -f ${LCK_FILE} ]; then
	print -u2 "Script with parameter ${CL_NAME} already running, exiting..."
	exit 1
else
	touch ${LCK_FILE}
fi

exec < ${FILE_CONFIG}
	while IFS=\| read -r name source_account source_port source_path target_account target_port target_path mask; do
	case ${name} in
		\#* | "")
			continue
			;;
	esac
		
			eval FILE_SOURCE_ACCOUNT=${source_account}
                        eval FILE_SOURCE_PORT=${source_port}
			eval FILE_SOURCE_PATH=${source_path}
			eval FILE_TARGET_ACCOUNT=${target_account}
                        eval FILE_TARGET_PORT=${target_port}
			eval FILE_TARGET_PATH=${target_path}
			eval FILE_MASK=${mask} 
			FOUND=true
			break

done

if [ ${FOUND} = false ]; then
	print -u2 "File to transfer not configured"
    rm ${LCK_FILE}
	exit 2
fi

mkdir ${FILES_PATH}

set -f
Remote2Local
Local2Remote
set +f

rm -Rf ${FILES_PATH}

rm ${LCK_FILE}

exit 0