Something went wrong. Try again.
Scalable and distributed custom feed generator, ott - on that topic
Something went wrong. Try again.
2.1 kB · 66 lines
Rust
at main
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667use eyre::eyre;use fluvio_smartmodule::{RecordData, Result, SmartModuleRecord, smartmodule};use serde_json::{Map, Value};
#[smartmodule(filter_map)]pub fn filter_map(record: &SmartModuleRecord) -> Result<Option<(Option<RecordData>, RecordData)>> { let key = record.key.clone();
let string = std::str::from_utf8(record.value.as_ref())?; let mut value: Value = serde_json::from_str(string)?; let obj = value .as_object_mut() .ok_or(eyre!("Failed to parse value"))?;
if let Ok(uri) = get_uri(obj) { let uri_value = Value::String(uri); obj.insert("uri".to_string(), uri_value);
Ok(Some((key, value.to_string().as_str().into()))) } else { Ok(None) }}
fn get_uri(obj: &Map<String, Value>) -> Result<String> { let collection = obj .get("commit") .and_then(|v| v.get("collection")) .and_then(|v| v.as_str()) .ok_or(eyre!("Missing commit.collection"))?; match collection { "app.bsky.feed.post" => { let did = obj .get("did") .and_then(|v| v.as_str()) .ok_or(eyre!("did missing or not a string"))?;
let commit = obj.get("commit").ok_or(eyre!("commit missing"))?;
let collection = commit .get("collection") .and_then(|v| v.as_str()) .ok_or(eyre!("commit.collection missing or not a string"))?;
let rkey = commit .get("rkey") .and_then(|v| v.as_str()) .ok_or(eyre!("commit.rkey missing or not a string"))?;
Ok(format!("at://{did}/{collection}/{rkey}")) }, "app.bsky.feed.like" => { let uri = obj .get("commit") .and_then(|v| v.get("record")) .and_then(|v| v.get("subject")) .and_then(|v| v.get("uri")) .and_then(|v| v.as_str()) .ok_or(eyre!("Likely not commit message"))?; Ok(uri.to_string()) } &_ => { Err(eyre!("Not supported yet")) } }}