pub mod abs;
pub mod aggregate;
pub mod basic_math;
pub mod random;
pub mod round;
pub mod trig;

use crate::shared::models::UserSession;
use crate::shared::state::AppState;
use log::debug;
use rhai::Engine;
use std::sync::Arc;

pub fn register_math_functions(state: &Arc<AppState>, user: UserSession, engine: &mut Engine) {
    abs::abs_keyword(state, user.clone(), engine);
    round::round_keyword(state, user.clone(), engine);
    basic_math::int_keyword(state, user.clone(), engine);
    basic_math::floor_keyword(state, user.clone(), engine);
    basic_math::ceil_keyword(state, user.clone(), engine);
    basic_math::max_keyword(state, user.clone(), engine);
    basic_math::min_keyword(state, user.clone(), engine);
    basic_math::mod_keyword(state, user.clone(), engine);
    basic_math::sgn_keyword(state, user.clone(), engine);
    basic_math::sqrt_keyword(state, user.clone(), engine);
    basic_math::pow_keyword(state, user.clone(), engine);
    random::random_keyword(state, user.clone(), engine);
    trig::sin_keyword(state, user.clone(), engine);
    trig::cos_keyword(state, user.clone(), engine);
    trig::tan_keyword(state, user.clone(), engine);
    trig::log_keyword(state, user.clone(), engine);
    trig::exp_keyword(state, user.clone(), engine);
    trig::pi_keyword(state, user.clone(), engine);
    aggregate::sum_keyword(state, user.clone(), engine);
    aggregate::avg_keyword(state, user, engine);

    debug!("Registered all math functions");
}