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
|
use crate::{
cmd_output,
cmds::{
arg::storage_write::JVStorageWriteArgument, collect::empty::JVEmptyCollect,
r#in::storage_rw::JVStorageRWInput, out::none::JVNoneOutput,
},
systems::cmd::{
cmd_system::JVCommandContext,
errors::{CmdExecuteError, CmdPrepareError},
},
};
use cli_utils::display::md;
use cmd_system_macros::exec;
use just_enough_vcs::system::{
constants::vault::values::vault_value_index_file_suffix,
storage_system::{
error::StorageIOError,
store::{ChunkingPolicy, StorageConfig, write_file},
},
};
use rust_i18n::t;
use std::any::TypeId;
pub struct JVStorageWriteCommand;
type Cmd = JVStorageWriteCommand;
type Arg = JVStorageWriteArgument;
type In = JVStorageRWInput;
type Collect = JVEmptyCollect;
fn help_str() -> String {
todo!()
}
async fn prepare(args: &Arg, _ctx: &JVCommandContext) -> Result<In, CmdPrepareError> {
let output_path = match &args.output_index {
Some(v) => v.clone(),
None => args
.file
.clone()
.with_extension(vault_value_index_file_suffix()),
};
// Default to using kb as the unit
let scale = if args.gb {
1024 * 1024 * 1024
} else if args.mb {
1024 * 1024
} else if args.b {
1
} else {
1024
};
let (input, storage, output) = just_enough_vcs::system::storage_system::store::precheck(
args.file.clone(),
args.storage.clone(),
output_path,
)
.await?;
let chunking_policy: ChunkingPolicy = if args.cdc_chunking > 0 {
ChunkingPolicy::Cdc(args.cdc_chunking * scale)
} else if args.fixed_chunking > 0 {
ChunkingPolicy::FixedSize(args.fixed_chunking * scale)
} else if args.line_chunking {
ChunkingPolicy::Line
} else {
return Err(CmdPrepareError::Error(md(t!(
"storage_write.unknown_chunking_policy"
))));
};
Ok(JVStorageRWInput {
input,
storage,
output,
chunking_policy: Some(chunking_policy),
})
}
async fn collect(_args: &Arg, _ctx: &JVCommandContext) -> Result<Collect, CmdPrepareError> {
Ok(JVEmptyCollect {})
}
#[exec]
async fn exec(
input: In,
_collect: Collect,
) -> Result<(Box<dyn std::any::Any + Send + 'static>, TypeId), CmdExecuteError> {
// There is no chance to return None in the Prepare phase, so unwrap is safe here
let chunking_policy = input.chunking_policy.unwrap();
write_file(
input.input,
input.storage,
input.output,
&StorageConfig { chunking_policy },
)
.await
.map_err(|e| match e {
StorageIOError::IOErr(error) => CmdExecuteError::Io(error),
StorageIOError::HashTooShort => {
CmdExecuteError::Error(md(t!("storage_write.hash_too_short")).to_string())
}
})?;
cmd_output!(JVNoneOutput => JVNoneOutput {})
}
crate::command_template!();
|
