Add DeviceCodes::poll_until_available method

examples/device_flow.rs

@@ -1,6 +1,4 @@
-use either::Either;
 use http::header::ACCEPT;
-use std::time::Duration;
 
 #[tokio::main]
 async fn main() -> octocrab::Result<()> {
@@ -17,28 +15,7 @@ async fn main() -> octocrab::Result<()> {
         "Go to {} and enter code {}",
         codes.verification_uri, codes.user_code
     );
-    let mut interval = Duration::from_secs(codes.interval);
-    let mut clock = tokio::time::interval(interval);
-    let auth = loop {
-        clock.tick().await;
-        match codes.poll_once(&crab, &client_id).await? {
-            Either::Left(auth) => break auth,
-            Either::Right(cont) => match cont {
-                octocrab::auth::Continue::SlowDown => {
-                    // We were request to slow down. We add five seconds to the polling
-                    // duration.
-                    interval += Duration::from_secs(5);
-                    clock = tokio::time::interval(interval);
-                    // The first tick happens instantly, so we tick that off immediately.
-                    clock.tick().await;
-                }
-                octocrab::auth::Continue::AuthorizationPending => {
-                    // The user has not clicked authorize yet, but nothing has gone wrong.
-                    // We keep polling.
-                }
-            },
-        }
-    };
+    let auth = codes.poll_until_available(&crab, &client_id).await?;
 
     println!("Authorization succeeded with access to {:?}", auth.scope);
     Ok(())

src/auth.rs

@@ -7,7 +7,7 @@ use jsonwebtoken::{Algorithm, EncodingKey, Header};
 use secrecy::{ExposeSecret, SecretString};
 use serde::{Deserialize, Serialize};
 use std::fmt;
-use std::time::SystemTime;
+use std::time::{Duration, SystemTime};
 
 use snafu::*;
 
@@ -204,7 +204,7 @@ pub struct DeviceCodes {
 impl DeviceCodes {
     /// Poll Github to see if authentication codes are available.
     ///
-    /// See `https://docs.github.com/en/developers/apps/building-oauth-apps/authorizing-oauth-apps#response-parameters` for details.
+    /// See `https://docs.github.com/en/developers/apps/building-oauth-apps/authorizing-oauth-apps` for details.
     pub async fn poll_once(
         &self,
         crab: &crate::Octocrab,
@@ -225,6 +225,36 @@ impl DeviceCodes {
             TokenResponse::Continue { error } => Either::Right(error),
         })
     }
+
+    /// Poll Github in a loop until authentication codes become available.
+    pub async fn poll_until_available(
+        &self,
+        crab: &crate::Octocrab,
+        client_id: &SecretString,
+    ) -> Result<OAuth> {
+        let mut interval = Duration::from_secs(self.interval);
+        let mut clock = tokio::time::interval(interval);
+
+        loop {
+            clock.tick().await;
+            match self.poll_once(crab, client_id).await? {
+                Either::Left(auth) => return Ok(auth),
+                Either::Right(cont) => match cont {
+                    Continue::SlowDown => {
+                        // We were requested to slow down, so add five seconds to the polling
+                        // duration.
+                        interval += Duration::from_secs(5);
+                        clock = tokio::time::interval(interval);
+                        // The first tick happens instantly, so we tick that off immediately.
+                        clock.tick().await;
+                    }
+                    Continue::AuthorizationPending => {
+                        // The user has not clicked authorize yet, so we keep polling as normal.
+                    }
+                },
+            }
+        }
+    }
 }
 
 /// See https://docs.github.com/en/developers/apps/building-oauth-apps/authorizing-oauth-apps#input-parameters