util-src/windows.c: Added get_consolecolor, set_consolecolor.

diff --git a/util-src/windows.c b/util-src/windows.c
index 12bd7ce9623259a05ddd83bbcc9f66f69f9fe966..e1e07608756ae1f9be01d019555fa4ef5d458f78 100644 (file)
--- a/util-src/windows.c
+++ b/util-src/windows.c
@@ -43,9 +43,38 @@ static int Lget_nameservers(lua_State *L) {
        }
 }
 
+static void lassert(lua_State *L, BOOL test, char* string) {
+       if (!test) {
+               luaL_error(L, "%s: %d", string, GetLastError());
+       }
+}
+
+static int Lget_consolecolor(lua_State *L) {
+       HWND console = GetStdHandle(STD_OUTPUT_HANDLE);
+       WORD color; DWORD read_len;
+       
+       CONSOLE_SCREEN_BUFFER_INFO info;
+       
+       lassert(L, console != INVALID_HANDLE_VALUE, "GetStdHandle");
+       lassert(L, GetConsoleScreenBufferInfo(console, &info), "GetConsoleScreenBufferInfo");
+       lassert(L, ReadConsoleOutputAttribute(console, &color, sizeof(WORD), info.dwCursorPosition, &read_len), "ReadConsoleOutputAttribute");
+
+       lua_pushnumber(L, color);
+       return 1;
+}
+static int Lset_consolecolor(lua_State *L) {
+       int color = luaL_checkint(L, 1);
+       HWND console = GetStdHandle(STD_OUTPUT_HANDLE);
+       lassert(L, console != INVALID_HANDLE_VALUE, "GetStdHandle");
+       lassert(L, SetConsoleTextAttribute(console, color), "SetConsoleTextAttribute");
+       return 0;
+}
+
 static const luaL_Reg Reg[] =
 {
        { "get_nameservers",    Lget_nameservers        },
+       { "get_consolecolor",   Lget_consolecolor       },
+       { "set_consolecolor",   Lset_consolecolor       },
        { NULL,         NULL    }
 };