pkgrank 0.2.5

Centrality analysis for dependency graphs and file-level import graphs
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

use clap::ValueEnum;
use petgraph::prelude::*;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, Hash, ValueEnum)]
#[serde(rename_all = "snake_case")]
pub(crate) enum Ecosystem {
    Cargo,
    Npm,
    Python,
    Go,
}

impl std::fmt::Display for Ecosystem {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Cargo => write!(f, "cargo"),
            Self::Npm => write!(f, "npm"),
            Self::Python => write!(f, "python"),
            Self::Go => write!(f, "go"),
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Hash)]
pub(crate) struct DepNode {
    pub name: String,
    pub version: Option<String>,
    pub ecosystem: Ecosystem,
}

pub(crate) type DepGraph = DiGraph<DepNode, f64>;
pub(crate) type DepNodeMap = HashMap<String, NodeIndex>;

/// Build a generic dependency graph from a list of packages and edges.
/// Edges are `(from_name, to_name)` where names match `DepNode.name`.
pub(crate) fn build_dep_graph(
    packages: &[DepNode],
    edges: &[(String, String)],
) -> (DepGraph, DepNodeMap) {
    let mut graph = DepGraph::new();
    let mut map = DepNodeMap::new();

    for pkg in packages {
        let idx = graph.add_node(pkg.clone());
        map.insert(pkg.name.clone(), idx);
    }

    for (from, to) in edges {
        if let (Some(&fi), Some(&ti)) = (map.get(from), map.get(to)) {
            graph.update_edge(fi, ti, 1.0);
        }
    }

    (graph, map)
}

/// FNV-1a 64-bit hash (deterministic, non-cryptographic).
pub(crate) fn fnv1a64(bytes: &[u8]) -> u64 {
    let mut h: u64 = 0xcbf29ce484222325;
    for &b in bytes {
        h ^= b as u64;
        h = h.wrapping_mul(0x100000001b3);
    }
    h
}