#!/bin/bash # Author: [王思揚] # Description: [Used to replace multiple files in the program.] start_time=$(date +%s) # Directory for storing new files, ...
#!/bin/bash
# Author: [王思揚]
# Description: [Used to replace multiple files in the program.]
start_time=$(date +%s)
# Directory for storing new files, backup files and program directory
NewFileDir=/home/test/new/
BacKFileDir=/home/test/bak/
ProDir=/home/test/program
# Set colors for text
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m' # No Color
# Backup function
backup() {
# Create backup directory with current date
mkdir -p $BacKFileDir$(date +%F) &>/dev/null
echo -e "${GREEN}Backup directory created successfully.${NC}"
# Copy each new file to backup directory
for i in $(ls $NewFileDir); do
cp -f $(find $ProDir -name $i) $BacKFileDir$(date +%F) &>/dev/null
echo -e "${GREEN}File $i backed up successfully.${NC}"
done
}
# Update function
update() {
files=($(ls $NewFileDir))
for i in "${files[@]}"; do
paths=($(find $ProDir -name $i))
for path in "${paths[@]}"; do
cp -f $NewFileDir$i $path
if [ $? -eq 0 ]; then
echo -e "${GREEN}File $i updated successfully.${NC}"
fi
done
done
}
# Rollback function
rollback() {
# Find backup directory with current date
backupDir=$BacKFileDir$(date +%F)
if [ -d $backupDir ]; then
for i in $(ls $backupDir); do
find $ProDir -name $i | while read path; do
cp -f $backupDir/$i $path
if [ $? -eq 0 ]; then
echo -e "${GREEN}File $i rolled back successfully.${NC}"
fi
done
done
fi
}
# Main function
while :; do
echo -e "${YELLOW}Please select an operation:${NC}"
echo -e "${YELLOW} 1. Backup${NC}"
echo -e "${YELLOW} 2. Update${NC}"
echo -e "${YELLOW} 3. Rollback${NC}"
echo -e "${YELLOW} 4. Quit${NC}"
read -r -p "$(echo -e ${YELLOW}Enter your choice:${NC} )" choice
case $choice in
1)
backup
;;
2)
update
;;
3)
rollback
;;
4)
echo -e "${YELLOW}Goodbye!${NC}"
exit 0
;;
*)
echo -e "${RED}Invalid choice, please try again.${NC}"
;;
esac
done
end_time=$(date +%s)
echo "Script completed in $(($end_time - $start_time)) seconds."
