archival 0.14.0

The simplest CMS in existence
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
120
121
122

use clap::{arg, value_parser, ArgMatches, Command};
use std::{
    env,
    error::Error,
    fs,
    path::PathBuf,
    sync::{atomic::AtomicBool, Arc, LazyLock},
};
mod build;
mod compat;
mod format;
mod import;
mod login;
mod manifest;
mod objects;
mod prebuild;
#[cfg(feature = "proto")]
mod proto;
mod run;
#[cfg(feature = "json-schema")]
mod schemas;
mod upload;

pub enum ExitStatus {
    Error,
    Ok,
}
impl ExitStatus {
    pub fn code(&self) -> i32 {
        match self {
            ExitStatus::Error => 1,
            ExitStatus::Ok => 0,
        }
    }
}

#[derive(Default, Debug)]
pub struct CommandConfig {
    upload_prefix: bool,
    site_path: bool,
}
impl CommandConfig {
    pub fn archival_site() -> Self {
        CommandConfig {
            upload_prefix: true,
            site_path: true,
        }
    }
    pub fn no_build() -> Self {
        CommandConfig {
            upload_prefix: false,
            site_path: true,
        }
    }
    pub fn upload_only() -> Self {
        CommandConfig {
            upload_prefix: true,
            site_path: false,
        }
    }
}

static CWD: LazyLock<PathBuf> = std::sync::LazyLock::new(|| env::current_dir().unwrap());

pub trait BinaryCommand {
    fn name(&self) -> &str;
    fn cli(&self, cmd: Command) -> Command;
    fn handler(
        &self,
        args: &ArgMatches,
        quit: Arc<AtomicBool>,
    ) -> Result<ExitStatus, Box<dyn Error>>;
}

pub fn add_args(mut cmd: Command, config: CommandConfig) -> Command {
    if config.upload_prefix {
        cmd = cmd.arg(
                // NOTE: weird long form quoting due to https://github.com/clap-rs/clap/issues/3586
                arg!(-u --"upload-prefix" <upload_prefix> "override the upload prefix. If no manifest.toml is present, this is required.")
                    .value_parser(value_parser!(String)),
            );
    }
    if config.site_path {
        cmd = cmd.arg(
                arg!([site_path] "an optional path to the archival site. Otherwise will be auto-detected from cwd.")
                    .required(false)
                    .default_value(CWD.to_str())
                    .value_parser(value_parser!(PathBuf)),
            );
    }
    cmd
}

pub fn command_root(args: &ArgMatches) -> PathBuf {
    let path = match args.try_get_one::<PathBuf>("site_path") {
        Ok(arg) => arg.unwrap(), // defaulted so unwrap is ok
        Err(_) => &CWD.to_path_buf(),
    };
    if *path != *CWD {
        // If not default, the path is relative
        fs::canonicalize(CWD.join(path)).unwrap()
    } else {
        CWD.to_path_buf()
    }
}

pub const COMMANDS: &[&'static dyn BinaryCommand] = &[
    &build::Command {},
    &run::Command {},
    &format::Command {},
    &manifest::Command {},
    &prebuild::Command {},
    &login::Command {},
    &compat::Command {},
    &upload::Command {},
    &import::Command {},
    &objects::Command {},
    #[cfg(feature = "json-schema")]
    &schemas::Command {},
    #[cfg(feature = "proto")]
    &proto::Command {},
];