about summary refs log tree commit diff
path: root/src/bin/pyindieblog_to_kittybox.rs
blob: b4e2b9707606f7b2504c4e8504461a5397fe15e1 (plain) (blame)
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
67
68
use anyhow::{anyhow, Context, Result};
use mobc_redis::redis;
use mobc_redis::redis::AsyncCommands;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::fs::File;

#[derive(Default, Serialize, Deserialize)]
struct PyindieblogData {
    posts: Vec<serde_json::Value>,
    cards: Vec<serde_json::Value>,
}

#[async_std::main]
async fn main() -> Result<()> {
    let mut args = std::env::args();
    args.next(); // skip argv[0] which is the name
    let redis_uri = args
        .next()
        .ok_or_else(|| anyhow!("No Redis URI provided"))?;
    let client = redis::Client::open(redis_uri.as_str())
        .with_context(|| format!("Failed to construct Redis client on {}", redis_uri))?;

    let filename = args
        .next()
        .ok_or_else(|| anyhow!("No filename provided for export"))?;

    let mut data: Vec<serde_json::Value>;

    let file = File::create(filename)?;

    let mut conn = client
        .get_async_std_connection()
        .await
        .with_context(|| "Failed to connect to the Redis server")?;

    data = conn
        .hgetall::<&str, HashMap<String, String>>("posts")
        .await?
        .values()
        .map(|s| {
            serde_json::from_str::<serde_json::Value>(s)
                .with_context(|| format!("Failed to parse the following entry: {:?}", s))
        })
        .collect::<std::result::Result<Vec<serde_json::Value>, anyhow::Error>>()
        .with_context(|| "Failed to export h-entries from pyindieblog")?;
    data.extend(
        conn.hgetall::<&str, HashMap<String, String>>("hcards")
            .await?
            .values()
            .map(|s| {
                serde_json::from_str::<serde_json::Value>(s)
                    .with_context(|| format!("Failed to parse the following card: {:?}", s))
            })
            .collect::<std::result::Result<Vec<serde_json::Value>, anyhow::Error>>()
            .with_context(|| "Failed to export h-cards from pyindieblog")?,
    );

    data.sort_by_key(|v| {
        v["properties"]["published"][0]
            .as_str()
            .map(|s| s.to_string())
    });

    serde_json::to_writer(file, &data)?;

    Ok(())
}