nuts_tool/cli/archive/add/
file.rs

1// MIT License
2//
3// Copyright (c) 2023,2024 Robin Doer
4//
5// Permission is hereby granted, free of charge, to any person obtaining a copy
6// of this software and associated documentation files (the "Software"), to
7// deal in the Software without restriction, including without limitation the
8// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
9// sell copies of the Software, and to permit persons to whom the Software is
10// furnished to do so, subject to the following conditions:
11//
12// The above copyright notice and this permission notice shall be included in
13// all copies or substantial portions of the Software.
14//
15// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21// IN THE SOFTWARE.
22
23use anyhow::Result;
24use clap::{ArgAction, Args};
25use log::debug;
26use std::io::{self, Read};
27
28use crate::cli::archive::add::{TimestampArgs, TSTAMP_HELP};
29use crate::cli::archive::open_archive;
30
31#[derive(Args, Debug)]
32#[clap(after_help(TSTAMP_HELP))]
33pub struct ArchiveAddFileArgs {
34    /// Name of the file.
35    name: String,
36
37    #[clap(flatten)]
38    timestamps: TimestampArgs,
39
40    /// Starts the migration when the container/archive is opened
41    #[clap(long, action = ArgAction::SetTrue)]
42    pub migrate: bool,
43
44    /// Specifies the name of the container
45    #[clap(short, long, env = "NUTS_CONTAINER")]
46    container: String,
47}
48
49impl ArchiveAddFileArgs {
50    pub fn run(&self) -> Result<()> {
51        debug!("args: {:?}", self);
52
53        let mut archive = open_archive(&self.container, self.migrate)?;
54        let block_size = archive.as_ref().block_size() as usize;
55        let mut builder = archive.append_file(&self.name);
56
57        if let Some(created) = self.timestamps.created {
58            builder.set_created(created);
59        }
60
61        if let Some(changed) = self.timestamps.changed {
62            builder.set_changed(changed);
63        }
64
65        if let Some(modified) = self.timestamps.modified {
66            builder.set_modified(modified);
67        }
68
69        let mut entry = builder.build()?;
70        let mut buf = vec![0; block_size];
71
72        loop {
73            let n = io::stdin().read(&mut buf)?;
74            debug!("{} bytes read from stdin", n);
75
76            if n > 0 {
77                entry.write_all(&buf[..n])?;
78            } else {
79                break;
80            }
81        }
82
83        Ok(())
84    }
85}
nuts_tool/cli/archive/add/file.rs

nuts_tool/cli/archive/add/
file.rs
