nab 0.7.1

//! Reddit content extraction via Reddit JSON API.
//!
//! Reddit provides a JSON API by appending `.json` to any URL.
//! This extracts post content, author info, and top comments.
//!
//! # Example
//!
//! ```rust,no_run
//! use nab::site::{SiteProvider, reddit::RedditProvider};
//! use nab::AcceleratedClient;
//!
//! # async fn example() -> anyhow::Result<()> {
//! let client = AcceleratedClient::new()?;
//! let provider = RedditProvider;
//!
//! let content = provider.extract(
//!     "https://reddit.com/r/rust/comments/abc123",
//!     &client,
//!     None,
//!     None
//! ).await?;
//!
//! println!("{}", content.markdown);
//! # Ok(())
//! # }
//! ```

use anyhow::{Context, Result};
use async_trait::async_trait;
use serde::Deserialize;
use std::fmt::Write as _;

use super::{Engagement, SiteContent, SiteMetadata, SiteProvider, format_number_compact};
use crate::http_client::AcceleratedClient;

/// Reddit content provider using Reddit JSON API.
pub struct RedditProvider;

#[async_trait]
impl SiteProvider for RedditProvider {
    fn name(&self) -> &'static str {
        "reddit"
    }

    fn matches(&self, url: &str) -> bool {
        let normalized = url.to_lowercase();
        let normalized = normalized.split('?').next().unwrap_or(&normalized);

        (normalized.contains("reddit.com/r/") || normalized.contains("old.reddit.com/r/"))
            && normalized.contains("/comments/")
    }

    async fn extract(
        &self,
        url: &str,
        _client: &AcceleratedClient,
        _cookies: Option<&str>,
        _prefetched_html: Option<&[u8]>,
    ) -> Result<SiteContent> {
        // Normalize URL and append .json
        let json_url = parse_reddit_url(url);
        tracing::debug!("Fetching from Reddit: {}", json_url);

        // Build a fresh client without http2_prior_knowledge. The AcceleratedClient
        // forces H2 without ALPN negotiation, which causes Reddit to return HTML
        // instead of JSON. A standard client negotiates the protocol via TLS ALPN.
        let reddit_client = reqwest::Client::builder()
            .use_rustls_tls()
            .gzip(true)
            .brotli(true)
            .connect_timeout(std::time::Duration::from_secs(10))
            .timeout(std::time::Duration::from_secs(30))
            .redirect(reqwest::redirect::Policy::limited(5))
            .build()
            .context("Failed to build Reddit HTTP client")?;

        let response = reddit_client
            .get(&json_url)
            .header("User-Agent", "nab/0.3.0 (by /u/nab-cli)")
            .header("Accept", "application/json")
            .send()
            .await
            .context("Failed to fetch from Reddit API")?
            .text()
            .await
            .context("Failed to read Reddit response body")?;

        let api_response: Vec<RedditListing> =
            serde_json::from_str(&response).context("Failed to parse Reddit response")?;

        // First listing is the post, second is comments
        let post_data = api_response
            .first()
            .and_then(|l| l.data.children.first())
            .context("No post data found")?;

        let empty_comments = vec![];
        let comments_data = api_response
            .get(1)
            .map_or(&empty_comments, |l| &l.data.children);

        let markdown = format_reddit_markdown(&post_data.data, comments_data);

        #[allow(clippy::cast_sign_loss)]
        let engagement = Engagement {
            likes: Some(post_data.data.score.max(0) as u64),
            reposts: None,
            replies: Some(post_data.data.num_comments),
            views: None,
        };

        let metadata = SiteMetadata {
            author: Some(format!("u/{}", post_data.data.author)),
            title: Some(post_data.data.title.clone()),
            published: Some(format_timestamp(post_data.data.created_utc)),
            platform: "Reddit".to_string(),
            canonical_url: post_data.data.url.clone(),
            media_urls: vec![],
            engagement: Some(engagement),
        };

        Ok(SiteContent { markdown, metadata })
    }
}

/// Parse Reddit URL and convert to JSON API endpoint.
fn parse_reddit_url(url: &str) -> String {
    let url = url.split('?').next().unwrap_or(url);
    let mut json_url = url.to_string();

    // Ensure it ends with .json (case-insensitive check)
    let has_json_ext = std::path::Path::new(&json_url)
        .extension()
        .is_some_and(|e| e.eq_ignore_ascii_case("json"));
    if !has_json_ext {
        json_url.push_str(".json");
    }

    json_url
}

/// Format Reddit post and comments as markdown.
fn format_reddit_markdown(post: &RedditPost, comments: &[RedditChild]) -> String {
    let mut md = String::new();

    // Title
    md.push_str("## ");
    md.push_str(&post.title);
    md.push_str("\n\n");

    // Metadata line
    let _ = writeln!(
        md,
        "by u/{} · {} points · {} comments\n",
        post.author,
        format_score(post.score),
        format_number_compact(post.num_comments)
    );

    // Post body (selftext for text posts, url for link posts)
    if let Some(selftext) = &post.selftext
        && !selftext.is_empty()
    {
        md.push_str(selftext);
        md.push_str("\n\n");
    }

    // If it's a link post, include the URL
    if !post.is_self {
        md.push_str("🔗 ");
        md.push_str(&post.url);
        md.push_str("\n\n");
    }

    // Top comments (up to 10)
    if !comments.is_empty() {
        md.push_str("### Top Comments\n\n");

        let mut count = 0;
        for comment in comments {
            if count >= 10 {
                break;
            }

            if let Some(body) = &comment.data.body {
                let _ = write!(
                    md,
                    "**u/{}** ({} points):\n\n{}\n\n---\n\n",
                    comment.data.author,
                    format_score(comment.data.score),
                    body
                );
                count += 1;
            }
        }
    }

    md
}

/// Format Unix timestamp as a relative time string (e.g., "3 hours ago").
///
/// More useful than ISO dates for Reddit posts where recency matters.
/// Falls back to "Unknown" for invalid timestamps.
fn format_timestamp(timestamp: f64) -> String {
    use std::time::{SystemTime, UNIX_EPOCH};

    #[allow(clippy::cast_possible_truncation, clippy::cast_sign_loss)]
    let secs = timestamp as u64;
    let post_time = UNIX_EPOCH + std::time::Duration::from_secs(secs);

    let Ok(elapsed) = SystemTime::now().duration_since(post_time) else {
        return "Unknown".to_string();
    };

    let mins = elapsed.as_secs() / 60;
    let hours = mins / 60;
    let days = hours / 24;

    if days > 365 {
        format!("{} years ago", days / 365)
    } else if days > 30 {
        format!("{} months ago", days / 30)
    } else if days > 0 {
        format!("{days} days ago")
    } else if hours > 0 {
        format!("{hours} hours ago")
    } else if mins > 0 {
        format!("{mins} minutes ago")
    } else {
        "just now".to_string()
    }
}

/// Format signed score values with K/M suffixes.
fn format_score(n: i64) -> String {
    let sign = if n < 0 { "-" } else { "" };
    let abs = n.unsigned_abs();
    format!("{sign}{}", format_number_compact(abs))
}

// ============================================================================
// Reddit API Response Types
// ============================================================================

#[derive(Debug, Deserialize)]
struct RedditListing {
    data: RedditListingData,
}

#[derive(Debug, Deserialize)]
struct RedditListingData {
    children: Vec<RedditChild>,
}

#[derive(Debug, Deserialize)]
struct RedditChild {
    data: RedditPost,
}

#[derive(Debug, Deserialize)]
struct RedditPost {
    #[serde(default)]
    title: String,
    #[serde(default)]
    author: String,
    #[serde(default)]
    score: i64,
    #[serde(default)]
    num_comments: u64,
    #[serde(default)]
    created_utc: f64,
    #[serde(default)]
    selftext: Option<String>,
    #[serde(default)]
    url: String,
    #[serde(default)]
    is_self: bool,
    #[serde(default)]
    body: Option<String>, // For comments
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn matches_reddit_dot_com_comments_urls() {
        let provider = RedditProvider;
        assert!(provider.matches("https://reddit.com/r/rust/comments/abc123"));
        assert!(
            provider.matches("https://www.reddit.com/r/programming/comments/xyz789/some_title")
        );
    }

    #[test]
    fn matches_old_reddit_dot_com_urls() {
        let provider = RedditProvider;
        assert!(provider.matches("https://old.reddit.com/r/rust/comments/abc123"));
        assert!(provider.matches("https://OLD.REDDIT.COM/r/rust/COMMENTS/123"));
    }

    #[test]
    fn does_not_match_non_comment_urls() {
        let provider = RedditProvider;
        assert!(!provider.matches("https://reddit.com/r/rust"));
        assert!(!provider.matches("https://reddit.com/user/someone"));
        assert!(!provider.matches("https://youtube.com/watch?v=abc"));
    }

    #[test]
    fn parse_reddit_url_appends_json() {
        let result = parse_reddit_url("https://reddit.com/r/rust/comments/abc123");
        assert_eq!(result, "https://reddit.com/r/rust/comments/abc123.json");
    }

    #[test]
    fn parse_reddit_url_strips_query() {
        let result = parse_reddit_url("https://reddit.com/r/rust/comments/abc123?utm_source=x");
        assert_eq!(result, "https://reddit.com/r/rust/comments/abc123.json");
    }

    // format_number_compact tests live in site/hackernews.rs + doc tests on
    // site::format_number_compact.

    #[test]
    fn format_reddit_markdown_self_post() {
        let post = RedditPost {
            title: "Test Post".to_string(),
            author: "testuser".to_string(),
            score: 42,
            num_comments: 5,
            created_utc: 1_700_000_000.0,
            selftext: Some("Hello world".to_string()),
            url: "https://reddit.com/r/test/comments/abc".to_string(),
            is_self: true,
            body: None,
        };
        let md = format_reddit_markdown(&post, &[]);
        assert!(md.contains("## Test Post"));
        assert!(md.contains("u/testuser"));
        assert!(md.contains("42 points"));
        assert!(md.contains("Hello world"));
        assert!(!md.contains("\u{1f517}")); // no link emoji for self posts
    }

    #[test]
    fn format_reddit_markdown_link_post() {
        let post = RedditPost {
            title: "Cool Link".to_string(),
            author: "linkposter".to_string(),
            score: 1_500,
            num_comments: 200,
            created_utc: 1_700_000_000.0,
            selftext: None,
            url: "https://example.com/article".to_string(),
            is_self: false,
            body: None,
        };
        let md = format_reddit_markdown(&post, &[]);
        assert!(md.contains("## Cool Link"));
        assert!(md.contains("1.5K points"));
        assert!(md.contains("200 comments"));
        assert!(md.contains("https://example.com/article"));
    }

    #[test]
    fn format_reddit_markdown_with_comments() {
        let post = RedditPost {
            title: "Discussion".to_string(),
            author: "op".to_string(),
            score: 10,
            num_comments: 2,
            created_utc: 1_700_000_000.0,
            selftext: None,
            url: "https://reddit.com/r/test/comments/x".to_string(),
            is_self: true,
            body: None,
        };
        let comments = vec![RedditChild {
            data: RedditPost {
                title: String::new(),
                author: "commenter1".to_string(),
                score: 8,
                num_comments: 0,
                created_utc: 1_700_000_100.0,
                selftext: None,
                url: String::new(),
                is_self: false,
                body: Some("Great post!".to_string()),
            },
        }];
        let md = format_reddit_markdown(&post, &comments);
        assert!(md.contains("### Top Comments"));
        assert!(md.contains("u/commenter1"));
        assert!(md.contains("Great post!"));
    }

    #[test]
    fn format_score_handles_negative() {
        assert_eq!(format_score(-5), "-5");
        assert_eq!(format_score(-1_500), "-1.5K");
        assert_eq!(format_score(42), "42");
        assert_eq!(format_score(0), "0");
    }

    #[test]
    fn deserialize_reddit_api_response() {
        let json = r#"[
            {
                "data": {
                    "children": [{
                        "data": {
                            "title": "Test",
                            "author": "user",
                            "score": 100,
                            "num_comments": 10,
                            "created_utc": 1700000000.0,
                            "selftext": "body text",
                            "url": "https://reddit.com/r/test/comments/abc",
                            "is_self": true
                        }
                    }]
                }
            },
            {
                "data": {
                    "children": [{
                        "data": {
                            "title": "",
                            "author": "replier",
                            "score": 5,
                            "num_comments": 0,
                            "created_utc": 1700000100.0,
                            "selftext": null,
                            "url": "",
                            "is_self": false,
                            "body": "Nice post!"
                        }
                    }]
                }
            }
        ]"#;
        let parsed: Vec<RedditListing> = serde_json::from_str(json).unwrap();
        assert_eq!(parsed.len(), 2);
        assert_eq!(parsed[0].data.children[0].data.title, "Test");
        assert_eq!(
            parsed[1].data.children[0].data.body.as_deref(),
            Some("Nice post!")
        );
    }

    #[test]
    fn deserialize_reddit_api_with_float_timestamps() {
        // Reddit API returns created_utc as floats (e.g., 1770617951.0)
        let json = r#"[{
            "data": {
                "children": [{
                    "data": {
                        "title": "Float Test",
                        "author": "user",
                        "score": -3,
                        "num_comments": 0,
                        "created_utc": 1770617951.0,
                        "url": "https://reddit.com/r/test/comments/x",
                        "is_self": true
                    }
                }]
            }
        }]"#;
        let parsed: Vec<RedditListing> = serde_json::from_str(json).unwrap();
        let post = &parsed[0].data.children[0].data;
        assert_eq!(post.score, -3);
        assert!((post.created_utc - 1_770_617_951.0).abs() < f64::EPSILON);
    }
}