1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
use crate::Extension;use crate::JsRuntime;use crate::RuntimeOptions;use crate::Snapshot;use std::path::Path;use std::path::PathBuf;
pub type CompressionCb = dyn Fn(&mut Vec<u8>, &[u8]);
pub struct CreateSnapshotOptions {  pub cargo_manifest_dir: &'static str,  pub snapshot_path: PathBuf,  pub startup_snapshot: Option<Snapshot>,  pub extensions: Vec<Extension>,  pub additional_files: Vec<PathBuf>,  pub compression_cb: Option<Box<CompressionCb>>,}
pub fn create_snapshot(create_snapshot_options: CreateSnapshotOptions) {  let mut js_runtime = JsRuntime::new(RuntimeOptions {    will_snapshot: true,    startup_snapshot: create_snapshot_options.startup_snapshot,    extensions: create_snapshot_options.extensions,    ..Default::default()  });
  // TODO(nayeemrmn): https://github.com/rust-lang/cargo/issues/3946 to get the  // workspace root.  let display_root = Path::new(create_snapshot_options.cargo_manifest_dir)    .parent()    .unwrap();  for file in create_snapshot_options.additional_files {    let display_path = file.strip_prefix(display_root).unwrap_or(&file);    let display_path_str = display_path.display().to_string();    js_runtime      .execute_script(        &("deno:".to_string() + &display_path_str.replace('\\', "/")),        &std::fs::read_to_string(&file).unwrap(),      )      .unwrap();  }
  let snapshot = js_runtime.snapshot();  let snapshot_slice: &[u8] = &snapshot;  println!("Snapshot size: {}", snapshot_slice.len());
  let maybe_compressed_snapshot: Box<dyn AsRef<[u8]>> =    if let Some(compression_cb) = create_snapshot_options.compression_cb {      let mut vec = vec![];
      vec.extend_from_slice(        &u32::try_from(snapshot.len())          .expect("snapshot larger than 4gb")          .to_le_bytes(),      );
      (compression_cb)(&mut vec, snapshot_slice);
      println!("Snapshot compressed size: {}", vec.len());
      Box::new(vec)    } else {      Box::new(snapshot_slice)    };
  std::fs::write(    &create_snapshot_options.snapshot_path,    &*maybe_compressed_snapshot,  )  .unwrap();  println!(    "Snapshot written to: {} ",    create_snapshot_options.snapshot_path.display()  );}
pub fn get_js_files(  cargo_manifest_dir: &'static str,  directory: &str,) -> Vec<PathBuf> {  let manifest_dir = Path::new(cargo_manifest_dir);  let mut js_files = std::fs::read_dir(directory)    .unwrap()    .map(|dir_entry| {      let file = dir_entry.unwrap();      manifest_dir.join(file.path())    })    .filter(|path| {      path.extension().unwrap_or_default() == "js"        && !path.ends_with("99_main.js")    })    .collect::<Vec<PathBuf>>();  js_files.sort();  js_files}