Non-blocking console

src/client/component/console.cpp

@@ -17,8 +17,198 @@ namespace console
 {
 	namespace
 	{
-		bool kill = false;
+		struct
-		std::thread console_thread;
+		{
+			bool kill;
+			std::thread thread;
+			HANDLE kill_event;
+			char buffer[512]{};
+			int cursor;
+			std::int32_t history_index = -1;
+			std::deque<std::string> history{};
+		} con{};
+
+		void set_cursor_pos(int x)
+		{
+			const auto handle = GetStdHandle(STD_OUTPUT_HANDLE);
+			CONSOLE_SCREEN_BUFFER_INFO info{};
+			GetConsoleScreenBufferInfo(handle, &info);
+			info.dwCursorPosition.X = static_cast<short>(x);
+			SetConsoleCursorPosition(handle, info.dwCursorPosition);
+		}
+
+		void show_cursor(const bool show)
+		{
+			const auto handle = GetStdHandle(STD_OUTPUT_HANDLE);
+			CONSOLE_CURSOR_INFO info{};
+			GetConsoleCursorInfo(handle, &info);
+			info.bVisible = show;
+			SetConsoleCursorInfo(handle, &info);
+		}
+
+		void update()
+		{
+			show_cursor(false);
+			set_cursor_pos(0);
+
+			printf("%s", con.buffer);
+
+			set_cursor_pos(con.cursor);
+			show_cursor(true);
+		}
+
+		void clear_output()
+		{
+			show_cursor(false);
+			set_cursor_pos(0);
+
+			for (auto i = 0; i < std::strlen(con.buffer); i++)
+			{
+				printf(" ");
+			}
+
+			set_cursor_pos(con.cursor);
+			show_cursor(true);
+		}
+
+		void clear()
+		{
+			clear_output();
+			strncpy_s(con.buffer, "", sizeof(con.buffer));
+			con.cursor = 0;
+			set_cursor_pos(0);
+		}
+
+		void handle_input(const INPUT_RECORD record)
+		{
+			if (record.EventType != KEY_EVENT || !record.Event.KeyEvent.bKeyDown)
+			{
+				return;
+			}
+
+			const auto key = record.Event.KeyEvent.wVirtualKeyCode;
+			switch (key)
+			{
+			case VK_UP:
+			{
+				if (++con.history_index >= con.history.size())
+				{
+					con.history_index = static_cast<int>(con.history.size()) - 1;
+				}
+
+				clear();
+
+				if (con.history_index != -1)
+				{
+					strncpy_s(con.buffer, con.history.at(con.history_index).data(), sizeof(con.buffer));
+					con.cursor = static_cast<int>(strlen(con.buffer));
+					update();
+				}
+
+				break;
+			}
+			case VK_DOWN:
+			{
+				if (--con.history_index < -1)
+				{
+					con.history_index = -1;
+				}
+
+				clear();
+
+				if (con.history_index != -1)
+				{
+					strncpy_s(con.buffer, con.history.at(con.history_index).data(), sizeof(con.buffer));
+					con.cursor = static_cast<int>(strlen(con.buffer));
+					update();
+				}
+
+				break;
+			}
+			case VK_LEFT:
+			{
+				if (con.cursor > 0)
+				{
+					con.cursor--;
+					set_cursor_pos(con.cursor);
+				}
+
+				break;
+			}
+			case VK_RIGHT:
+			{
+				if (con.cursor < std::strlen(con.buffer))
+				{
+					con.cursor++;
+					set_cursor_pos(con.cursor);
+				}
+
+				break;
+			}
+			case VK_RETURN:
+			{
+				if (con.history_index != -1)
+				{
+					const auto itr = con.history.begin() + con.history_index;
+
+					if (*itr == con.buffer)
+					{
+						con.history.erase(con.history.begin() + con.history_index);
+					}
+				}
+
+				con.history.push_front(con.buffer);
+
+				if (con.history.size() > 10)
+				{
+					con.history.erase(con.history.begin() + 10);
+				}
+
+				con.history_index = -1;
+
+				game::Cbuf_AddText(0, 0, con.buffer);
+
+				strncpy_s(con.buffer, "", sizeof(con.buffer));
+				con.cursor = 0;
+
+				printf("\r\n");
+				break;
+			}
+			case VK_BACK:
+			{
+				clear_output();
+
+				std::memmove(con.buffer + con.cursor - 1, con.buffer + con.cursor,
+					strlen(con.buffer) + 1 - con.cursor);
+				con.cursor--;
+
+				update();
+
+				break;
+			}
+			default:
+			{
+				const auto c = record.Event.KeyEvent.uChar.AsciiChar;
+				if (!c)
+				{
+					break;
+				}
+
+				if (std::strlen(con.buffer) + 1 >= sizeof(con.buffer))
+				{
+					break;
+				}
+
+				std::memmove(con.buffer + con.cursor + 1, 
+					con.buffer + con.cursor, std::strlen(con.buffer) + 1 - con.cursor);
+				con.buffer[con.cursor] = c;
+				con.cursor++;
+				
+				update();
+				break;
+			}
+			}
+		}
 	}
 
 	std::string format(va_list* ap, const char* message)
@@ -26,14 +216,37 @@ namespace console
 		static thread_local char buffer[0x1000];
 
 		const auto count = _vsnprintf_s(buffer, sizeof(buffer), sizeof(buffer), message, *ap);
+		if (count < 0)
+		{
+			return {};
+		}
 
-		if (count < 0) return {};
 		return {buffer, static_cast<size_t>(count)};
 	}
 
+	uint8_t get_attribute(const int type)
+	{
+		switch (type)
+		{
+		case con_type_info:
+			return 7; // white
+		case con_type_warning:
+			return 6; // yellow
+		case con_type_error:
+			return 4; // red
+		case con_type_debug:
+			return 3; // cyan
+		}
+
+		return 7;
+	}
+
 	void dispatch_message(const int type, const std::string& message)
 	{
+		const auto handle = GetStdHandle(STD_OUTPUT_HANDLE);
+		SetConsoleTextAttribute(handle, get_attribute(type));
 		printf("%s", message.data());
+		SetConsoleTextAttribute(handle, get_attribute(con_type_info));
 		game_console::print(type, message);
 	}
 
@@ -60,25 +273,67 @@ namespace console
 			ShowWindow(GetConsoleWindow(), SW_SHOW);
 			SetConsoleTitle("H1-Mod");
 
-			console_thread = utils::thread::create_named_thread("Console", []()
+			con.kill_event = CreateEvent(NULL, TRUE, FALSE, NULL);
-			{
-				std::string cmd;
 
-				while (!kill)
+			con.thread = utils::thread::create_named_thread("Console", []()
+			{
+				const auto handle = GetStdHandle(STD_INPUT_HANDLE);
+				HANDLE handles[2] = {handle, con.kill_event};
+				MSG msg{};
+
+				INPUT_RECORD record{};
+				DWORD num_events{};
+
+				while (!con.kill)
 				{
-					std::getline(std::cin, cmd);
+					const auto result = MsgWaitForMultipleObjects(2, handles, FALSE, INFINITE, QS_ALLINPUT);
-					game::Cbuf_AddText(0, 0, cmd.data());
+					if (con.kill)
+					{
+						return;
+					}
+
+					switch (result)
+					{
+					case WAIT_OBJECT_0:
+					{
+						if (!ReadConsoleInput(handle, &record, 1, &num_events) || num_events == 0)
+						{
+							break;
+						}
+
+						handle_input(record);
+						break;
+					}
+					case WAIT_OBJECT_0 + 1:
+					{
+						if (!PeekMessageA(&msg, GetConsoleWindow(), NULL, NULL, PM_REMOVE))
+						{
+							break;
+						}
+
+						if (msg.message == WM_QUIT)
+						{
+							command::execute("quit", false);
+							break;
+						}
+
+						TranslateMessage(&msg);
+						DispatchMessage(&msg);
+						break;
+					}
+					}
 				}
 			});
 		}
 
 		void pre_destroy() override
 		{
-			kill = true;
+			con.kill = true;
+			SetEvent(con.kill_event);
 
-			if (console_thread.joinable())
+			if (con.thread.joinable())
 			{
-				console_thread.join();
+				con.thread.join();
 			}
 		}
 	};