one shift for args in main case

winehelper

@@ -23,14 +23,30 @@ check_variables () { [[ -z ${!1} ]] && export $1="$2" ;}
 
 ##### CHECK DEBUG #####
 if [[ "$1" == "--debug" ]] ; then
-    export DEBUG="full"
+    check_variables WINEDEBUG "+loaddll,+seh"
-    export LOG_FILE="$HOME/winehelper.log"
+    export DXVK_LOG_LEVEL="error"
+    export VKD3D_DEBUG "error"
+    export WINE_MONO_TRACE "E:System.NotImplementedException"
+    export VK_LOADER_DEBUG "error"
+    export VKBASALT_LOG_LEVEL "error"
+    export DXVK_NVAPI_LOG_LEVEL "error"
     shift
-    print_warning "Включен режим подробного логирования работы WINE."
+else
+    check_variables WINEDEBUG "-all"
+    check_variables DXVK_LOG_LEVEL "none"
+    check_variables VKD3D_SHADER_DEBUG "none"
+    check_variables VKD3D_DEBUG "none"
+    check_variables DXVK_LOG_PATH "none"
+    check_variables VKBASALT_LOG_LEVEL "none"
+    check_variables DXVK_NVAPI_LOG_LEVEL "none"
+fi
+
+if [[ $WINEDEBUG != "-all" ]] ; then
+    export LOG_FILE="$HOME/winehelper.log"
+    date > "$LOG_FILE"
+    print_warning "Включен режим логирования работы WINE."
     print_warning "Лог будет сохранен по пути: $LOG_FILE"
     sleep 3
-else
-    check_variables DEBUG "0"
 fi
 
 ##### WINETRICKS VERSION #####
@@ -345,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\""
@@ -369,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"
@@ -384,14 +405,19 @@ create_desktop () {
 
     cp -f "$USER_WORK_PATH/$desktop_filename.desktop" "$WH_MENU_DIR/"
 
-    # добавляем информацию о приложении в "$WINEPREFIX/desktop.list"
+    if [[ "$RESTORE_FROM_BACKUP" == "1" ]] ; then
-    if [[ -f "$WINEPREFIX/desktop.list" ]] \
+        print_info "Пропускаем обновление desktop.list (режим восстановления из бэкапа)"
-    && grep -qe "^${name_desktop}=" "$WINEPREFIX/desktop.list"
+    else
-    then sed -i "/^$name_desktop=/d" "$WINEPREFIX/desktop.list"
+        # добавляем информацию о приложении в "$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"
@@ -679,27 +705,6 @@ init_wine_ver () {
         export WINESERVER="wineserver"
     fi
 
-    check_variables DXVK_LOG_LEVEL "none"
-    check_variables VKD3D_SHADER_DEBUG "none"
-    check_variables VKD3D_DEBUG "none"
-    check_variables DXVK_LOG_PATH "none"
-    check_variables VKBASALT_LOG_LEVEL "none"
-    check_variables DXVK_NVAPI_LOG_LEVEL "none"
-
-    if [[ $DEBUG == "0" ]] ; then
-        check_variables WINEDEBUG "-all"
-    elif [[ $DEBUG == "full" ]] ; then
-        check_variables WINEDEBUG "+loaddll,+seh"
-        export DXVK_LOG_LEVEL="error"
-        export VKD3D_DEBUG "error"
-        export WINE_MONO_TRACE "E:System.NotImplementedException"
-        export VK_LOADER_DEBUG "error"
-        export VKBASALT_LOG_LEVEL "error"
-        export DXVK_NVAPI_LOG_LEVEL "error"
-    else
-        check_variables WINEDEBUG "err+all"
-    fi
-
     print_info "Используется версия wine: $WH_WINE_USE"
 }
 
@@ -842,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")"
@@ -850,14 +855,12 @@ check_prefix_var () {
             fi
         done
 
-        if [[ ${#prefixes[@]} -eq 0 ]]; then
+        [[ ${#prefixes[@]} -eq 0 ]] && fatal "Не найдено ни одного префикса!"
-            fatal "Не найдено ни одного префикса!"
-        fi
 
         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 "Неверный выбор."
@@ -869,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
 }
@@ -1139,13 +1143,15 @@ wine_run () {
     WIN_FILE_PATH="$(dirname "$WIN_FILE_EXEC")"
     [[ -d "$WIN_FILE_PATH" ]] && cd "$WIN_FILE_PATH"
 
-    if [[ $DEBUG == "0" ]] ; then
+    if [[ -n $LOG_FILE ]] && [[ -f "$LOG_FILE" ]] ; then
-        $MANGOHUD_RUN "$WINELOADER" "$@" $LAUNCH_PARAMETERS
+        echo "##### Основные переменные #####" | tee -a "$LOG_FILE"
-    else
+        env | grep -e "WH_" -e "WINE" -e "DXVK" -e "VKD3D" | tee -a "$LOG_FILE"
-        print_var WH_WINE_USE WINEARCH WINEPREFIX WINEDLLOVERRIDES WH_WINDOWS_VER | tee "$LOG_FILE"
-        $MANGOHUD_RUN "$WINELOADER" "$@" $LAUNCH_PARAMETERS 2>&1 | tee -a "$LOG_FILE"
-    fi
 
+        echo "##### Лог WINE #####" | tee -a "$LOG_FILE"
+        $MANGOHUD_RUN "$WINELOADER" "$@" $LAUNCH_PARAMETERS 2>&1 | tee -a "$LOG_FILE"
+    else
+        $MANGOHUD_RUN "$WINELOADER" "$@" $LAUNCH_PARAMETERS
+    fi
     wait_wineserver
 }
 
@@ -1316,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 list                    список возможных установочных скриптов
-    install [скрипт]              запустить скрипт установки программы
+    install [скрипт]                запустить скрипт установки программы
-    install [скрипт] --clear-pfx  не использовать подготовленный префикс для установки ПО
+    install [скрипт] --clear-pfx    не использовать готовый префикс для установки ПО
 
-    installed                     список установленных программ
+    installed                       список установленных программ
-    run [программа.exe]           запуск программы (отладка)
+    run [программа.exe]             запуск программы (отладка)
-    remove-all                    удалить WineHelper и все связанные данные
+    remove-all                      удалить WineHelper и все связанные данные
-    remove-prefix [имя_префикса]  удалить префикс и все связанные данные
+    remove-prefix [имя_префикса]    удалить префикс и все связанные данные
+    backup-prefix [имя_префикса]    создать резерную копию префикса
+    restore-prefix \"путь/до/whpack\" восстановить префикс из резервной копии
 
 Параметры:
-    --help                        показать эту справку и выйти
+    --help                          показать эту справку и выйти
-    --version                     показать информацию о пакете и его версии
+    --version                       показать информацию о пакете и его версии
-    --debug [команда]             включить режим логирования работы WINE
+    --debug [команда]               включить режим логирования работы WINE
 
 "
 }
@@ -1348,7 +1497,14 @@ if [[ -d "$HOME/.local/share/$SCRIPT_NAME" ]] \
 then try_force_link_dir "$HOME/.local/share/$SCRIPT_NAME" "$HOME/.winehelper"
 fi
 
-case "${1,,}" in
+if [[ -n "$1" ]] ; then
+    arg1="$1"
+    shift
+else
+    arg1="--help"
+fi
+
+case "$arg1" in
     --version) rpm -qi "$SCRIPT_NAME" ; exit 0 ;;
     --help) wh_info ; exit 0 ;;
     killall) kill_wine ;;
@@ -1356,18 +1512,19 @@ case "${1,,}" in
     winereg|regedit) prepair_wine ; wine_run "regedit" ;;
     winefile|explorer) prepair_wine ; wine_run "winefile" ;;
     wineconsole|cmd) prepair_wine ; wine_run "wineconsole" ;;
-    winetricks) shift ; prepair_wine ; "$WINETRICKS" -q "$@" ;;
+    winetricks) prepair_wine ; "$WINETRICKS" -q "$@" ;;
-    desktop) shift ; create_desktop "$@" ; exit 0 ;;
+    desktop) create_desktop "$@" ; exit 0 ;;
-    install|-i) shift ; run_autoinstall "$@" ;;
+    install|-i) run_autoinstall "$@" ;;
-    installed) shift ; check_installed_programs "$1" ;;
+    installed) check_installed_programs "$1" ;;
-    run|-r) shift ; run_installed_programs "$1" ;;
+    run|-r) run_installed_programs "$1" ;;
-    remove-all) shift ; remove_winehelper "$@" ;;
+    backup-prefix) backup_prefix "$@" ;;
-    remove-prefix) shift ; remove_prefix "$@" ;;
+    restore-prefix) restore_prefix "$@" ;;
+    remove-all) remove_winehelper "$@" ;;
+    remove-prefix) remove_prefix "$@" ;;
     *)
-        if [[ -f "$1" ]] ; then
+        if [[ -f "$arg1" ]] ; then
-            WIN_FILE_EXEC="$(readlink -f "$1")"
+            WIN_FILE_EXEC="$(readlink -f "$arg1")"
-            WIN_FILE_NAME="$(basename "$WIN_FILE_EXEC")"
+            WIN_FILE_NAME="$(basename "$arg1")"
-            shift
             case "${WIN_FILE_NAME,,}" in
                 *.exe) prepair_wine ; wine_run $WINE_WIN_START "$WIN_FILE_EXEC" "$@" ;;
                 *.msi) prepair_wine ; wine_run msiexec /i "$WIN_FILE_EXEC" "$@" ;;
@@ -1375,8 +1532,9 @@ case "${1,,}" in
                 *) fatal "Тип файла не поддерживается." ;;
             esac
         else
+            print_error "Команды $arg1 не существует."
             wh_info
             exit 1
         fi
-        ;;
+    ;;
 esac