123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220
// Copyright 2018-2022 the Deno authors. All rights reserved. MIT license.
use std::env;use std::path::Path;use std::path::PathBuf;
// This is a shim that allows to generate documentation on docs.rs#[cfg(not(feature = "docsrs"))]mod not_docs {  use super::*;  use deno_core::Extension;  use deno_core::JsRuntime;  use deno_core::RuntimeOptions;
  // TODO(bartlomieju): this module contains a lot of duplicated  // logic with `cli/build.rs`, factor out to `deno_core`.  fn create_snapshot(    mut js_runtime: JsRuntime,    snapshot_path: &Path,    files: Vec<PathBuf>,  ) {    // TODO(nayeemrmn): https://github.com/rust-lang/cargo/issues/3946 to get the    // workspace root.    let display_root = Path::new(env!("CARGO_MANIFEST_DIR")).parent().unwrap();    for file in files {      println!("cargo:rerun-if-changed={}", file.display());      let display_path = file.strip_prefix(display_root).unwrap();      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 compressed_snapshot_with_size = {      let mut vec = vec![];
      vec.extend_from_slice(        &u32::try_from(snapshot.len())          .expect("snapshot larger than 4gb")          .to_le_bytes(),      );
      lzzzz::lz4_hc::compress_to_vec(        snapshot_slice,        &mut vec,        lzzzz::lz4_hc::CLEVEL_MAX,      )      .expect("snapshot compression failed");
      vec    };
    println!(      "Snapshot compressed size: {}",      compressed_snapshot_with_size.len()    );
    std::fs::write(&snapshot_path, compressed_snapshot_with_size).unwrap();    println!("Snapshot written to: {} ", snapshot_path.display());  }
  struct Permissions;
  impl deno_fetch::FetchPermissions for Permissions {    fn check_net_url(      &mut self,      _url: &deno_core::url::Url,    ) -> Result<(), deno_core::error::AnyError> {      unreachable!("snapshotting!")    }
    fn check_read(      &mut self,      _p: &Path,    ) -> Result<(), deno_core::error::AnyError> {      unreachable!("snapshotting!")    }  }
  impl deno_websocket::WebSocketPermissions for Permissions {    fn check_net_url(      &mut self,      _url: &deno_core::url::Url,    ) -> Result<(), deno_core::error::AnyError> {      unreachable!("snapshotting!")    }  }
  impl deno_web::TimersPermission for Permissions {    fn allow_hrtime(&mut self) -> bool {      unreachable!("snapshotting!")    }
    fn check_unstable(      &self,      _state: &deno_core::OpState,      _api_name: &'static str,    ) {      unreachable!("snapshotting!")    }  }
  impl deno_ffi::FfiPermissions for Permissions {    fn check(      &mut self,      _path: Option<&Path>,    ) -> Result<(), deno_core::error::AnyError> {      unreachable!("snapshotting!")    }  }
  impl deno_net::NetPermissions for Permissions {    fn check_net<T: AsRef<str>>(      &mut self,      _host: &(T, Option<u16>),    ) -> Result<(), deno_core::error::AnyError> {      unreachable!("snapshotting!")    }
    fn check_read(      &mut self,      _p: &Path,    ) -> Result<(), deno_core::error::AnyError> {      unreachable!("snapshotting!")    }
    fn check_write(      &mut self,      _p: &Path,    ) -> Result<(), deno_core::error::AnyError> {      unreachable!("snapshotting!")    }  }
  fn create_runtime_snapshot(snapshot_path: &Path, files: Vec<PathBuf>) {    let extensions: Vec<Extension> = vec![      deno_webidl::init(),      deno_console::init(),      deno_url::init(),      deno_tls::init(),      deno_web::init::<Permissions>(        deno_web::BlobStore::default(),        Default::default(),      ),      deno_fetch::init::<Permissions>(Default::default()),      deno_websocket::init::<Permissions>("".to_owned(), None, None),      deno_webstorage::init(None),      deno_crypto::init(None),      deno_webgpu::init(false),      deno_broadcast_channel::init(        deno_broadcast_channel::InMemoryBroadcastChannel::default(),        false, // No --unstable.      ),      deno_ffi::init::<Permissions>(false),      deno_net::init::<Permissions>(        None, false, // No --unstable.        None,      ),      deno_http::init(),    ];
    let js_runtime = JsRuntime::new(RuntimeOptions {      will_snapshot: true,      extensions,      ..Default::default()    });    create_snapshot(js_runtime, snapshot_path, files);  }
  fn get_js_files(d: &str) -> Vec<PathBuf> {    let manifest_dir = Path::new(env!("CARGO_MANIFEST_DIR"));    let mut js_files = std::fs::read_dir(d)      .unwrap()      .map(|dir_entry| {        let file = dir_entry.unwrap();        manifest_dir.join(file.path())      })      .filter(|path| path.extension().unwrap_or_default() == "js")      .collect::<Vec<PathBuf>>();    js_files.sort();    js_files  }
  pub fn build_snapshot(runtime_snapshot_path: PathBuf) {    let js_files = get_js_files("js");    create_runtime_snapshot(&runtime_snapshot_path, js_files);  }}
fn main() {  // To debug snapshot issues uncomment:  // op_fetch_asset::trace_serializer();
  println!("cargo:rustc-env=TARGET={}", env::var("TARGET").unwrap());  println!("cargo:rustc-env=PROFILE={}", env::var("PROFILE").unwrap());  let o = PathBuf::from(env::var_os("OUT_DIR").unwrap());
  // Main snapshot  let runtime_snapshot_path = o.join("CLI_SNAPSHOT.bin");
  // If we're building on docs.rs we just create  // and empty snapshot file and return, because `rusty_v8`  // doesn't actually compile on docs.rs  if env::var_os("DOCS_RS").is_some() {    let snapshot_slice = &[];    std::fs::write(&runtime_snapshot_path, snapshot_slice).unwrap();    return;  }
  #[cfg(not(feature = "docsrs"))]  not_docs::build_snapshot(runtime_snapshot_path)}