fixed body text boundary truncation

This commit is contained in:
delorean 2024-05-23 19:41:09 -05:00
parent d514dd5a1c
commit d9b0eef4df
2 changed files with 11 additions and 10 deletions

View File

@ -38,7 +38,7 @@ pub struct Config {
/// follow redirects
pub follow: bool,
#[clap(long = "body")]
#[clap(default_value_t = 0, long = "body")]
/// read n bytes of the response document body
pub bodysize: usize,
}

View File

@ -17,15 +17,16 @@ pub fn get_title(body: &String) -> String {
pub fn read_body(body: &String, lim: usize) -> String {
let document = Document::from(body.as_str());
let mut text = String::new();
for tag in document.find(Name("body")) {
text.push_str(&tag.text());
let mut bodytext = document.find(Name("body")).next().map(|n| n.text()).unwrap_or_else(|| "".to_string());
bodytext = fmtwhitespace(bodytext);
if bodytext.len() > lim {
bodytext = bodytext
.char_indices()
.take_while(|(i, _)| *i < lim)
.map(|(_, c)| c)
.collect();
}
text = fmtwhitespace(text);
if text.len() > lim {
text.truncate(lim);
}
parsebody(text)
parsebody(bodytext)
}