1 files changed, 86 insertions, 0 deletions
diff --git a/hid/lib/drivers-stm32/usb/keyboard-stm32.h b/hid/lib/drivers-stm32/usb/keyboard-stm32.h
new file mode 100644
index 00000000..b788be19
--- /dev/null
+++ b/hid/lib/drivers-stm32/usb/keyboard-stm32.h
@@ -0,0 +1,86 @@
+/*****************************************************************************
+#                                                                            #
+#    KVMD - The main PiKVM daemon.                                           #
+#                                                                            #
+#    Copyright (C) 2018-2022  Maxim Devaev <[email protected]>               #
+#                                                                            #
+#    This program is free software: you can redistribute it and/or modify    #
+#    it under the terms of the GNU General Public License as published by    #
+#    the Free Software Foundation, either version 3 of the License, or       #
+#    (at your option) any later version.                                     #
+#                                                                            #
+#    This program is distributed in the hope that it will be useful,         #
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of          #
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           #
+#    GNU General Public License for more details.                            #
+#                                                                            #
+#    You should have received a copy of the GNU General Public License       #
+#    along with this program.  If not, see <https://www.gnu.org/licenses/>.  #
+#                                                                            #
+*****************************************************************************/
+
+#pragma once
+
+#include "keyboard.h"
+#include "hid-wrapper-stm32.h"
+#include <USBComposite.h>
+#include "keymap.h"
+#include "tools.h"
+
+namespace DRIVERS {
+
+	const uint8_t reportDescriptionKeyboard[] = {
+		HID_KEYBOARD_REPORT_DESCRIPTOR(),
+	};
+
+	class UsbKeyboard : public Keyboard {
+		public:
+			UsbKeyboard(HidWrapper& _hidWrapper) : Keyboard(USB_KEYBOARD),
+				_hidWrapper(_hidWrapper), _keyboard(_hidWrapper.usbHid) {
+				_hidWrapper.addReportDescriptor(reportDescriptionKeyboard, sizeof(reportDescriptionKeyboard));
+			}
+
+			void begin() override {
+				_hidWrapper.begin();
+				_keyboard.begin();
+			}
+
+			void clear() override {
+				_keyboard.releaseAll();
+			}
+
+			void sendKey(uint8_t code, bool state) override {
+				uint16_t usb_code = keymapUsb(code);
+				if (usb_code == KEY_ERROR_UNDEFINED) {
+					return;
+				}
+
+				if(usb_code >= KEY_LEFT_CTRL && usb_code <= KEY_RIGHT_WINDOWS) {
+					usb_code = usb_code - KEY_LEFT_CTRL + 0x80;
+				} else {
+					usb_code += KEY_HID_OFFSET;
+				}
+
+				state ? _keyboard.press(usb_code) : _keyboard.release(usb_code);
+			}
+
+			bool isOffline() override {
+				return USBComposite == false;
+			}
+
+			KeyboardLedsState getLeds() override {
+				uint8 leds = _keyboard.getLEDs();
+				KeyboardLedsState result = {
+					.caps = leds & 0b00000010,
+					.scroll = leds & 0b00000100,
+					.num = leds & 0b00000001,
+				};
+				return result;
+			}
+
+		private:
+			HidWrapper& _hidWrapper;
+			HIDKeyboard _keyboard;
+			static constexpr uint8 KEY_ERROR_UNDEFINED = 3;
+	};
+}