use crate::{
custom_serde::{duration_second, space_separated_scopes},
ModelResult,
};
use std::{
collections::{HashMap, HashSet},
fs,
io::{Read, Write},
path::Path,
};
use chrono::{DateTime, Duration, TimeDelta, Utc};
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
pub struct Token {
pub access_token: String,
#[serde(with = "duration_second")]
pub expires_in: Duration,
pub expires_at: Option<DateTime<Utc>>,
pub refresh_token: Option<String>,
#[serde(default, with = "space_separated_scopes", rename = "scope")]
pub scopes: HashSet<String>,
}
impl Default for Token {
fn default() -> Self {
Self {
access_token: String::new(),
expires_in: Duration::try_seconds(0).unwrap(),
expires_at: Some(Utc::now()),
refresh_token: None,
scopes: HashSet::new(),
}
}
}
impl Token {
pub fn from_cache<T: AsRef<Path>>(path: T) -> ModelResult<Self> {
let mut file = fs::File::open(path)?;
let mut tok_str = String::new();
file.read_to_string(&mut tok_str)?;
let tok = serde_json::from_str(&tok_str)?;
Ok(tok)
}
pub fn write_cache<T: AsRef<Path>>(&self, path: T) -> ModelResult<()> {
let token_info = serde_json::to_string(&self)?;
let mut file = fs::OpenOptions::new()
.write(true)
.create(true)
.truncate(true)
.open(path)?;
file.set_len(0)?;
file.write_all(token_info.as_bytes())?;
Ok(())
}
#[must_use]
pub fn is_expired(&self) -> bool {
self.expires_at.map_or(true, |expiration| {
Utc::now() + TimeDelta::try_seconds(10).unwrap() >= expiration
})
}
#[must_use]
pub fn auth_headers(&self) -> HashMap<String, String> {
let auth = "authorization".to_owned();
let value = format!("Bearer {}", self.access_token);
let mut headers = HashMap::new();
headers.insert(auth, value);
headers
}
}
#[cfg(test)]
mod test {
use std::collections::HashSet;
use crate::Token;
use serde_json::json;
#[test]
fn test_bearer_auth() {
let tok = Token {
access_token: "access_token".to_string(),
..Default::default()
};
let headers = tok.auth_headers();
assert_eq!(headers.len(), 1);
assert_eq!(
headers.get("authorization"),
Some(&"Bearer access_token".to_owned())
);
}
#[test]
fn test_token_deserialize() {
let mut scopes = HashSet::<String>::new();
scopes.insert("user-read-email".to_owned());
let tok = Token {
access_token: "access_token".to_string(),
scopes,
..Default::default()
};
let value = json!(tok);
let token = serde_json::from_value::<Token>(value);
assert!(token.is_ok());
assert_eq!(token.unwrap().scopes, tok.scopes);
}
}