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
|
use clap::Subcommand;
use eyre::{Result, WrapErr};
use crate::atuin_client::{
database::Database,
encryption,
history::store::HistoryStore,
record::{sqlite_store::SqliteStore, store::Store, sync},
settings::Settings,
};
mod status;
#[derive(Subcommand, Debug)]
#[command(infer_subcommands = true)]
pub(crate) enum Cmd {
/// Sync with the configured server
Sync {
/// Force re-download everything
#[arg(long, short)]
force: bool,
},
/// Print the encryption key for transfer to another machine
Key {},
/// Display the sync status
Status,
}
impl Cmd {
pub(crate) async fn run(
self,
settings: Settings,
db: &impl Database,
store: SqliteStore,
) -> Result<()> {
match self {
Self::Sync { force } => run(&settings, force, db, store).await,
Self::Status => status::run(&settings).await,
Self::Key {} => {
use crate::atuin_client::encryption::{encode_key, load_key};
let key = load_key(&settings).wrap_err("could not load encryption key")?;
let encode = encode_key(&key).wrap_err("could not encode encryption key")?;
println!("{encode}");
Ok(())
}
}
}
}
async fn run(
settings: &Settings,
force: bool,
db: &impl Database,
store: SqliteStore,
) -> Result<()> {
let encryption_key: [u8; 32] = encryption::load_key(settings)
.context("could not load encryption key")?
.into();
let host_id = Settings::host_id().await?;
let history_store = HistoryStore::new(store.clone(), host_id, encryption_key);
let (uploaded, downloaded) = sync::sync(settings, &store, &encryption_key)
.await
.map_err(crate::print_error::format_sync_error)?;
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?;
#[expect(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, &encryption_key)
.await
.map_err(crate::print_error::format_sync_error)?;
crate::sync::build(settings, &store, db, Some(&downloaded)).await?;
println!("{uploaded}/{} up/down to record store", downloaded.len());
}
println!(
"Sync complete! {} items in history database, force: {}",
db.history_count(true).await?,
force
);
Ok(())
}
|
