mirror of
https://github.com/ankitects/anki.git
synced 2026-01-16 23:39:01 -05:00
7216032e82
* Prepare to switch Rust import style * Run nightly format Closes #2320 * Clean up a few imports * Enable comment wrapping * Wrap comments
63 lines
2.3 KiB
Rust
63 lines
2.3 KiB
Rust
// Copyright: Ankitects Pty Ltd and contributors
|
|
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
use crate::collection::CollectionBuilder;
|
|
use crate::io::atomic_rename;
|
|
use crate::io::new_tempfile_in_parent_of;
|
|
use crate::io::read_file;
|
|
use crate::io::write_file;
|
|
use crate::prelude::*;
|
|
use crate::storage::SchemaVersion;
|
|
use crate::sync::collection::progress::FullSyncProgressFn;
|
|
use crate::sync::collection::protocol::EmptyInput;
|
|
use crate::sync::collection::protocol::SyncProtocol;
|
|
use crate::sync::error::HttpResult;
|
|
use crate::sync::error::OrHttpErr;
|
|
use crate::sync::http_client::HttpSyncClient;
|
|
use crate::sync::login::SyncAuth;
|
|
|
|
impl Collection {
|
|
/// Download collection from AnkiWeb. Caller must re-open afterwards.
|
|
pub async fn full_download(
|
|
self,
|
|
auth: SyncAuth,
|
|
progress_fn: FullSyncProgressFn,
|
|
) -> Result<()> {
|
|
let mut server = HttpSyncClient::new(auth);
|
|
server.set_full_sync_progress_fn(Some(progress_fn));
|
|
self.full_download_with_server(server).await
|
|
}
|
|
|
|
pub(crate) async fn full_download_with_server(self, server: HttpSyncClient) -> Result<()> {
|
|
let col_path = self.col_path.clone();
|
|
let _col_folder = col_path.parent().or_invalid("couldn't get col_folder")?;
|
|
self.close(None)?;
|
|
let out_data = server.download(EmptyInput::request()).await?.data;
|
|
// check file ok
|
|
let temp_file = new_tempfile_in_parent_of(&col_path)?;
|
|
write_file(temp_file.path(), out_data)?;
|
|
let col = CollectionBuilder::new(temp_file.path())
|
|
.set_check_integrity(true)
|
|
.build()?;
|
|
col.storage.db.execute_batch("update col set ls=mod")?;
|
|
col.close(None)?;
|
|
atomic_rename(temp_file, &col_path, true)?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
pub fn server_download(
|
|
col: &mut Option<Collection>,
|
|
schema_version: SchemaVersion,
|
|
) -> HttpResult<Vec<u8>> {
|
|
let col_path = {
|
|
let mut col = col.take().or_internal_err("take col")?;
|
|
let path = col.col_path.clone();
|
|
col.transact_no_undo(|col| col.storage.increment_usn())
|
|
.or_internal_err("incr usn")?;
|
|
col.close(Some(schema_version)).or_internal_err("close")?;
|
|
path
|
|
};
|
|
let data = read_file(col_path).or_internal_err("read col")?;
|
|
Ok(data)
|
|
}
|