detect/
triage_shim.rs

1// Copyright 2020 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5// Interface to the Triage library.
6
7use crate::diagnostics::Selectors;
8use anyhow::Error;
9use fuchsia_triage::{ActionTagDirective, ParseResult, SnapshotTrigger};
10use std::collections::HashMap;
11
12pub fn evaluate_int_math(expression: &str) -> Result<i64, Error> {
13    fuchsia_triage::evaluate_int_math(expression)
14}
15
16type ConfigFiles = HashMap<String, String>;
17
18type DiagnosticData = fuchsia_triage::DiagnosticData;
19
20pub struct TriageLib {
21    triage_config: fuchsia_triage::ParseResult,
22}
23
24impl TriageLib {
25    pub fn new(configs: ConfigFiles) -> Result<TriageLib, Error> {
26        let triage_config = ParseResult::new(&configs, &ActionTagDirective::AllowAll)?;
27        Ok(TriageLib { triage_config })
28    }
29
30    pub fn selectors(&self) -> Selectors {
31        Selectors::new().with_inspect_selectors(fuchsia_triage::all_selectors(&self.triage_config))
32    }
33
34    pub fn evaluate(
35        &self,
36        data: Vec<DiagnosticData>,
37    ) -> (Vec<SnapshotTrigger>, fuchsia_triage::WarningVec) {
38        fuchsia_triage::snapshots(&data, &self.triage_config)
39    }
40}
41
42#[cfg(test)]
43mod test {
44    use super::*;
45    use maplit::hashmap;
46
47    const CONFIG: &str = r#"{
48            select: {
49                foo: "INSPECT:foo.cm:path/to:leaf",
50            },
51            act: {
52                yes: {
53                    type: "Snapshot",
54                    trigger: "foo==8",
55                    repeat: "Micros(42)",
56                    signature: "got-it",
57                },
58            },
59        }"#;
60
61    const INSPECT: &str = r#"[
62        {
63            "moniker": "foo.cm",
64            "metadata": {},
65            "payload": {"path": {"to": {"leaf": 8}}}
66        }
67    ]"#;
68
69    #[fuchsia::test]
70    fn library_calls_work() -> Result<(), Error> {
71        let configs = hashmap! { "foo.triage".to_string() => CONFIG.to_string() };
72        let lib = TriageLib::new(configs)?;
73        let data = vec![DiagnosticData::new(
74            "inspect.json".to_string(),
75            fuchsia_triage::Source::Inspect,
76            INSPECT.to_string(),
77        )?];
78        let expected_trigger =
79            vec![SnapshotTrigger { signature: "got-it".to_string(), interval: 42_000 }];
80
81        assert_eq!(
82            lib.selectors().inspect_selectors,
83            vec!["INSPECT:foo.cm:path/to:leaf".to_string()]
84        );
85        assert_eq!(lib.evaluate(data), (expected_trigger, vec![]));
86        Ok(())
87    }
88}
detect/triage_shim.rs

detect/
triage_shim.rs
