tauri-plugins-workspace/plugins/updater/tests/app-updater/src/main.rs

// Copyright 2019-2023 Tauri Programme within The Commons Conservancy
// SPDX-License-Identifier: Apache-2.0
// SPDX-License-Identifier: MIT

#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]

use tauri_plugin_updater::UpdaterExt;

fn main() {
    #[allow(unused_mut)]
    let mut context = tauri::generate_context!();

    let mut updater = tauri_plugin_updater::Builder::new();
    if std::env::var("TARGET").unwrap_or_default() == "nsis" {
        // /D sets the default installation directory ($INSTDIR),
        // overriding InstallDir and InstallDirRegKey.
        // It must be the last parameter used in the command line and must not contain any quotes, even if the path contains spaces.
        // Only absolute paths are supported.
        // NOTE: we only need this because this is an integration test and we don't want to install the app in the programs folder
        updater = updater.installer_args(vec![format!(
            "/D={}",
            tauri::utils::platform::current_exe()
                .unwrap()
                .parent()
                .unwrap()
                .display()
        )]);
    }

    tauri::Builder::default()
        .plugin(updater.build())
        .setup(|app| {
            let handle = app.handle();
            tauri::async_runtime::spawn(async move {
                match handle.updater().check().await {
                    Ok(update) => {
                        if update.is_update_available() {
                            if let Err(e) = update.download_and_install(|_event| {}).await {
                                println!("{e}");
                                std::process::exit(1);
                            }
                        }
                        std::process::exit(0);
                    }
                    Err(e) => {
                        println!("{e}");
                        std::process::exit(1);
                    }
                }
            });
            Ok(())
        })
        .run(context)
        .expect("error while running tauri application");
}