Reverse sort and filter to calling user
This commit is contained in:
parent
cd08775783
commit
01335b6929
1 changed files with 27 additions and 7 deletions
34
src/main.rs
34
src/main.rs
|
@ -29,28 +29,30 @@ struct Page<'a> {
|
||||||
struct Config {
|
struct Config {
|
||||||
#[arg(required = true)]
|
#[arg(required = true)]
|
||||||
date: String,
|
date: String,
|
||||||
#[arg(short)]
|
#[arg(short, long)]
|
||||||
verbose: bool,
|
verbose: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> Result<()> {
|
async fn main() -> Result<()> {
|
||||||
env_logger::init();
|
|
||||||
let Config { date, verbose } = Config::parse();
|
let Config { date, verbose } = Config::parse();
|
||||||
if verbose {
|
if verbose {
|
||||||
env::set_var("RUST_LOG", format!("{}=debug", module_path!()));
|
env::set_var("RUST_LOG", format!("{}=debug", module_path!()));
|
||||||
} else {
|
} else {
|
||||||
env::set_var("RUST_LOG", format!("{}=none", module_path!()));
|
env::set_var("RUST_LOG", format!("{}=none", module_path!()));
|
||||||
}
|
}
|
||||||
|
env_logger::init();
|
||||||
|
|
||||||
let day = try_create_range(date)?;
|
let day = try_create_range(date)?;
|
||||||
|
|
||||||
debug!("Date {}", day.end.format("%Y-%m-%d"));
|
debug!("Date {}", day.end.format("%Y-%m-%d"));
|
||||||
|
|
||||||
let client = create_client()?;
|
let client = create_client()?;
|
||||||
|
let Response { json: account, .. } = client.verify_account_credentials().await?;
|
||||||
|
|
||||||
let mut last_id_on_page: Option<String> = None;
|
let mut last_id_on_page: Option<String> = None;
|
||||||
debug!("Fetching posts");
|
debug!("Fetching posts");
|
||||||
|
let mut reversed = Vec::new();
|
||||||
loop {
|
loop {
|
||||||
let json = fetch_page(&client, &last_id_on_page).await?;
|
let json = fetch_page(&client, &last_id_on_page).await?;
|
||||||
let page = Page {
|
let page = Page {
|
||||||
|
@ -72,35 +74,51 @@ async fn main() -> Result<()> {
|
||||||
let json = json
|
let json = json
|
||||||
.clone()
|
.clone()
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.filter(|json| day.start <= json.created_at && json.created_at <= day.end)
|
.filter(|json| {
|
||||||
|
json.account.id == account.id
|
||||||
|
&& day.start <= json.created_at
|
||||||
|
&& json.created_at <= day.end
|
||||||
|
})
|
||||||
.collect::<Vec<Status>>();
|
.collect::<Vec<Status>>();
|
||||||
trace!("Filtered to {} post(s)", json.len());
|
trace!("Filtered to {} post(s)", json.len());
|
||||||
|
|
||||||
let mut stream = iter(json);
|
let mut stream = iter(json);
|
||||||
|
|
||||||
while let Some(status) = stream.next().await {
|
while let Some(status) = stream.next().await {
|
||||||
println!(
|
let ancestor = format!(
|
||||||
"{}
|
"{}
|
||||||
> {}",
|
> {}",
|
||||||
status.created_at.with_timezone(&Local).format("%H:%M"),
|
status.created_at.with_timezone(&Local).format("%H:%M"),
|
||||||
parse_html(&status.content)
|
parse_html(&status.content).trim()
|
||||||
);
|
);
|
||||||
let Response { json, .. } = client.get_status_context(status.id, None).await?;
|
let Response { json, .. } = client.get_status_context(status.id, None).await?;
|
||||||
let thread = json
|
let thread = json
|
||||||
.descendants
|
.descendants
|
||||||
.into_iter()
|
.into_iter()
|
||||||
|
.filter(|s| {
|
||||||
|
s.in_reply_to_account_id == Some(account.id.clone())
|
||||||
|
&& s.account.id == account.id
|
||||||
|
})
|
||||||
.map(|status| {
|
.map(|status| {
|
||||||
format!(
|
format!(
|
||||||
">
|
">
|
||||||
> {}
|
> {}
|
||||||
>> {}",
|
>> {}",
|
||||||
status.created_at.with_timezone(&Local).format("%H:%M"),
|
status.created_at.with_timezone(&Local).format("%H:%M"),
|
||||||
parse_html(&status.content)
|
parse_html(&status.content).trim()
|
||||||
)
|
)
|
||||||
})
|
})
|
||||||
.collect::<Vec<String>>()
|
.collect::<Vec<String>>()
|
||||||
.join("\n");
|
.join("\n");
|
||||||
println!("{}", thread);
|
reversed.push(format!(
|
||||||
|
"{}{}",
|
||||||
|
ancestor,
|
||||||
|
if !thread.is_empty() {
|
||||||
|
thread
|
||||||
|
} else {
|
||||||
|
String::default()
|
||||||
|
}
|
||||||
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
if page_end_older_than(&page, &day) {
|
if page_end_older_than(&page, &day) {
|
||||||
|
@ -112,6 +130,8 @@ async fn main() -> Result<()> {
|
||||||
last_id_on_page.replace(id.clone());
|
last_id_on_page.replace(id.clone());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
reversed.reverse();
|
||||||
|
println!("{}", reversed.join("\n\n"));
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue