use std::path::PathBuf; use std::sync::Arc; use chrono::{DateTime, Utc}; use reqwest::Client; use serde::{Deserialize, Serialize}; use tokio::sync::Mutex; use tokio::time::{sleep, Duration}; use crate::domain::models::*; #[derive(Debug, Clone, Serialize, Deserialize)] pub struct OAuthToken { pub access_token: String, pub refresh_token: Option, pub expires_at: Option>, } #[derive(Debug)] #[allow(dead_code)] pub enum ApiError { Network(String), Auth(String), Api(String), } pub struct ApiClient { client: Client, client_id: String, client_secret: String, token: Arc>>, token_path: PathBuf, } impl ApiClient { pub fn new(client_id: String, client_secret: String) -> Self { let token_path = dirs::config_dir() .unwrap_or_else(|| PathBuf::from(".")) .join("task_app") .join("token.json"); Self { client: Client::new(), client_id, client_secret, token: Arc::new(Mutex::new(None)), token_path, } } pub fn token_file_exists(&self) -> bool { self.token_path.exists() && std::fs::read_to_string(&self.token_path) .ok() .and_then(|s| serde_json::from_str::(&s).ok()) .is_some() } pub async fn load_token(&self) -> Option { let content = std::fs::read_to_string(&self.token_path).ok()?; serde_json::from_str(&content).ok() } pub async fn save_token(&self, token: &OAuthToken) { if let Some(parent) = self.token_path.parent() { std::fs::create_dir_all(parent).ok(); } if let Ok(content) = serde_json::to_string_pretty(token) { std::fs::write(&self.token_path, content).ok(); } } pub async fn authenticate(&self) -> Result<(String, String), ApiError> { let params = serde_json::json!({ "client_id": self.client_id, "scope": "https://www.googleapis.com/auth/tasks", }); let resp = self .client .post("https://oauth2.googleapis.com/device/code") .json(¶ms) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; let data: serde_json::Value = resp .json() .await .map_err(|e| ApiError::Network(e.to_string()))?; let url = data["verification_url"] .as_str() .unwrap_or(&data["verification_url"].to_string()) .to_string(); let code = data["user_code"] .as_str() .unwrap_or("") .to_string(); let device_code = data["device_code"].as_str().unwrap_or("").to_string(); let interval = data["interval"].as_u64().unwrap_or(5); tokio::spawn({ let client = self.client.clone(); let client_id = self.client_id.clone(); let client_secret = self.client_secret.clone(); let device_code = device_code.clone(); let token = self.token.clone(); let token_path = self.token_path.clone(); async move { loop { sleep(Duration::from_secs(interval)).await; let poll = serde_json::json!({ "client_id": client_id, "client_secret": client_secret, "device_code": device_code, "grant_type": "urn:ietf:params:oauth:grant_type:device_code", }); if let Ok(resp) = client .post("https://oauth2.googleapis.com/token") .json(&poll) .send() .await { if let Ok(data) = resp.json::().await { if let Some(access_token) = data["access_token"].as_str() { let expires_in = data["expires_in"].as_i64().unwrap_or(3600); let oauth_token = OAuthToken { access_token: access_token.to_string(), refresh_token: data["refresh_token"].as_str().map(|s| s.to_string()), expires_at: Some(Utc::now() + chrono::Duration::seconds(expires_in)), }; if let Ok(content) = serde_json::to_string_pretty(&oauth_token) { std::fs::write(&token_path, content).ok(); } let mut t = token.lock().await; *t = Some(oauth_token); break; } } } } } }); Ok((url, code)) } pub async fn refresh_access_token(&self, refresh_token: &str) -> Result<(), ApiError> { let params = serde_json::json!({ "client_id": self.client_id, "client_secret": self.client_secret, "refresh_token": refresh_token, "grant_type": "refresh_token", }); let resp = self .client .post("https://oauth2.googleapis.com/token") .json(¶ms) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; let data: serde_json::Value = resp .json() .await .map_err(|e| ApiError::Network(e.to_string()))?; if let Some(access_token) = data["access_token"].as_str() { let expires_in = data["expires_in"].as_i64().unwrap_or(3600); let token = OAuthToken { access_token: access_token.to_string(), refresh_token: Some(refresh_token.to_string()), expires_at: Some(Utc::now() + chrono::Duration::seconds(expires_in)), }; self.save_token(&token).await; let mut t = self.token.lock().await; *t = Some(token); Ok(()) } else { Err(ApiError::Auth("Failed to refresh token".to_string())) } } pub async fn ensure_token(&self) -> Result { let mut token = self.token.lock().await; if let Some(ref t) = *token { if let Some(expires_at) = t.expires_at { if Utc::now() < expires_at { return Ok(t.access_token.clone()); } } if let Some(ref refresh) = t.refresh_token { let refresh_token = refresh.clone(); drop(token); self.refresh_access_token(&refresh_token).await?; let t2 = self.token.lock().await; if let Some(ref t) = *t2 { return Ok(t.access_token.clone()); } } Err(ApiError::Auth("Token expired and no refresh token".to_string())) } else if let Some(saved) = self.load_token().await { *token = Some(saved); if let Some(ref t) = *token { Ok(t.access_token.clone()) } else { Err(ApiError::Auth("No token available".to_string())) } } else { Err(ApiError::Auth("Not authenticated".to_string())) } } pub async fn fetch_lists(&self) -> Result, ApiError> { let token = self.ensure_token().await?; let resp = self .client .get("https://tasks.googleapis.com/tasks/v1/users/@me/lists") .bearer_auth(&token) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; let data: serde_json::Value = resp .json() .await .map_err(|e| ApiError::Api(e.to_string()))?; let empty = vec![]; let items = data["items"].as_array().unwrap_or(&empty); let lists = items .iter() .map(|item| TaskList { id: item["id"].as_str().unwrap_or("").to_string(), title: item["title"].as_str().unwrap_or("").to_string(), }) .collect(); Ok(lists) } pub async fn fetch_tasks(&self, list_id: &str) -> Result, ApiError> { let token = self.ensure_token().await?; let url = format!( "https://tasks.googleapis.com/tasks/v1/lists/{}/tasks?showCompleted=true&showHidden=true", list_id ); let resp = self .client .get(&url) .bearer_auth(&token) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; let data: serde_json::Value = resp .json() .await .map_err(|e| ApiError::Api(e.to_string()))?; let empty = vec![]; let items = data["items"].as_array().unwrap_or(&empty); let tasks = items .iter() .enumerate() .map(|(i, item)| { let due_str = item["due"].as_str().and_then(|s| { chrono::NaiveDateTime::parse_from_str( &s.replace("T", " ").replace("Z", "").chars().take(16).collect::(), "%Y-%m-%d %H:%M", ) .ok() }); Task { id: item["id"].as_str().unwrap_or("").to_string(), list_id: list_id.to_string(), title: item["title"].as_str().unwrap_or("").to_string(), notes: item["notes"].as_str().map(|s| s.to_string()), status: if item["status"].as_str() == Some("completed") { TaskStatus::Completed } else { TaskStatus::NeedsAction }, due: due_str, position: i as i64, } }) .collect(); Ok(tasks) } pub async fn create_task(&self, list_id: &str, task: &Task) -> Result<(), ApiError> { let token = self.ensure_token().await?; let mut body = serde_json::json!({ "title": task.title, }); if let Some(ref notes) = task.notes { body["notes"] = serde_json::Value::String(notes.clone()); } if let Some(due) = task.due { body["due"] = serde_json::Value::String(due.format("%Y-%m-%dT%H:%M:00.000Z").to_string()); } if task.status == TaskStatus::Completed { body["status"] = serde_json::Value::String("completed".to_string()); } let url = format!( "https://tasks.googleapis.com/tasks/v1/lists/{}/tasks", list_id ); let resp = self .client .post(&url) .bearer_auth(&token) .json(&body) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; if !resp.status().is_success() { return Err(ApiError::Api(format!("Create failed: {}", resp.status()))); } Ok(()) } pub async fn update_task(&self, list_id: &str, task: &Task) -> Result<(), ApiError> { let token = self.ensure_token().await?; let mut body = serde_json::json!({ "title": task.title, }); if let Some(ref notes) = task.notes { body["notes"] = serde_json::Value::String(notes.clone()); } if let Some(due) = task.due { body["due"] = serde_json::Value::String(due.format("%Y-%m-%dT%H:%M:00.000Z").to_string()); } body["status"] = serde_json::Value::String(match task.status { TaskStatus::Completed => "completed".to_string(), TaskStatus::NeedsAction => "needsAction".to_string(), }); let url = format!( "https://tasks.googleapis.com/tasks/v1/lists/{}/tasks/{}", list_id, task.id ); let resp = self .client .patch(&url) .bearer_auth(&token) .json(&body) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; if !resp.status().is_success() { return Err(ApiError::Api(format!("Update failed: {}", resp.status()))); } Ok(()) } pub async fn delete_task(&self, list_id: &str, task_id: &str) -> Result<(), ApiError> { let token = self.ensure_token().await?; let url = format!( "https://tasks.googleapis.com/tasks/v1/lists/{}/tasks/{}", list_id, task_id ); let resp = self .client .delete(&url) .bearer_auth(&token) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; if !resp.status().is_success() { return Err(ApiError::Api(format!("Delete failed: {}", resp.status()))); } Ok(()) } pub async fn move_task( &self, list_id: &str, task_id: &str, prev: Option<&str>, sibling: Option<&str>, ) -> Result<(), ApiError> { let token = self.ensure_token().await?; let mut url = format!( "https://tasks.googleapis.com/tasks/v1/lists/{}/tasks/{}/move", list_id, task_id ); if let Some(p) = prev { url.push_str(&format!("&previous={}", p)); } if let Some(s) = sibling { url.push_str(&format!("&destinationTaskList={}", s)); } let resp = self .client .post(&url) .bearer_auth(&token) .send() .await .map_err(|e| ApiError::Network(e.to_string()))?; if !resp.status().is_success() { return Err(ApiError::Api(format!("Move failed: {}", resp.status()))); } Ok(()) } }