kaniop-data-mover 0.16.2

S3-compatible data mover for Kanidm backup payload and manifest transfer
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

mod checksum;
mod commands;
mod crypto;
mod s3;

use std::process::ExitCode;

use clap::{Parser, Subcommand};
use tracing_subscriber::EnvFilter;

#[derive(Parser)]
#[command(name = "kaniop-data-mover", about = "S3 data mover for Kanidm backups")]
struct Cli {
    #[arg(long, default_value = "info", env = "DATA_MOVER_LOG_FILTER")]
    log_filter: String,

    #[arg(long, default_value = "json", env)]
    log_format: String,

    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    Upload {
        #[arg(
            long,
            env = "OPERATION_DOC_PATH",
            default_value = "/run/kaniop/operation.json"
        )]
        operation_doc: String,
    },
    Download {
        #[arg(
            long,
            env = "OPERATION_DOC_PATH",
            default_value = "/run/kaniop/operation.json"
        )]
        operation_doc: String,
    },
    DeletePlan {
        #[arg(
            long,
            env = "OPERATION_DOC_PATH",
            default_value = "/run/kaniop/operation.json"
        )]
        operation_doc: String,
    },
    Discover {
        #[arg(
            long,
            env = "OPERATION_DOC_PATH",
            default_value = "/run/kaniop/operation.json"
        )]
        operation_doc: String,
    },
    Check {
        #[arg(
            long,
            env = "OPERATION_DOC_PATH",
            default_value = "/run/kaniop/operation.json"
        )]
        operation_doc: String,
    },
    Transport {
        #[arg(
            long,
            env = "OPERATION_DOC_PATH",
            default_value = "/run/kaniop/operation.json"
        )]
        operation_doc: String,
    },
}

#[tokio::main]
async fn main() -> ExitCode {
    let cli = Cli::parse();

    let filter = EnvFilter::try_new(&cli.log_filter).unwrap_or_else(|_| EnvFilter::new("info"));
    match cli.log_format.as_str() {
        "json" => {
            tracing_subscriber::fmt()
                .json()
                .with_env_filter(filter)
                .init();
        }
        _ => {
            tracing_subscriber::fmt().with_env_filter(filter).init();
        }
    }

    let exit_code = match cli.command {
        Commands::Upload { operation_doc } => commands::upload::run(&operation_doc).await,
        Commands::Download { operation_doc } => commands::download::run(&operation_doc).await,
        Commands::DeletePlan { operation_doc } => commands::delete_plan::run(&operation_doc).await,
        Commands::Discover { operation_doc } => commands::discover::run(&operation_doc).await,
        Commands::Check { operation_doc } => commands::check::run(&operation_doc).await,
        Commands::Transport { operation_doc } => commands::transport::run(&operation_doc).await,
    };

    match exit_code {
        Ok(_) => ExitCode::SUCCESS,
        Err(code) => ExitCode::from(code as u8),
    }
}