indicate 0.2.0

Library behind cargo-indicate
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

//! Includes the tokens that correspond to the types and relationships
//! defined by [`SCHEMA`](crate::SCHEMA).

use std::{rc::Rc, sync::Arc};

use cargo_metadata::Package;
use octorust::types::{FullRepository, PublicUser};
use rustsec::{advisory::affected::FunctionPath, Advisory, VersionReq};
use trustfall::provider::TrustfallEnumVertex;

use crate::{
    code_stats::{LanguageBlob, LanguageCodeStats},
    geiger::{GeigerCategories, GeigerCount, GeigerUnsafety},
    NameVersion,
};

/// A node in the GraphQL schema as defined in the schema.
///
/// Each node wraps a reference to some type of actual data.
#[allow(dead_code)]
#[derive(Debug, Clone, TrustfallEnumVertex)]
pub enum Vertex {
    Package(Rc<Package>),
    CratesIoStats(NameVersion),

    #[trustfall(skip_conversion)]
    Webpage(String),

    #[trustfall(skip_conversion)]
    Repository(String),
    GitHubRepository(Arc<FullRepository>),
    GitHubUser(Arc<PublicUser>),
    Advisory(Rc<Advisory>),
    AffectedFunctionVersions((FunctionPath, Vec<VersionReq>)),
    // CvssBase(Rc<cvss::v3::base::Base>), // TODO: Add when Trustfall supports enums?

    // Geiger types implement `Copy` and does not to be inside an Rc
    GeigerUnsafety(GeigerUnsafety),
    GeigerCategories(GeigerCategories),
    GeigerCount(GeigerCount),

    LanguageCodeStats(Rc<LanguageCodeStats>),
    LanguageBlob(Rc<LanguageBlob>),
}

impl Vertex {
    pub fn as_webpage(&self) -> Option<&str> {
        match self {
            Vertex::Webpage(url) | Vertex::Repository(url) => {
                Some(url.as_ref())
            }
            Vertex::GitHubRepository(r) => Some(&r.html_url),
            _ => None,
        }
    }

    pub fn as_repository(&self) -> Option<&str> {
        match self {
            Vertex::Repository(url) => Some(url.as_ref()),
            Vertex::GitHubRepository(r) => Some(&r.html_url),
            _ => None,
        }
    }
}

impl From<Package> for Vertex {
    fn from(value: Package) -> Self {
        Self::Package(Rc::new(value))
    }
}

// #[cfg(test)]
// mod test {
//     /// Verify that all `Vertex` variants are types in
//     /// the schema, and that all types are nodes variants
//     #[test]
//     #[ignore = "not possible at this point"]
//     fn verify_nodes_in_schema() {
//         // TODO: Use trustfall_core::schema::Schema to
//         // access vertexes
//         todo!()
//     }
// }