about summary refs log tree commit diff
path: root/src/bin/pyindieblog_to_kittybox.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/bin/pyindieblog_to_kittybox.rs')
-rw-r--r--src/bin/pyindieblog_to_kittybox.rs48
1 files changed, 48 insertions, 0 deletions
diff --git a/src/bin/pyindieblog_to_kittybox.rs b/src/bin/pyindieblog_to_kittybox.rs
new file mode 100644
index 0000000..7935da5
--- /dev/null
+++ b/src/bin/pyindieblog_to_kittybox.rs
@@ -0,0 +1,48 @@
+use std::collections::HashMap;
+use std::fs::File;
+use anyhow::{Error, Result, Context, anyhow, bail};
+use mobc_redis::redis;
+use mobc_redis::redis::AsyncCommands;
+use serde::{Serialize, Deserialize};
+use serde_json::json;
+
+#[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(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(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(())
+}