mrubyedge 1.1.12

use std::rc::Rc;

use crate::{
    Error,
    yamrb::{
        helpers::{mrb_call_block, mrb_define_cmethod, mrb_funcall},
        value::*,
        vm::VM,
    },
};

pub(crate) fn initialize_object(vm: &mut VM) {
    let object_class = vm.object_class.clone();
    let klass = RObject::class(object_class.clone(), vm);
    vm.consts.insert("Object".to_string(), klass);
    vm.builtin_class_table
        .insert("Object", object_class.clone());

    #[cfg(feature = "wasi")]
    {
        mrb_define_cmethod(vm, object_class.clone(), "puts", Box::new(mrb_kernel_puts));
        mrb_define_cmethod(vm, object_class.clone(), "p", Box::new(mrb_kernel_p));
        mrb_define_cmethod(
            vm,
            object_class.clone(),
            "debug",
            Box::new(mrb_kernel_debug),
        );
    }

    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "initialize",
        Box::new(mrb_object_initialize),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "==",
        Box::new(mrb_object_double_eq),
    );
    mrb_define_cmethod(vm, object_class.clone(), "!=", Box::new(mrb_object_not_eq));
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "===",
        Box::new(mrb_object_triple_eq),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "object_id",
        Box::new(mrb_object_object_id),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "__id__",
        Box::new(mrb_object_object_id),
    );
    mrb_define_cmethod(vm, object_class.clone(), "to_s", Box::new(mrb_object_to_s));
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "inspect",
        Box::new(mrb_object_to_s),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "raise",
        Box::new(mrb_object_raise),
    );
    mrb_define_cmethod(vm, object_class.clone(), "nil?", Box::new(mrb_object_nil_p));
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "block_given?",
        Box::new(mrb_object_block_given),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "lambda",
        Box::new(mrb_object_lambda),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "proc",
        Box::new(mrb_object_lambda),
    );
    mrb_define_cmethod(vm, object_class.clone(), "is_a?", Box::new(mrb_object_is_a));
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "kind_of?",
        Box::new(mrb_object_is_a),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "class",
        Box::new(mrb_object_class),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "<=>",
        Box::new(mrb_object_compare),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "method_missing",
        Box::new(mrb_object_method_missing),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "extend",
        Box::new(mrb_object_extend),
    );
    mrb_define_cmethod(vm, object_class.clone(), "loop", Box::new(mrb_object_loop));
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "respond_to?",
        Box::new(mrb_object_respond_to),
    );
    mrb_define_cmethod(
        vm,
        object_class.clone(),
        "public_send",
        Box::new(mrb_object_public_send),
    );

    // define global consts:
    vm.consts.insert(
        "RUBY_VERSION".to_string(),
        Rc::new(RObject::string(crate::yamrb::vm::VERSION.to_string())),
    );
    vm.consts.insert(
        "MRUBY_VERSION".to_string(),
        Rc::new(RObject::string(crate::yamrb::vm::VERSION.to_string())),
    );
    vm.consts.insert(
        "MRUBY_EDGE_VERSION".to_string(),
        Rc::new(RObject::string(crate::yamrb::vm::VERSION.to_string())),
    );
    vm.consts.insert(
        "RUBY_ENGINE".to_string(),
        Rc::new(RObject::string(crate::yamrb::vm::ENGINE.to_string())),
    );
    mrb_define_cmethod(vm, object_class.clone(), "wasm?", Box::new(mrb_is_wasm));
}

pub fn mrb_self(vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    vm.getself()
}

#[cfg(feature = "wasi")]
pub fn mrb_kernel_puts(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let msg = args[0].clone();
    match &msg.value {
        RValue::String(s, _) => {
            println!("{}", String::from_utf8_lossy(&s.borrow()));
        }
        RValue::Integer(i) => {
            println!("{}", i);
        }
        _ => {
            let inspect = mrb_funcall(vm, Some(msg), "to_s", &[])?;
            let inspect: String = inspect.as_ref().try_into()?;
            println!("{}", inspect);
        }
    }
    Ok(Rc::new(RObject::nil()))
}

#[cfg(feature = "wasi")]
pub fn mrb_kernel_p(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let msg = args[0].clone();
    let inspect = mrb_funcall(vm, Some(msg), "inspect", &[])?;
    let inspect: String = inspect.as_ref().try_into()?;
    println!("{}", inspect);
    Ok(Rc::new(RObject::nil()))
}

#[cfg(feature = "wasi")]
pub fn mrb_kernel_debug(_vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    for (i, obj) in args.iter().enumerate() {
        dbg!(i, obj.clone());
    }
    Ok(Rc::new(RObject::nil()))
}

pub fn mrb_object_is_equal(_vm: &mut VM, lhs: Rc<RObject>, rhs: Rc<RObject>) -> Rc<RObject> {
    RObject::boolean(lhs.as_eq_value() == rhs.as_eq_value()).to_refcount_assigned()
}

pub fn mrb_object_is_not_equal(_vm: &mut VM, lhs: Rc<RObject>, rhs: Rc<RObject>) -> Rc<RObject> {
    RObject::boolean(lhs.as_eq_value() != rhs.as_eq_value()).to_refcount_assigned()
}

pub fn mrb_object_double_eq(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let lhs = vm.getself()?;
    let rhs = args[0].clone();
    Ok(mrb_object_is_equal(vm, lhs, rhs))
}

pub fn mrb_object_not_eq(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let lhs = vm.getself()?;
    let rhs = args[0].clone();
    Ok(mrb_object_is_not_equal(vm, lhs, rhs))
}

pub fn mrb_object_triple_eq(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let lhs = vm.getself()?;
    let rhs = args[0].clone();

    match (&lhs.value, &rhs.value) {
        (RValue::Integer(i1), RValue::Integer(i2)) => Ok(Rc::new(RObject::boolean(*i1 == *i2))),
        (RValue::Float(f1), RValue::Float(f2)) => Ok(Rc::new(RObject::boolean(*f1 == *f2))),
        (RValue::Symbol(sym1), RValue::Symbol(sym2)) => Ok(Rc::new(RObject::boolean(sym1 == sym2))),
        (RValue::String(s1, _), RValue::String(s2, _)) => Ok(Rc::new(RObject::boolean(s1 == s2))),
        (RValue::Class(c1), _) => match &lhs.value {
            RValue::Class(c2) => Ok(Rc::new(RObject::boolean(c1.sym_id == c2.sym_id))),
            _ => {
                let c2 = lhs.get_class(vm);
                Ok(Rc::new(RObject::boolean(c1.sym_id == c2.sym_id)))
            }
        },
        (RValue::Range(_s, _e, _v), _) => {
            let arg = vec![rhs];
            mrb_funcall(vm, Some(lhs), "include?", &arg)
        }
        // TODO: Implement object id for generic instance
        _ => Ok(Rc::new(RObject::boolean(false))),
    }
}

// Object#<=>: Comparison operator (returns -1, 0, or 1)
pub fn mrb_object_compare(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let lhs = vm.getself()?;
    let rhs = &args[0];

    // Use as_eq_value for comparison
    let lhs_val = lhs.as_eq_value();
    let rhs_val = rhs.as_eq_value();

    use ValueEquality::*;
    let result = match (&lhs_val, &rhs_val) {
        (Integer(a), Integer(b)) => {
            if a < b {
                -1
            } else if a > b {
                1
            } else {
                0
            }
        }
        (Float(a), Float(b)) => {
            if a < b {
                -1
            } else if a > b {
                1
            } else {
                0
            }
        }
        (String(a), String(b)) => {
            if a < b {
                -1
            } else if a > b {
                1
            } else {
                0
            }
        }
        (Integer(a), Float(b)) => {
            let a_float = *a as f64;
            if a_float < *b {
                -1
            } else if a_float > *b {
                1
            } else {
                0
            }
        }
        (Float(a), Integer(b)) => {
            let b_float = *b as f64;
            if a < &b_float {
                -1
            } else if a > &b_float {
                1
            } else {
                0
            }
        }
        (Symbol(a), Symbol(b)) => {
            if a < b {
                -1
            } else if a > b {
                1
            } else {
                0
            }
        }
        _ => {
            return Err(Error::ArgumentError(
                "comparison of incompatible types".to_string(),
            ));
        }
    };

    Ok(Rc::new(RObject::integer(result)))
}

pub fn mrb_object_object_id(vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    // Abstract method; do nothing
    let x = vm.getself()?.object_id.get();
    // ref: https://stackoverflow.com/questions/74491204/how-do-i-represent-an-i64-in-the-u64-domain
    let to_i64 = ((x as i64) ^ (1 << 63)) & (1 << 63) | (x & (u64::MAX >> 1)) as i64;
    Ok(Rc::new(RObject::integer(to_i64)))
}

pub fn mrb_object_to_s(vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let obj = vm.getself()?;
    if obj.is_main() {
        return Ok(RObject::string("main".to_string()).to_refcount_assigned());
    }
    let class = obj.get_class(vm);
    let addr = format!("{:018p}", Rc::as_ptr(&obj));
    Ok(RObject::string(format!("#<{}:{}>", class.full_name(), addr)).to_refcount_assigned())
}

pub fn mrb_object_raise(_vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    // TODO: accept exception class
    let msg = args[0].as_ref().try_into()?;
    let err = Error::RuntimeError(msg);
    Err(err)
}

fn mrb_object_nil_p(_vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    Ok(Rc::new(RObject::boolean(false)))
}

fn mrb_object_block_given(vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    // CALLINFO の has_block フラグをチェック
    let has_block = if let Some(ci) = vm.current_callinfo.as_ref() {
        ci.has_block.get()
    } else {
        false
    };

    Ok(Rc::new(RObject::boolean(has_block)))
}

pub fn mrb_object_initialize(_vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    // Abstract method; do nothing
    Ok(Rc::new(RObject::nil()))
}

pub fn mrb_object_lambda(_vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let proc = args[args.len() - 1].clone();
    if matches!(proc.value, RValue::Proc(_)) {
        Ok(proc)
    } else {
        Err(Error::RuntimeError(
            "Object#lambda expects a Proc as the last argument".to_string(),
        ))
    }
}

fn mrb_object_is_a(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let obj = vm.getself()?;
    let class_arg = &args[0];
    let is_a = match &class_arg.value {
        RValue::Class(c) => mrb_is_a(vm, obj, c.clone()),
        RValue::Module(m) => mrb_is_a(vm, obj, m.clone()),
        _ => {
            return Err(Error::ArgumentError(
                "Object#is_a? expects a Class or Module".to_string(),
            ));
        }
    };
    Ok(Rc::new(RObject::boolean(is_a)))
}

fn mrb_object_class(vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let obj = vm.getself()?;
    let class = obj.get_class(vm);
    Ok(RObject::class_or_module(class.as_module(), vm))
}

fn mrb_object_loop(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let block = args[0].clone();
    if !matches!(block.value, RValue::Proc(_)) {
        return Err(Error::ArgumentError(
            "Object#loop expects a block".to_string(),
        ));
    }

    let this = vm.getself()?;
    loop {
        mrb_call_block(vm, block.clone(), Some(this.clone()), &[], 0)?;
    }
}

fn mrb_object_respond_to(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let method_name: String = args[0].as_ref().try_into()?;
    let obj = vm.getself()?;
    let klass = obj.singleton_or_this_class(vm);
    let has_method = resolve_method(&klass, &method_name).is_some();
    Ok(Rc::new(RObject::boolean(has_method)))
}

fn mrb_object_public_send(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    if args.is_empty() {
        return Err(Error::ArgumentError(
            "wrong number of arguments (given 0, expected 1+)".to_string(),
        ));
    }

    let method_name: String = args[0].as_ref().try_into()?;
    let obj = vm.getself()?;
    let method_args = &args[1..];

    // For now, public_send behaves the same as send since we don't have visibility modifiers
    mrb_funcall(vm, Some(obj), &method_name, method_args)
}

fn mrb_object_method_missing(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let method_name_obj = &args
        .first()
        .ok_or_else(|| Error::Internal("[BUG] method_missing without any args".to_string()))?;
    let method_name: String = method_name_obj.as_ref().try_into()?;
    Err(Error::NoMethodError(format!(
        "undefined method `{}` for {}",
        method_name,
        vm.getself()?.get_class(vm).full_name()
    )))
}

pub fn mrb_is_a(vm: &mut VM, obj: Rc<RObject>, class: impl AsModule) -> bool {
    let obj_class = obj.get_class(vm);
    let target_module = class.as_module();
    for module in build_lookup_chain(&obj_class).iter() {
        if Rc::ptr_eq(module, &target_module) {
            return true;
        }
    }
    false
}

fn mrb_is_wasm(_vm: &mut VM, _args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let is_wasm = cfg!(target_arch = "wasm32");
    Ok(Rc::new(RObject::boolean(is_wasm)))
}

#[test]
fn test_mrb_object_is_equal() {
    let mut vm = VM::empty();

    let lhs = RObject::integer(1).to_refcount_assigned();
    let rhs = RObject::integer(1).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::integer(1).to_refcount_assigned();
    let rhs = RObject::integer(3).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let lhs = RObject::string("mruby/edge is Ruby".into()).to_refcount_assigned();
    let rhs = RObject::string("mruby/edge is Ruby".into()).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::string("mruby/edge is Ruby".into()).to_refcount_assigned();
    let rhs = RObject::string("mruby/edge is not Ruby".into()).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let lhs = RObject::symbol("some".into()).to_refcount_assigned();
    let rhs = RObject::symbol("some".into()).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::symbol("some".into()).to_refcount_assigned();
    let rhs = RObject::symbol("other".into()).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let lhs = RObject::boolean(true).to_refcount_assigned();
    let rhs = RObject::boolean(true).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::boolean(false).to_refcount_assigned();
    let rhs = RObject::boolean(false).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::boolean(true).to_refcount_assigned();
    let rhs = RObject::boolean(false).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let lhs = RObject::float(0.1).to_refcount_assigned();
    let rhs = RObject::float(0.1).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::float(0.2).to_refcount_assigned();
    let rhs = RObject::float(0.1).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let lhs = RObject::nil().to_refcount_assigned();
    let rhs = RObject::nil().to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::integer(100).to_refcount_assigned();
    let rhs = RObject::nil().to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let lhs = RObject::integer(100).to_refcount_assigned();
    let rhs = RObject::nil().to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);
}

#[test]
fn test_mrb_object_is_equal_range() {
    let mut vm = VM::empty();

    let s = RObject::integer(1).to_refcount_assigned();
    let e = RObject::integer(10).to_refcount_assigned();
    let lhs = RObject::range(s.clone(), e.clone(), true).to_refcount_assigned();
    let rhs = RObject::range(s.clone(), e.clone(), true).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let s = RObject::integer(1).to_refcount_assigned();
    let e = RObject::integer(10).to_refcount_assigned();
    let lhs = RObject::range(s.clone(), e.clone(), true).to_refcount_assigned();
    let rhs = RObject::range(s.clone(), e.clone(), false).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let s = RObject::integer(1).to_refcount_assigned();
    let e = RObject::integer(10).to_refcount_assigned();
    let e2 = RObject::integer(11).to_refcount_assigned();
    let lhs = RObject::range(s.clone(), e.clone(), true).to_refcount_assigned();
    let rhs = RObject::range(s.clone(), e2.clone(), true).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let s = RObject::string("a".into()).to_refcount_assigned();
    let e = RObject::string("z".into()).to_refcount_assigned();
    let lhs = RObject::range(s.clone(), e.clone(), true).to_refcount_assigned();
    let rhs = RObject::range(s.clone(), e.clone(), true).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let s = RObject::string("a".into()).to_refcount_assigned();
    let e = RObject::string("z".into()).to_refcount_assigned();
    let e2 = RObject::string("A".into()).to_refcount_assigned();
    let lhs = RObject::range(s.clone(), e.clone(), true).to_refcount_assigned();
    let rhs = RObject::range(s.clone(), e2.clone(), true).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);
}

#[test]
fn test_mrb_object_is_equal_array() {
    let mut vm = VM::empty();

    let lhs = RObject::array(vec![]).to_refcount_assigned();
    let rhs = RObject::array(vec![]).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = vec![
        RObject::integer(1).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
        RObject::integer(3).to_refcount_assigned(),
    ];
    let rhs = vec![
        RObject::integer(1).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
        RObject::integer(3).to_refcount_assigned(),
    ];
    let lhs = RObject::array(lhs).to_refcount_assigned();
    let rhs = RObject::array(rhs).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = vec![
        RObject::integer(1).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
        RObject::integer(3).to_refcount_assigned(),
    ];
    let rhs = vec![
        RObject::integer(1).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
        RObject::integer(4).to_refcount_assigned(),
    ];
    let lhs = RObject::array(lhs).to_refcount_assigned();
    let rhs = RObject::array(rhs).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);
}

#[test]
fn test_mrb_object_is_equal_hash() {
    use crate::yamrb::prelude::hash::*;

    let mut vm = VM::empty();

    let lhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    let rhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    mrb_hash_set_index(
        lhs.clone(),
        RObject::symbol("key1".into()).to_refcount_assigned(),
        RObject::integer(1).to_refcount_assigned(),
    )
    .expect("set index failed");
    mrb_hash_set_index(
        lhs.clone(),
        RObject::symbol("key2".into()).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
    )
    .expect("set index failed");

    let rhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    mrb_hash_set_index(
        rhs.clone(),
        RObject::symbol("key2".into()).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
    )
    .expect("set index failed");
    mrb_hash_set_index(
        rhs.clone(),
        RObject::symbol("key1".into()).to_refcount_assigned(),
        RObject::integer(1).to_refcount_assigned(),
    )
    .expect("set index failed");

    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    mrb_hash_set_index(
        lhs.clone(),
        RObject::symbol("key1".into()).to_refcount_assigned(),
        RObject::integer(1).to_refcount_assigned(),
    )
    .expect("set index failed");
    mrb_hash_set_index(
        lhs.clone(),
        RObject::symbol("key2".into()).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
    )
    .expect("set index failed");

    let rhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    mrb_hash_set_index(
        rhs.clone(),
        RObject::symbol("key2".into()).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
    )
    .expect("set index failed");
    mrb_hash_set_index(
        rhs.clone(),
        RObject::symbol("key1".into()).to_refcount_assigned(),
        RObject::integer(3).to_refcount_assigned(),
    )
    .expect("set index failed");

    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);

    let lhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    mrb_hash_set_index(
        lhs.clone(),
        RObject::symbol("key1".into()).to_refcount_assigned(),
        RObject::integer(1).to_refcount_assigned(),
    )
    .expect("set index failed");
    mrb_hash_set_index(
        lhs.clone(),
        RObject::symbol("key2".into()).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
    )
    .expect("set index failed");

    let rhs = RObject::hash(RHashMap::default()).to_refcount_assigned();
    mrb_hash_set_index(
        rhs.clone(),
        RObject::symbol("key2".into()).to_refcount_assigned(),
        RObject::integer(2).to_refcount_assigned(),
    )
    .expect("set index failed");
    mrb_hash_set_index(
        rhs.clone(),
        RObject::symbol("key1-b".into()).to_refcount_assigned(),
        RObject::integer(1).to_refcount_assigned(),
    )
    .expect("set index failed");

    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);
}

#[test]
fn test_mrb_object_is_equal_klass() {
    let mut vm = VM::empty();

    let lhs: Rc<RClass> = vm.get_class_by_name("String");
    let rhs: Rc<RClass> = RObject::string("String".into()).get_class(&vm);
    let lhs = RObject::class(lhs.clone(), &mut vm);
    let rhs = RObject::class(rhs.clone(), &mut vm);
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs: Rc<RClass> = RObject::integer(5471).get_class(&vm);
    let rhs: Rc<RClass> = RObject::string("String".into()).get_class(&vm);
    let lhs = RObject::class(lhs.clone(), &mut vm);
    let rhs = RObject::class(rhs.clone(), &mut vm);
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);
}

#[test]
fn test_mrb_object_is_equal_instance() {
    let mut vm = VM::empty();

    let lhs = RObject::instance(vm.object_class.clone()).to_refcount_assigned();
    let rhs = lhs.clone();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(ret);

    let lhs = RObject::instance(vm.object_class.clone()).to_refcount_assigned();
    let rhs = RObject::instance(vm.object_class.clone()).to_refcount_assigned();
    let ret: bool = mrb_object_is_equal(&mut vm, lhs, rhs)
        .as_ref()
        .try_into()
        .expect("must return bool");
    assert!(!ret);
}

fn mrb_object_extend(vm: &mut VM, args: &[Rc<RObject>]) -> Result<Rc<RObject>, Error> {
    let this = vm.getself()?;

    if args.is_empty() {
        return Err(Error::ArgumentError(
            "wrong number of arguments (given 0, expected 1+)".to_string(),
        ));
    }

    // Initialize or get singleton class
    let singleton_class = this.initialize_or_get_singleton_class(vm);

    // Extend with each module argument
    for arg in args.iter().rev() {
        let module = match &arg.value {
            RValue::Module(m) => m.clone(),
            RValue::Class(c) => c.module.clone(),
            RValue::Nil => {
                continue;
            }
            _ => {
                return Err(Error::ArgumentError(
                    "wrong argument type (expected Module)".to_string(),
                ));
            }
        };

        // Add module to extended_modules
        singleton_class
            .extended_modules
            .borrow_mut()
            .insert(0, module);
    }

    Ok(this)
}