/***** BEGIN LICENSE BLOCK *****

* Version: EPL 2.0/GPL 2.0/LGPL 2.1

*

* The contents of this file are subject to the Eclipse Public

* License Version 2.0 (the "License"); you may not use this file

* except in compliance with the License. You may obtain a copy of

* the License at http://www.eclipse.org/legal/epl-v20.html

*

* Software distributed under the License is distributed on an "AS

* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or

* implied. See the License for the specific language governing

* rights and limitations under the License.

*

* Copyright (C) 2013 JRuby Contributors

*

* Alternatively, the contents of this file may be used under the terms of

* either of the GNU General Public License Version 2 or later (the "GPL"),

* or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),

* in which case the provisions of the GPL or the LGPL are applicable instead

* of those above. If you wish to allow use of your version of this file only

* under the terms of either the GPL or the LGPL, and not to allow others to

* use your version of this file under the terms of the EPL, indicate your

* decision by deleting the provisions above and replace them with the notice

* and other provisions required by the GPL or the LGPL. If you do not delete

* the provisions above, a recipient may use your version of this file under

* the terms of any one of the EPL, the GPL or the LGPL.

***** END LICENSE BLOCK *****/

package org.jruby;

import org.jruby.anno.JRubyClass;

import org.jruby.anno.JRubyMethod;

import org.jruby.api.Convert;

import org.jruby.common.IRubyWarnings;

import org.jruby.runtime.Arity;

import org.jruby.runtime.Block;

import org.jruby.runtime.ThreadContext;

import org.jruby.runtime.Visibility;

import org.jruby.runtime.builtin.IRubyObject;

import org.jruby.util.ArraySupport;

import static org.jruby.api.Convert.castAsProc;

import static org.jruby.api.Error.typeError;

import static org.jruby.api.Warn.warn;

@JRubyClass(name = "Enumerator::Generator")

public class RubyGenerator extends RubyObject {

public static RubyClass createGeneratorClass(ThreadContext context, RubyClass Object, RubyClass Enumerator, RubyModule Enumerable) {

return Enumerator.defineClassUnder(context, "Generator", Object, RubyGenerator::new).

include(context, Enumerable).

defineMethods(context, RubyGenerator.class);

}

public RubyGenerator(Ruby runtime, RubyClass klass) {

super(runtime, klass);

}

// generator_initialize

@JRubyMethod(visibility = Visibility.PRIVATE, optional = 1, checkArity = false)

public IRubyObject initialize(ThreadContext context, IRubyObject[] args, Block block) {

if (Arity.checkArgumentCount(context, args, 0, 1) == 0) {

proc = RubyProc.newProc(context.runtime, block, Block.Type.PROC);

} else {

proc = Convert.castAsProc(context, args[0]);

if (block.isGiven()) warn(context, "given block not used");

}

return this;

}

// generator_init_copy

@JRubyMethod(visibility = Visibility.PRIVATE)

public IRubyObject initialize_copy(ThreadContext context, IRubyObject other) {

if (other instanceof RubyGenerator generator) {

checkFrozen();

this.proc = generator.proc;

return this;

}

throw typeError(context, other, context.runtime.getGenerator());

}

// generator_each

@JRubyMethod(rest = true, keywords = true)

public IRubyObject each(ThreadContext context, IRubyObject[] args, Block block) {

return proc.call(context, ArraySupport.newCopy(RubyYielder.newYielder(context, block), args));

}

public RubyProc getProc() {

return proc;

}

private RubyProc proc;

}