conduwuit/src/utils.rs

93 lines
2.7 KiB
Rust
Raw Normal View History

2020-04-19 12:14:47 +00:00
use argon2::{Config, Variant};
use cmp::Ordering;
use rand::prelude::*;
use sled::IVec;
use std::{
cmp,
convert::TryInto,
time::{SystemTime, UNIX_EPOCH},
};
2020-03-29 11:48:44 +00:00
2020-04-12 19:12:50 +00:00
pub fn millis_since_unix_epoch() -> u64 {
SystemTime::now()
2020-03-29 11:48:44 +00:00
.duration_since(UNIX_EPOCH)
2020-06-09 13:13:17 +00:00
.expect("time is valid")
2020-04-12 19:12:50 +00:00
.as_millis() as u64
2020-03-29 11:48:44 +00:00
}
2020-03-29 19:05:20 +00:00
pub fn increment(old: Option<&[u8]>) -> Option<Vec<u8>> {
2020-06-09 13:13:17 +00:00
let number = match old.map(|bytes| bytes.try_into()) {
Some(Ok(bytes)) => {
let number = u64::from_be_bytes(bytes);
number + 1
}
2020-06-09 13:13:17 +00:00
_ => 1, // Start at one. since 0 should return the first event in the db
};
2020-03-29 19:05:20 +00:00
Some(number.to_be_bytes().to_vec())
2020-03-29 19:05:20 +00:00
}
2020-04-19 12:14:47 +00:00
pub fn generate_keypair(old: Option<&[u8]>) -> Option<Vec<u8>> {
2020-06-09 13:13:17 +00:00
Some(old.map(|s| s.to_vec()).unwrap_or_else(|| {
ruma::signatures::Ed25519KeyPair::generate()
.expect("Ed25519KeyPair generation always works (?)")
}))
2020-04-19 12:14:47 +00:00
}
/// Parses the bytes into an u64.
2020-06-09 13:13:17 +00:00
pub fn u64_from_bytes(bytes: &[u8]) -> Result<u64, std::array::TryFromSliceError> {
let array: [u8; 8] = bytes.try_into()?;
Ok(u64::from_be_bytes(array))
}
/// Parses the bytes into a string.
2020-06-09 13:13:17 +00:00
pub fn string_from_bytes(bytes: &[u8]) -> Result<String, std::string::FromUtf8Error> {
String::from_utf8(bytes.to_vec())
2020-03-29 19:05:20 +00:00
}
pub fn random_string(length: usize) -> String {
thread_rng()
.sample_iter(&rand::distributions::Alphanumeric)
.take(length)
.collect()
}
/// Calculate a new hash for the given password
2020-06-09 13:13:17 +00:00
pub fn calculate_hash(password: &str) -> Result<String, argon2::Error> {
let hashing_config = Config {
variant: Variant::Argon2id,
..Default::default()
};
let salt = random_string(32);
2020-04-19 12:14:47 +00:00
argon2::hash_encoded(password.as_bytes(), salt.as_bytes(), &hashing_config)
}
pub fn common_elements(
mut iterators: impl Iterator<Item = impl Iterator<Item = IVec>>,
check_order: impl Fn(&IVec, &IVec) -> Ordering,
) -> Option<impl Iterator<Item = IVec>> {
let first_iterator = iterators.next()?;
let mut other_iterators = iterators.map(|i| i.peekable()).collect::<Vec<_>>();
Some(first_iterator.filter(move |target| {
other_iterators
.iter_mut()
.map(|it| {
while let Some(element) = it.peek() {
match check_order(element, target) {
Ordering::Greater => return false, // We went too far
Ordering::Equal => return true, // Element is in both iters
Ordering::Less => {
// Keep searching
it.next();
}
}
}
false
})
.all(|b| b)
}))
}