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

use chrono::NaiveDate;
use indexmap::IndexMap;
use serde_derive::{Deserialize, Serialize};
use std::collections::HashMap;

#[derive(Debug, Eq, PartialEq, Serialize, Deserialize)]
pub struct ScreenNameResult {
    pub accounts: Vec<Account>,
}

#[derive(Debug, Eq, PartialEq, Serialize, Deserialize)]
pub struct Account {
    pub id: u64,
    #[serde(rename = "screen-names")]
    pub screen_names: IndexMap<String, Option<Vec<NaiveDate>>>,
}

impl Account {
    pub fn from_raw_result(id: u64, result: HashMap<String, Vec<NaiveDate>>) -> Self {
        let mut sorted = result
            .into_iter()
            .map(|(screen_name, mut dates)| {
                dates.sort();

                let value = match dates.len() {
                    0 => None,
                    1 => Some(vec![dates[0]]),
                    n => Some(vec![dates[0], dates[n - 1]]),
                };

                (screen_name, value)
            })
            .collect::<IndexMap<_, _>>();

        sorted.sort_by(|screen_name_a, dates_a, screen_name_b, dates_b| {
            dates_a
                .as_ref()
                .and_then(|dates| dates.get(0))
                .cmp(&dates_b.as_ref().and_then(|dates| dates.get(0)))
                .then_with(|| screen_name_a.cmp(screen_name_b))
        });

        Self {
            id,
            screen_names: sorted,
        }
    }
}