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

仓库连接: 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(())

}