♻️ rs 代码重构

src-tauri/src/main.rs

@@ -1,142 +1,58 @@
 //! @file src/main.rs
 //! @desc 主模块，用于启动应用
-//! @since Beta v0.4.2
+//! @since Beta v0.4.3
 
 // Prevents additional console window on Windows in release, DO NOT REMOVE!!
 #![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]
 
-use log::LevelFilter;
-use tauri::{Manager, WindowBuilder};
-use tauri_plugin_log::{LogTarget, TimezoneStrategy};
-use tauri_utils::config::WindowConfig;
+use tauri::Manager;
 mod client;
+mod commands;
+mod plugins;
 mod utils;
 
-// 放一个常数，用来判断应用是否初始化
-static mut APP_INITIALIZED: bool = false;
-static mut DEEP_LINK_REGISTERED: bool = false;
-
-#[tauri::command]
-async fn init_app(app_handle: tauri::AppHandle) {
-  unsafe {
-    if APP_INITIALIZED == true && DEEP_LINK_REGISTERED == true {
-      return;
+// 窗口事件处理
+fn window_event_handler(event: tauri::GlobalWindowEvent) -> () {
+  match event.event() {
+    tauri::WindowEvent::CloseRequested { api, .. } => {
+      api.prevent_close();
+      let window = event.window().clone();
+      if window.label() == "TeyvatGuide" {
+        // 子窗口 label 的数组
+        const SUB_WINDOW_LABELS: [&str; 3] = ["Sub_window", "Dev_JSON", "mhy_client"];
+        for label in SUB_WINDOW_LABELS.iter() {
+          let sub = window.get_window(label);
+          if sub.is_some() {
+            sub.unwrap().close().unwrap();
+          }
+        }
+      }
+      window.close().unwrap();
     }
+    _ => {}
   }
-  dbg!("init_app");
-  let _mhy = app_handle.get_window("mhy_client");
-  if _mhy.is_some() {
-    std::thread::sleep(std::time::Duration::from_millis(1000));
-    _mhy.unwrap().close().unwrap();
-  }
-  app_handle.emit_all("initApp", ()).unwrap();
-  unsafe {
-    APP_INITIALIZED = true;
-  }
-}
-
-#[tauri::command]
-async fn register_deep_link(app_handle: tauri::AppHandle) {
-  unsafe {
-    if DEEP_LINK_REGISTERED == true {
-      return;
-    }
-  }
-  dbg!("register_deep_link");
-  tauri_plugin_deep_link::register("teyvatguide", move |request| {
-    dbg!(&request);
-    app_handle.emit_all("active_deep_link", request).unwrap();
-  })
-  .unwrap();
-  unsafe {
-    DEEP_LINK_REGISTERED = true;
-  }
-}
-
-// 执行 js
-#[tauri::command]
-async fn execute_js(app_handle: tauri::AppHandle, label: String, js: String) {
-  let window = app_handle.get_window(&label);
-  if window.is_none() {
-    return;
-  }
-  window.unwrap().eval(&js).ok().unwrap();
-}
-
-// 创建窗口
-#[tauri::command]
-async fn create_window(app_handle: tauri::AppHandle, label: String, mut option: WindowConfig) {
-  let window_old = app_handle.get_window(&label);
-  option.label = label.clone();
-  if window_old.is_some() {
-    dbg!("window exists");
-    window_old.unwrap().close().unwrap();
-    return;
-  }
-  let window_new =
-    Some(WindowBuilder::from_config(&app_handle, option).build().expect("failed to create window"));
-  window_new.unwrap();
-}
-
-// 读取目录大小
-#[tauri::command]
-async fn get_dir_size(path: String) -> u64 {
-  let walk_dir = walkdir::WalkDir::new(path);
-  let mut size = 0;
-  for entry in walk_dir {
-    let entry = entry.unwrap();
-    let file_type = entry.file_type();
-    if file_type.is_file() {
-      size += entry.metadata().unwrap().len();
-    }
-  }
-  size
 }
 
 fn main() {
   tauri_plugin_deep_link::prepare("teyvatguide");
   tauri::Builder::default()
-    .on_window_event(|event| {
-      match event.event() {
-        tauri::WindowEvent::CloseRequested { api, .. } => {
-          api.prevent_close();
-          let window = event.window().clone();
-          if window.label() == "TeyvatGuide" {
-            // 子窗口 label 的数组
-            const SUB_WINDOW_LABELS: [&str; 3] = ["Sub_window", "Dev_JSON", "mhy_client"];
-            for label in SUB_WINDOW_LABELS.iter() {
-              let sub = window.get_window(label);
-              if sub.is_some() {
-                sub.unwrap().close().unwrap();
-              }
-            }
-          }
-          window.close().unwrap();
-        }
-        _ => {}
-      }
-    })
-    .plugin(tauri_plugin_sql::Builder::default().build())
-    .plugin(
-      tauri_plugin_log::Builder::default()
-        .targets([LogTarget::LogDir, LogTarget::Stdout])
-        .timezone_strategy(TimezoneStrategy::UseLocal)
-        .level(LevelFilter::Info)
-        .log_name(utils::get_current_date())
-        .build(),
-    )
+    .on_window_event(|event| window_event_handler(event))
+    .plugin(plugins::build_sql_plugin())
+    .plugin(plugins::build_log_plugin())
     .invoke_handler(tauri::generate_handler![
-      register_deep_link,
-      init_app,
-      execute_js,
-      create_window,
-      get_dir_size,
+      commands::init_app,
+      commands::register_deep_link,
+      commands::create_window,
+      commands::execute_js,
+      commands::get_dir_size,
       client::create_mhy_client,
     ])
     .setup(|_app| {
-      let _window = _app.get_window("TeyvatGuide").unwrap();
-      #[cfg(debug_assertions)] // only include this code on debug builds
-      _window.open_devtools(); // open the devtools on startup
+      #[cfg(debug_assertions)]
+      let _window = _app.get_window("TeyvatGuide");
+      if _window.is_some() {
+        _window.unwrap().open_devtools();
+      }
       Ok(())
     })
     .run(tauri::generate_context!())