/*
 *  supplement/terminal.c  --  Terminal methods
 */

#include "terminal.h"

#if   HAVE_HEADER_RUBYIO_H
    #include <rubyio.h>
#elif HAVE_HEADER_RUBY_IO_H
    #include <ruby/io.h>
#endif

#include <sys/ioctl.h>
#include <termios.h>

static VALUE io_unget( VALUE);
static VALUE io_reset( VALUE);


#ifndef RUBY_VM
  #define RB_SYS_FAIL( fptr)     rb_sys_fail( ((OpenFile *) fptr)->path);
#else
  #define RB_SYS_FAIL( fptr)     rb_sys_fail_str( ((rb_io_t *) fptr)->pathv);
#endif

/*
 *  call-seq:
 *     io.unget(str, ...)   -> nil
 *
 *  Feed <em>str</em> into the TTY's input queue.
 *
 *     $stdin.unget "hello\n"    #=> nil
 */

VALUE
rb_io_unget( int argc, VALUE *argv, VALUE io)
{
#ifndef RUBY_VM
    OpenFile *fptr;
#else
    rb_io_t *fptr;
#endif
    int fd;
    struct termios oldtio, newtio;
    void *v[5];

    GetOpenFile( io, fptr);
#ifndef RUBY_VM
    fd = fileno( fptr->f);
#else
    fd = fptr->fd;
#endif

    if (tcgetattr( fd, &oldtio) < 0)
        RB_SYS_FAIL( fptr);
    newtio = oldtio;
    newtio.c_iflag &= ~ICRNL;
    if (tcsetattr( fd, TCSANOW, &newtio) < 0)
        RB_SYS_FAIL( fptr);

    v[0] = &fd, v[1] = fptr, v[2] = &oldtio,
        v[3] = &argc, v[4] = (void *) argv;
    return rb_ensure( io_unget, (VALUE) v, io_reset, (VALUE) v);
}

VALUE
io_unget( VALUE v)
{
    void **vp = (void **) v;
    VALUE str;
    long i;
    int j;
    char *p;
    int *argc;
    VALUE *argv;

    argc = (int *)vp[3], argv = (VALUE *) vp[4];
    for (j = 0 ; j < *argc; ++j) {
        str = argv[ j];
        p = RSTRING_PTR(str);
        for (i = RSTRING_LEN(str); i; --i, ++p)
            if (ioctl( *(int *) vp[0], TIOCSTI, p) < 0)
                RB_SYS_FAIL( vp[1]);
    }
    return Qnil;
}

VALUE
io_reset( VALUE v)
{
    void **vp = (void **) v;

    if (tcsetattr( *(int *) vp[0], TCSANOW, (struct termios *) vp[2]) < 0)
        RB_SYS_FAIL( vp[1]);

    return Qnil;
}


/*
 *  call-seq:
 *     io.winsize()   -> ary
 *
 *  Get the available window space.
 *
 *     cols, rows, xpixel, ypixel = $stdout.winsize
 */

VALUE
rb_io_winsize( VALUE self)
{
#ifndef RUBY_VM
    OpenFile *fptr;
#else
    rb_io_t *fptr;
#endif
    int fd;
    struct winsize w;
    VALUE r;

    GetOpenFile( self, fptr);
#ifndef RUBY_VM
    fd = fileno( fptr->f);
#else
    fd = fptr->fd;
#endif

    if (ioctl( fd, TIOCGWINSZ, &w) < 0)
        RB_SYS_FAIL( fptr);
    r = rb_ary_new2( 4);
    rb_ary_store( r, 0, INT2NUM( w.ws_col));
    rb_ary_store( r, 1, INT2NUM( w.ws_row));
    rb_ary_store( r, 2, INT2NUM( w.ws_xpixel));
    rb_ary_store( r, 3, INT2NUM( w.ws_ypixel));
    return r;
}


void Init_terminal( void)
{
    rb_define_method( rb_cIO, "unget", rb_io_unget, -1);
    rb_define_method( rb_cIO, "winsize", rb_io_winsize, 0);
}