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(())
}