Nekomata/site_icons
Nekomata/site_icons
Archived
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

0.4.2: ignore logos with search / menu in ancestor tree

Browse source
This commit is contained in:
Sam Denty 2022-12-25 21:10:15 +00:00
parent b460b4415a
commit d95824a11d
No known key found for this signature in database
GPG key ID: 7B4EAF7B9E291B79
3 changed files with 35 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Cargo.lock generated
View file

@ -1638,7 +1638,7 @@ checksum = "7bd3e3206899af3f8b12af284fafc038cc1dc2b41d1b89dd17297221c5d225de"
[[package]]
name = "site_icons"
version = "0.4.1"
version = "0.4.2"
dependencies = [
"byteorder",
"clap",

2
Cargo.toml
View file

@ -1,6 +1,6 @@
[package]
name = "site_icons"
version = "0.4.1"
version = "0.4.2"
authors = ["Sam Denty <sam@samdenty.com>"]
edition = "2018"
license = "GPL-3.0"

47
src/icons.rs
View file

@ -187,25 +187,44 @@ impl Icons {
"img[alt*=logo], svg[alt*=logo]",
"img[class*=logo], svg[class*=logo]",
))
.map(|elem_ref| {
let elem = elem_ref.value();
let mut weight = 0;
// if in the header
if elem_ref
.filter_map(|elem_ref| {
let mut ancestors = elem_ref
.ancestors()
.map(ElementRef::wrap)
.flatten()
.any(|element| element.value().name() == "header")
{
.map(|elem_ref| elem_ref.value());
let skip_classnames = regex!("menu|search");
let should_skip = ancestors.any(|ancestor| {
ancestor
.attr("class")
.map(|attr| skip_classnames.is_match(&attr.to_lowercase()))
.or_else(|| {
ancestor
.attr("id")
.map(|attr| skip_classnames.is_match(&attr.to_lowercase()))
})
.unwrap_or(false)
});
if should_skip {
return None;
}
let mut weight = 0;
// if in the header
if ancestors.any(|element| element.name() == "header") {
weight += 2;
}
let mentions_logo = |attr_name| {
elem
.attr(attr_name)
.map(|attr| regex!("logo([^s]|$)").is_match(&attr.to_lowercase()))
.unwrap_or(false)
let mut mentions_logo = |attr_name| {
ancestors.any(|ancestor| {
ancestor
.attr(attr_name)
.map(|attr| regex!("logo([^s]|$)").is_match(&attr.to_lowercase()))
.unwrap_or(false)
})
};
if mentions_logo("class") || mentions_logo("id") {
weight += 3;
@ -217,7 +236,7 @@ impl Icons {
weight += 1;
}
(elem_ref, weight)
Some((elem_ref, weight))
})
.collect();