ruby/goruby.c

69 строки
1.5 KiB
C

static void Init_golf_prelude(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*);
#include "golf_prelude.rbbin"
static VALUE
init_golf(VALUE arg)
{
Init_golf_prelude();
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 (ruby_executable_node(arg, NULL) &&
NIL_P(rb_protect(init_golf, Qtrue, &state))) {
return state == EXIT_SUCCESS ? EXIT_FAILURE : state;
}
return ruby_run_node(arg);
}