Something went wrong. Try again.
Constellation, Spacedust, Slingshot, UFOs: atproto crates and services for microcosm
Something went wrong. Try again.
5.4 kB · 163 lines
Rust
at proxy-blobby
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164use tinyjson::JsonValue;
use crate::{parse_any_link, CollectedLink};
pub fn walk_record(path: &str, v: &JsonValue, found: &mut Vec<CollectedLink>) { match v { JsonValue::Object(o) => { for (key, child) in o { walk_record(&format!("{path}.{key}"), child, found) } } JsonValue::Array(a) => { for child in a { let child_p = match child { JsonValue::Object(o) => { if let Some(JsonValue::String(t)) = o.get("$type") { format!("{path}[{t}]") } else { format!("{path}[]") } } _ => format!("{path}[]"), }; walk_record(&child_p, child, found) } } JsonValue::String(s) => { if let Some(link) = parse_any_link(s) { found.push(CollectedLink { path: path.to_string(), target: link, }); } } _ => {} }}
pub fn collect_links(v: &JsonValue) -> Vec<CollectedLink> { let mut found = vec![]; walk_record("", v, &mut found); found}
#[cfg(test)]mod tests { use super::*; use crate::Link;
fn l(path: &str, target: Link) -> CollectedLink { CollectedLink { path: path.into(), target, } }
#[test] fn test_collect_links() { let rec = r#"{"a": "https://example.com", "b": "not a link"}"#; let json = collect_links(&rec.parse().unwrap()); assert_eq!(json, vec![l(".a", Link::Uri("https://example.com".into()))]); }
#[test] fn test_bsky_feed_post_record_reply() { let rec = r#"{ "$type": "app.bsky.feed.post", "createdAt": "2025-01-08T20:52:43.041Z", "langs": [ "en" ], "reply": { "parent": { "cid": "bafyreifk3bwnmulk37ezrarg4ouheqnhgucypynftqafl4limssogvzk6i", "uri": "at://did:plc:b3rzzkblqsxhr3dgcueymkqe/app.bsky.feed.post/3lf6yc4drhk2f" }, "root": { "cid": "bafyreifk3bwnmulk37ezrarg4ouheqnhgucypynftqafl4limssogvzk6i", "uri": "at://did:plc:b3rzzkblqsxhr3dgcueymkqe/app.bsky.feed.post/3lf6yc4drhk2f" } }, "text": "Yup!" }"#; let mut json = collect_links(&rec.parse().unwrap()); json.sort_by_key(|c| (c.path.clone(), c.target.clone())); assert_eq!( json, vec![ l( ".reply.parent.uri", Link::AtUri( "at://did:plc:b3rzzkblqsxhr3dgcueymkqe/app.bsky.feed.post/3lf6yc4drhk2f" .into() ) ), l( ".reply.root.uri", Link::AtUri( "at://did:plc:b3rzzkblqsxhr3dgcueymkqe/app.bsky.feed.post/3lf6yc4drhk2f" .into() ) ), ] ) }
#[test] fn test_bsky_feed_post_record_embed() { let rec = r#"{ "$type": "app.bsky.feed.post", "createdAt": "2025-01-08T20:52:39.539Z", "embed": { "$type": "app.bsky.embed.external", "external": { "description": "YouTube video by More Perfect Union", "thumb": { "$type": "blob", "ref": { "$link": "bafkreifxuvkbqksq5usi4cryex37o4absjexuouvgenlb62ojsx443b2tm" }, "mimeType": "image/jpeg", "size": 477460 }, "title": "Corporations & Wealthy Elites Are Coopting Our Government. Who Can Stop Them?", "uri": "https://youtu.be/oKXm4szEP1Q?si=_0n_uPu4qNKokMnq" } }, "facets": [ { "features": [ { "$type": "app.bsky.richtext.facet#link", "uri": "https://youtu.be/oKXm4szEP1Q?si=_0n_uPu4qNKokMnq" } ], "index": { "byteEnd": 24, "byteStart": 0 } } ], "langs": [ "en" ], "text": "youtu.be/oKXm4szEP1Q?..." }"#; let mut json = collect_links(&rec.parse().unwrap()); json.sort_by_key(|c| (c.path.clone(), c.target.clone())); assert_eq!( json, vec![ l( ".embed.external.uri", Link::Uri("https://youtu.be/oKXm4szEP1Q?si=_0n_uPu4qNKokMnq".into()), ), l( ".facets[].features[app.bsky.richtext.facet#link].uri", Link::Uri("https://youtu.be/oKXm4szEP1Q?si=_0n_uPu4qNKokMnq".into()), ), ] ) }}