Add native save file dialog

Author: timon-schelling

desktop/src/app.rs

@@ -1,6 +1,7 @@
 use crate::CustomEvent;
 use crate::WindowSize;
 use crate::dialogs::dialog_open_graphite_file;
+use crate::dialogs::dialog_save_graphite_file;
 use crate::render::GraphicsState;
 use crate::render::WgpuContext;
 use graph_craft::wasm_application_io::WasmApplicationIo;
@@ -80,6 +81,31 @@ impl WinitApp {
 			});
 		}
 
+		for message in responses.extract_if(.., |m| matches!(m, FrontendMessage::TriggerSaveDocument { .. })) {
+			let FrontendMessage::TriggerSaveDocument { document_id, name, path, document } = message else {
+				unreachable!()
+			};
+			if let Some(path) = path {
+				let _ = std::fs::write(&path, document);
+			} else {
+				let event_loop_proxy = self.event_loop_proxy.clone();
+				let _ = thread::spawn(move || {
+					let path = futures::executor::block_on(dialog_save_graphite_file(name));
+					if let Some(path) = path {
+						if let Err(e) = std::fs::write(&path, document) {
+							tracing::error!("Failed to save file: {}: {}", path.display(), e);
+						} else {
+							let message = Message::Portfolio(PortfolioMessage::DocumentPassMessage {
+								document_id,
+								message: DocumentMessage::SavedDocument { path: Some(path) },
+							});
+							let _ = event_loop_proxy.send_event(CustomEvent::DispatchMessage(message));
+						}
+					}
+				});
+			}
+		}
+
 		if responses.is_empty() {
 			return;
 		}

desktop/src/dialogs.rs

@@ -3,6 +3,20 @@ use std::path::PathBuf;
 use rfd::AsyncFileDialog;
 
 pub(crate) async fn dialog_open_graphite_file() -> Option<PathBuf> {
-	let file = AsyncFileDialog::new().add_filter("Graphite", &["graphite"]).set_title("Open Graphite Document").pick_file().await;
-	file.map(|f| f.path().to_path_buf())
+	AsyncFileDialog::new()
+		.add_filter("Graphite", &["graphite"])
+		.set_title("Open Graphite Document")
+		.pick_file()
+		.await
+		.map(|f| f.path().to_path_buf())
+}
+
+pub(crate) async fn dialog_save_graphite_file(name: String) -> Option<PathBuf> {
+	AsyncFileDialog::new()
+		.add_filter("Graphite", &["graphite"])
+		.set_title("Save Graphite Document")
+		.set_file_name(name)
+		.save_file()
+		.await
+		.map(|f| f.path().to_path_buf())
 }

editor/src/messages/frontend/frontend_message.rs

@@ -12,6 +12,7 @@ use graph_craft::document::NodeId;
 use graphene_std::raster::Image;
 use graphene_std::raster::color::Color;
 use graphene_std::text::{Font, TextAlign};
+use std::path::PathBuf;
 
 #[cfg(not(target_family = "wasm"))]
 use crate::messages::portfolio::document::overlays::utility_types::OverlayContext;
@@ -63,6 +64,12 @@ pub enum FrontendMessage {
 		#[serde(rename = "commitDate")]
 		commit_date: String,
 	},
+	TriggerSaveDocument {
+		document_id: DocumentId,
+		name: String,
+		path: Option<PathBuf>,
+		document: String,
+	},
 	TriggerDownloadImage {
 		svg: String,
 		name: String,

editor/src/messages/portfolio/document/document_message.rs

@@ -1,3 +1,5 @@
+use std::path::PathBuf;
+
 use super::utility_types::misc::{GroupFolderType, SnappingState};
 use crate::messages::input_mapper::utility_types::input_keyboard::Key;
 use crate::messages::portfolio::document::overlays::utility_types::OverlayContext;
@@ -105,6 +107,9 @@ pub enum DocumentMessage {
 	RenderRulers,
 	RenderScrollbars,
 	SaveDocument,
+	SavedDocument {
+		path: Option<PathBuf>,
+	},
 	SelectParentLayer,
 	SelectAllLayers,
 	SelectedLayersLower,

editor/src/messages/portfolio/document/document_message_handler.rs

@@ -37,6 +37,7 @@ use graphene_std::table::Table;
 use graphene_std::vector::PointId;
 use graphene_std::vector::click_target::{ClickTarget, ClickTargetType};
 use graphene_std::vector::style::ViewMode;
+use std::path::PathBuf;
 use std::time::Duration;
 
 #[derive(ExtractField)]
@@ -115,6 +116,9 @@ pub struct DocumentMessageHandler {
 	/// Stack of document network snapshots for future history states.
 	#[serde(skip)]
 	document_redo_history: VecDeque<NodeNetworkInterface>,
+	/// The path of the to the document file.
+	#[serde(skip)]
+	path: Option<PathBuf>,
 	/// Hash of the document snapshot that was most recently saved to disk by the user.
 	#[serde(skip)]
 	saved_hash: Option<u64>,
@@ -162,6 +166,7 @@ impl Default for DocumentMessageHandler {
 			selection_network_path: Vec::new(),
 			document_undo_history: VecDeque::new(),
 			document_redo_history: VecDeque::new(),
+			path: None,
 			saved_hash: None,
 			auto_saved_hash: None,
 			layer_range_selection_reference: None,
@@ -996,11 +1001,16 @@ impl MessageHandler<DocumentMessage, DocumentMessageContext<'_>> for DocumentMes
 					true => self.name.clone(),
 					false => self.name.clone() + FILE_SAVE_SUFFIX,
 				};
-				responses.add(FrontendMessage::TriggerDownloadTextFile {
-					document: self.serialize_document(),
+				responses.add(FrontendMessage::TriggerSaveDocument {
+					document_id,
 					name,
+					path: self.path.clone(),
+					document: self.serialize_document(),
 				})
 			}
+			DocumentMessage::SavedDocument { path } => {
+				self.path = path;
+			}
 			DocumentMessage::SelectParentLayer => {
 				let selected_nodes = self.network_interface.selected_nodes();
 				let selected_layers = selected_nodes.selected_layers(self.metadata());

frontend/src/messages.ts

@@ -791,6 +791,16 @@ export class TriggerImport extends JsMessage {}
 
 export class TriggerPaste extends JsMessage {}
 
+export class TriggerSaveDocument extends JsMessage {
+	readonly documentId!: bigint;
+
+	readonly name!: string;
+
+	readonly path!: string | undefined;
+
+	readonly document!: string;
+}
+
 export class TriggerDownloadImage extends JsMessage {
 	readonly svg!: string;
 
@@ -1647,6 +1657,7 @@ export const messageMakers: Record<string, MessageMaker> = {
 	DisplayRemoveEditableTextbox,
 	SendUIMetadata,
 	TriggerAboutGraphiteLocalizedCommitDate,
+	TriggerSaveDocument,
 	TriggerDownloadImage,
 	TriggerDownloadTextFile,
 	TriggerFetchAndOpenDocument,

frontend/src/state-providers/portfolio.ts

@@ -6,6 +6,7 @@ import { type Editor } from "@graphite/editor";
 import {
 	type FrontendDocumentDetails,
 	TriggerFetchAndOpenDocument,
+	TriggerSaveDocument,
 	TriggerDownloadImage,
 	TriggerDownloadTextFile,
 	TriggerImport,
@@ -84,6 +85,9 @@ export function createPortfolioState(editor: Editor) {
 		const imageData = await extractPixelData(new Blob([data.content.data], { type: data.type }));
 		editor.handle.pasteImage(data.filename, new Uint8Array(imageData.data), imageData.width, imageData.height);
 	});
+	editor.subscriptions.subscribeJsMessage(TriggerSaveDocument, (triggerSaveDocument) => {
+		downloadFileText(triggerSaveDocument.name, triggerSaveDocument.document);
+	});
 	editor.subscriptions.subscribeJsMessage(TriggerDownloadTextFile, (triggerFileDownload) => {
 		downloadFileText(triggerFileDownload.name, triggerFileDownload.document);
 	});