1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

//! Glue between mruby FFI and `securerandom` Rust implementation.

use crate::convert::implicitly_convert_to_int;
use crate::extn::prelude::*;
use crate::extn::stdlib::securerandom;

#[inline]
pub fn alphanumeric(interp: &mut Artichoke, len: Option<Value>) -> Result<Value, Error> {
    let alpha = if let Some(len) = len {
        let len = implicitly_convert_to_int(interp, len)?;
        securerandom::alphanumeric(Some(len))?
    } else {
        securerandom::alphanumeric(None)?
    };
    Ok(interp.convert_mut(alpha))
}

#[inline]
pub fn base64(interp: &mut Artichoke, len: Option<Value>) -> Result<Value, Error> {
    let base64 = if let Some(len) = len {
        let len = implicitly_convert_to_int(interp, len)?;
        securerandom::base64(Some(len))?
    } else {
        securerandom::base64(None)?
    };
    Ok(interp.convert_mut(base64))
}

#[inline]
pub fn urlsafe_base64(interp: &mut Artichoke, len: Option<Value>, padding: Option<Value>) -> Result<Value, Error> {
    let padding = match padding {
        None => false,
        Some(val) if val.is_nil() => false,
        Some(padding) => {
            // All truthy values evaluate to `true` for this argument. So either
            // `padding` is a `bool` and we can extract it, or we default to
            // `true` since only `nil` (handled above) and `false` are falsy.
            padding.try_into::<bool>(interp).unwrap_or(true)
        }
    };
    let base64 = if let Some(len) = len {
        let len = implicitly_convert_to_int(interp, len)?;
        securerandom::urlsafe_base64(Some(len), padding)?
    } else {
        securerandom::urlsafe_base64(None, padding)?
    };
    Ok(interp.convert_mut(base64))
}

#[inline]
pub fn hex(interp: &mut Artichoke, len: Option<Value>) -> Result<Value, Error> {
    let hex = if let Some(len) = len {
        let len = implicitly_convert_to_int(interp, len)?;
        securerandom::hex(Some(len))?
    } else {
        securerandom::hex(None)?
    };
    Ok(interp.convert_mut(hex))
}

#[inline]
pub fn random_bytes(interp: &mut Artichoke, len: Option<Value>) -> Result<Value, Error> {
    let bytes = if let Some(len) = len {
        // Upstream uses `n.to_int`, which means we must implicitly convert to
        // int.
        //
        // https://github.com/ruby/ruby/blob/v2_6_3/lib/securerandom.rb#L135
        let len = implicitly_convert_to_int(interp, len)?;
        securerandom::random_bytes(Some(len))?
    } else {
        securerandom::random_bytes(None)?
    };
    Ok(interp.convert_mut(bytes))
}

#[inline]
pub fn random_number(interp: &mut Artichoke, max: Option<Value>) -> Result<Value, Error> {
    let max = interp.try_convert_mut(max)?;
    let num = securerandom::random_number(max)?;
    Ok(interp.convert_mut(num))
}

#[inline]
pub fn uuid(interp: &mut Artichoke) -> Result<Value, Error> {
    let uuid = securerandom::uuid()?;
    Ok(interp.convert_mut(uuid))
}