mirror of
https://github.com/ankitects/anki.git
synced 2025-12-01 00:47:12 -05:00
a83c4a7da7
Due to the orphan rule, this meant removing our usages of impl ProtoStruct, or converting them to a trait when they were used commonly. rslib now directly references anki_proto and anki_i18n, instead of 'pub use'-ing them, and we can put the generated files back in OUT_DIR.
101 lines
2.9 KiB
Rust
101 lines
2.9 KiB
Rust
// Copyright: Ankitects Pty Ltd and contributors
|
|
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
use async_trait::async_trait;
|
|
use reqwest::Url;
|
|
use serde::de::DeserializeOwned;
|
|
use serde::Deserialize;
|
|
use serde::Serialize;
|
|
use strum::IntoStaticStr;
|
|
|
|
use crate::error;
|
|
use crate::error::AnkiError;
|
|
use crate::sync::collection::protocol::AsSyncEndpoint;
|
|
use crate::sync::error::HttpResult;
|
|
use crate::sync::media::begin::SyncBeginRequest;
|
|
use crate::sync::media::begin::SyncBeginResponse;
|
|
use crate::sync::media::changes::MediaChangesRequest;
|
|
use crate::sync::media::changes::MediaChangesResponse;
|
|
use crate::sync::media::download::DownloadFilesRequest;
|
|
use crate::sync::media::sanity::MediaSanityCheckResponse;
|
|
use crate::sync::media::sanity::SanityCheckRequest;
|
|
use crate::sync::media::upload::MediaUploadResponse;
|
|
use crate::sync::request::SyncRequest;
|
|
use crate::sync::response::SyncResponse;
|
|
|
|
#[derive(IntoStaticStr, Deserialize, PartialEq, Eq, Debug)]
|
|
#[serde(rename_all = "camelCase")]
|
|
#[strum(serialize_all = "camelCase")]
|
|
pub enum MediaSyncMethod {
|
|
Begin,
|
|
MediaChanges,
|
|
UploadChanges,
|
|
DownloadFiles,
|
|
MediaSanity,
|
|
}
|
|
|
|
impl AsSyncEndpoint for MediaSyncMethod {
|
|
fn as_sync_endpoint(&self, base: &Url) -> Url {
|
|
base.join("msync/").unwrap().join(self.into()).unwrap()
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
pub trait MediaSyncProtocol: Send + Sync + 'static {
|
|
async fn begin(
|
|
&self,
|
|
req: SyncRequest<SyncBeginRequest>,
|
|
) -> HttpResult<SyncResponse<JsonResult<SyncBeginResponse>>>;
|
|
async fn media_changes(
|
|
&self,
|
|
req: SyncRequest<MediaChangesRequest>,
|
|
) -> HttpResult<SyncResponse<JsonResult<MediaChangesResponse>>>;
|
|
async fn upload_changes(
|
|
&self,
|
|
req: SyncRequest<Vec<u8>>,
|
|
) -> HttpResult<SyncResponse<JsonResult<MediaUploadResponse>>>;
|
|
async fn download_files(
|
|
&self,
|
|
req: SyncRequest<DownloadFilesRequest>,
|
|
) -> HttpResult<SyncResponse<Vec<u8>>>;
|
|
async fn media_sanity_check(
|
|
&self,
|
|
req: SyncRequest<SanityCheckRequest>,
|
|
) -> HttpResult<SyncResponse<JsonResult<MediaSanityCheckResponse>>>;
|
|
}
|
|
|
|
/// Media endpoints wrap their returns in a JSON result, and legacy
|
|
/// clients expect it to always have an err field, even if it's empty.
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
#[serde(untagged)]
|
|
pub enum JsonResult<T> {
|
|
Ok {
|
|
data: T,
|
|
#[serde(default)]
|
|
err: String,
|
|
},
|
|
Err {
|
|
err: String,
|
|
},
|
|
}
|
|
|
|
impl<T> JsonResult<T> {
|
|
pub fn ok(inner: T) -> Self {
|
|
Self::Ok {
|
|
data: inner,
|
|
err: String::new(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<T> SyncResponse<JsonResult<T>>
|
|
where
|
|
T: DeserializeOwned,
|
|
{
|
|
pub fn json_result(&self) -> error::Result<T> {
|
|
match serde_json::from_slice(&self.data)? {
|
|
JsonResult::Ok { data, .. } => Ok(data),
|
|
JsonResult::Err { err } => Err(AnkiError::server_message(err)),
|
|
}
|
|
}
|
|
}
|