feat: Add server ping check

Author: LeanderBB

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "proton-api-rs"
 authors = ["Leander Beernaert <lbb-dev@pm.me>"]
-version = "0.3.0"
+version = "0.4.0"
 edition = "2021"
 license = "AGPL-3.0-only"
 description = "Unofficial implemention of proton REST API in rust"

examples/user_id.rs

@@ -7,12 +7,11 @@ async fn main() {
     let user_password = std::env::var("PAPI_USER_PASSWORD").unwrap();
     let app_version = std::env::var("PAPI_APP_VERSION").unwrap();
 
-    let client = match ClientBuilder::new()
-        .app_version(&app_version)
-        .login(&user_email, &user_password)
-        .await
-        .unwrap()
-    {
+    let builder = ClientBuilder::new().app_version(&app_version);
+
+    builder.ping().await.unwrap();
+
+    let client = match builder.login(&user_email, &user_password).await.unwrap() {
         ClientLoginState::Authenticated(c) => c,
 
         ClientLoginState::AwaitingTotp(mut t) => {

src/client/client_builder.rs

@@ -5,6 +5,7 @@ use crate::client::types::{
 use crate::client::{HttpClientBuilder, X_PM_UID_HEADER};
 use crate::domain::{SecretString, UserUid};
 use crate::{impl_error_conversion, Client, RequestError};
+use anyhow::anyhow;
 use go_srp::SRPAuth;
 use secrecy::ExposeSecret;
 use std::time::Duration;
@@ -214,6 +215,17 @@ impl ClientBuilder {
 
         Ok(Client { http_client, user })
     }
+
+    /// Check connectivity
+    pub async fn ping(&self) -> Result<(), RequestError> {
+        let http_client = self.0.clone().build()?;
+        let r = http_client.get("tests/ping").execute().await?;
+        if r.status_code() != 200 {
+            return Err(RequestError::Other(anyhow!("Service not pingable")));
+        }
+
+        Ok(())
+    }
 }
 
 #[derive(Debug)]