xmaster 1.6.0

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

use serde::Serialize;

/// Trait for types that can be rendered as CSV rows.
///
/// A default implementation is provided that serializes the value as a single
/// JSON cell — types with tabular data should override with proper columns.
pub trait CsvRenderable {
    fn csv_headers() -> Vec<&'static str> {
        vec!["data"]
    }

    fn csv_rows(&self) -> Vec<Vec<String>>
    where
        Self: Serialize,
    {
        // Fallback: emit the whole value as a single JSON cell.
        let json = serde_json::to_string(self).unwrap_or_default();
        vec![vec![json]]
    }
}

/// Escape a CSV field: quote if it contains commas, quotes, or newlines.
fn escape_field(value: &str) -> String {
    if value.contains(',') || value.contains('"') || value.contains('\n') || value.contains('\r') {
        format!("\"{}\"", value.replace('"', "\"\""))
    } else {
        value.to_string()
    }
}

pub fn render<T: Serialize + CsvRenderable>(data: &T, _metadata: Option<serde_json::Value>) {
    let headers = T::csv_headers();
    println!(
        "{}",
        headers
            .iter()
            .map(|h| escape_field(h))
            .collect::<Vec<_>>()
            .join(",")
    );
    for row in data.csv_rows() {
        println!(
            "{}",
            row.iter()
                .map(|f| escape_field(f))
                .collect::<Vec<_>>()
                .join(",")
        );
    }
}