Add file assosiation for Tauri

src-tauri/src/main.rs

@@ -420,6 +420,47 @@ async fn streamed_fetch(
 }
 
 
+use std::path::PathBuf;
+// Copyright 2019-2024 Tauri Programme within The Commons Conservancy
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-License-Identifier: MIT
+fn handle_file_associations(app: AppHandle, files: Vec<PathBuf>) {
+    // -- Scope handling start --
+  
+    // You can remove this block if you only want to know about the paths, but not actually "use" them in the frontend.
+  
+    // This requires the `fs` tauri plugin and is required to make the plugin's frontend work:
+    // use tauri_plugin_fs::FsExt;
+    // let fs_scope = app.fs_scope();
+  
+    // This is for the `asset:` protocol to work:
+    let asset_protocol_scope = app.asset_protocol_scope();
+  
+    for file in &files {
+      // This requires the `fs` plugin:
+      // let _ = fs_scope.allow_file(file);
+  
+      // This is for the `asset:` protocol:
+      let _ = asset_protocol_scope.allow_file(file);
+    }
+  
+    // -- Scope handling end --
+  
+    let files = files
+      .into_iter()
+      .map(|f| {
+        let file = f.to_string_lossy().replace("\\", "\\\\"); // escape backslash
+        format!("\"{file}\"",) // wrap in quotes for JS array
+      })
+      .collect::<Vec<_>>()
+      .join(",");
+  
+    tauri::WebviewWindowBuilder::new(&app, "main", Default::default())
+      .initialization_script(&format!("window.tauriOpenedFiles = [{files}]"))
+      .build()
+      .unwrap();
+  }
+
 fn main() {
     tauri::Builder::default()
         .plugin(tauri_plugin_http::init())
@@ -429,6 +470,35 @@ fn main() {
         .plugin(tauri_plugin_dialog::init())
         .plugin(tauri_plugin_os::init())
         .plugin(tauri_plugin_fs::init())
+        .setup(|#[allow(unused_variables)] app| {
+            #[cfg(any(windows, target_os = "linux"))]
+            {
+                let mut files = Vec::new();
+        
+                // NOTICE: `args` may include URL protocol (`your-app-protocol://`)
+                // or arguments (`--`) if your app supports them.
+                // files may aslo be passed as `file://path/to/file`
+                for maybe_file in std::env::args().skip(1) {
+                // skip flags like -f or --flag
+                if maybe_file.starts_with("-") {
+                    continue;
+                }
+        
+                // handle `file://` path urls and skip other urls
+                if let Ok(url) = url::Url::parse(&maybe_file) {
+                    if let Ok(path) = url.to_file_path() {
+                    files.push(path);
+                    }
+                } else {
+                    files.push(PathBuf::from(maybe_file))
+                }
+                }
+        
+                handle_file_associations(app.handle().clone(), files);
+            }
+        
+            Ok(())
+        })
         .invoke_handler(tauri::generate_handler![
             greet,
             native_request,
@@ -441,8 +511,26 @@ fn main() {
             install_py_dependencies,
             streamed_fetch,
         ])
-        .run(tauri::generate_context!())
-        .expect("error while running tauri application");
+        .build(tauri::generate_context!())
+        .expect("error while running tauri application")
+        .run(
+            // Copyright 2019-2024 Tauri Programme within The Commons Conservancy
+            // SPDX-License-Identifier: Apache-2.0
+            // SPDX-License-Identifier: MIT
+            #[allow(unused_variables)]
+            |app, event| {
+              #[cfg(any(target_os = "macos", target_os = "ios"))]
+              if let tauri::RunEvent::Opened { urls } = event {
+                let files = urls
+                  .into_iter()
+                  .filter_map(|url| url.to_file_path().ok())
+                  .collect::<Vec<_>>();
+      
+                handle_file_associations(app.clone(), files);
+              }
+            },
+        );
+        
 }
 
 fn header_map_to_json(header_map: &HeaderMap) -> serde_json::Value {

src-tauri/tauri.conf.json

@@ -18,7 +18,14 @@
       "src-python/*"
     ],
     "targets": "all",
-    "createUpdaterArtifacts": "v1Compatible"
+    "createUpdaterArtifacts": "v1Compatible",
+    "fileAssociations": [
+      {
+        "ext": [
+          "risum", "risup", "charx"
+        ]
+      }
+    ]
   },
   "productName": "RisuAI",
   "mainBinaryName": "RisuAI",