ruby/goruby.c at master · ruby/ruby

Cannot retrieve contributors at this time

68 lines (61 sloc) 1.52 KB

	static void Init_golf(void);
	static void goruby_options(int argc, char *argv);
	static int goruby_run_node(void *arg);
	#define ruby_options goruby_options
	#define ruby_run_node goruby_run_node
	#include "main.c"
	#undef ruby_options
	#undef ruby_run_node

	#if defined _WIN32
	#include <io.h>
	#include <fcntl.h>
	#define pipe(p) _pipe(p, 32L, _O_NOINHERIT)
	#elif defined HAVE_UNISTD_H
	#include <unistd.h>
	#endif

	RUBY_EXTERN void ruby_options(int argc, char *argv);
	RUBY_EXTERN int ruby_run_node(void*);
	RUBY_EXTERN void ruby_init_ext(const char name, void (init)(void));

	#include "golf_prelude.c"

	static VALUE
	init_golf(VALUE arg)
	{
	Init_golf();
	rb_provide("golf.so");
	return arg;
	}

	void *
	goruby_options(int argc, char **argv)
	{
	static const char cmd[] = "END{require 'irb';IRB.start}";
	int rw[2], infd;
	void *ret;

	if ((isatty(0) && isatty(1) && isatty(2)) && (pipe(rw) == 0)) {
	ssize_t n;
	infd = dup(0);
	if (infd < 0) {
	close(rw[0]);
	close(rw[1]);
	goto no_irb;
	}
	dup2(rw[0], 0);
	close(rw[0]);
	n = write(rw[1], cmd, sizeof(cmd) - 1);
	close(rw[1]);
	ret = n > 0 ? ruby_options(argc, argv) : NULL;
	dup2(infd, 0);
	close(infd);
	return ret;
	}
	no_irb:
	return ruby_options(argc, argv);
	}

	int
	goruby_run_node(void *arg)
	{
	int state;
	if (NIL_P(rb_protect(init_golf, Qtrue, &state))) {
	return state == EXIT_SUCCESS ? EXIT_FAILURE : state;
	}
	return ruby_run_node(arg);
	}