1 files changed, 86 insertions, 0 deletions
diff --git a/src/isolate_init.rs b/src/isolate_init.rs
new file mode 100644
index 000000000..c77f0c998
--- /dev/null
+++ b/src/isolate_init.rs
@@ -0,0 +1,86 @@
+// Copyright 2018-2019 the Deno authors. All rights reserved. MIT license.
+use crate::libdeno::deno_buf;
+
+pub struct IsolateInitScript {
+  pub source: String,
+  pub filename: String,
+}
+
+pub struct IsolateInit {
+  pub snapshot: Option<deno_buf>,
+  pub init_script: Option<IsolateInitScript>,
+}
+
+pub fn deno_isolate_init() -> IsolateInit {
+  if cfg!(not(feature = "check-only")) {
+    if cfg!(feature = "use-snapshot-init") {
+      let data =
+        include_bytes!(concat!(env!("GN_OUT_DIR"), "/gen/snapshot_deno.bin"));
+
+      unsafe {
+        IsolateInit {
+          snapshot: Some(deno_buf::from_raw_parts(data.as_ptr(), data.len())),
+          init_script: None,
+        }
+      }
+    } else {
+      #[cfg(not(feature = "check-only"))]
+      let source_bytes =
+        include_bytes!(concat!(env!("GN_OUT_DIR"), "/gen/bundle/main.js"));
+
+      #[cfg(feature = "check-only")]
+      let source_bytes = vec![];
+
+      IsolateInit {
+        snapshot: None,
+        init_script: Some(IsolateInitScript {
+          filename: "gen/bundle/main.js".to_string(),
+          source: std::str::from_utf8(source_bytes).unwrap().to_string(),
+        }),
+      }
+    }
+  } else {
+    IsolateInit {
+      snapshot: None,
+      init_script: None,
+    }
+  }
+}
+
+pub fn compiler_isolate_init() -> IsolateInit {
+  if cfg!(not(feature = "check-only")) {
+    if cfg!(feature = "use-snapshot-init") {
+      let data = include_bytes!(concat!(
+        env!("GN_OUT_DIR"),
+        "/gen/snapshot_compiler.bin"
+      ));
+
+      unsafe {
+        IsolateInit {
+          snapshot: Some(deno_buf::from_raw_parts(data.as_ptr(), data.len())),
+          init_script: None,
+        }
+      }
+    } else {
+      #[cfg(not(feature = "check-only"))]
+      let source_bytes =
+        include_bytes!(concat!(env!("GN_OUT_DIR"), "/gen/bundle/compiler.js"));
+
+      #[cfg(feature = "check-only")]
+      let source_bytes = vec![];
+
+      IsolateInit {
+        snapshot: None,
+        init_script: Some(IsolateInitScript {
+          filename: "gen/bundle/compiler.js".to_string(),
+          source: std::str::from_utf8(source_bytes).unwrap().to_string(),
+        }),
+      }
+    }
+  } else {
+    IsolateInit {
+      snapshot: None,
+      init_script: None,
+    }
+  }
+}