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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
use std::path::PathBuf;
use eyre::Result;
use structopt::StructOpt;
use atuin_client::database::Sqlite;
use atuin_client::settings::Settings as ClientSettings;
use atuin_common::utils::uuid_v4;
use atuin_server::settings::Settings as ServerSettings;
mod event;
mod history;
mod import;
mod init;
mod login;
mod register;
mod search;
mod server;
mod stats;
mod sync;
#[derive(StructOpt)]
pub enum AtuinCmd {
#[structopt(
about="manipulate shell history",
aliases=&["h", "hi", "his", "hist", "histo", "histor"],
)]
History(history::Cmd),
#[structopt(about = "import shell history from file")]
Import(import::Cmd),
#[structopt(about = "start an atuin server")]
Server(server::Cmd),
#[structopt(about = "calculate statistics for your history")]
Stats(stats::Cmd),
#[structopt(about = "output shell setup")]
Init(init::Cmd),
#[structopt(about = "generates a UUID")]
Uuid,
#[structopt(about = "interactive history search")]
Search {
#[structopt(long, short, about = "filter search result by directory")]
cwd: Option<String>,
#[structopt(long = "exclude-cwd", about = "exclude directory from results")]
exclude_cwd: Option<String>,
#[structopt(long, short, about = "filter search result by exit code")]
exit: Option<i64>,
#[structopt(long = "exclude-exit", about = "exclude results with this exit code")]
exclude_exit: Option<i64>,
#[structopt(long, short, about = "only include results added before this date")]
before: Option<String>,
#[structopt(long, about = "only include results after this date")]
after: Option<String>,
#[structopt(long, short, about = "open interactive search UI")]
interactive: bool,
#[structopt(long, short, about = "use human-readable formatting for time")]
human: bool,
query: Vec<String>,
#[structopt(long, about = "Show only the text of the command")]
cmd_only: bool,
},
#[structopt(about = "sync with the configured server")]
Sync {
#[structopt(long, short, about = "force re-download everything")]
force: bool,
},
#[structopt(about = "login to the configured server")]
Login(login::Cmd),
#[structopt(about = "register with the configured server")]
Register(register::Cmd),
#[structopt(about = "print the encryption key for transfer to another machine")]
Key,
}
impl AtuinCmd {
pub async fn run(self) -> Result<()> {
let client_settings = ClientSettings::new()?;
let server_settings = ServerSettings::new()?;
let db_path = PathBuf::from(client_settings.db_path.as_str());
let mut db = Sqlite::new(db_path).await?;
match self {
Self::History(history) => history.run(&client_settings, &mut db).await,
Self::Import(import) => import.run(&mut db).await,
Self::Server(server) => server.run(&server_settings).await,
Self::Stats(stats) => stats.run(&mut db, &client_settings).await,
Self::Init(init) => init.run(),
Self::Search {
cwd,
exit,
interactive,
human,
exclude_exit,
exclude_cwd,
before,
after,
query,
cmd_only,
} => {
search::run(
&client_settings,
cwd,
exit,
interactive,
human,
exclude_exit,
exclude_cwd,
before,
after,
cmd_only,
&query,
&mut db,
)
.await
}
Self::Sync { force } => sync::run(&client_settings, force, &mut db).await,
Self::Login(l) => l.run(&client_settings),
Self::Register(r) => register::run(
&client_settings,
r.username.as_str(),
r.email.as_str(),
r.password.as_str(),
),
Self::Key => {
let key = std::fs::read(client_settings.key_path.as_str())?;
println!("{}", base64::encode(key));
Ok(())
}
Self::Uuid => {
println!("{}", uuid_v4());
Ok(())
}
}
}
}
|
