rspack_plugin_html 0.100.2

use core::fmt;

use itertools::Itertools;
use rspack_util::fx_hash::FxHashMap;
use serde::{
  Deserialize, Deserializer, Serialize, Serializer,
  de::{MapAccess, Visitor},
  ser::SerializeMap,
};
use swc_core::{atoms::Atom, common::DUMMY_SP};
use swc_html::ast::{Attribute, Child, Element, Namespace, Text};

use crate::config::{HtmlRspackPluginBaseOptions, HtmlScriptLoading};

// attributes are presented as plain string.
// namespace is not supported currently.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct HtmlPluginAttribute {
  pub attr_name: String,
  // None is ``
  pub attr_value: Option<String>,
}

impl From<HtmlPluginAttribute> for Attribute {
  fn from(attr: HtmlPluginAttribute) -> Self {
    Attribute {
      span: Default::default(),
      namespace: None,
      prefix: None,
      name: attr.attr_name.into(),
      raw_name: None,
      value: attr.attr_value.as_ref().map(|str| Atom::from(str.as_str())),
      raw_value: None,
    }
  }
}

#[derive(Debug, Default, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct HtmlPluginTag {
  pub tag_name: String,
  #[serde(
    serialize_with = "serialize_attributes",
    deserialize_with = "deserialize_attributes"
  )]
  pub attributes: Vec<HtmlPluginAttribute>,
  pub void_tag: bool,
  #[serde(rename = "innerHTML")]
  pub inner_html: Option<String>,
  pub asset: Option<String>,
}

fn serialize_attributes<S>(x: &Vec<HtmlPluginAttribute>, s: S) -> Result<S::Ok, S::Error>
where
  S: Serializer,
{
  let mut map = s.serialize_map(Some(x.len()))?;
  for attr in x {
    let attr_value = attr.attr_value.clone().unwrap_or("true".to_string());
    map.serialize_entry(&attr.attr_name, &attr_value)?;
  }
  map.end()
}

fn deserialize_attributes<'de, D>(d: D) -> Result<Vec<HtmlPluginAttribute>, D::Error>
where
  D: Deserializer<'de>,
{
  struct DataVisitor;

  impl<'de> Visitor<'de> for DataVisitor {
    type Value = Vec<HtmlPluginAttribute>;

    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
      formatter.write_str("html attributes")
    }

    fn visit_map<A>(self, mut access: A) -> Result<Self::Value, A::Error>
    where
      A: MapAccess<'de>,
    {
      let mut res = vec![];

      while let Some((k, v)) = access.next_entry::<String, Option<String>>()? {
        res.push(HtmlPluginAttribute {
          attr_name: k,
          attr_value: v.filter(|value| value != "true"),
        });
      }

      Ok(res)
    }
  }

  d.deserialize_map(DataVisitor)
}

impl HtmlPluginTag {
  pub fn create_style(href: &str) -> HtmlPluginTag {
    HtmlPluginTag {
      tag_name: "link".to_string(),
      attributes: vec![
        HtmlPluginAttribute {
          attr_name: "href".to_string(),
          attr_value: Some(href.to_string()),
        },
        HtmlPluginAttribute {
          attr_name: "rel".to_string(),
          attr_value: Some("stylesheet".to_string()),
        },
      ],
      void_tag: true,
      asset: Some(href.to_string()),
      ..Default::default()
    }
  }

  pub fn create_script(src: &str, script_loading: &HtmlScriptLoading) -> HtmlPluginTag {
    let mut attributes = vec![];
    match script_loading {
      HtmlScriptLoading::Defer => {
        attributes.push(HtmlPluginAttribute {
          attr_name: "defer".to_string(),
          attr_value: None,
        });
      }
      HtmlScriptLoading::Module => {
        attributes.push(HtmlPluginAttribute {
          attr_name: "type".to_string(),
          attr_value: Some("module".to_string()),
        });
      }
      HtmlScriptLoading::SystemjsModule => {
        attributes.push(HtmlPluginAttribute {
          attr_name: "type".to_string(),
          attr_value: Some("systemjs-module".to_string()),
        });
      }
      _ => {}
    }

    attributes.push(HtmlPluginAttribute {
      attr_name: "src".to_string(),
      attr_value: Some(src.to_string()),
    });

    HtmlPluginTag {
      tag_name: "script".to_string(),
      attributes,
      asset: Some(src.to_string()),
      ..Default::default()
    }
  }

  pub fn create_base(base: &HtmlRspackPluginBaseOptions) -> Option<HtmlPluginTag> {
    let mut attributes = vec![];

    if let Some(href) = &base.href {
      attributes.push(HtmlPluginAttribute {
        attr_name: "href".to_string(),
        attr_value: Some(href.clone()),
      });
    }

    if let Some(target) = &base.target {
      attributes.push(HtmlPluginAttribute {
        attr_name: "target".to_string(),
        attr_value: Some(target.clone()),
      });
    }

    if !attributes.is_empty() {
      Some(HtmlPluginTag {
        tag_name: "base".to_string(),
        attributes,
        void_tag: true,
        ..Default::default()
      })
    } else {
      None
    }
  }

  pub fn create_title(title: &str) -> HtmlPluginTag {
    HtmlPluginTag {
      tag_name: "title".to_string(),
      void_tag: true,
      inner_html: Some(title.to_string()),
      ..Default::default()
    }
  }

  pub fn create_meta(meta: &FxHashMap<String, FxHashMap<String, String>>) -> Vec<HtmlPluginTag> {
    meta
      .values()
      .map(|value| HtmlPluginTag {
        tag_name: "meta".to_string(),
        attributes: value
          .iter()
          .sorted()
          .map(|(key, value)| HtmlPluginAttribute {
            attr_name: key.clone(),
            attr_value: Some(value.clone()),
          })
          .collect_vec(),
        void_tag: true,
        ..Default::default()
      })
      .collect_vec()
  }

  pub fn create_favicon(favicon: &str) -> HtmlPluginTag {
    HtmlPluginTag {
      tag_name: "link".to_string(),
      attributes: vec![
        HtmlPluginAttribute {
          attr_name: "rel".to_string(),
          attr_value: Some("icon".into()),
        },
        HtmlPluginAttribute {
          attr_name: "href".to_string(),
          attr_value: Some(favicon.into()),
        },
      ],
      void_tag: true,
      ..Default::default()
    }
  }
}

impl fmt::Display for HtmlPluginTag {
  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
    let mut attributes = self
      .attributes
      .iter()
      .map(|attr| {
        if let Some(attr_value) = &attr.attr_value {
          format!(r#"{}="{}""#, attr.attr_name, attr_value)
        } else {
          attr.attr_name.clone()
        }
      })
      .collect::<Vec<String>>();

    attributes.sort();

    let res = format!(
      "<{} {}{}>{}{}",
      self.tag_name,
      attributes.join(" "),
      if self.void_tag && self.inner_html.is_none() {
        "/"
      } else {
        ""
      },
      if let Some(inner_html) = &self.inner_html {
        inner_html
      } else {
        ""
      },
      if !self.void_tag || self.inner_html.is_some() {
        format!("</{}>", self.tag_name)
      } else {
        String::new()
      }
    );
    write!(f, "{res}")
  }
}

impl From<HtmlPluginTag> for Element {
  fn from(tag: HtmlPluginTag) -> Self {
    Element {
      tag_name: Atom::from(&*tag.tag_name),
      attributes: tag
        .attributes
        .into_iter()
        .sorted_unstable_by(|a, b| a.attr_name.cmp(&b.attr_name))
        .map(Attribute::from)
        .collect::<Vec<_>>(),
      children: tag.inner_html.map_or(vec![], |inner_html| {
        vec![Child::Text(Text {
          span: DUMMY_SP,
          data: Atom::from(inner_html),
          raw: None,
        })]
      }),
      content: None,
      is_self_closing: tag.void_tag,
      namespace: Namespace::HTML,
      span: DUMMY_SP,
    }
  }
}