use dasl::drisl::Value as DrislValue;
use 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 walk_drisl(path: &str, v: &DrislValue, found: &mut Vec<CollectedLink>) {
    match v {
        DrislValue::Map(o) => {
            for (key, child) in o {
                walk_drisl(&format!("{path}.{key}"), child, found)
            }
        }
        DrislValue::Array(a) => {
            for child in a {
                let child_p = match child {
                    DrislValue::Map(o) => {
                        if let Some(DrislValue::Text(t)) = o.get("$type") {
                            format!("{path}[{t}]")
                        } else {
                            format!("{path}[]")
                        }
                    }
                    _ => format!("{path}[]"),
                };
                walk_drisl(&child_p, child, found)
            }
        }
        DrislValue::Text(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
}

pub fn collect_links_drisl(v: &DrislValue) -> Vec<CollectedLink> {
    let mut found = vec![];
    walk_drisl("", 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()),
                ),
            ]
        )
    }
}