rspack_loader_runner 0.100.0-rc.3

use std::fmt;

use rspack_cacheable::cacheable;

#[cacheable]
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Scheme {
  None,
  Data,
  File,
  Http,
  Https,
  Custom(String),
}

impl Scheme {
  pub fn is_file(&self) -> bool {
    matches!(self, Self::File)
  }

  pub fn is_data(&self) -> bool {
    matches!(self, Self::Data)
  }

  pub fn is_none(&self) -> bool {
    matches!(self, Self::None)
  }

  pub fn is_http(&self) -> bool {
    matches!(self, Self::Http)
  }

  pub fn is_https(&self) -> bool {
    matches!(self, Self::Https)
  }
}

impl From<&str> for Scheme {
  fn from(value: &str) -> Self {
    if value.is_empty() || value.eq_ignore_ascii_case("builtin") {
      Self::None
    } else if value.eq_ignore_ascii_case("data") {
      Self::Data
    } else if value.eq_ignore_ascii_case("file") {
      Self::File
    } else if value.eq_ignore_ascii_case("http") {
      Self::Http
    } else if value.eq_ignore_ascii_case("https") {
      Self::Https
    } else {
      Self::Custom(value.to_string())
    }
  }
}

impl Scheme {
  pub fn as_str(&self) -> &str {
    match self {
      Self::None => "",
      Self::Data => "data",
      Self::File => "file",
      Self::Http => "http",
      Self::Https => "https",
      Self::Custom(v) => v,
    }
  }
}

impl fmt::Display for Scheme {
  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
    write!(f, "{}", self.as_str())
  }
}

const BACK_SLASH: char = '\\';
const SLASH: char = '/';
const A_LOWER_CASE: char = 'a';
const Z_LOWER_CASE: char = 'z';
const A_UPPER_CASE: char = 'A';
const Z_UPPER_CASE: char = 'Z';
const ZERO: char = '0';
const NINE: char = '9';
const PLUS: char = '+';
const HYPHEN: char = '-';
const COLON: char = ':';
const HASH: char = '#';
const QUERY: char = '?';

pub fn get_scheme(specifier: &str) -> Scheme {
  // Fast path: most modules don't have a scheme
  let bytes = specifier.as_bytes();
  if memchr::memchr(b':', bytes).is_none() {
    return Scheme::None;
  }

  let mut chars = specifier.chars().enumerate().peekable();

  // First char maybe only a letter
  let start = chars.next();
  if start.is_none()
    || matches!(start, Some((_, c)) if (c < A_LOWER_CASE || c > Z_LOWER_CASE) && (c < A_UPPER_CASE || c > Z_UPPER_CASE))
  {
    return Scheme::None;
  }

  let mut next = chars.next();
  while let Some((_, ch)) = next
    && ((ch >= A_LOWER_CASE && ch <= Z_LOWER_CASE)
      || (ch >= A_UPPER_CASE && ch <= Z_UPPER_CASE)
      || (ch >= ZERO && ch <= NINE)
      || ch == PLUS
      || ch == HYPHEN)
  {
    if chars.peek().is_none() {
      return Scheme::None;
    }
    next = chars.next();
  }

  // Scheme must end with colon
  let maybe_colon = next;
  if maybe_colon.is_none() || matches!(maybe_colon, Some((_, c)) if c != COLON) {
    return Scheme::None;
  }

  // Check for Windows absolute path
  // https://url.spec.whatwg.org/#url-miscellaneous
  let (i, _) = maybe_colon.expect("should not be None");
  if i == 1 {
    let next_ch = chars.next();
    if next_ch.is_none()
      || matches!(next_ch, Some((_, ch)) if ch == BACK_SLASH || ch == SLASH || ch == HASH || ch == QUERY)
    {
      return Scheme::None;
    }
  }

  Scheme::from(&specifier[..i])
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn none_for_windows_path() {
    assert_eq!(get_scheme("D:\\a\\rspack\\index.js"), Scheme::None);
  }

  #[test]
  fn data_for_data_uri() {
    assert_eq!(get_scheme("data:text/javascript"), Scheme::Data);
  }

  #[test]
  fn http_for_http_url() {
    assert_eq!(get_scheme("http://localhost"), Scheme::Http);
    assert_eq!(get_scheme("https://localhost"), Scheme::Https);
  }

  #[test]
  fn file_for_file_url() {
    assert_eq!(get_scheme("file:/a.js"), Scheme::File);
  }

  #[test]
  fn custom_for_custom_url() {
    assert_eq!(
      get_scheme("native:/Users/a.js"),
      Scheme::Custom("native".to_owned())
    );
  }
}