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
use clap::{ Args, Parser, Subcommand }; use std::fmt::{ self }; use std::io::{ self }; use std::path::{ PathBuf }; use std::process::{ ExitCode }; // ------------------------------------------------------------------------------------------------- // Command line arguments // ------------------------------------------------------------------------------------------------- #[derive(Parser)] #[command(author, version, about, propagate_version = true)] struct Arguments { #[command(subcommand)] command: Command, } #[derive(Subcommand)] enum Command { /// Generates a new database file CreateDatabase(CreateDatabaseArguments), ShowDatabase(ShowDatabaseArguments), } #[derive(Args)] struct CreateDatabaseArguments { path: PathBuf, } #[derive(Args)] struct ShowDatabaseArguments { path: PathBuf, } impl Command { fn run(&self) -> Result<()> { match *self { Command::CreateDatabase(ref arguments) => { let _ = basalt::create_database(&arguments.path)?; Ok(()) }, Command::ShowDatabase(ref arguments) => { let db = basalt::open_database(&arguments.path)?; println!("Database: {}", db.path().display()); println!("Table count: {}", db.table_count()); 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() { Ok(_) => ExitCode::SUCCESS, Err(e) => { eprintln!("{}", e); ExitCode::FAILURE } } }