spikes 0.2.0

Drop-in feedback collection for static HTML mockups
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

use std::io::{self, Write};

use crate::error::Result;
use crate::storage::load_spikes;

#[derive(Debug, Clone, Copy, PartialEq)]
pub enum ExportFormat {
    Json,
    Csv,
    Jsonl,
}

impl std::str::FromStr for ExportFormat {
    type Err = String;

    fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
        match s.to_lowercase().as_str() {
            "json" => Ok(ExportFormat::Json),
            "csv" => Ok(ExportFormat::Csv),
            "jsonl" => Ok(ExportFormat::Jsonl),
            _ => Err(format!("Invalid format: {}. Use json, csv, or jsonl", s)),
        }
    }
}

pub fn run(format: ExportFormat) -> Result<()> {
    let spikes = load_spikes()?;
    let stdout = io::stdout();
    let mut handle = stdout.lock();

    match format {
        ExportFormat::Json => {
            let json = serde_json::to_string_pretty(&spikes)?;
            writeln!(handle, "{}", json)?;
        }
        ExportFormat::Jsonl => {
            for spike in &spikes {
                let json = serde_json::to_string(spike)?;
                writeln!(handle, "{}", json)?;
            }
        }
        ExportFormat::Csv => {
            let mut wtr = csv::Writer::from_writer(handle);
            wtr.write_record([
                "id",
                "type",
                "project_key",
                "page",
                "url",
                "reviewer_id",
                "reviewer_name",
                "selector",
                "element_text",
                "rating",
                "comments",
                "timestamp",
                "viewport_width",
                "viewport_height",
            ])?;

            for spike in &spikes {
                wtr.write_record([
                    &spike.id,
                    spike.type_str(),
                    &spike.project_key,
                    &spike.page,
                    &spike.url,
                    &spike.reviewer.id,
                    &spike.reviewer.name,
                    spike.selector.as_deref().unwrap_or(""),
                    spike.element_text.as_deref().unwrap_or(""),
                    spike.rating_str(),
                    &spike.comments,
                    &spike.timestamp,
                    &spike.viewport.width.to_string(),
                    &spike.viewport.height.to_string(),
                ])?;
            }
            wtr.flush()?;
        }
    }

    Ok(())
}