Rust 使用egui創建一個簡單的下載器demo

#[derive(Default)]
struct MainWindow {
    window_download_url: DownloadUrl,
}
#[derive(Default, Clone)]
pub struct DownloadUrl {
    pub is_show: bool,
    pub is_start: bool,
    pub url: String,
    pub local_path: String,
}

impl DownloadUrl {
    pub fn show_window(&mut self, ctx: &Context) {
        let _ = Window::new("Download Url")
            .open(&mut self.is_show.clone())
            .show(ctx, |ui| {
		// 這里在為下載視窗添加一些ui元素
                ui.heading("Download Url");
                ui.text_edit_singleline(&mut self.url);
                ui.text_edit_singleline(&mut self.local_path);
		// 這里將用戶輸入的資料保存在MainWindow中, 這樣當我們點擊下載按鈕時就會開始下載檔案
                if ui.button("Select Folder").clicked() {
                    if let Some(path) = rfd::FileDialog::new().pick_folder() {
                        self.local_path = path.display().to_string();
                    }
                }
		// 關閉當前視窗. 因為是即時模式, 因此下一幀這個視窗不會出現. 我們只需要修改布林值即可.
                if ui.button("Download").clicked() {
                    self.is_show = false;
                    self.is_start = true;
                }
            });
    }
}

egui = "0.19.0"
eframe = "0.19.0"
reqwest = "0.11.12"
tokio = { version = "1.21.2" , features=["full"]}
rfd = "0.10.0"

use eframe::{run_native, App, Frame, NativeOptions};
use egui::{CentralPanel, Context};
use std::fs::File;
use std::io::Write;
use std::path::Path;

//首先使用tokio的main方法
#[tokio::main]
async fn main() {
    println!("Hello, world!");
    let option = NativeOptions {
	// 定義視窗大小
        initial_window_size: Some(egui::vec2(640.0, 480.0)),
        ..Default::default()
    };
	// 啟動egui的主視窗, MainWindow就是我們保持狀態的結構體
    run_native(
        "egui download util",
        option,
        Box::new(|_c| Box::<MainWindow>::default()),
    );
}
// 實作App Trait, 因為egui是即時模式, 因此狀態資料只能從self(MainWindow)拿
impl App for MainWindow {
    fn update(&mut self, ctx: &Context, frame: &mut Frame) {
	// 這里是創建了一個面板, 并且面板里有一個下載的按鈕, 當點擊按鈕后, 會展示一個子視窗
        CentralPanel::default().show(ctx, |ui| {
            if ui.button("Download").clicked() {
                self.window_download_url.is_show = true;
            }
        });

        if self.window_download_url.is_show {
            self.window_download_url.show_window(ctx);
        }

	// 在這里開始執行下載檔案的邏輯, 因為所有權問題, 因此我直接clone了這個結構
        let url = &mut self.window_download_url;
        let target = url.clone();

        if !(target.url.is_empty() || target.local_path.is_empty()) && url.is_start {
            url.is_start = false;
            tokio::spawn(async move {
		// 執行下載檔案的邏輯, 失敗的處理感覺沒啥必要, 其實可以考慮出個dialog
                download_file_to_local_path(&target)
                    .await
                    .expect("TODO: panic message");
            });
        }
    }
}

async fn download_file_to_local_path(
    target: &DownloadUrl,
) -> Result<(), Box<dyn std::error::Error>> {
// 獲取檔案夾路徑選擇器的路徑, 因為不打算太精細, 就直接生成了當前時間戳的檔案名, 連檔案后綴都不給.
    let file_path = Path::new(&target.local_path).join(
        SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .unwrap()
            .as_millis()
            .to_string(),
    );

    let mut file = File::create(file_path)?;
    let response = reqwest::get(&target.url).await?;
	// 寫入檔案, 下載檔案邏輯完成
    file.write_all(&response.bytes().await?)?;

    Ok(())
}