feat(cmds): 添加toast命令

Author: develon2015

Cargo.lock

@@ -23,3 +23,7 @@ path = "./cmds/limit"
 
 [dependencies.core]
 path = "./cmds/remote-bind/core"
+
+[build-dependencies]
+cc = "1.2.50"
+macro-log = "0.3.0"

Cargo.toml

@@ -0,0 +1,16 @@
+use macro_log::read_dir;
+
+fn debug(obj: &impl std::fmt::Debug, tag: &str) {
+    println!("cargo:warning=[Debug] {}: {tag} -> {obj:?}", file!());
+}
+
+fn main() {
+    let mut cc = cc::Build::new();
+    read_dir!("src/ffi").iter().for_each(|(file_name, _)| {
+        if file_name.ends_with(".c") || file_name.ends_with(".cpp") {
+            cc.file(format!("src/ffi/{file_name}"));
+            debug(file_name, "compile file");
+        }
+    });
+    cc.compile("ffi");
+}

build.rs

@@ -15,5 +15,6 @@ pub enum Commands {
     RBS(crate::mods::rbs::Param),
     RBC(crate::mods::rbc::Param),
     Oneport(crate::mods::oneport::Param),
+    Toast(crate::mods::toast::Param),
     Install(crate::mods::install::Param),
 }

src/args.rs

@@ -0,0 +1,3 @@
+file(GLOB_RECURSE files RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" *.c)
+message(STATUS "构建Win32Helper: ${files}")
+add_library(Win32Helper STATIC ${files})

src/ffi/Win32Helper/CMakeLists.txt

@@ -0,0 +1,9 @@
+#include "include.h"
+#include <stdio.h>
+
+void alloc_console() {
+    // 分配控制台，并使用特殊文件名"CONOUT$"重新打开输入流stdout: FILE *
+	AllocConsole();
+	FILE* file = NULL;
+	freopen_s(&file, "CONOUT$", "w+", stdout);
+}

src/ffi/Win32Helper/alloc_console.c

@@ -0,0 +1,27 @@
+#include "include.h"
+
+void balloon_notification(HWND hwnd, UINT id, const wchar_t *message) {
+    NOTIFYICONDATAW nid = {0};
+    nid.uID = id;
+    nid.cbSize = sizeof nid;
+    nid.hWnd = hwnd; // 必须要设置hWnd
+
+    nid.uFlags |= NIF_ICON;
+    nid.hIcon = LoadIconA(NULL, IDI_SHIELD);
+
+    // nid.uFlags |= NIF_MESSAGE;
+    // nid.uCallbackMessage = WM_USER + 2;
+
+    nid.uFlags |= NIF_INFO;
+    nid.dwInfoFlags = NIIF_ICON_MASK; // 必须要设置hIcon
+    lstrcpyW(nid.szInfoTitle, L"通知");
+    lstrcpyW(nid.szInfo, message);
+    SYSCALL(Shell_NotifyIconW(NIM_ADD, &nid));
+
+    Sleep(80); // 延迟隐藏以显示进程名，似乎会导致无法接收消息
+    // 隐藏图标
+    nid.uFlags |= NIF_STATE;
+    nid.dwState = NIS_HIDDEN;
+    nid.dwStateMask = NIS_HIDDEN;
+    SYSCALL(Shell_NotifyIconW(NIM_MODIFY, &nid));
+}

src/ffi/Win32Helper/balloon_notification.c

@@ -0,0 +1,51 @@
+#ifndef __WIN32_HELPER__
+#define __WIN32_HELPER__
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#include <Windows.h>
+
+// 检查表达式是否为真
+#define CHECK(call) ;\
+    if (!(call)) {\
+        SetLastError(-1);\
+        Perror(#call);\
+    }\
+;
+// 检查Win32系统调用
+#define SYSCALL(call) ;\
+    SetLastError(0);\
+    call;\
+    Perror(#call);\
+;
+// 检查Win32系统调用，同时将返回值赋值给某变量
+#define SYSCALL2(call, recv) ;\
+    SetLastError(0);\
+    recv = call;\
+    Perror(#call);\
+;
+#define Perror(x) _Perror(x, __FILE__, __LINE__)
+
+void win32_init();
+void init_locale(const char *locale);
+long _Perror(char *op, const char *file, int line);
+
+// ANSI转义序列
+#define GREEN "\x1B[32m"
+#define RED "\x1B[91m"
+#define RESET "\x1B[0m"
+void enable_virtual_terminal_processing();
+
+void alloc_console();
+
+void balloon_notification(HWND hwnd, UINT id, const wchar_t *message);
+
+wchar_t *utf8to16(const char *str);
+ 
+char *utf16to8(const wchar_t *str);
+
+#ifdef __cplusplus
+}
+#endif
+#endif // __WIN32_HELPER__

src/ffi/Win32Helper/include.h

@@ -0,0 +1,14 @@
+#include "include.h"
+
+void win32_init() {
+#ifndef NDEBUG
+    // 执行字符集是UTF-8
+    init_locale("zh_CN.UTF-8");
+    alloc_console();
+#else
+    // 执行字符集是GBK
+    init_locale("zh_CN.BGK");
+    // alloc_console();
+#endif
+    enable_virtual_terminal_processing();
+}

src/ffi/Win32Helper/init.c

@@ -0,0 +1,5 @@
+#include <locale.h>
+
+void init_locale(const char *locale) {
+    setlocale(LC_ALL, locale);
+}