fix: ensure user is initialized after login (#38)

Author: scarmuega

rustfmt.toml

@@ -0,0 +1,2 @@
+edition = "2015"
+wrap_comments = true

src/api/account.rs

@@ -30,6 +30,34 @@ where
     Ok(response)
 }
 
+pub async fn initialize_user(access_token: &str) -> Result<String, Error> {
+    let url = format!("{}/users", build_api_url());
+
+    let mut sp = Spinner::new(Spinners::Dots, "".into());
+
+    let client = Client::new();
+    let resp = client
+        .post(url)
+        .header("Authorization", format!("Bearer {}", access_token))
+        .header("agent", build_agent_header())
+        .send()
+        .await?;
+
+    sp.stop_with_symbol("".into());
+
+    check_response_update_header(&resp)?;
+    let response = resp.json::<serde_json::Value>().await?;
+
+    let userid = response
+        .as_object()
+        .and_then(|o| o.get("userId"))
+        .and_then(|x| x.as_str())
+        .unwrap_or_default()
+        .to_owned();
+
+    Ok(userid)
+}
+
 pub async fn create_api_key(
     access_token: &str,
     project: &str,

src/init/login.rs

@@ -1,8 +1,10 @@
-use miette::bail;
+use miette::{bail, Context as _, IntoDiagnostic as _};
 use reqwest::StatusCode;
 use serde::{Deserialize, Serialize};
 use std::{collections::HashMap, time::Duration};
 
+use crate::api;
+
 async fn find_login_url() -> (String, String) {
     let mut params = HashMap::new();
     params.insert("client_id", "gpJ63MG5g1V1PKufM9WHGjjeAe7yCT8L");
@@ -94,6 +96,11 @@ pub async fn run() -> miette::Result<String> {
         let (status, access_token) = poll_token(&device_code).await;
 
         if status.is_success() {
+            api::account::initialize_user(&access_token)
+                .await
+                .into_diagnostic()
+                .context("initializing user")?;
+
             println!("login successful!");
             return Ok(access_token);
         }