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
|
use clap::Subcommand;
use eyre::{Result, WrapErr};
use atuin_client::{
database::Database,
encryption,
history::store::HistoryStore,
record::{sqlite_store::SqliteStore, store::Store, sync},
settings::Settings,
};
mod status;
use crate::command::client::account;
#[derive(Subcommand, Debug)]
#[command(infer_subcommands = true)]
pub enum Cmd {
/// Sync with the configured server
Sync {
/// Force re-download everything
#[arg(long, short)]
force: bool,
},
/// Login to the configured server
Login(account::login::Cmd),
/// Log out
Logout,
/// Register with the configured server
Register(account::register::Cmd),
/// Print the encryption key for transfer to another machine
Key {
/// Switch to base64 output of the key
#[arg(long)]
base64: bool,
},
/// Display the sync status
Status,
}
impl Cmd {
pub async fn run(
self,
settings: Settings,
db: &impl Database,
store: SqliteStore,
) -> Result<()> {
match self {
Self::Sync { force } => run(&settings, force, db, store).await,
Self::Login(l) => l.run(&settings, &store).await,
Self::Logout => account::logout::run(&settings),
Self::Register(r) => r.run(&settings).await,
Self::Status => status::run(&settings, db).await,
Self::Key { base64 } => {
use atuin_client::encryption::{encode_key, load_key};
let key = load_key(&settings).wrap_err("could not load encryption key")?;
if base64 {
let encode = encode_key(&key).wrap_err("could not encode encryption key")?;
println!("{encode}");
} else {
let mnemonic = bip39::Mnemonic::from_entropy(&key, bip39::Language::English)
.map_err(|_| eyre::eyre!("invalid key"))?;
println!("{mnemonic}");
}
Ok(())
}
}
}
}
async fn run(
settings: &Settings,
force: bool,
db: &impl Database,
store: SqliteStore,
) -> Result<()> {
if settings.sync.records {
let encryption_key: [u8; 32] = encryption::load_key(settings)
.context("could not load encryption key")?
.into();
let host_id = Settings::host_id().expect("failed to get host_id");
let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);
let (uploaded, downloaded) = sync::sync(settings, &store).await?;
crate::sync::build(settings, &store, db, Some(&downloaded)).await?;
println!("{uploaded}/{} up/down to record store", downloaded.len());
let history_length = db.history_count(true).await?;
let store_history_length = store.len_tag("history").await?;
#[allow(clippy::cast_sign_loss)]
if history_length as u64 > store_history_length {
println!(
"{history_length} in history index, but {store_history_length} in history store"
);
println!("Running automatic history store init...");
// Internally we use the global filter mode, so this context is ignored.
// don't recurse or loop here.
history_store.init_store(db).await?;
println!("Re-running sync due to new records locally");
// we'll want to run sync once more, as there will now be stuff to upload
let (uploaded, downloaded) = sync::sync(settings, &store).await?;
crate::sync::build(settings, &store, db, Some(&downloaded)).await?;
println!("{uploaded}/{} up/down to record store", downloaded.len());
}
} else {
atuin_client::sync::sync(settings, force, db).await?;
}
println!(
"Sync complete! {} items in history database, force: {}",
db.history_count(true).await?,
force
);
Ok(())
}
|
