1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
use crate::{Algorithm, OTPMethod, Steam, HOTP, OTP, TOTP};
use anyhow::Result;
#[derive(Debug, Eq, PartialEq, Clone)]
pub struct Account {
pub secret: String,
pub digits: u32,
pub period: Option<u32>,
pub counter: Option<u32>,
pub method: OTPMethod,
pub algorithm: Algorithm,
pub issuer: Option<String>,
pub label: String,
}
pub trait Restorable: Sized {
fn restore(bytes: &[u8]) -> Result<Vec<Account>>;
}
pub trait Backupable: Sized {
fn backup(codes: Vec<Account>) -> Result<Vec<u8>>;
}
impl OTP for Account {
fn compute(&self, counter: u64) -> Result<String> {
match self.method {
OTPMethod::TOTP => TOTP {
algorithm: self.algorithm,
digits: self.digits,
period: self.period.unwrap(),
secret: self.secret.clone(),
}
.compute(counter),
OTPMethod::HOTP => HOTP {
algorithm: self.algorithm,
digits: self.digits,
secret: self.secret.clone(),
}
.compute(counter),
OTPMethod::Steam => Steam {
secret: self.secret.clone(),
}
.compute(counter),
}
}
}
mod andotp;
pub use andotp::AndOTP;