weavatrix-rust 2.1.1

Protocol-independent Rust repository intelligence: typed evidence graphs for impact, architecture, APIs, Git, search, and memory
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

use super::{
    Call, Certainty, Detection, Entity, ProviderHints, Role, Transport, bindings_none,
    non_empty_resources, positional_values, property, resource_values,
};

pub(super) fn detect_kafka(
    call: &Call<'_, '_>,
    name: &str,
    chain: &str,
    hints: ProviderHints,
) -> Option<Detection> {
    let explicit = [
        "producerrecord",
        "futurerecord",
        "baserecord",
        "consumepartition",
        "subscribetopics",
        "addconsumetopics",
        "kafkaconsumer",
        "newwriter",
        "newreader",
    ]
    .iter()
    .any(|marker| chain.contains(marker));
    if !hints.kafka && !explicit {
        return None;
    }
    let group = property(call, &["groupid", "group_id", "group"]);
    let (role, entity, resources) = match name {
        "send" | "produce" if property(call, &["topic"]).is_some() || hints.kafka => (
            Role::Producer,
            Entity::Topic,
            resource_values(call, &["topic"], 0, bindings_none()),
        ),
        "to" if chain.contains("record") => (
            Role::Producer,
            Entity::Topic,
            positional_values(call, 0, bindings_none()),
        ),
        "producerrecord" => (
            Role::Producer,
            Entity::Topic,
            positional_values(call, 0, bindings_none()),
        ),
        "subscribe" | "subscribetopics" | "addconsumetopics" | "kafkalistener" => (
            Role::Consumer,
            Entity::Topic,
            resource_values(call, &["topic", "topics"], 0, bindings_none()),
        ),
        "kafkaconsumer" | "consumepartition" => (
            Role::Consumer,
            Entity::Topic,
            positional_values(call, 0, bindings_none()),
        ),
        "newwriter" => (
            Role::Producer,
            Entity::Topic,
            resource_values(call, &["topic"], 0, bindings_none()),
        ),
        "newreader" => (
            Role::Consumer,
            Entity::Topic,
            resource_values(call, &["topic"], 0, bindings_none()),
        ),
        "consumer" if property(call, &["groupid", "group_id"]).is_some() => {
            return Some(Detection {
                transport: Transport::Kafka,
                entity: Entity::Topic,
                role: Role::Declare,
                resources: Vec::new(),
                exchange: None,
                routing_key: None,
                consumer_group: group,
                certainty: Certainty::Exact,
                uncertainty: None,
            });
        }
        _ => return None,
    };
    Some(Detection {
        transport: Transport::Kafka,
        entity,
        role,
        resources: non_empty_resources(resources),
        exchange: None,
        routing_key: None,
        consumer_group: group,
        certainty: Certainty::Exact,
        uncertainty: None,
    })
}