canic-cli 0.35.11

Operator CLI for Canic fleet backup and restore workflows
Documentation 
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use crate::metrics::model::{MetricEntry, MetricValue};
use canic_host::response_parse::{
    RECORD_MARKER, candid_record_blocks, find_field, parse_json_u64, parse_json_u128,
    parse_u64_digits, parse_u128_digits, quoted_strings, text_after,
};

pub fn parse_metrics_page(output: &str) -> Option<Vec<MetricEntry>> {
    let value = serde_json::from_str::<serde_json::Value>(output).ok()?;
    if let Some(entries) = parse_metrics_page_json(&value) {
        return Some(entries);
    }
    find_field(&value, "response_candid")
        .and_then(serde_json::Value::as_str)
        .and_then(parse_metrics_page_text)
}

fn parse_metrics_page_json(value: &serde_json::Value) -> Option<Vec<MetricEntry>> {
    Some(
        find_field(value, "entries")?
            .as_array()?
            .iter()
            .filter_map(parse_metric_entry_json)
            .collect::<Vec<_>>(),
    )
}

fn parse_metric_entry_json(value: &serde_json::Value) -> Option<MetricEntry> {
    Some(MetricEntry {
        labels: find_field(value, "labels")?
            .as_array()?
            .iter()
            .filter_map(|value| value.as_str().map(str::to_string))
            .collect(),
        principal: find_field(value, "principal").and_then(parse_principal_json),
        value: find_field(value, "value").and_then(parse_metric_value_json)?,
    })
}

fn parse_metrics_page_text(output: &str) -> Option<Vec<MetricEntry>> {
    let mut entries = Vec::new();
    for chunk in candid_record_blocks(output) {
        if !(chunk[RECORD_MARKER.len()..]
            .trim_start()
            .starts_with("\"principal\" =")
            && chunk.contains("labels = vec")
            && chunk.contains("value = variant"))
        {
            continue;
        }
        entries.push(MetricEntry {
            labels: parse_candid_labels(chunk)?,
            principal: parse_candid_principal(chunk),
            value: parse_candid_metric_value(chunk)?,
        });
    }
    Some(entries)
}

fn parse_candid_labels(chunk: &str) -> Option<Vec<String>> {
    let (_, after_field) = chunk.split_once("labels = vec")?;
    let (_, after_open) = after_field.split_once('{')?;
    let (labels, _) = after_open.split_once("};")?;
    Some(quoted_strings(labels))
}

fn parse_candid_principal(chunk: &str) -> Option<String> {
    let (_, after_field) = chunk.split_once("\"principal\" =")?;
    let value = after_field.trim_start();
    if value.starts_with("null") {
        return None;
    }
    quoted_strings(value).into_iter().next()
}

fn parse_candid_metric_value(chunk: &str) -> Option<MetricValue> {
    if let Some(value) = text_after(chunk, "Count =").and_then(parse_u64_digits) {
        return Some(MetricValue::Count { count: value });
    }
    if let Some(value) = text_after(chunk, "U128 =").and_then(parse_u128_digits) {
        return Some(MetricValue::U128 { value });
    }
    if chunk.contains("CountAndU64") {
        let count = text_after(chunk, "count =").and_then(parse_u64_digits)?;
        let value_u64 = text_after(chunk, "value_u64 =").and_then(parse_u64_digits)?;
        return Some(MetricValue::CountAndU64 { count, value_u64 });
    }
    None
}

fn parse_principal_json(value: &serde_json::Value) -> Option<String> {
    match value {
        serde_json::Value::String(value) => Some(value.clone()),
        serde_json::Value::Object(map) => map.values().find_map(parse_principal_json),
        serde_json::Value::Array(values) => values.iter().find_map(parse_principal_json),
        _ => None,
    }
}

fn parse_metric_value_json(value: &serde_json::Value) -> Option<MetricValue> {
    match value {
        serde_json::Value::Object(map) => {
            if let Some(value) = map.get("Count").and_then(parse_json_u64) {
                return Some(MetricValue::Count { count: value });
            }
            if let Some(value) = map.get("U128").and_then(parse_json_u128) {
                return Some(MetricValue::U128 { value });
            }
            if let Some(value) = map.get("CountAndU64") {
                let count = find_field(value, "count").and_then(parse_json_u64)?;
                let value_u64 = find_field(value, "value_u64").and_then(parse_json_u64)?;
                return Some(MetricValue::CountAndU64 { count, value_u64 });
            }
            if let (Some(count), Some(value_u64)) = (
                map.get("count").and_then(parse_json_u64),
                map.get("value_u64").and_then(parse_json_u64),
            ) {
                return Some(MetricValue::CountAndU64 { count, value_u64 });
            }
            map.values().find_map(parse_metric_value_json)
        }
        serde_json::Value::Array(values) => values.iter().find_map(parse_metric_value_json),
        _ => None,
    }
}