aver-lang 0.9.4

/// Random service — effectful random number generation.
///
/// Methods:
///   Random.int(min, max)  — random Int in [min, max] inclusive; ! [Random]
///   Random.float()        — random Float in [0.0, 1.0); ! [Random]
///
/// Backed by `aver_rt::random` (OS entropy via `rand` crate).
use std::collections::HashMap;
use std::sync::Arc as Rc;

use crate::nan_value::{Arena, NanValue};
use crate::value::{RuntimeError, Value};

pub fn register(global: &mut HashMap<String, Value>) {
    let mut members = HashMap::new();
    for method in &["int", "float"] {
        members.insert(
            method.to_string(),
            Value::Builtin(format!("Random.{}", method)),
        );
    }
    global.insert(
        "Random".to_string(),
        Value::Namespace {
            name: "Random".to_string(),
            members,
        },
    );
}

pub const DECLARED_EFFECTS: &[&str] = &["Random.int", "Random.float"];

pub fn effects(name: &str) -> &'static [&'static str] {
    match name {
        "Random.int" => &["Random.int"],
        "Random.float" => &["Random.float"],
        _ => &[],
    }
}

pub fn call(name: &str, args: &[Value]) -> Option<Result<Value, RuntimeError>> {
    match name {
        "Random.int" => Some(random_int(args)),
        "Random.float" => Some(random_float(args)),
        _ => None,
    }
}

fn random_int(args: &[Value]) -> Result<Value, RuntimeError> {
    if args.len() != 2 {
        return Err(RuntimeError::Error(format!(
            "Random.int takes 2 arguments (min, max), got {}",
            args.len()
        )));
    }
    let Value::Int(min) = &args[0] else {
        return Err(RuntimeError::Error(
            "Random.int: first argument must be an Int".to_string(),
        ));
    };
    let Value::Int(max) = &args[1] else {
        return Err(RuntimeError::Error(
            "Random.int: second argument must be an Int".to_string(),
        ));
    };

    match aver_rt::random::random_int(*min, *max) {
        Ok(value) => Ok(Value::Int(value)),
        Err(msg) => Err(RuntimeError::Error(msg)),
    }
}

fn random_float(args: &[Value]) -> Result<Value, RuntimeError> {
    if !args.is_empty() {
        return Err(RuntimeError::Error(format!(
            "Random.float takes 0 arguments, got {}",
            args.len()
        )));
    }
    Ok(Value::Float(aver_rt::random::random_float()))
}

// ─── NanValue-native API ─────────────────────────────────────────────────────

pub fn register_nv(global: &mut HashMap<String, NanValue>, arena: &mut Arena) {
    let methods = &["int", "float"];
    let mut members: Vec<(Rc<str>, NanValue)> = Vec::with_capacity(methods.len());
    for method in methods {
        let idx = arena.push_builtin(&format!("Random.{}", method));
        members.push((Rc::from(*method), NanValue::new_builtin(idx)));
    }
    let ns_idx = arena.push(crate::nan_value::ArenaEntry::Namespace {
        name: Rc::from("Random"),
        members,
    });
    global.insert("Random".to_string(), NanValue::new_namespace(ns_idx));
}

pub fn call_nv(
    name: &str,
    args: &[NanValue],
    arena: &mut Arena,
) -> Option<Result<NanValue, RuntimeError>> {
    match name {
        "Random.int" => Some(random_int_nv(args, arena)),
        "Random.float" => Some(random_float_nv(args, arena)),
        _ => None,
    }
}

fn random_int_nv(args: &[NanValue], arena: &mut Arena) -> Result<NanValue, RuntimeError> {
    if args.len() != 2 {
        return Err(RuntimeError::Error(format!(
            "Random.int takes 2 arguments (min, max), got {}",
            args.len()
        )));
    }
    if !args[0].is_int() {
        return Err(RuntimeError::Error(
            "Random.int: first argument must be an Int".to_string(),
        ));
    }
    if !args[1].is_int() {
        return Err(RuntimeError::Error(
            "Random.int: second argument must be an Int".to_string(),
        ));
    }
    let min = args[0].as_int(arena);
    let max = args[1].as_int(arena);
    match aver_rt::random::random_int(min, max) {
        Ok(value) => Ok(NanValue::new_int(value, arena)),
        Err(msg) => Err(RuntimeError::Error(msg)),
    }
}

fn random_float_nv(args: &[NanValue], _arena: &mut Arena) -> Result<NanValue, RuntimeError> {
    if !args.is_empty() {
        return Err(RuntimeError::Error(format!(
            "Random.float takes 0 arguments, got {}",
            args.len()
        )));
    }
    Ok(NanValue::new_float(aver_rt::random::random_float()))
}