just so much groundwork

lib/input.ts

@@ -3,15 +3,21 @@ export class InputManager {
   private mouseStates: Map<string | number, boolean> = new Map();
   private mouseLocation: { x: number; y: number } = { x: 0, y: 0 };
   private mouseDelta: { x: number; y: number } = { x: 0, y: 0 };
+
+  private keyEvents: Map<string | number, () => void> = new Map();
+  private mouseEvents: Map<string | number, () => void> = new Map();
+
   constructor() {
     document.addEventListener("keydown", (e) => {
       this.keyStates.set(e.key, true);
+      this.keyEvents.get(e.key)?.call(e);
     });
     document.addEventListener("keyup", (e) => {
       this.keyStates.set(e.key, false);
     });
     document.addEventListener("mousedown", (e) => {
       this.mouseStates.set(e.button, true);
+      this.mouseEvents.get(e.button)?.call(e);
     });
     document.addEventListener("mouseup", (e) => {
       this.mouseStates.set(e.button, false);
@@ -38,4 +44,18 @@ export class InputManager {
   getMouseDelta() {
     return this.mouseDelta;
   }
+
+  onKey(key: string | number, cb: () => void) {
+    this.keyEvents.set(key, cb);
+  }
+  onMouse(key: string | number, cb: () => void) {
+    this.mouseEvents.set(key, cb);
+  }
+
+  offKey(key: string | number) {
+    this.keyEvents.delete(key);
+  }
+  offMouse(key: string | number) {
+    this.mouseEvents.delete(key);
+  }
 }