xmaster 1.6.3

Enterprise-grade X/Twitter CLI — post, reply, like, retweet, DM, search, and more
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use crate::context::AppContext;
use crate::errors::XmasterError;
use crate::output::{self, OutputFormat, Tableable};
use crate::providers::xai::XaiSearch;
use serde::Serialize;
use std::sync::Arc;

#[derive(Serialize)]
struct AiSearchResult {
    text: String,
    citations: Vec<String>,
}

impl Tableable for AiSearchResult {
    fn to_table(&self) -> comfy_table::Table {
        let mut table = comfy_table::Table::new();
        table.set_header(vec!["AI Search Results"]);
        table.add_row(vec![&self.text]);
        if !self.citations.is_empty() {
            table.add_row(vec![""]);
            table.add_row(vec!["--- Citations ---"]);
            for url in &self.citations {
                table.add_row(vec![url.as_str()]);
            }
        }
        table
    }
}

pub async fn execute(
    ctx: Arc<AppContext>,
    format: OutputFormat,
    query: &str,
    count: usize,
    from_date: Option<&str>,
    to_date: Option<&str>,
) -> Result<(), XmasterError> {
    let xai = XaiSearch::new(ctx.clone());
    let result = xai.search_posts(query, count, None, from_date, to_date).await?;
    let display = AiSearchResult {
        text: result.text,
        citations: result.citations,
    };
    output::render(format, &display, None);
    Ok(())
}

pub async fn trending(
    ctx: Arc<AppContext>,
    format: OutputFormat,
    region: Option<&str>,
    category: Option<&str>,
) -> Result<(), XmasterError> {
    let xai = XaiSearch::new(ctx.clone());
    let result = xai.get_trending(region, category).await?;
    let display = AiSearchResult {
        text: result.text,
        citations: result.citations,
    };
    output::render(format, &display, None);
    Ok(())
}

/// Personalized trends via the X API (not xAI). Requires X Premium.
pub async fn personalized_trends(
    ctx: Arc<AppContext>,
    format: OutputFormat,
) -> Result<(), XmasterError> {
    let api = crate::providers::xapi::XApi::new(ctx.clone());
    let trends = api.get_personalized_trends().await?;

    if trends.is_empty() {
        return Err(XmasterError::NotFound(
            "No personalized trends returned. Ensure your account has X Premium.".into(),
        ));
    }

    #[derive(Serialize)]
    struct TrendsResult {
        source: String,
        count: usize,
        trends: Vec<crate::providers::xapi::TrendData>,
    }
    impl Tableable for TrendsResult {
        fn to_table(&self) -> comfy_table::Table {
            let mut table = comfy_table::Table::new();
            table.set_header(vec!["Trend", "Tweets", "Description"]);
            for t in &self.trends {
                table.add_row(vec![
                    t.name.clone(),
                    t.tweet_count.map(|c| c.to_string()).unwrap_or_else(|| "—".into()),
                    t.description.clone().unwrap_or_default(),
                ]);
            }
            table
        }
    }

    let display = TrendsResult {
        source: "personalized".into(),
        count: trends.len(),
        trends,
    };
    output::render(format, &display, None);
    Ok(())
}