mirror of
https://github.com/denoland/deno.git
synced 2024-11-25 15:29:32 -05:00
117 lines
3.6 KiB
Rust
117 lines
3.6 KiB
Rust
// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
|
|
|
|
use deno_core::op2;
|
|
use deno_core::url::Url;
|
|
use deno_core::OpState;
|
|
use deno_fs::FileSystemRc;
|
|
use node_resolver::NodeResolution;
|
|
use std::borrow::Cow;
|
|
use std::path::Path;
|
|
use std::path::PathBuf;
|
|
|
|
use crate::NodePermissions;
|
|
use crate::NodeRequireResolverRc;
|
|
use crate::NodeResolverRc;
|
|
|
|
#[must_use = "the resolved return value to mitigate time-of-check to time-of-use issues"]
|
|
fn ensure_read_permission<'a, P>(
|
|
state: &mut OpState,
|
|
file_path: &'a Path,
|
|
) -> Result<Cow<'a, Path>, deno_core::error::AnyError>
|
|
where
|
|
P: NodePermissions + 'static,
|
|
{
|
|
let resolver = state.borrow::<NodeRequireResolverRc>().clone();
|
|
let permissions = state.borrow_mut::<P>();
|
|
resolver.ensure_read_permission(permissions, file_path)
|
|
}
|
|
|
|
#[derive(Debug, thiserror::Error)]
|
|
pub enum WorkerThreadsFilenameError {
|
|
#[error(transparent)]
|
|
Permission(deno_core::error::AnyError),
|
|
#[error("{0}")]
|
|
UrlParse(#[from] url::ParseError),
|
|
#[error("Relative path entries must start with '.' or '..'")]
|
|
InvalidRelativeUrl,
|
|
#[error("URL from Path-String")]
|
|
UrlFromPathString,
|
|
#[error("URL to Path-String")]
|
|
UrlToPathString,
|
|
#[error("URL to Path")]
|
|
UrlToPath,
|
|
#[error("File not found [{0:?}]")]
|
|
FileNotFound(PathBuf),
|
|
#[error("Neither ESM nor CJS")]
|
|
NeitherEsmNorCjs,
|
|
#[error("{0}")]
|
|
UrlToNodeResolution(node_resolver::errors::UrlToNodeResolutionError),
|
|
#[error(transparent)]
|
|
Fs(#[from] deno_io::fs::FsError),
|
|
}
|
|
|
|
#[op2]
|
|
#[string]
|
|
pub fn op_worker_threads_filename<P>(
|
|
state: &mut OpState,
|
|
#[string] specifier: String,
|
|
) -> Result<String, WorkerThreadsFilenameError>
|
|
where
|
|
P: NodePermissions + 'static,
|
|
{
|
|
if specifier.starts_with("data:") {
|
|
return Ok(specifier);
|
|
}
|
|
let url: Url = if specifier.starts_with("file:") {
|
|
Url::parse(&specifier)?
|
|
} else {
|
|
let path = PathBuf::from(&specifier);
|
|
if path.is_relative() && !specifier.starts_with('.') {
|
|
return Err(WorkerThreadsFilenameError::InvalidRelativeUrl);
|
|
}
|
|
let path = ensure_read_permission::<P>(state, &path)
|
|
.map_err(WorkerThreadsFilenameError::Permission)?;
|
|
let fs = state.borrow::<FileSystemRc>();
|
|
let canonicalized_path =
|
|
deno_path_util::strip_unc_prefix(fs.realpath_sync(&path)?);
|
|
Url::from_file_path(canonicalized_path)
|
|
.map_err(|_| WorkerThreadsFilenameError::UrlFromPathString)?
|
|
};
|
|
let url_path = url
|
|
.to_file_path()
|
|
.map_err(|_| WorkerThreadsFilenameError::UrlToPathString)?;
|
|
let url_path = ensure_read_permission::<P>(state, &url_path)
|
|
.map_err(WorkerThreadsFilenameError::Permission)?;
|
|
let fs = state.borrow::<FileSystemRc>();
|
|
if !fs.exists_sync(&url_path) {
|
|
return Err(WorkerThreadsFilenameError::FileNotFound(
|
|
url_path.to_path_buf(),
|
|
));
|
|
}
|
|
let node_resolver = state.borrow::<NodeResolverRc>();
|
|
match node_resolver
|
|
.url_to_node_resolution(url)
|
|
.map_err(WorkerThreadsFilenameError::UrlToNodeResolution)?
|
|
{
|
|
NodeResolution::Esm(u) => Ok(u.to_string()),
|
|
NodeResolution::CommonJs(u) => wrap_cjs(u),
|
|
NodeResolution::BuiltIn(_) => {
|
|
Err(WorkerThreadsFilenameError::NeitherEsmNorCjs)
|
|
}
|
|
}
|
|
}
|
|
|
|
///
|
|
/// Wrap a CJS file-URL and the required setup in a stringified `data:`-URL
|
|
///
|
|
fn wrap_cjs(url: Url) -> Result<String, WorkerThreadsFilenameError> {
|
|
let path = url
|
|
.to_file_path()
|
|
.map_err(|_| WorkerThreadsFilenameError::UrlToPath)?;
|
|
let filename = path.file_name().unwrap().to_string_lossy();
|
|
Ok(format!(
|
|
"data:text/javascript,import {{ createRequire }} from \"node:module\";\
|
|
const require = createRequire(\"{}\"); require(\"./{}\");",
|
|
url, filename,
|
|
))
|
|
}
|