updated backup and recovery pfx

winehelper

@@ -361,7 +361,13 @@ create_desktop () {
     local name_desktop="$1"
     local exe_file="$2"
     local desktop_filename="$(basename "$exe_file" .exe | sed "s| |_|")"
-    local icon_file="$WH_IMAGE_PATH/$3.png"
+    local icon_file
+    if [[ "$RESTORE_FROM_BACKUP" == "1" ]] && [[ -f "$3" ]]
+    then icon_file="$3"
+    elif [[ -f "$WH_IMAGE_PATH/$3.png" ]]
+    then icon_file="$WH_IMAGE_PATH/$3.png"
+    else icon_file="wine"
+    fi
 
     if [[ -z "$name_desktop" ]] || [[ -z "$exe_file" ]] ; then
         fatal "Used: $SCRIPT_NAME --desktop \"desktop_name\" \"path_to_exe\" \"name_png_from_image\""
@@ -385,7 +391,6 @@ create_desktop () {
 
     # создаем .desktop файл
     create_new_dir "$WH_MENU_DIR"
-    [[ ! -f "$icon_file" ]] && icon_file=wine
     {
         echo "[Desktop Entry]"
         echo "Name=$name_desktop"
@@ -400,14 +405,19 @@ create_desktop () {
 
     cp -f "$USER_WORK_PATH/$desktop_filename.desktop" "$WH_MENU_DIR/"
 
-    # добавляем информацию о приложении в "$WINEPREFIX/desktop.list"
-    if [[ -f "$WINEPREFIX/desktop.list" ]] \
-    && grep -qe "^${name_desktop}=" "$WINEPREFIX/desktop.list"
-    then sed -i "/^$name_desktop=/d" "$WINEPREFIX/desktop.list"
+    if [[ "$RESTORE_FROM_BACKUP" == "1" ]] ; then
+        print_info "Пропускаем обновление desktop.list (режим восстановления из бэкапа)"
+    else
+        # добавляем информацию о приложении в "$WINEPREFIX/desktop.list"
+        if [[ -f "$WINEPREFIX/desktop.list" ]] \
+        && grep -qe "^${name_desktop}=" "$WINEPREFIX/desktop.list"
+        then sed -i "/^$name_desktop=/d" "$WINEPREFIX/desktop.list"
+        fi
+        create_new_dir "$WINEPREFIX/icons"
+        try_copy_file "$icon_file" "$WINEPREFIX/icons/"
+        echo "$name_desktop=${exe_file//$WINEPREFIX/}=$(basename "$icon_file")" >> "$WINEPREFIX/desktop.list"
     fi
-    create_new_dir "$WINEPREFIX/icons"
-    try_copy_file "$icon_file" "$WINEPREFIX/icons/"
-    echo "$name_desktop=${exe_file//$WINEPREFIX/}=$(basename "$icon_file")" >> "$WINEPREFIX/desktop.list"
+
 
     # создаем файл категории для меню
     create_new_dir "$HOME/.local/share/desktop-directories"
@@ -837,7 +847,7 @@ check_prefix_var () {
         local prefixes=()
         local count=1
         print_info "Доступные префиксы WineHelper:"
-        for prefix in "$WH_PREFIXES_DIR"/*; do
+        for prefix in "$WH_PREFIXES_DIR"/* ; do
             if [[ -d "$prefix" ]]; then
                 prefixes+=("$prefix")
                 echo "$count. $(basename "$prefix")"
@@ -845,14 +855,12 @@ check_prefix_var () {
             fi
         done
 
-        if [[ ${#prefixes[@]} -eq 0 ]]; then
-            fatal "Не найдено ни одного префикса!"
-        fi
+        [[ ${#prefixes[@]} -eq 0 ]] && fatal "Не найдено ни одного префикса!"
 
         local max_choice=${#prefixes[@]}
         read -p "Выберите префикс (1-$max_choice): " choice
 
-        if [[ "$choice" -ge 1 && "$choice" -le "$max_choice" ]]; then
+        if [[ "$choice" -ge 1 && "$choice" -le "$max_choice" ]] ; then
             export WINEPREFIX="${prefixes[$choice-1]}"
         else
             fatal "Неверный выбор."
@@ -864,6 +872,7 @@ check_prefix_var () {
     if echo "$WINEPREFIX" | grep -qv '/' ; then
         export WINEPREFIX="$WH_PREFIXES_DIR/$WINEPREFIX"
     fi
+    export PREFIX_NAME="$(basename "$WINEPREFIX")"
 
     return 0
 }
@@ -1313,23 +1322,166 @@ remove_winehelper () {
     fi
 }
 
+backup_prefix() {
+    export WINEPREFIX="$1"
+    check_prefix_var
+
+    local backup_base_dir="$(xdg-user-dir DESKTOP)"
+    local backup_archive_name="backup_${PREFIX_NAME}_$(date +%Y.%m.%d).whpack"
+    local backup_dest_path="$backup_base_dir/$backup_archive_name"
+    local temp_backup_dir="$WH_TMP_DIR/backup_${PREFIX_NAME}_$(date +%Y.%m.%d)"
+
+    print_info "Начало резервного копирования префикса: $PREFIX_NAME"
+    create_new_dir "$temp_backup_dir"
+    create_new_dir "$temp_backup_dir/prefixes/"
+    create_new_dir "$temp_backup_dir/dist/"
+
+    print_info "Подготовка префикса к упаковке..."
+    if cp -a "$WINEPREFIX" "$temp_backup_dir/prefixes/$PREFIX_NAME" ; then
+        print_ok "Директория префикса подготовлена."
+    else
+        print_error "Не удалось подготовить директорию префикса."
+        try_remove_dir "$temp_backup_dir"
+        return 1
+    fi
+
+    if [[ -f "$WINEPREFIX/last.conf" ]]; then
+        source "$WINEPREFIX/last.conf"
+
+        if [[ -n "$WH_WINE_USE" ]] \
+        && [[ $WH_WINE_USE != system* ]] \
+        && [[ -d "$WH_DIST_DIR/$WH_WINE_USE" ]]
+        then
+            print_info "Копирование используемой версии WINE: $WH_WINE_USE"
+
+            if cp -a "$WH_DIST_DIR/$WH_WINE_USE" "$temp_backup_dir/dist/"
+            then print_ok "WINE скопирован."
+            else fatal "Не удалось скопировать WINE."
+            fi
+        fi
+    fi
+
+    print_info "Создание squashfs архива..."
+    if mksquashfs "$temp_backup_dir" "$backup_dest_path" -comp zstd ; then
+        print_ok "Архив префикса $PREFIX_NAME успешно создан по пути:\n$backup_dest_path"
+    else
+        print_error "Не удалось создать архив префикса $PREFIX_NAME."
+        try_remove_dir "$temp_backup_dir"
+        return 1
+    fi
+
+    try_remove_dir "$temp_backup_dir"
+    print_info "Создание резервной копии префикса завершено."
+    return 0
+}
+
+restore_prefix() {
+    local backup_archive_path="$1"
+    local temp_extract_dir
+
+    if [[ -z "$backup_archive_path" ]] ; then
+        read -e -p "Укажите путь к архиву резервной копии (/путь/к/архиву.whpack): " backup_archive_path
+        if [[ -z "$backup_archive_path" ]] ; then
+            fatal "Путь к архиву не указан. Восстановление отменено."
+        fi
+    fi
+
+    backup_archive_path=$(echo "$backup_archive_path" | sed "s/'//g; s/\"//g")
+    if [[ ! -f "$backup_archive_path" ]]
+    then fatal "Файл архива не найден: $backup_archive_path"
+    fi
+
+    print_info "Восстановление из резервной копии: $backup_archive_path"
+
+    temp_extract_dir="$WH_TMP_DIR/restore_$(basename "$backup_archive_path" .whpack)"
+    create_new_dir "$temp_extract_dir"
+
+    print_info "Распаковка архива..."
+    if unsquashfs -f -d "$temp_extract_dir" "$backup_archive_path" ; then
+        print_ok "Архив успешно распакован."
+    else
+        try_remove_dir "$temp_extract_dir"
+        fatal "Не удалось распаковать архив."
+    fi
+
+    if [[ -d "$temp_extract_dir/prefixes" ]] ; then
+        local prefix_name prefix_dir
+        prefix_name="$(ls -1 "$temp_extract_dir/prefixes")"
+        prefix_dir="$temp_extract_dir/prefixes/$prefix_name"
+
+        print_info "Восстановление префикса: $prefix_name"
+
+        if [[ -d "$WH_PREFIXES_DIR/$prefix_name" ]] ; then
+            try_remove_dir "$temp_extract_dir"
+            fatal "Префикс $prefix_name уже существует!
+Удалите его если действительно желаете востановить префикс заново.
+Команда для удаления: $SCRIPT_NAME remove-prefix $prefix_name"
+        fi
+
+        if [[ -d "$temp_extract_dir/dist" ]] ; then
+            print_info "Восстановление версий WINE"
+            if cp -fr "$temp_extract_dir/dist"/* "$WH_DIST_DIR/"
+            then print_ok "Версии WINE восстановлены."
+            else print_warning "Не удалось восстановить версии WINE."
+            fi
+            try_remove_dir "$temp_extract_dir/dist"
+        fi
+
+        if ! mv "$prefix_dir" "$WH_PREFIXES_DIR/" ; then
+            try_remove_dir "$temp_extract_dir"
+            fatal "Не удалось восстановить префикс $prefix_name."
+        else
+            print_ok "Префикс $prefix_name восстановлен."
+
+            print_info "Восстановление ярлыков для префикса $prefix_name..."
+            export RESTORE_FROM_BACKUP="1"  # Устанавливаем флаг восстановления
+            while IFS='=' read -r line; do
+                local name_desktop=$(echo "$line" | cut -d'=' -f1)
+                local exe_path=$(echo "$line" | cut -d'=' -f2)
+                local icon_name=$(echo "$line" | cut -d'=' -f3)
+
+                local full_exe_path="$WH_PREFIXES_DIR/$prefix_name${exe_path}"
+                local full_icon_path="$WH_PREFIXES_DIR/$prefix_name/icons/$icon_name"
+
+                if [[ -f "$full_exe_path" ]]; then
+                    print_info "Создание ярлыка для $name_desktop"
+                    create_desktop "$name_desktop" "$full_exe_path" "$full_icon_path"
+                else
+                    print_warning "Исполняемый файл для ярлыка '$name_desktop' не найден: $full_exe_path"
+                fi
+            done < "$WH_PREFIXES_DIR/$prefix_name/desktop.list"
+            unset RESTORE_FROM_BACKUP  # Снимаем флаг после завершения
+            print_ok "Ярлыки для префикса $prefix_name восстановлены."
+        fi
+    else
+        try_remove_dir "$temp_extract_dir"
+        fatal "Директория prefixes не найдена в архиве."
+    fi
+
+    try_remove_dir "$temp_extract_dir"
+    print_ok "Полное восстановление префикса $prefix_name завершено."
+    return 0
+}
+
 wh_info () {
     echo "Использование: $SCRIPT_NAME [команда]
 
 Команды:
-    install list                  список возможных установочных скриптов
-    install [скрипт]              запустить скрипт установки программы
-    install [скрипт] --clear-pfx  не использовать подготовленный префикс для установки ПО
+    install list                   список возможных установочных скриптов
+    install [скрипт]                запустить скрипт установки программы
+    install [скрипт] --clear-pfx    не использовать готовый префикс для установки ПО
 
-    installed                     список установленных программ
-    run [программа.exe]           запуск программы (отладка)
-    remove-all                    удалить WineHelper и все связанные данные
-    remove-prefix [имя_префикса]  удалить префикс и все связанные данные
+    installed                       список установленных программ
+    run [программа.exe]             запуск программы (отладка)
+    remove-all                      удалить WineHelper и все связанные данные
+    remove-prefix [имя_префикса]    удалить префикс и все связанные данные
+    backup-prefix [имя_префикса]    создать резерную копию префикса
+    restore-prefix \"путь/до/whpack\" восстановить префикс из резервной копии
 
 Параметры:
-    --help                        показать эту справку и выйти
-    --version                     показать информацию о пакете и его версии
-    --debug [команда]             включить режим логирования работы WINE
+    --help                          показать эту справку и выйти
+    --version                       показать информацию о пакете и его версии
+    --debug [команда]               включить режим логирования работы WINE
 
 "
 }
@@ -1358,6 +1510,8 @@ case "${1,,}" in
     install|-i) shift ; run_autoinstall "$@" ;;
     installed) shift ; check_installed_programs "$1" ;;
     run|-r) shift ; run_installed_programs "$1" ;;
+    backup-prefix) shift ; backup_prefix "$@" ;;
+    restore-prefix) shift ; restore_prefix "$@" ;;
     remove-all) shift ; remove_winehelper "$@" ;;
     remove-prefix) shift ; remove_prefix "$@" ;;
     *)