1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use anyhow;
use sha2::{Digest, Sha512};

mod simple_file;
pub use simple_file::FileDatabase;
mod dummy_database;
pub use dummy_database::DummyDatabase;

pub type Hash = [u8; 64];

/// Database which stores mappings from hashes to matrix media urls,
/// to avoid duplicate uploads of the same file.
pub trait Database {
	async fn get(&self, hash: &Hash) -> anyhow::Result<Option<String>>;
	async fn add(&self, hash: Hash, url: String) -> anyhow::Result<()>;
}

pub fn hash(value: &[u8]) -> Hash {
	let mut hasher = Sha512::new();
	hasher.update(value);
	hasher.finalize().into()
}