nvidia-sdk 0.1.0

Verified acquisition and staging for NVIDIA graphics SDKs
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
112
113
114
115
116
117
118
119

#![forbid(unsafe_code)]

use {
    anyhow::Result,
    clap::{Parser, Subcommand},
    nvidia_sdk::{ResolveOptions, RuntimeProfile, Sdk},
    std::path::PathBuf,
};

#[derive(Parser)]
#[command(about = "Acquire and stage pinned NVIDIA SDK files")]
struct Args {
    #[command(subcommand)]
    command: Command,
}

#[derive(Subcommand)]
enum Command {
    /// Resolve an SDK and print its validated path.
    Path {
        sdk: Sdk,
        #[arg(long)]
        target: String,
        #[arg(long)]
        json: bool,
        #[arg(long)]
        offline: bool,
    },
    /// Resolve an SDK and stage its redistributable runtime files.
    Stage {
        sdk: Sdk,
        #[arg(long)]
        target: String,
        #[arg(long)]
        output: PathBuf,
        #[arg(long, default_value = "release")]
        profile: RuntimeProfile,
        #[arg(long)]
        json: bool,
        #[arg(long)]
        offline: bool,
    },
    /// Resolve and validate an SDK without staging files.
    Verify {
        sdk: Sdk,
        #[arg(long)]
        target: String,
        #[arg(long)]
        offline: bool,
    },
}

impl Command {
    fn run(self) -> Result<()> {
        match self {
            Command::Path {
                sdk,
                target,
                json,
                offline,
            } => {
                let resolved = sdk.resolve(&Self::options(target, offline))?;

                if json {
                    println!("{}", serde_json::to_string_pretty(&resolved)?);
                } else {
                    println!("{}", resolved.path.display());
                }
            }
            Command::Stage {
                sdk,
                target,
                output,
                profile,
                json,
                offline,
            } => {
                let resolved = sdk.resolve(&Self::options(target, offline))?;
                let files = resolved.stage_runtime(&output, profile)?;

                if json {
                    println!("{}", serde_json::to_string_pretty(&files)?);
                } else {
                    for file in files {
                        println!("{}", file.path.display());
                    }
                }
            }
            Command::Verify {
                sdk,
                target,
                offline,
            } => {
                let resolved = sdk.resolve(&Self::options(target, offline))?;

                println!(
                    "verified {} {} for {} at {}",
                    resolved.sdk.name(),
                    resolved.version,
                    resolved.target,
                    resolved.path.display()
                );
            }
        }

        Ok(())
    }

    fn options(target: String, offline: bool) -> ResolveOptions {
        let mut options = ResolveOptions::for_target(target);
        options.offline |= offline;

        options
    }
}

fn main() -> Result<()> {
    Args::parse().command.run()
}