371 lines
13 KiB
Rust
371 lines
13 KiB
Rust
|
use cssparser::{ParseError, Parser, ParserInput, SourcePosition, Token};
|
||
|
|
use reqwest::blocking::Client;
|
||
|
|
use std::collections::HashMap;
|
||
|
|
|
||
|
|
use crate::utils::{data_to_data_url, decode_url, get_url_fragment, resolve_url, retrieve_asset};
|
||
|
|
|
||
|
|
const CSS_PROPS_WITH_IMAGE_URLS: &[&str] = &[
|
||
|
|
"background",
|
||
|
|
"background-image",
|
||
|
|
"border",
|
||
|
|
"border-image",
|
||
|
|
"border-image-source",
|
||
|
|
"content",
|
||
|
|
"cursor",
|
||
|
|
"list-style",
|
||
|
|
"list-style-image",
|
||
|
|
"mask",
|
||
|
|
"mask-image",
|
||
|
|
];
|
||
|
|
|
||
|
|
const TRANSPARENT_PIXEL: &str = "data:image/png;base64,\
|
||
|
|
iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNkYAAAAAYAAjCB0C8AAAAASUVORK5CYII=";
|
||
|
|
|
||
|
|
pub fn is_image_url_prop(prop_name: &str) -> bool {
|
||
|
|
CSS_PROPS_WITH_IMAGE_URLS
|
||
|
|
.iter()
|
||
|
|
.find(|p| prop_name.eq_ignore_ascii_case(p))
|
||
|
|
.is_some()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn enquote(input: String, double: bool) -> String {
|
||
|
|
if double {
|
||
|
|
format!("\"{}\"", input.replace("\"", "\\\""))
|
||
|
|
} else {
|
||
|
|
format!("'{}'", input.replace("'", "\\'"))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn process_css<'a>(
|
||
|
|
cache: &mut HashMap<String, String>,
|
||
|
|
client: &Client,
|
||
|
|
parent_url: &str,
|
||
|
|
parser: &mut Parser,
|
||
|
|
rule_name: &str,
|
||
|
|
prop_name: &str,
|
||
|
|
func_name: &str,
|
||
|
|
opt_no_images: bool,
|
||
|
|
opt_silent: bool,
|
||
|
|
) -> Result<String, ParseError<'a, String>> {
|
||
|
|
let mut result: String = str!();
|
||
|
|
|
||
|
|
let mut curr_rule: String = str!(rule_name.clone());
|
||
|
|
let mut curr_prop: String = str!(prop_name.clone());
|
||
|
|
let mut token: &Token;
|
||
|
|
let mut token_offset: SourcePosition;
|
||
|
|
|
||
|
|
loop {
|
||
|
|
token_offset = parser.position();
|
||
|
|
token = match parser.next_including_whitespace_and_comments() {
|
||
|
|
Ok(token) => token,
|
||
|
|
Err(_) => {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
match *token {
|
||
|
|
Token::Comment(_) => {
|
||
|
|
let token_slice = parser.slice_from(token_offset);
|
||
|
|
result.push_str(str!(token_slice).as_str());
|
||
|
|
}
|
||
|
|
Token::Semicolon => result.push_str(";"),
|
||
|
|
Token::Colon => result.push_str(":"),
|
||
|
|
Token::Comma => result.push_str(","),
|
||
|
|
Token::ParenthesisBlock | Token::SquareBracketBlock | Token::CurlyBracketBlock => {
|
||
|
|
let closure: &str;
|
||
|
|
if token == &Token::ParenthesisBlock {
|
||
|
|
result.push_str("(");
|
||
|
|
closure = ")";
|
||
|
|
} else if token == &Token::SquareBracketBlock {
|
||
|
|
result.push_str("[");
|
||
|
|
closure = "]";
|
||
|
|
} else {
|
||
|
|
result.push_str("{");
|
||
|
|
closure = "}";
|
||
|
|
}
|
||
|
|
|
||
|
|
let block_css: String = parser
|
||
|
|
.parse_nested_block(|parser| {
|
||
|
|
process_css(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
parent_url,
|
||
|
|
parser,
|
||
|
|
rule_name,
|
||
|
|
curr_prop.as_str(),
|
||
|
|
func_name,
|
||
|
|
opt_no_images,
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
})
|
||
|
|
.unwrap();
|
||
|
|
result.push_str(block_css.as_str());
|
||
|
|
|
||
|
|
result.push_str(closure);
|
||
|
|
}
|
||
|
|
Token::CloseParenthesis => result.push_str(")"),
|
||
|
|
Token::CloseSquareBracket => result.push_str("]"),
|
||
|
|
Token::CloseCurlyBracket => result.push_str("}"),
|
||
|
|
Token::IncludeMatch => result.push_str("~="),
|
||
|
|
Token::DashMatch => result.push_str("|="),
|
||
|
|
Token::PrefixMatch => result.push_str("^="),
|
||
|
|
Token::SuffixMatch => result.push_str("$="),
|
||
|
|
Token::SubstringMatch => result.push_str("*="),
|
||
|
|
Token::CDO => result.push_str("<!--"),
|
||
|
|
Token::CDC => result.push_str("-->"),
|
||
|
|
Token::WhiteSpace(ref value) => {
|
||
|
|
result.push_str(value);
|
||
|
|
}
|
||
|
|
Token::Ident(ref value) => {
|
||
|
|
curr_prop = str!(value);
|
||
|
|
result.push_str(value);
|
||
|
|
}
|
||
|
|
Token::AtKeyword(ref value) => {
|
||
|
|
curr_rule = str!(value);
|
||
|
|
result.push_str("@");
|
||
|
|
result.push_str(value);
|
||
|
|
}
|
||
|
|
Token::Hash(ref value) => {
|
||
|
|
result.push_str("#");
|
||
|
|
result.push_str(value);
|
||
|
|
}
|
||
|
|
Token::QuotedString(ref value) => {
|
||
|
|
let is_import: bool = curr_rule == "import";
|
||
|
|
if is_import {
|
||
|
|
// Reset current at-rule value
|
||
|
|
curr_rule = str!();
|
||
|
|
}
|
||
|
|
|
||
|
|
if is_import {
|
||
|
|
// Skip empty import values
|
||
|
|
if value.len() < 1 {
|
||
|
|
result.push_str("''");
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
let full_url = resolve_url(&parent_url, value).unwrap_or_default();
|
||
|
|
let url_fragment = get_url_fragment(full_url.clone());
|
||
|
|
let full_url_decoded = decode_url(full_url);
|
||
|
|
let (css, final_url) = retrieve_asset(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
&parent_url,
|
||
|
|
&full_url_decoded,
|
||
|
|
false,
|
||
|
|
"",
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
.unwrap_or_default();
|
||
|
|
|
||
|
|
result.push_str(
|
||
|
|
enquote(
|
||
|
|
data_to_data_url(
|
||
|
|
"text/css",
|
||
|
|
embed_css(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
final_url.as_str(),
|
||
|
|
&css,
|
||
|
|
opt_no_images,
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
.as_bytes(),
|
||
|
|
&final_url,
|
||
|
|
url_fragment.as_str(),
|
||
|
|
),
|
||
|
|
false,
|
||
|
|
)
|
||
|
|
.as_str(),
|
||
|
|
);
|
||
|
|
} else {
|
||
|
|
if func_name == "url" {
|
||
|
|
// Skip empty url()'s
|
||
|
|
if value.len() < 1 {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
if opt_no_images && is_image_url_prop(curr_prop.as_str()) {
|
||
|
|
result.push_str(enquote(str!(TRANSPARENT_PIXEL), false).as_str());
|
||
|
|
} else {
|
||
|
|
let resolved_url = resolve_url(&parent_url, value).unwrap_or_default();
|
||
|
|
let (data_url, _final_url) = retrieve_asset(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
&parent_url,
|
||
|
|
&resolved_url,
|
||
|
|
true,
|
||
|
|
"",
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
.unwrap_or_default();
|
||
|
|
result.push_str(enquote(data_url, false).as_str());
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
result.push_str(enquote(str!(value), false).as_str());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
Token::Number {
|
||
|
|
ref has_sign,
|
||
|
|
ref value,
|
||
|
|
..
|
||
|
|
} => {
|
||
|
|
if *has_sign && *value >= 0. {
|
||
|
|
result.push_str("+");
|
||
|
|
}
|
||
|
|
result.push_str(&value.to_string())
|
||
|
|
}
|
||
|
|
Token::Percentage {
|
||
|
|
ref has_sign,
|
||
|
|
ref unit_value,
|
||
|
|
..
|
||
|
|
} => {
|
||
|
|
if *has_sign {
|
||
|
|
result.push_str("-");
|
||
|
|
}
|
||
|
|
result.push_str(str!(unit_value * 100.).as_str());
|
||
|
|
result.push_str("%");
|
||
|
|
}
|
||
|
|
Token::Dimension {
|
||
|
|
ref value,
|
||
|
|
ref unit,
|
||
|
|
..
|
||
|
|
} => {
|
||
|
|
result.push_str(str!(value).as_str());
|
||
|
|
result.push_str(str!(unit).as_str());
|
||
|
|
}
|
||
|
|
Token::IDHash(ref value) => {
|
||
|
|
result.push_str("#");
|
||
|
|
result.push_str(value);
|
||
|
|
}
|
||
|
|
Token::UnquotedUrl(ref value) => {
|
||
|
|
let is_import: bool = curr_rule == "import";
|
||
|
|
if is_import {
|
||
|
|
// Reset current at-rule value
|
||
|
|
curr_rule = str!();
|
||
|
|
}
|
||
|
|
|
||
|
|
// Skip empty url()'s
|
||
|
|
if value.len() < 1 {
|
||
|
|
result.push_str("url()");
|
||
|
|
continue;
|
||
|
|
} else if value.starts_with("#") {
|
||
|
|
result.push_str("url(");
|
||
|
|
result.push_str(value);
|
||
|
|
result.push_str(")");
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
result.push_str("url(");
|
||
|
|
if is_import {
|
||
|
|
let full_url = resolve_url(&parent_url, value).unwrap_or_default();
|
||
|
|
let url_fragment = get_url_fragment(full_url.clone());
|
||
|
|
let full_url_decoded = decode_url(full_url);
|
||
|
|
let (css, final_url) = retrieve_asset(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
&parent_url,
|
||
|
|
&full_url_decoded,
|
||
|
|
false,
|
||
|
|
"",
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
.unwrap_or_default();
|
||
|
|
|
||
|
|
result.push_str(
|
||
|
|
enquote(
|
||
|
|
data_to_data_url(
|
||
|
|
"text/css",
|
||
|
|
embed_css(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
final_url.as_str(),
|
||
|
|
&css,
|
||
|
|
opt_no_images,
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
.as_bytes(),
|
||
|
|
&final_url,
|
||
|
|
url_fragment.as_str(),
|
||
|
|
),
|
||
|
|
false,
|
||
|
|
)
|
||
|
|
.as_str(),
|
||
|
|
);
|
||
|
|
} else {
|
||
|
|
if opt_no_images && is_image_url_prop(curr_prop.as_str()) {
|
||
|
|
result.push_str(enquote(str!(TRANSPARENT_PIXEL), false).as_str());
|
||
|
|
} else {
|
||
|
|
let full_url = resolve_url(&parent_url, value).unwrap_or_default();
|
||
|
|
let (data_url, _final_url) = retrieve_asset(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
&parent_url,
|
||
|
|
&full_url,
|
||
|
|
true,
|
||
|
|
"",
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
.unwrap_or_default();
|
||
|
|
result.push_str(enquote(data_url, false).as_str());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
result.push_str(")");
|
||
|
|
}
|
||
|
|
Token::Delim(ref value) => result.push_str(&value.to_string()),
|
||
|
|
Token::Function(ref name) => {
|
||
|
|
let function_name: &str = &name.clone();
|
||
|
|
result.push_str(function_name);
|
||
|
|
result.push_str("(");
|
||
|
|
|
||
|
|
let block_css: String = parser
|
||
|
|
.parse_nested_block(|parser| {
|
||
|
|
process_css(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
parent_url,
|
||
|
|
parser,
|
||
|
|
curr_rule.as_str(),
|
||
|
|
curr_prop.as_str(),
|
||
|
|
function_name,
|
||
|
|
opt_no_images,
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
})
|
||
|
|
.unwrap();
|
||
|
|
result.push_str(block_css.as_str());
|
||
|
|
|
||
|
|
result.push_str(")");
|
||
|
|
}
|
||
|
|
Token::BadUrl(_) | Token::BadString(_) => {}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(result)
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn embed_css(
|
||
|
|
cache: &mut HashMap<String, String>,
|
||
|
|
client: &Client,
|
||
|
|
parent_url: &str,
|
||
|
|
css: &str,
|
||
|
|
opt_no_images: bool,
|
||
|
|
opt_silent: bool,
|
||
|
|
) -> String {
|
||
|
|
let mut input = ParserInput::new(&css);
|
||
|
|
let mut parser = Parser::new(&mut input);
|
||
|
|
|
||
|
|
process_css(
|
||
|
|
cache,
|
||
|
|
client,
|
||
|
|
parent_url,
|
||
|
|
&mut parser,
|
||
|
|
"",
|
||
|
|
"",
|
||
|
|
"",
|
||
|
|
opt_no_images,
|
||
|
|
opt_silent,
|
||
|
|
)
|
||
|
|
.unwrap()
|
||
|
|
}
|