1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
use anyhow::{anyhow, bail, Context, Result};
use std::fs::File;
use std::io;
#[async_std::main]
async fn main() -> Result<()> {
let args = std::env::args().collect::<Vec<String>>();
if args.iter().skip(1).any(|s| s == "--help") {
println!("Usage: {} <url> [file]", args[0]);
println!("\nIf launched with no arguments, reads from stdin.");
println!(
"\nUse KITTYBOX_AUTH_TOKEN environment variable to authorize to the Micropub endpoint."
);
std::process::exit(0);
}
let token = std::env::var("KITTYBOX_AUTH_TOKEN")
.map_err(|_| anyhow!("No auth token found! Use KITTYBOX_AUTH_TOKEN env variable."))?;
let data: Vec<serde_json::Value> = (if args.len() == 2 || (args.len() == 3 && args[2] == "-") {
serde_json::from_reader(io::stdin())
} else if args.len() == 3 {
serde_json::from_reader(File::open(&args[2]).with_context(|| "Error opening input file")?)
} else {
bail!("See `{} --help` for usage.", args[0]);
})
.with_context(|| "Error while loading the input file")?;
let url = surf::Url::parse(&args[1])?;
let client = surf::Client::new();
let iter = data.into_iter();
for post in iter {
println!(
"Processing {}...",
post["properties"]["url"][0]
.as_str()
.or_else(|| post["properties"]["published"][0]
.as_str()
.or_else(|| post["properties"]["name"][0]
.as_str()
.or(Some("<unidentified post>"))))
.unwrap()
);
match client
.post(&url)
.body(surf::http::Body::from_string(serde_json::to_string(&post)?))
.header("Content-Type", "application/json")
.header("Authorization", format!("Bearer {}", &token))
.send()
.await
{
Ok(mut response) => {
if response.status() == 201 || response.status() == 202 {
println!("Posted at {}", response.header("location").unwrap().last());
} else {
println!("Error: {:?}", response.body_string().await);
}
}
Err(err) => {
println!("{}", err);
}
}
}
Ok(())
}
|
