当前位置：首页 > article >正文

Rust练手项目，写个有趣的小工具定时从一言网获取一段有趣的话并推送通知

article 2024/11/16 3:13:20

Rust练手项目，写个有趣的小工具

- 代码

继续练习Rust, 写个小工具定时从一言网获取一段有趣的话并提示，如下

练习以下Rust点

并发编程 Mutex, Arc指针使用
HTTP请求
Windows Gui

代码

Cargo.toml

[package]
name = "funny_word"
edition = "2021"
version.workspace = true
authors.workspace = true
description.workspace = true

[dependencies]
trayicon = "0.2.0"
winapi = { version = "0.3.9", features = [
    "winuser",
    "windef",
    "minwindef",
    "shellapi",
    "libloaderapi",
    "commctrl",
    "basetsd",
] }
win-toast-notify = "0.1.6"
windows = { version = "0.58.0", features = [
    "Win32_Security",
    "Win32_System_Threading",
    "Win32_UI_WindowsAndMessaging",
] }
reqwest = { version = "0.11", features = ["blocking"]  }
[package]
name = "funny_word"
edition = "2021"
version.workspace = true
authors.workspace = true
description.workspace = true

[dependencies]
trayicon = "0.2.0"
winapi = { version = "0.3.9", features = [
    "winuser",
    "windef",
    "minwindef",
    "shellapi",
    "libloaderapi",
    "commctrl",
    "basetsd",
] }
win-toast-notify = "0.1.6"
windows = { version = "0.58.0", features = [
    "Win32_Security",
    "Win32_System_Threading",
    "Win32_UI_WindowsAndMessaging",
] }
reqwest = { version = "0.11", features = ["blocking"]  }

main.rs

#![windows_subsystem = "windows"]
use core::mem::MaybeUninit;
use std::{
    sync::{Arc, Mutex},
    thread::sleep,
    time::Duration,
};
use trayicon::*;
use win_toast_notify::WinToastNotify;
use winapi::um::winuser;

fn main() {
    #[derive(Copy, Clone, Eq, PartialEq, Debug)]
    enum Events {
        RightClickTrayIcon,
        LeftClickTrayIcon,
        OneMinute,
        FiveMinute,
        FifteenMinute,
        HalfHour,
        OneHour,
        Exit,
    }

    let (sender, receiver) = std::sync::mpsc::channel::<Events>();
    let icon_bytes = include_bytes!("rust.ico");

    let mut tray_icon = TrayIconBuilder::new()
        .sender(move |e: &Events| {
            let _ = sender.send(*e);
        })
        .icon(Icon::from_buffer(icon_bytes, None, None).unwrap())
        .tooltip("💪没事喝点鸡汤吧！")
        .on_right_click(Events::RightClickTrayIcon)
        .on_click(Events::LeftClickTrayIcon)
        .menu(
            MenuBuilder::new()
                .separator()
                .submenu(
                    "提醒间隔",
                    MenuBuilder::new()
                        .checkable("1分钟", true, Events::OneMinute)
                        .checkable("5分钟", false, Events::FiveMinute)
                        .checkable("15分钟", false, Events::FifteenMinute)
                        .checkable("半小时", false, Events::HalfHour)
                        .checkable("一小时", false, Events::OneHour),
                )
                .separator()
                .item("退出", Events::Exit),
        )
        .build()
        .unwrap();

    let _interval = Arc::new(Mutex::new(1));
    let interval = _interval.clone();
    std::thread::spawn(move || {
        receiver.iter().for_each(|m| match m {
            Events::RightClickTrayIcon => {
                tray_icon.show_menu().unwrap();
            }
            Events::LeftClickTrayIcon => {
                tray_icon.show_menu().unwrap();
            }
            Events::Exit => {
                std::process::exit(0);
            }
            Events::OneMinute => {
                let mut i = interval.lock().unwrap();
                *i = 1;
                tray_icon
                    .set_menu_item_checkable(Events::OneMinute, true)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FiveMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FifteenMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::HalfHour, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::OneHour, false)
                    .unwrap();
            }
            Events::FiveMinute => {
                let mut i = interval.lock().unwrap();
                *i = 5;
                tray_icon
                    .set_menu_item_checkable(Events::OneMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FiveMinute, true)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FifteenMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::HalfHour, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::OneHour, false)
                    .unwrap();
            }
            Events::FifteenMinute => {
                let mut i = interval.lock().unwrap();
                *i = 15;
                tray_icon
                    .set_menu_item_checkable(Events::OneMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FiveMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FifteenMinute, true)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::HalfHour, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::OneHour, false)
                    .unwrap();
            }
            Events::HalfHour => {
                let mut i = interval.lock().unwrap();
                *i = 30;
                tray_icon
                    .set_menu_item_checkable(Events::OneMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FiveMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::FifteenMinute, false)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::HalfHour, true)
                    .unwrap();
                tray_icon
                    .set_menu_item_checkable(Events::OneHour, false)
                    .unwrap();
            }
            Events::OneHour => {
                let mut i = interval.lock().unwrap();
                *i = 60;
                tray_icon.set_menu_item_checkable(Events::OneMinute, false);
                tray_icon.set_menu_item_checkable(Events::FiveMinute, false);
                tray_icon.set_menu_item_checkable(Events::FifteenMinute, false);
                tray_icon.set_menu_item_checkable(Events::HalfHour, false);
                tray_icon.set_menu_item_checkable(Events::OneHour, true);
            }
        })
    });

    std::thread::spawn(move || loop {
        match reqwest::blocking::get("https://v1.hitokoto.cn?&encode=text") {
            Ok(resp) => match resp.text() {
                Ok(text) => {
                    WinToastNotify::new()
                        .set_title("来自一言网")
                        .set_messages(vec![&text])
                        .show()
                        .expect("发送通知失败!");
                }
                Err(e) => {
                    WinToastNotify::new()
                        .set_title("错误")
                        .set_messages(vec![&format!(
                            "HTTP错误 {}",
                            e.status().unwrap_or_default()
                        )])
                        .show()
                        .expect("发送通知失败!");
                }
            },
            Err(e) => {
                WinToastNotify::new()
                    .set_title("获取网站内容失败错误")
                    .set_messages(vec![&format!(
                        "HTTP错误 {}",
                        e.status().unwrap_or_default()
                    )])
                    .show()
                    .expect("发送通知失败!");
            }
        }

        let mut _b = 1;
        {
            // 加个大括号是为了尽快释放lock, 免得睡眠中(60 * (*i))一直占用锁，点击右键菜单无反应
            let interval = _interval.clone();
            let i = interval.lock().unwrap();
            _b = *i;
        }
        sleep(Duration::from_secs(60 * _b));
    });

    loop {
        unsafe {
            let mut msg = MaybeUninit::uninit();
            let bret = winuser::GetMessageA(msg.as_mut_ptr(), 0 as _, 0, 0);
            if bret > 0 {
                winuser::TranslateMessage(msg.as_ptr());
                winuser::DispatchMessageA(msg.as_ptr());
            } else {
                break;
            }
        }
    }
}