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
120
121
122
123

// MIT License
//
// Copyright (c) 2023 Robin Doer
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.

use anyhow::Result;
use clap::{ArgAction, Args};
use log::debug;
use nuts_archive::{Archive, Entry, Mode};
use nuts_directory::DirectoryBackend;
use std::ops::Deref;
use std::path::PathBuf;

use crate::cli::open_container;

struct ListEntry {
    name: String,
    size: u64,
    mode: Mode,
}

impl<'a> From<&'a Entry<'a, DirectoryBackend<PathBuf>>> for ListEntry {
    fn from(entry: &Entry<DirectoryBackend<PathBuf>>) -> Self {
        ListEntry {
            name: entry.name().to_string(),
            size: entry.size(),
            mode: *entry.deref(),
        }
    }
}

fn collect_entries<'a>(
    archive: &'a mut Archive<DirectoryBackend<PathBuf>>,
) -> Result<Vec<ListEntry>> {
    let mut vec = vec![];
    let mut entry_opt = archive.first();

    loop {
        match entry_opt {
            Some(Ok(entry)) => {
                vec.push((&entry).into());
                entry_opt = entry.next();
            }
            Some(Err(err)) => return Err(err.into()),
            None => break,
        }
    }

    Ok(vec)
}

fn print_short(entries: Vec<ListEntry>) {
    for entry in entries {
        println!("{}", entry.name);
    }
}

fn print_long(entries: Vec<ListEntry>) {
    let max_size = entries.iter().max_by_key(|e| e.size).map_or(0, |e| e.size);
    let max_n = max_size.checked_ilog10().unwrap_or(0) as usize + 1;

    for entry in entries {
        let type_char = if entry.mode.is_file() {
            '-'
        } else if entry.mode.is_directory() {
            'd'
        } else if entry.mode.is_symlink() {
            'l'
        } else {
            '?'
        };

        println!("{} {:>max_n$} {}", type_char, entry.size, entry.name);
    }
}

#[derive(Args, Debug)]
pub struct ArchiveListArgs {
    /// Lists the content in the long format
    #[clap(short, long, action = ArgAction::SetTrue)]
    long: bool,

    /// Specifies the name of the container
    #[clap(short, long, env = "NUTS_CONTAINER")]
    container: String,
}

impl ArchiveListArgs {
    pub fn run(&self) -> Result<()> {
        debug!("container: {}", self.container);
        debug!("long: {}", self.long);

        let container = open_container(&self.container)?;
        let mut archive = Archive::open(container)?;

        let entries = collect_entries(&mut archive)?;

        if self.long {
            print_long(entries);
        } else {
            print_short(entries);
        }

        Ok(())
    }
}