Change snapshot path

build.rs

@@ -1,19 +1,19 @@
-use std::env;
 use std::fs::File;
 use std::path::PathBuf;
 
-const RUNTIME_SNAPSHOT_PATH: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/target/RUNTIME_SNAPSHOT.bin");
+const RUNTIME_SNAPSHOT_PATH: Option<&str> = option_env!("RUNTIME_SNAPSHOT_PATH");
+const DEFAULT_SNAPSHOT_PATH: &str = "/tmp/openworkers-runtime-snapshot.bin";
 
 fn main () {
     println!("cargo:rerun-if-changed=build.rs");
     println!("cargo:rerun-if-changed=src/ext/*");
     println!("cargo:rerun-if-changed=src/runtime.rs");
     println!("cargo:rerun-if-changed=src/extensions.rs");
 
-    let path = PathBuf::from(RUNTIME_SNAPSHOT_PATH);
+    let path = PathBuf::from(RUNTIME_SNAPSHOT_PATH.unwrap_or(DEFAULT_SNAPSHOT_PATH));
 
     // Create the file if it doesn't exist
     if !path.exists() {
         File::create(&path).unwrap();
     }
-}
+}

src/snapshot.rs

@@ -9,13 +9,14 @@ use std::env;
 use std::path::PathBuf;
 use std::fs::File;
 
-const RUNTIME_SNAPSHOT_PATH: &str = concat!(env!("CARGO_MANIFEST_DIR"), "/target/RUNTIME_SNAPSHOT.bin");
+const RUNTIME_SNAPSHOT_PATH: Option<&str> = option_env!("RUNTIME_SNAPSHOT_PATH");
+const DEFAULT_SNAPSHOT_PATH: &str = "/tmp/openworkers-runtime-snapshot.bin";
 
 pub fn create_runtime_snapshot() {
     println!("Building snapshot");
 
     // Build the file path to the snapshot.
-    let snapshot_path = PathBuf::from(RUNTIME_SNAPSHOT_PATH);
+    let snapshot_path = PathBuf::from(RUNTIME_SNAPSHOT_PATH.unwrap_or(DEFAULT_SNAPSHOT_PATH));
 
     let serializer: SnapshotFileSerializer = SnapshotFileSerializer::new(File::create(snapshot_path).unwrap());