z2m-copy-data.sh 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. #!/usr/bin/env bash
  2. # Use to copy all data from one Zigbee2MQTT LXC to another
  3. # run from the Proxmox Shell
  4. # bash -c "$(wget -qLO - https://raw.githubusercontent.com/tteck/Proxmox/main/misc/z2m-copy-data.sh)"
  5. while true; do
  6. read -p "Use to copy all data from one Zigbee2MQTT LXC to another. Proceed(y/n)?" yn
  7. case $yn in
  8. [Yy]* ) break;;
  9. [Nn]* ) exit;;
  10. * ) echo "Please answer yes or no.";;
  11. esac
  12. done
  13. set -o errexit
  14. set -o errtrace
  15. set -o nounset
  16. set -o pipefail
  17. shopt -s expand_aliases
  18. alias die='EXIT=$? LINE=$LINENO error_exit'
  19. trap die ERR
  20. trap cleanup EXIT
  21. function error_exit() {
  22. trap - ERR
  23. local DEFAULT='Unknown failure occured.'
  24. local REASON="\e[97m${1:-$DEFAULT}\e[39m"
  25. local FLAG="\e[91m[ERROR] \e[93m$EXIT@$LINE"
  26. msg "$FLAG $REASON"
  27. exit $EXIT
  28. }
  29. function warn() {
  30. local REASON="\e[97m$1\e[39m"
  31. local FLAG="\e[93m[WARNING]\e[39m"
  32. msg "$FLAG $REASON"
  33. }
  34. function info() {
  35. local REASON="$1"
  36. local FLAG="\e[36m[INFO]\e[39m"
  37. msg "$FLAG $REASON"
  38. }
  39. function msg() {
  40. local TEXT="$1"
  41. echo -e "$TEXT"
  42. }
  43. function cleanup() {
  44. [ -d "${CTID_FROM_PATH:-}" ] && pct unmount $CTID_FROM
  45. [ -d "${CTID_TO_PATH:-}" ] && pct unmount $CTID_TO
  46. popd >/dev/null
  47. rm -rf $TEMP_DIR
  48. }
  49. TEMP_DIR=$(mktemp -d)
  50. pushd $TEMP_DIR >/dev/null
  51. TITLE="Zigbee2MQTT LXC Data Copy"
  52. while read -r line; do
  53. TAG=$(echo "$line" | awk '{print $1}')
  54. ITEM=$(echo "$line" | awk '{print substr($0,36)}')
  55. OFFSET=2
  56. if [[ $((${#ITEM} + $OFFSET)) -gt ${MSG_MAX_LENGTH:-} ]]; then
  57. MSG_MAX_LENGTH=$((${#ITEM} + $OFFSET))
  58. fi
  59. CTID_MENU+=( "$TAG" "$ITEM " "OFF" )
  60. done < <(pct list | awk 'NR>1')
  61. while [ -z "${CTID_FROM:+x}" ]; do
  62. CTID_FROM=$(whiptail --title "$TITLE" --radiolist \
  63. "\nWhich Zigbee2MQTT LXC would you like to copy FROM?\n" \
  64. 16 $(($MSG_MAX_LENGTH + 23)) 6 \
  65. "${CTID_MENU[@]}" 3>&1 1>&2 2>&3) || exit
  66. done
  67. while [ -z "${CTID_TO:+x}" ]; do
  68. CTID_TO=$(whiptail --title "$TITLE" --radiolist \
  69. "\nWhich Zigbee2MQTT LXC would you like to copy TO?\n" \
  70. 16 $(($MSG_MAX_LENGTH + 23)) 6 \
  71. "${CTID_MENU[@]}" 3>&1 1>&2 2>&3) || exit
  72. done
  73. for i in ${!CTID_MENU[@]}; do
  74. [ "${CTID_MENU[$i]}" == "$CTID_FROM" ] && \
  75. CTID_FROM_HOSTNAME=$(sed 's/[[:space:]]*$//' <<<${CTID_MENU[$i+1]})
  76. [ "${CTID_MENU[$i]}" == "$CTID_TO" ] && \
  77. CTID_TO_HOSTNAME=$(sed 's/[[:space:]]*$//' <<<${CTID_MENU[$i+1]})
  78. done
  79. whiptail --defaultno --title "$TITLE" --yesno \
  80. "Are you sure you want to copy data between the following LXCs?
  81. $CTID_FROM (${CTID_FROM_HOSTNAME}) -> $CTID_TO (${CTID_TO_HOSTNAME})
  82. Version: 2022.01.23" 13 50 || exit
  83. info "Zigbee2MQTT Data from '$CTID_FROM' to '$CTID_TO'"
  84. if [ $(pct status $CTID_TO | sed 's/.* //') == 'running' ]; then
  85. msg "Stopping '$CTID_TO'..."
  86. pct stop $CTID_TO
  87. fi
  88. msg "Mounting Container Disks..."
  89. DATA_PATH=/opt/zigbee2mqtt/data/
  90. CTID_FROM_PATH=$(pct mount $CTID_FROM | sed -n "s/.*'\(.*\)'/\1/p") || \
  91. die "There was a problem mounting the root disk of LXC '${CTID_FROM}'."
  92. [ -d "${CTID_FROM_PATH}${DATA_PATH}" ] || \
  93. die "Zigbee2igbee2MQTT directories in '$CTID_FROM' not found."
  94. CTID_TO_PATH=$(pct mount $CTID_TO | sed -n "s/.*'\(.*\)'/\1/p") || \
  95. die "There was a problem mounting the root disk of LXC '${CTID_TO}'."
  96. [ -d "${CTID_TO_PATH}${DATA_PATH}" ] || \
  97. die "Zigbee2MQTT directories in '$CTID_TO' not found."
  98. #rm -rf ${CTID_TO_PATH}${DATA_PATH}
  99. #mkdir ${CTID_TO_PATH}${DATA_PATH}
  100. msg "Copying Data Between Containers..."
  101. RSYNC_OPTIONS=(
  102. --archive
  103. --hard-links
  104. --sparse
  105. --xattrs
  106. --no-inc-recursive
  107. --info=progress2
  108. )
  109. msg "<======== Zigbee2MQTT Data ========>"
  110. rsync ${RSYNC_OPTIONS[*]} ${CTID_FROM_PATH}${DATA_PATH} ${CTID_TO_PATH}${DATA_PATH}
  111. echo -en "\e[1A\e[0K\e[1A\e[0K"
  112. info "Successfully Transferred Data."
  113. # Use to copy all data from one Zigbee2MQTT LXC to another
  114. # run from the Proxmox Shell
  115. # bash -c "$(wget -qLO - https://raw.githubusercontent.com/tteck/Proxmox/main/misc/z2m-copy-data.sh)"