muntjac 0.1.0

Translate uv.lock into Buck2 build rules
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

//! Lockfile domain types.
//!
//! Mirrors the shape of `uv.lock` but typed and normalized. The parser
//! (`src/lock/parser.rs`) converts the raw TOML into these.

use pep440_rs::Version;
use pep508_rs::{MarkerTree, PackageName};
use url::Url;

// ---- Types ----

#[derive(Debug, Clone)]
pub struct Package {
    pub name: PackageName,
    pub version: Version,
    pub source: Source,
    pub dependencies: Vec<DepEdge>,
    pub sdist: Option<Sdist>,
    pub wheels: Vec<Wheel>,
    pub metadata: Option<Metadata>,
}

#[derive(Debug, Clone)]
pub enum Source {
    Registry {
        url: Url,
    },
    Git {
        url: Url,
        rev: String,
        subdirectory: Option<String>,
    },
    FirstParty {
        kind: FirstPartyKind,
        path: String,
    },
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum FirstPartyKind {
    Virtual,
    Editable,
    Directory,
    Path,
}

#[derive(Debug, Clone)]
pub struct DepEdge {
    pub name: PackageName,
    pub extra: Vec<String>,
    pub marker: Option<MarkerTree>,
}

#[derive(Debug, Clone)]
pub struct Sdist {
    pub url: Url,
    pub hash: String,
    pub size: Option<u64>,
}

#[derive(Debug, Clone)]
pub struct Wheel {
    pub url: Url,
    pub hash: String,
    pub size: Option<u64>,
    pub filename: String,
}

#[derive(Debug, Clone)]
pub struct Metadata {
    pub requires_dist: Vec<DepEdge>,
    pub provides_extras: Vec<String>,
}

#[derive(Debug, Clone)]
pub struct Lockfile {
    pub version: u32,
    pub revision: u32,
    pub requires_python: String,
    pub packages: Vec<Package>,
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::str::FromStr;

    #[test]
    fn first_party_kind_round_trips() {
        let k = FirstPartyKind::Editable;
        assert_eq!(k, FirstPartyKind::Editable);
        assert_ne!(k, FirstPartyKind::Virtual);
    }

    #[test]
    fn package_constructs() {
        let pkg = Package {
            name: PackageName::from_str("numpy").unwrap(),
            version: Version::from_str("2.4.6").unwrap(),
            source: Source::Registry {
                url: Url::parse("https://pypi.org/simple").unwrap(),
            },
            dependencies: vec![],
            sdist: None,
            wheels: vec![],
            metadata: None,
        };
        assert_eq!(pkg.name.as_ref(), "numpy");
        assert_eq!(pkg.version.to_string(), "2.4.6");
    }
}