rspack_binding_api 0.101.0-rc.0

//! Port rspack_resolver napi
//!
//! This module is generally copied from https://github.com/rstackjs/rspack-resolver/blob/main/napi/src/lib.rs
mod options;

use std::{
  path::{Path, PathBuf},
  sync::Arc,
};

use napi::{bindgen_prelude::Either, tokio::runtime};
use napi_derive::napi;
use rspack_resolver::{ResolveOptions, Resolver};

use self::options::{AliasRawValue, NapiResolveOptions, StrOrStrList};

#[napi(object)]
pub struct ResolveResult {
  pub path: Option<String>,
  pub error: Option<String>,
  /// "type" field in the package.json file
  pub module_type: Option<String>,
}

async fn resolve(resolver: &Resolver, path: &Path, request: &str) -> ResolveResult {
  match resolver.resolve(path, request).await {
    Ok(resolution) => ResolveResult {
      path: Some(resolution.full_path().to_string_lossy().to_string()),
      error: None,
      module_type: resolution
        .package_json()
        .and_then(|p| p.r#type.as_ref())
        .map(|t| t.to_string()),
    },
    Err(err) => ResolveResult {
      path: None,
      module_type: None,
      error: Some(err.to_string()),
    },
  }
}

#[napi]
pub fn sync(path: String, request: String) -> ResolveResult {
  let path = PathBuf::from(path);
  let resolver = Resolver::new(ResolveOptions::default());
  napi::bindgen_prelude::within_runtime_if_available(|| {
    runtime::Handle::current().block_on(resolve(&resolver, &path, &request))
  })
}

#[napi(js_name = "async")]
pub async fn async_(path: String, request: String) -> ResolveResult {
  let path = PathBuf::from(path);
  let resolver = Resolver::new(ResolveOptions::default());
  resolve(&resolver, &path, &request).await
}

#[napi]
pub struct ResolverFactory {
  resolver: Arc<Resolver>,
}

#[napi]
impl ResolverFactory {
  #[napi(constructor)]
  pub fn new(options: Option<NapiResolveOptions>) -> Self {
    let options = options.map_or_else(ResolveOptions::default, Self::normalize_options);
    Self {
      resolver: Arc::new(Resolver::new(options)),
    }
  }

  #[napi]
  #[allow(clippy::should_implement_trait)]
  pub fn default() -> Self {
    let default_options = ResolveOptions::default();
    Self {
      resolver: Arc::new(Resolver::new(default_options)),
    }
  }

  /// Clone the resolver using the same underlying cache.
  #[napi]
  pub fn clone_with_options(&self, options: NapiResolveOptions) -> Self {
    Self {
      resolver: Arc::new(
        self
          .resolver
          .clone_with_options(Self::normalize_options(options)),
      ),
    }
  }

  /// Clear the underlying cache.
  #[napi]
  pub fn clear_cache(&self) {
    self.resolver.clear_cache();
  }

  /// Synchronously resolve `specifier` at an absolute path to a `directory`.
  #[allow(clippy::needless_pass_by_value)]
  #[napi]
  pub fn sync(&self, directory: String, request: String) -> ResolveResult {
    let path = PathBuf::from(directory);
    napi::bindgen_prelude::within_runtime_if_available(|| {
      runtime::Handle::current().block_on(resolve(&self.resolver, &path, &request))
    })
  }

  /// Asynchronously resolve `specifier` at an absolute path to a `directory`.
  #[allow(clippy::needless_pass_by_value)]
  #[napi(js_name = "async")]
  pub async fn resolve_async(&self, directory: String, request: String) -> ResolveResult {
    let path = PathBuf::from(directory);
    let resolver = self.resolver.clone();
    resolve(&resolver, &path, &request).await
  }

  fn normalize_options(op: NapiResolveOptions) -> ResolveOptions {
    let default = ResolveOptions::default();
    // merging options
    ResolveOptions {
      tsconfig: op.tsconfig.map(|tsconfig| tsconfig.into()),
      alias: op
        .alias
        .map(|alias| {
          alias
            .into_iter()
            .map(|(k, v)| (k, AliasRawValue(v).into()))
            .collect()
        })
        .unwrap_or(default.alias),
      alias_fields: op
        .alias_fields
        .map(|o| {
          o.into_iter()
            .map(|x| StrOrStrList(x).into())
            .collect::<Vec<_>>()
        })
        .unwrap_or(default.alias_fields),
      condition_names: op.condition_names.unwrap_or(default.condition_names),
      description_files: op.description_files.unwrap_or(default.description_files),
      enforce_extension: op
        .enforce_extension
        .map_or(default.enforce_extension, |enforce_extension| {
          enforce_extension.into()
        }),
      exports_fields: op
        .exports_fields
        .map(|o| {
          o.into_iter()
            .map(|x| StrOrStrList(x).into())
            .collect::<Vec<_>>()
        })
        .unwrap_or(default.exports_fields),
      imports_fields: op
        .imports_fields
        .map(|o| {
          o.into_iter()
            .map(|x| StrOrStrList(x).into())
            .collect::<Vec<_>>()
        })
        .unwrap_or(default.imports_fields),
      extension_alias: op
        .extension_alias
        .map(|extension_alias| extension_alias.into_iter().collect::<Vec<_>>())
        .unwrap_or(default.extension_alias),
      extensions: op.extensions.unwrap_or(default.extensions),
      fallback: op
        .fallback
        .map(|fallback| {
          fallback
            .into_iter()
            .map(|(k, v)| {
              let v = v
                .into_iter()
                .map(|item| match item {
                  Some(path) => rspack_resolver::AliasValue::from(path),
                  None => rspack_resolver::AliasValue::Ignore,
                })
                .collect();
              (k, v)
            })
            .collect::<Vec<_>>()
        })
        .unwrap_or(default.fallback),
      fully_specified: op.fully_specified.unwrap_or(default.fully_specified),
      main_fields: op
        .main_fields
        .map(|o| StrOrStrList(o).into())
        .unwrap_or(default.main_fields),
      main_files: op.main_files.unwrap_or(default.main_files),
      modules: op
        .modules
        .map(|o| StrOrStrList(o).into())
        .unwrap_or(default.modules),
      resolve_to_context: op.resolve_to_context.unwrap_or(default.resolve_to_context),
      prefer_relative: op.prefer_relative.unwrap_or(default.prefer_relative),
      prefer_absolute: op.prefer_absolute.unwrap_or(default.prefer_absolute),
      restrictions: op
        .restrictions
        .map(|restrictions| {
          restrictions
            .into_iter()
            .map(|restriction| restriction.into())
            .collect::<Vec<_>>()
        })
        .unwrap_or(default.restrictions),
      roots: op
        .roots
        .map(|roots| roots.into_iter().map(PathBuf::from).collect::<Vec<_>>())
        .unwrap_or(default.roots),
      symlinks: op.symlinks.unwrap_or(default.symlinks),
      builtin_modules: op.builtin_modules.unwrap_or(default.builtin_modules),
      enable_pnp: op.enable_pnp.unwrap_or_default(),
      node_path: false,
    }
  }
}