mirror of https://github.com/realaravinth/gitpad
49 lines
1.4 KiB
Rust
49 lines
1.4 KiB
Rust
/*
|
|
* Copyright (C) 2022 Aravinth Manivannan <realaravinth@batsense.net>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
/// Get random string of specific length
|
|
pub(crate) fn get_random(len: usize) -> String {
|
|
use rand::{distributions::Alphanumeric, rngs::ThreadRng, thread_rng, Rng};
|
|
use std::iter;
|
|
|
|
let mut rng: ThreadRng = thread_rng();
|
|
|
|
iter::repeat(())
|
|
.map(|()| rng.sample(Alphanumeric))
|
|
.map(char::from)
|
|
.take(len)
|
|
.collect::<String>()
|
|
}
|
|
|
|
pub fn escape_spaces(name: &str) -> String {
|
|
if name.contains(' ') {
|
|
name.replace(' ', "\\ ")
|
|
} else {
|
|
name.to_string()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn space_escape() {
|
|
let space = "do re mi";
|
|
assert_eq!(&escape_spaces(space), ("do\\ re\\ mi"));
|
|
}
|
|
}
|