Compare commits
2 commits
b8a1c5461b
...
01335b6929
Author | SHA1 | Date | |
---|---|---|---|
01335b6929 | |||
cd08775783 |
1 changed files with 42 additions and 8 deletions
48
src/main.rs
48
src/main.rs
|
@ -1,5 +1,6 @@
|
|||
use anyhow::{bail, format_err, Result};
|
||||
use chrono::{DateTime, Local, LocalResult, NaiveDate, TimeZone, Utc};
|
||||
use clap::{arg, command, Parser};
|
||||
use html2md::parse_html;
|
||||
use log::{debug, trace};
|
||||
use megalodon::{
|
||||
|
@ -23,20 +24,35 @@ struct Page<'a> {
|
|||
newest: Option<&'a DateTime<Utc>>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Parser)]
|
||||
#[command()]
|
||||
struct Config {
|
||||
#[arg(required = true)]
|
||||
date: String,
|
||||
#[arg(short, long)]
|
||||
verbose: bool,
|
||||
}
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> Result<()> {
|
||||
let Config { date, verbose } = Config::parse();
|
||||
if verbose {
|
||||
env::set_var("RUST_LOG", format!("{}=debug", module_path!()));
|
||||
} else {
|
||||
env::set_var("RUST_LOG", format!("{}=none", module_path!()));
|
||||
}
|
||||
env_logger::init();
|
||||
|
||||
// TODO add clap and argument for date
|
||||
let day = try_create_range("2023-07-01")?;
|
||||
let day = try_create_range(date)?;
|
||||
|
||||
debug!("Date {}", day.end.format("%Y-%m-%d"));
|
||||
|
||||
let client = create_client()?;
|
||||
let Response { json: account, .. } = client.verify_account_credentials().await?;
|
||||
|
||||
let mut last_id_on_page: Option<String> = None;
|
||||
debug!("Fetching posts");
|
||||
let mut reversed = Vec::new();
|
||||
loop {
|
||||
let json = fetch_page(&client, &last_id_on_page).await?;
|
||||
let page = Page {
|
||||
|
@ -58,35 +74,51 @@ async fn main() -> Result<()> {
|
|||
let json = json
|
||||
.clone()
|
||||
.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>>();
|
||||
trace!("Filtered to {} post(s)", json.len());
|
||||
|
||||
let mut stream = iter(json);
|
||||
|
||||
while let Some(status) = stream.next().await {
|
||||
println!(
|
||||
let ancestor = format!(
|
||||
"{}
|
||||
> {}",
|
||||
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 thread = json
|
||||
.descendants
|
||||
.into_iter()
|
||||
.filter(|s| {
|
||||
s.in_reply_to_account_id == Some(account.id.clone())
|
||||
&& s.account.id == account.id
|
||||
})
|
||||
.map(|status| {
|
||||
format!(
|
||||
">
|
||||
> {}
|
||||
>> {}",
|
||||
status.created_at.with_timezone(&Local).format("%H:%M"),
|
||||
parse_html(&status.content)
|
||||
parse_html(&status.content).trim()
|
||||
)
|
||||
})
|
||||
.collect::<Vec<String>>()
|
||||
.join("\n");
|
||||
println!("{}", thread);
|
||||
reversed.push(format!(
|
||||
"{}{}",
|
||||
ancestor,
|
||||
if !thread.is_empty() {
|
||||
thread
|
||||
} else {
|
||||
String::default()
|
||||
}
|
||||
));
|
||||
}
|
||||
|
||||
if page_end_older_than(&page, &day) {
|
||||
|
@ -98,6 +130,8 @@ async fn main() -> Result<()> {
|
|||
last_id_on_page.replace(id.clone());
|
||||
}
|
||||
}
|
||||
reversed.reverse();
|
||||
println!("{}", reversed.join("\n\n"));
|
||||
Ok(())
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue