use alloc::vec::Vec;
pub use azul_core::keyring::{KeyringRequest, KeyringResult};
#[derive(Debug, Clone, PartialEq, Default)]
pub struct KeyringManager {
pub last_result: Option<KeyringResult>,
}
impl KeyringManager {
pub fn new() -> Self {
Self::default()
}
pub fn last_result(&self) -> Option<&KeyringResult> {
self.last_result.as_ref()
}
pub fn set_last_result(&mut self, result: KeyringResult) -> bool {
let changed = self.last_result.as_ref() != Some(&result);
self.last_result = Some(result);
changed
}
}
static PENDING_REQUESTS: std::sync::Mutex<Vec<KeyringRequest>> =
std::sync::Mutex::new(Vec::new());
pub fn push_keyring_request(request: KeyringRequest) {
let mut q = PENDING_REQUESTS.lock().unwrap_or_else(|e| e.into_inner());
q.push(request);
}
pub fn drain_keyring_requests() -> Vec<KeyringRequest> {
let mut q = PENDING_REQUESTS.lock().unwrap_or_else(|e| e.into_inner());
core::mem::take(&mut *q)
}
static PENDING_RESULTS: std::sync::Mutex<Vec<KeyringResult>> =
std::sync::Mutex::new(Vec::new());
pub fn push_keyring_result(result: KeyringResult) {
let mut q = PENDING_RESULTS.lock().unwrap_or_else(|e| e.into_inner());
q.push(result);
}
pub fn drain_keyring_results() -> Vec<KeyringResult> {
let mut q = PENDING_RESULTS.lock().unwrap_or_else(|e| e.into_inner());
core::mem::take(&mut *q)
}
#[cfg(test)]
mod tests {
use super::*;
use azul_css::AzString;
#[test]
fn manager_defaults_to_no_result() {
let mgr = KeyringManager::new();
assert_eq!(mgr.last_result(), None);
}
#[test]
fn set_last_result_returns_change_flag() {
let mut mgr = KeyringManager::new();
assert!(mgr.set_last_result(KeyringResult::Stored));
assert_eq!(mgr.last_result(), Some(&KeyringResult::Stored));
assert!(!mgr.set_last_result(KeyringResult::Stored));
assert!(mgr.set_last_result(KeyringResult::Deleted));
}
#[test]
fn result_helpers() {
let secret = KeyringResult::Retrieved(AzString::from_const_str("hunter2"));
assert_eq!(secret.secret().map(|s| s.as_str()), Some("hunter2"));
assert!(secret.is_ok());
assert!(KeyringResult::Stored.is_ok());
assert!(KeyringResult::Deleted.is_ok());
for r in [
KeyringResult::NotFound,
KeyringResult::Denied,
KeyringResult::Unavailable,
KeyringResult::Error,
] {
assert!(!r.is_ok(), "{:?} must not be ok", r);
assert_eq!(r.secret(), None);
}
}
#[test]
fn requests_round_trip_through_channel() {
let _ = drain_keyring_requests();
push_keyring_request(KeyringRequest::Store {
key: AzString::from_const_str("token"),
secret: AzString::from_const_str("abc"),
require_biometry: true,
});
push_keyring_request(KeyringRequest::Get {
key: AzString::from_const_str("token"),
});
let drained = drain_keyring_requests();
assert_eq!(drained.len(), 2, "both queued requests drain in order");
assert!(matches!(drained[0], KeyringRequest::Store { .. }));
assert!(matches!(drained[1], KeyringRequest::Get { .. }));
assert!(drain_keyring_requests().is_empty());
}
#[test]
fn results_round_trip_through_manager() {
let _ = drain_keyring_results();
push_keyring_result(KeyringResult::NotFound);
push_keyring_result(KeyringResult::Retrieved(AzString::from_const_str("s"))); let drained = drain_keyring_results();
assert_eq!(drained.len(), 2);
let mut mgr = KeyringManager::new();
for r in drained {
mgr.set_last_result(r);
}
assert_eq!(
mgr.last_result().and_then(|r| r.secret()).map(|s| s.as_str()),
Some("s"),
"the last applied result wins"
);
assert!(drain_keyring_results().is_empty());
}
}