z2m-copy-data.sh 3.3 KB

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