farmfe_plugin_html 2.1.0

Html plugin of farm.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use farmfe_core::swc_html_ast::{Document, Element};
use farmfe_toolkit::swc_html_visit::{VisitMut, VisitMutWith};

pub struct AbsolutePathHandler {
  pub public_path: String,
}

impl AbsolutePathHandler {
  /// Process the script and link with absolute paths manually added by users in HTML.
  /// Add the prefix of "publicPath" to its path.
  /// eg: <script src="/test.js"></script>  to  <script src="/publicPath/test.js"></script>
  pub fn add_public_path_prefix(&mut self, html_ast: &mut Document) {
    html_ast.visit_mut_with(self)
  }
}

impl VisitMut for AbsolutePathHandler {
  fn visit_mut_element(&mut self, element: &mut Element) {
    if matches!(element.tag_name.to_lowercase().as_str(), "script" | "link") {
      for attr in &mut element.attributes {
        let value = attr.value.clone().unwrap_or_default();
        // determine if the path start with /.
        if matches!(attr.name.to_lowercase().as_str(), "src" | "href")
          && !value.starts_with(&self.public_path)
          && value.starts_with("/")
        {
          let normalized_value = value.trim_start_matches("/");
          attr.value = Some(format!("{}{}", self.public_path, normalized_value).into());
        }
      }
    }
    element.visit_mut_children_with(self);
  }
}