mirror of https://github.com/w4/bin.git
55 lines
1.7 KiB
Rust
55 lines
1.7 KiB
Rust
use actix_web::web::Bytes;
|
|
use linked_hash_map::LinkedHashMap;
|
|
use once_cell::sync::Lazy;
|
|
use parking_lot::RwLock;
|
|
use rand::{distributions::Alphanumeric, thread_rng, Rng};
|
|
use std::cell::RefCell;
|
|
|
|
pub type PasteStore = RwLock<LinkedHashMap<String, Bytes>>;
|
|
|
|
static BUFFER_SIZE: Lazy<usize> = Lazy::new(|| argh::from_env::<crate::BinArgs>().buffer_size);
|
|
|
|
/// Ensures `ENTRIES` is less than the size of `BIN_BUFFER_SIZE`. If it isn't then
|
|
/// `ENTRIES.len() - BIN_BUFFER_SIZE` elements will be popped off the front of the map.
|
|
///
|
|
/// During the purge, `ENTRIES` is locked and the current thread will block.
|
|
fn purge_old(entries: &mut LinkedHashMap<String, Bytes>) {
|
|
if entries.len() > *BUFFER_SIZE {
|
|
let to_remove = entries.len() - *BUFFER_SIZE;
|
|
|
|
for _ in 0..to_remove {
|
|
entries.pop_front();
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Generates a 'pronounceable' random ID using gpw
|
|
pub fn generate_id() -> String {
|
|
thread_local!(static KEYGEN: RefCell<gpw::PasswordGenerator> = RefCell::new(gpw::PasswordGenerator::default()));
|
|
|
|
KEYGEN.with(|k| k.borrow_mut().next()).unwrap_or_else(|| {
|
|
thread_rng()
|
|
.sample_iter(&Alphanumeric)
|
|
.take(6)
|
|
.map(char::from)
|
|
.collect()
|
|
})
|
|
}
|
|
|
|
/// Stores a paste under the given id
|
|
pub fn store_paste(entries: &PasteStore, id: String, content: Bytes) {
|
|
let mut entries = entries.write();
|
|
|
|
purge_old(&mut entries);
|
|
|
|
entries.insert(id, content);
|
|
}
|
|
|
|
/// Get a paste by id.
|
|
///
|
|
/// Returns `None` if the paste doesn't exist.
|
|
pub fn get_paste(entries: &PasteStore, id: &str) -> Option<Bytes> {
|
|
// need to box the guard until owning_ref understands Pin is a stable address
|
|
entries.read().get(id).map(Bytes::clone)
|
|
}
|