mirror of
https://github.com/denoland/deno.git
synced 2024-12-23 07:44:48 -05:00
286e5d0be9
This is a proof of concept for being able to snapshot TypeScript files. Currently only a single runtime file is authored in TypeScript - "runtime/js/01_version.ts". Not needed infrastructure was removed from "core/snapshot_util.rs". --------- Co-authored-by: Bartek Iwańczuk <biwanczuk@gmail.com>
89 lines
2.5 KiB
Rust
89 lines
2.5 KiB
Rust
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
|
|
|
|
use std::path::Path;
|
|
use std::path::PathBuf;
|
|
|
|
use crate::Extension;
|
|
use crate::InternalModuleLoaderCb;
|
|
use crate::JsRuntime;
|
|
use crate::RuntimeOptions;
|
|
use crate::Snapshot;
|
|
|
|
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 extensions_with_js: Vec<Extension>,
|
|
pub compression_cb: Option<Box<CompressionCb>>,
|
|
pub snapshot_module_load_cb: Option<InternalModuleLoaderCb>,
|
|
}
|
|
|
|
pub fn create_snapshot(create_snapshot_options: CreateSnapshotOptions) {
|
|
let js_runtime = JsRuntime::new(RuntimeOptions {
|
|
will_snapshot: true,
|
|
startup_snapshot: create_snapshot_options.startup_snapshot,
|
|
extensions: create_snapshot_options.extensions,
|
|
extensions_with_js: create_snapshot_options.extensions_with_js,
|
|
snapshot_module_load_cb: create_snapshot_options.snapshot_module_load_cb,
|
|
..Default::default()
|
|
});
|
|
|
|
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 type FilterFn = Box<dyn Fn(&PathBuf) -> bool>;
|
|
|
|
pub fn get_js_files(
|
|
cargo_manifest_dir: &'static str,
|
|
directory: &str,
|
|
filter: Option<FilterFn>,
|
|
) -> 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"
|
|
&& filter.as_ref().map(|filter| filter(path)).unwrap_or(true)
|
|
})
|
|
.collect::<Vec<PathBuf>>();
|
|
js_files.sort();
|
|
js_files
|
|
}
|