Merge pull request #9 from chardon55/8-add-chrono-support-as-an-optio…

…nal-feature

Add chrono timestamp feature

Cargo.toml

@@ -20,13 +20,15 @@ homepage = "https://github.com/chardon55/minotp"
 
 [dependencies]
 hmac = "0.12"
+chrono = { version = "0.4", optional = true }
 
 [dev-dependencies]
-chrono = "^0.4.33"
 data-encoding = "2.5.0"
 sha1 = { version = "0.10", default-features = false }
 sha2 = { version = "0.10", default-features = false }
 sha3 = { version = "0.10", default-features = false }
 
 [features]
+default = []
+chrono = ["dep:chrono"]
 

src/util.rs

@@ -1,3 +1,4 @@
+#[cfg(not(feature = "chrono"))]
 use std::time::{SystemTime, UNIX_EPOCH};
 
 const ENS_BIT: usize = 8;
@@ -29,20 +30,17 @@ pub(crate) fn calc_totp_counter(interval: u32, timestamp: u64) -> (u64, u32) {
 }
 
 #[inline]
+#[cfg(feature = "chrono")]
+pub(crate) fn time_now() -> u64 {
+    chrono::Utc::now().timestamp() as u64
+}
+
+#[inline]
+#[cfg(not(feature = "chrono"))]
 pub(crate) fn time_now() -> u64 {
     SystemTime::now()
         .duration_since(UNIX_EPOCH)
         .expect("[minotp::Totp] FATAL: Your system time is probably incorrectly set.")
         .as_secs()
 }
 
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn test_time_now() {
-        // Compare our own function with the chrono's
-        assert_eq!(time_now(), chrono::Utc::now().timestamp() as u64)
-    }
-}