implement improvements suggested by @robatipoor
This commit is contained in:
parent
c011f90b76
commit
643c4ce7ef
13
src/html.rs
13
src/html.rs
@ -28,18 +28,15 @@ pub fn get_parent_node(node: &Handle) -> Handle {
|
|||||||
parent.and_then(|node| node.upgrade()).unwrap()
|
parent.and_then(|node| node.upgrade()).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_node_name(node: &Handle) -> &'_ str {
|
pub fn get_node_name(node: &Handle) -> Option<&'_ str> {
|
||||||
match &node.data {
|
match &node.data {
|
||||||
NodeData::Element { ref name, .. } => name.local.as_ref(),
|
NodeData::Element { ref name, .. } => Some(name.local.as_ref()),
|
||||||
_ => "",
|
_ => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn is_icon(attr_value: &str) -> bool {
|
pub fn is_icon(attr_value: &str) -> bool {
|
||||||
ICON_VALUES
|
ICON_VALUES.contains(&attr_value.to_lowercase().as_str())
|
||||||
.iter()
|
|
||||||
.find(|a| attr_value.eq_ignore_ascii_case(a))
|
|
||||||
.is_some()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn walk_and_embed_assets(
|
pub fn walk_and_embed_assets(
|
||||||
@ -310,7 +307,7 @@ pub fn walk_and_embed_assets(
|
|||||||
attr.value.clear();
|
attr.value.clear();
|
||||||
attr.value.push_slice(src_full_url.as_str());
|
attr.value.push_slice(src_full_url.as_str());
|
||||||
} else if attr_name == "srcset" {
|
} else if attr_name == "srcset" {
|
||||||
if get_node_name(&get_parent_node(&node)) == "picture" {
|
if get_node_name(&get_parent_node(&node)) == Some("picture") {
|
||||||
if opt_no_images {
|
if opt_no_images {
|
||||||
attr.value.clear();
|
attr.value.clear();
|
||||||
attr.value.push_slice(TRANSPARENT_PIXEL);
|
attr.value.push_slice(TRANSPARENT_PIXEL);
|
||||||
|
@ -28,11 +28,11 @@ fn get_node_name() {
|
|||||||
let parent = html::get_parent_node(node);
|
let parent = html::get_parent_node(node);
|
||||||
let parent_node_name = html::get_node_name(&parent);
|
let parent_node_name = html::get_node_name(&parent);
|
||||||
if node_name == "head" || node_name == "body" {
|
if node_name == "head" || node_name == "body" {
|
||||||
assert_eq!(parent_node_name, "html");
|
assert_eq!(parent_node_name, Some("html"));
|
||||||
} else if node_name == "div" {
|
} else if node_name == "div" {
|
||||||
assert_eq!(parent_node_name, "body");
|
assert_eq!(parent_node_name, Some("body"));
|
||||||
} else if node_name == "p" {
|
} else if node_name == "p" {
|
||||||
assert_eq!(parent_node_name, "div");
|
assert_eq!(parent_node_name, Some("div"));
|
||||||
}
|
}
|
||||||
|
|
||||||
for child in node.children.borrow().iter() {
|
for child in node.children.borrow().iter() {
|
||||||
|
Loading…
Reference in New Issue
Block a user