1 files changed, 14 insertions, 1 deletions

diff --git a/src/lisp/prelude.rs b/src/lisp/prelude.rs
index 7cb46ff..8088510 100644
--- a/src/lisp/prelude.rs
+++ b/src/lisp/prelude.rs
@@ -14,7 +14,7 @@ use crate::{
     utils::{load_script, rect_coords},
 };
 
-use std::convert::TryInto;
+use std::{convert::TryInto, fs::File, io::BufWriter, path::Path};
 
 #[macro_export]
 macro_rules! primitive {
@@ -260,6 +260,18 @@ pub fn new_env() -> Result<Environment, LispError> {
         }
     });
 
+    primitive!(env, Arity::Exact(1), "export-png", |args, app| {
+        if let LispExpr::StringLit(s) = &args[0] {
+            let export_path = Path::new(&s);
+            let file = File::create(export_path).map_err(EvalError::FileError)?;
+            let w = BufWriter::new(file);
+            app.export().write_png(w);
+            Ok(LispExpr::Unit)
+        } else {
+            Err(EvalError::TypeMismatch.into())
+        }
+    });
+
     primitive!(env, Arity::Exact(0), "grid-enabled?", |_, app| {
         Ok(LispExpr::BoolLit(app.grid.enabled))
     });
@@ -524,5 +536,6 @@ pub fn new_env() -> Result<Environment, LispError> {
     primitive!(env, Arity::Exact(1), "id", |args, _| {
         Ok(args[0].clone())
     });
+
     Ok(env)
 }