spikes 0.3.2

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

use std::collections::HashMap;

use crate::error::Result;
use crate::output::{print_hotspots_table, print_json};
use crate::spike::SpikeType;
use crate::storage::load_spikes;

pub fn run(json: bool) -> Result<()> {
    let spikes = load_spikes()?;

    let mut counts: HashMap<String, usize> = HashMap::new();
    for spike in spikes {
        if spike.spike_type == SpikeType::Element {
            if let Some(selector) = spike.selector {
                *counts.entry(selector).or_insert(0) += 1;
            }
        }
    }

    let mut hotspots: Vec<(String, usize)> = counts.into_iter().collect();
    hotspots.sort_by_key(|item| std::cmp::Reverse(item.1));

    if json {
        let output: Vec<serde_json::Value> = hotspots
            .iter()
            .map(|(selector, count)| {
                serde_json::json!({
                    "selector": selector,
                    "count": count
                })
            })
            .collect();
        print_json(&output);
    } else {
        print_hotspots_table(&hotspots);
    }

    Ok(())
}