mirror of
https://github.com/vmstan/gravity-sync.git
synced 2024-08-30 18:22:11 +00:00
126 lines
4.2 KiB
Bash
Executable File
126 lines
4.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Gravity Sync by vmstan
|
|
VERSION='1.1.1'
|
|
|
|
# Must execute from a location in the home folder of the user who own's it (ex: /home/pi/gravity-sync)
|
|
# Configure certificate based SSH authentication between the Pihole HA nodes - it does not use passwords
|
|
# Tested against Pihole 5.0 GA on Raspbian Buster and Ubuntu 20.04, but it should work on most configs
|
|
# More installation instructions available at https://vmstan.com/gravity-sync
|
|
# For the latest version please visit https://github.com/vmstan/gravity-sync under Releases
|
|
|
|
# You must change the REMOTE_HOST and REMOTE_USER variables below to match your configuration
|
|
|
|
# STANDARD SETTINGS
|
|
REMOTE_HOST='192.168.7.5'
|
|
REMOTE_USER='ubuntu'
|
|
|
|
# PUSH SETTINGS
|
|
|
|
|
|
# CUSTOMIZATION
|
|
LOCAL_FOLDR='gravity-sync' # must exist in running user home folder
|
|
SYNCING_LOG='gravity-sync.log' # will be created in above folder
|
|
|
|
# PIHOLE DEFAULTS
|
|
PIHOLE_DIR='/etc/pihole' # default install directory
|
|
GRAVITY_FI='gravity.db' # this should not change
|
|
|
|
# SCRIPT COLORS
|
|
RED='\033[0;91m'
|
|
GREEN='\033[0;92m'
|
|
CYAN='\033[0;96m'
|
|
YELLOW='033[0;93m'
|
|
PURPLE='033[0;95m'
|
|
NC='\033[0m'
|
|
|
|
##############################################
|
|
|
|
# print title
|
|
# echo -e "${GREEN}Gravity Sync ${VERSION}${NC}"
|
|
|
|
echo -e "${CYAN}Validating sync folder configuration${NC}"
|
|
|
|
# check to see if logging/backup directory is available
|
|
if [ -d ~/${LOCAL_FOLDR} ]
|
|
then
|
|
echo -e "${GREEN}Success${NC}: Required directory ~/${LOCAL_FOLDR} is present"
|
|
else
|
|
echo -e "${RED}Failure${NC}: Required directory ~/${LOCAL_FOLDR} is missing"
|
|
exit
|
|
fi
|
|
|
|
# check to see if current pihole directory is correct
|
|
if [ -f ${PIHOLE_DIR}/${GRAVITY_FI} ]
|
|
then
|
|
echo -e "${GREEN}Success${NC}: Required file ${PIHOLE_DIR}/${GRAVITY_FI} is present"
|
|
else
|
|
echo -e "${RED}Failure${NC}: Required file ${PIHOLE_DIR}/${GRAVITY_FI} is missing"
|
|
exit
|
|
fi
|
|
|
|
case $# in
|
|
0)
|
|
echo -e "${RED}Failure${NC}: ${GRAVITY_FI} replication direction required"
|
|
echo "Usage: $0 {pull|push}"
|
|
echo -e "> ${YELLOW}Pull${NC} will copy the ${GRAVITY_FI} configuration on $REMOTE_HOST to this server"
|
|
echo -e "> ${YELLOW}Push${NC} will force any changes made on this server to the primary"
|
|
exit 1
|
|
;;
|
|
1)
|
|
case $1 in
|
|
pull)
|
|
|
|
echo -e "${CYAN}Copying remote ${GRAVITY_FI} from ${REMOTE_HOST}${NC}"
|
|
rsync -v -e 'ssh -p 22' ${REMOTE_USER}@${REMOTE_HOST}:${PIHOLE_DIR}/${GRAVITY_FI} ~/${LOCAL_FOLDR}/${GRAVITY_FI}
|
|
echo -e "${CYAN}Replacing gravity.db on HA secondary${NC}"
|
|
sudo mv -v ${PIHOLE_DIR}/${GRAVITY_FI} ${PIHOLE_DIR}/${GRAVITY_FI}.backup
|
|
sudo cp -v ~/${LOCAL_FOLDR}/${GRAVITY_FI} ${PIHOLE_DIR}
|
|
sudo chmod 644 ${PIHOLE_DIR}/${GRAVITY_FI}
|
|
sudo chown pihole:pihole ${PIHOLE_DIR}/${GRAVITY_FI}
|
|
echo -e "${CYAN}Reloading configuration of HA secondary FTLDNS from new gravity.db${NC}"
|
|
pihole restartdns reloadlists
|
|
pihole restartdns
|
|
echo -e "${CYAN}Cleaning up things${NC}"
|
|
mv -v ~/${LOCAL_FOLDR}/${GRAVITY_FI} ~/${LOCAL_FOLDR}/${GRAVITY_FI}.last
|
|
date >> ~/${LOCAL_FOLDR}/${SYNCING_LOG}
|
|
echo -e "${GREEN}gravity.db pull completed${NC}"
|
|
;;
|
|
|
|
push)
|
|
echo -e "${RED}WARNING - DATA LOSS POSSIBLE${NC}"
|
|
echo -e "Are you sure you want to overwrite the primary node configuration on ${REMOTE_HOST}?"
|
|
select yn in "Yes" "No"; do
|
|
case $yn in
|
|
Yes )
|
|
echo "Replacing gravity.db on primary"
|
|
echo -e "${CYAN}Copying local ${GRAVITY_FI} to ${REMOTE_HOST}${NC}"
|
|
rsync --rsync-path="sudo rsync" -v -e 'ssh -p 22' ${PIHOLE_DIR}/${GRAVITY_FI} ${REMOTE_USER}@${REMOTE_HOST}:${PIHOLE_DIR}/${GRAVITY_FI}
|
|
echo -e "${CYAN}Applying permissions to remote gravity.db${NC}"
|
|
ssh ${REMOTE_USER}@${REMOTE_HOST} "sudo chmod 644 ${PIHOLE_DIR}/${GRAVITY_FI}"
|
|
ssh ${REMOTE_USER}@${REMOTE_HOST} "sudo chown pihole:pihole ${PIHOLE_DIR}/${GRAVITY_FI}"
|
|
echo -e "${CYAN}Reloading configuration on remote FTLDNS${NC}"
|
|
ssh ${REMOTE_USER}@${REMOTE_HOST} 'pihole restartdns reloadlists'
|
|
ssh ${REMOTE_USER}@${REMOTE_HOST} 'pihole restartdns'
|
|
echo -e "${GREEN}gravity.db push completed${NC}"
|
|
break;;
|
|
No )
|
|
echo "No changes have been made"
|
|
exit;;
|
|
esac
|
|
done
|
|
;;
|
|
|
|
*)
|
|
echo -e "${RED}'$1' is not a valid argument${NC}"
|
|
echo "Usage: $0 {pull|push}"
|
|
exit 2
|
|
;;
|
|
esac
|
|
;;
|
|
*)
|
|
echo -e "${RED}Too many arguments provided ($#)${NC}"
|
|
echo "Usage: $0 {pull|push}"
|
|
exit 3
|
|
;;
|
|
esac |