1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
use std::ffi::CStr;

use crate::extn::prelude::*;

const ENUMERATOR_CSTR: &CStr = qed::const_cstr_from_str!("Enumerator\0");
static ENUMERATOR_RUBY_SOURCE: &[u8] = include_bytes!("enumerator.rb");
static ENUMERATOR_LAZY_RUBY_SOURCE: &[u8] = include_bytes!("lazy.rb");

pub fn init(interp: &mut Artichoke) -> InitializeResult<()> {
    if interp.is_class_defined::<Enumerator>() {
        return Ok(());
    }

    let spec = class::Spec::new("Enumerator", ENUMERATOR_CSTR, None, None)?;
    interp.def_class::<Enumerator>(spec)?;
    interp.eval(ENUMERATOR_RUBY_SOURCE)?;
    interp.eval(ENUMERATOR_LAZY_RUBY_SOURCE)?;

    Ok(())
}

#[derive(Debug, Clone, Copy)]
pub struct Enumerator;