Rust 使用egui创建一个简单的下载器demo

发布时间 2023-07-10 23:39:06作者: GaN601

仓库连接: https://github.com/GaN601/egui-demo-download-util
这是我第一个rust gui demo, 学习rust有挺长时间了, 但是一直没有落实到实践中, 本着对桌面应用的兴趣, 考察了slint、egui两种框架, 最后还是选择了egui.

这篇博客同时包含我当前的一些理解, 但是自身技术有限, 可能有不少错误的地方. 有意者请在评论区指正.

这个demo的效果就是通过主窗口的按钮, 呼出子窗口的输入框, 点击下载按钮后就可以下载文件, 因为只是demo, 下载功能不详细, 只是用reqwest请求下载了而已.

egui要求我们创建一个自己的结构体来进行状态保存, 因此我们需要以下结构体:

点击查看结构体
#[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(())
}