basalt_cli.rs
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
use clap::{ Args, Parser, Subcommand }; use std::fmt::{ self }; use std::io::{ self }; use std::path::{ Path, PathBuf }; use std::process::{ ExitCode }; // ------------------------------------------------------------------------------------------------- // Command line arguments // ------------------------------------------------------------------------------------------------- #[derive(Parser)] #[command(author, version, about, propagate_version = true)] struct Arguments { path: PathBuf, #[command(subcommand)] command: Command, } #[derive(Subcommand)] enum Command { /// Generates a new database file CreateDatabase, ShowDatabase, AddTable(AddTableArguments), } #[derive(Args)] struct AddTableArguments { name: String, #[arg(id="column", short, long)] columns: Vec<String>, } impl Command { fn run(&self, path: &Path) -> Result<()> { match *self { Command::CreateDatabase => { let _ = basalt::create_database(path)?; Ok(()) }, Command::ShowDatabase => { let db = basalt::open_database(path)?; println!("Database: {}", db.path().display()); Ok(()) }, Command::AddTable(ref arguments) => { let columns = arguments.columns .iter() .map(|column_definition| { let column = basalt::parse_column(column_definition)?; Ok(column) }) .collect::<Result<Vec<_>>>()?; let mut db = basalt::open_database(path)?; db.add_table(&arguments.name, &columns)?; db.synchronize()?; Ok(()) } } } } // ------------------------------------------------------------------------------------------------- // Error handling // ------------------------------------------------------------------------------------------------- type Result<T> = ::std::result::Result<T, Error>; #[derive(Debug)] enum Error { FromBasalt(basalt::Error), FromStdIo(io::Error), } impl fmt::Display for Error { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { use Error::*; match *self { FromBasalt(ref e) => write!(f, "Basalt error: {}", e), FromStdIo(ref e) => write!(f, "IO error: {}", e), } } } impl From<io::Error> for Error { fn from(e: io::Error) -> Error { Error::FromStdIo(e) } } impl From<basalt::Error> for Error { fn from(e: basalt::Error) -> Error { Error::FromBasalt(e) } } // ------------------------------------------------------------------------------------------------- // Entry point // ------------------------------------------------------------------------------------------------- fn main() -> ExitCode { let arguments = Arguments::parse(); match arguments.command.run(&arguments.path) { Ok(_) => ExitCode::SUCCESS, Err(e) => { eprintln!("{}", e); ExitCode::FAILURE } } }