file I/O

2 files changed, 88 insertions, 27 deletions

diff --git a/src/main.rs b/src/main.rs
index 0ca84de..25c6062 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -13,6 +13,9 @@ mod node;
 mod preview;
 mod snarl_ext;
 mod types;
+#[cfg(target_arch = "wasm32")]
+mod wasm;
+
 use node::{AnyNode, ConcreteNode};
 use snarl_ext::{Compilable, WithSignatures};
 use types::{Dimension, FloatPrecision, ScalarType, Type};
@@ -199,8 +202,6 @@ impl App {
         });
         // let snarl = Snarl::new();
 
-        info!("about to init");
-
         App {
             snarl,
             viewer: Viewer { dirty: true },
@@ -212,24 +213,57 @@ impl App {
 impl eframe::App for App {
     fn update(&mut self, ctx: &egui::Context, _frame: &mut eframe::Frame) {
         egui::TopBottomPanel::top("top_panel").show(ctx, |ui| {
-            // The top panel is often a good place for a menu bar:
-
             egui::MenuBar::new().ui(ui, |ui| {
-                #[cfg(not(target_arch = "wasm32"))]
-                {
-                    ui.menu_button("File", |ui| {
-                        if ui.button("Quit").clicked() {
-                            ctx.send_viewport_cmd(egui::ViewportCommand::Close);
+                egui::widgets::global_theme_preference_switch(ui);
+
+                ui.menu_button("File", |ui| {
+                    if ui.button("Open").clicked() {
+                        #[cfg(not(target_arch = "wasm32"))]
+                        if let Some(path) = rfd::FileDialog::new()
+                            .add_filter("Nodetoy shader", &["nt.json"])
+                            .pick_file()
+                        {
+                            let file = std::fs::File::open(path).unwrap();
+                            self.snarl =
+                                serde_json::from_reader::<_, Snarl<AnyNode>>(file).unwrap();
+                            self.viewer.dirty = true;
                         }
-                    });
-                    ui.add_space(16.0);
-                }
 
-                egui::widgets::global_theme_preference_switch(ui);
+                        #[cfg(target_arch = "wasm32")]
+                        web_sys::window()
+                            .unwrap()
+                            .alert_with_message(
+                                "To open an .nt.json file, please drag it onto the canvas.",
+                            )
+                            .unwrap();
+                    }
 
-                if ui.button("Clear All").clicked() {
-                    self.snarl = Snarl::default();
-                }
+                    #[cfg(not(target_arch = "wasm32"))]
+                    if ui.button("Save").clicked() {
+                        if let Some(path) = rfd::FileDialog::new()
+                            .add_filter("Nodetoy shader", &["nt.json"])
+                            .save_file()
+                        {
+                            let file = std::fs::File::create(path).unwrap();
+                            serde_json::to_writer(file, &self.snarl).unwrap();
+                        }
+                    }
+
+                    #[cfg(target_arch = "wasm32")]
+                    if ui.button("Download").clicked() {
+                        wasm::download_file(&self.snarl, "shader.nt.json")
+                            .expect("successful download");
+                    }
+
+                    if ui.button("Clear All").clicked() {
+                        self.snarl = Snarl::default();
+                    }
+
+                    #[cfg(not(target_arch = "wasm32"))]
+                    if ui.button("Quit").clicked() {
+                        ctx.send_viewport_cmd(egui::ViewportCommand::Close);
+                    }
+                });
             });
         });
 
@@ -254,6 +288,13 @@ impl eframe::App for App {
                     }
                 }
             }
+
+            ui.input(|i| {
+                if let Some(data) = i.raw.dropped_files.first().and_then(|f| f.bytes.as_ref()) {
+                    self.snarl = serde_json::from_slice(data).unwrap();
+                    self.viewer.dirty = true;
+                }
+            });
         });
 
         egui::SidePanel::right("preview").show(ctx, |ui| {
@@ -283,27 +324,18 @@ fn main() -> eframe::Result<()> {
     };
 
     eframe::run_native(
-        "egui-snarl demo",
+        "nodetoy",
         native_options,
         Box::new(|cx| Ok(Box::new(App::new(cx)))),
     )
 }
 
-#[cfg(target_arch = "wasm32")]
-fn get_canvas_element() -> Option<web_sys::HtmlCanvasElement> {
-    use eframe::wasm_bindgen::JsCast;
-
-    let document = web_sys::window()?.document()?;
-    let canvas = document.get_element_by_id("egui_snarl_demo")?;
-    canvas.dyn_into::<web_sys::HtmlCanvasElement>().ok()
-}
-
 // When compiling to web using trunk:
 #[cfg(target_arch = "wasm32")]
 fn main() {
     eframe::WebLogger::init(log::LevelFilter::Debug).ok();
 
-    let canvas = get_canvas_element().expect("Failed to find canvas with id 'egui_snarl_demo'");
+    let canvas = wasm::get_canvas_element().expect("failed to find canvas");
 
     let web_options = eframe::WebOptions::default();
 
diff --git a/src/wasm.rs b/src/wasm.rs
new file mode 100644
index 0000000..6ce45e0
--- /dev/null
+++ b/src/wasm.rs
@@ -0,0 +1,29 @@
+use web_sys::js_sys::Array;
+use web_sys::wasm_bindgen::{JsCast, JsValue};
+use web_sys::{Blob, BlobPropertyBag, HtmlAnchorElement, HtmlCanvasElement, Url};
+
+pub fn get_canvas_element() -> Option<HtmlCanvasElement> {
+    let document = web_sys::window()?.document()?;
+    let canvas = document.get_element_by_id("nodetoy")?;
+    canvas.dyn_into::<HtmlCanvasElement>().ok()
+}
+
+pub fn download_file<T: serde::Serialize>(value: &T, filename: &str) -> Result<(), JsValue> {
+    let json = serde_json::to_string(value).unwrap();
+    let parts = Array::of1(&JsValue::from_str(&json));
+    let opts = BlobPropertyBag::new();
+    opts.set_type("application/json");
+
+    let blob = Blob::new_with_str_sequence_and_options(&parts, &opts)?;
+    let url = Url::create_object_url_with_blob(&blob)?;
+
+    let document = web_sys::window().unwrap().document().unwrap();
+    let a: HtmlAnchorElement = document.create_element("a")?.dyn_into()?;
+
+    a.set_href(&url);
+    a.set_download(filename);
+    a.click();
+
+    Url::revoke_object_url(&url)?;
+    Ok(())
+}