From 11b97855c4c1ef756fa1546ff0137aad1a910f34 Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sun, 3 May 2015 18:13:10 +0000
Subject: console.c: pressed? on Windows

* ext/io/console/console.c (console_key_pressed_p): new method
  IO#pressed? on Windows.  [EXPERIMENTAL]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@50427 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 ext/io/console/console.c | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'ext/io/console/console.c')

diff --git a/ext/io/console/console.c b/ext/io/console/console.c
index 2bf6dc9468..8f54b5dd09 100644
--- a/ext/io/console/console.c
+++ b/ext/io/console/console.c
@@ -697,10 +697,36 @@ console_cursor_set(VALUE io, VALUE cpos)
     if (RARRAY_LEN(cpos) != 2) rb_raise(rb_eArgError, "expected 2D coordinate");
     return console_goto(io, RARRAY_AREF(cpos, 0), RARRAY_AREF(cpos, 1));
 }
+
+#include "win32_vk.h"
+
+static VALUE
+console_key_pressed_p(VALUE io, VALUE k)
+{
+    int vk = -1;
+
+    if (FIXNUM_P(k)) {
+	vk = NUM2UINT(k);
+    }
+    else {
+	const struct vktable *t;
+	if (SYMBOL_P(k)) {
+	    k = rb_sym2str(k);
+	}
+	else {
+	    StringValueCStr(k);
+	}
+	t = console_win32_vk(RSTRING_PTR(k), RSTRING_LEN(k));
+	if (!t) rb_raise(rb_eArgError, "unknown virtual key code: %"PRIsVALUE, k);
+	vk = t->vk;
+    }
+    return GetKeyState(vk) & 0x80 ? Qtrue : Qfalse;
+}
 #else
 # define console_goto rb_f_notimplement
 # define console_cursor_pos rb_f_notimplement
 # define console_cursor_set rb_f_notimplement
+# define console_key_pressed_p rb_f_notimplement
 #endif
 
 /*
@@ -850,6 +876,7 @@ InitVM_console(void)
     rb_define_method(rb_cIO, "goto", console_goto, 2);
     rb_define_method(rb_cIO, "cursor", console_cursor_pos, 0);
     rb_define_method(rb_cIO, "cursor=", console_cursor_set, 1);
+    rb_define_method(rb_cIO, "pressed?", console_key_pressed_p, 1);
     rb_define_singleton_method(rb_cIO, "console", console_dev, -1);
     {
 	VALUE mReadable = rb_define_module_under(rb_cIO, "generic_readable");
-- 
cgit v1.2.3