feat: command hint

Author: LittleSwift

electron/main.js

@@ -48,6 +48,9 @@ export function parseRules(wasmBytes, rules) {
       ) {
         if (rule.parsedName) {
           if (matches[rule.parsedName] !== undefined) {
+            if (rule.first) {
+              continue;
+            }
             throw new Error(
               `Rule "${rule.parsedName}" is matching multiple functions: [${matches[rule.parsedName]}, ${funcs[i].name.value}]`
             );

electron/parse.js

@@ -13,5 +13,26 @@
       "u32.load8_u",
       "i32.eqz"
     ]
+  },
+  {
+    "parsedName": "commandHint",
+    "body": [
+      "get_global",
+      "i32.const",
+      "i32.sub",
+      "tee_local",
+      "set_global",
+      "get_local",
+      "u32.load",
+      "set_local",
+      "block",
+      "get_local",
+      "i32.const",
+      "i32.add",
+      "set_global",
+      "get_local",
+      "end"
+    ],
+    "first": true
   }
 ]

electron/rules.json

@@ -1,4 +1,4 @@
-import {dispatcher} from "@/commands/dispatcher.ts";
+import {commandManager, dispatcher} from "@/commands/dispatcher.ts";
 import {literal} from "@jsprismarine/brigadier";
 import {addChatMessage} from "@/gameChat/gameChatUtility.ts";
 
@@ -23,4 +23,6 @@ dispatcher.register(
       ]);
     }
   )
-)
+);
+
+commandManager.registerCommand('ahelp', 'Show help message for AyuScript commands');

src/commands/builtin/ahelp.ts

@@ -1,7 +1,90 @@
 import {CommandDispatcher} from "@jsprismarine/brigadier";
+import {writeString} from "@/memory/utils.ts";
+import {canvasTextWidth} from "@/gameChat/gameChatUtility.ts";
+
+class CommandManager {
+  private commands: Map<string, {
+    name: string,
+    description: string,
+    ptr: number
+  }>;
+  constructor() {
+    this.commands = new Map();
+  }
+  registerCommand(name: string, description: string) {
+    if (window.electron) {
+      (async () => {
+        // return;
+        while (!window.Module?.HEAPU8) {
+          await new Promise(r => setTimeout(r, 50));
+        }
+
+        const ptr = window.Module._malloc(68);
+        const desPtr = window.Module._malloc(name.length + description.length + 2);
+        const desPtr2 = window.Module._malloc(name.length + description.length + 5);
+
+        window.Module.HEAPU32[ptr / 4] = desPtr;
+        window.Module.HEAPU32[(ptr + 4) / 4] = 4;
+        window.Module.HEAPU32[(ptr + 8) / 4] = 0;
+        window.Module.HEAPF32[(ptr + 12) / 4] = canvasTextWidth(`/${name} - ${description}`);
+        window.Module.HEAPU32[(ptr + 16) / 4] = 1;
+        window.Module.HEAPU32[(ptr + 20) / 4] = 0;
+        window.Module.HEAPU32[(ptr + 24) / 4] = 0;
+        window.Module.HEAPU32[(ptr + 28) / 4] = 0;
+        window.Module.HEAPU32[(ptr + 32) / 4] = desPtr2;
+        window.Module.HEAPU32[(ptr + 36) / 4] = name.length + description.length + 5;
+        window.Module.HEAPU32[(ptr + 40) / 4] = 2147483728;
+        window.Module.HEAPU32[(ptr + 44) / 4] = 0x00f3c4a3; // color
+        window.Module.HEAPU32[(ptr + 48) / 4] = 0;
+        window.Module.HEAPU32[(ptr + 52) / 4] = 1039516303;
+
+        window.Module.HEAPF32[(ptr + 56) / 4] = 14.00;
+
+        window.Module.HEAPU32[(ptr + 60) / 4] = 264;
+        window.Module.HEAPU32[(ptr + 64) / 4] = 0;
+        writeString(name, desPtr);
+        writeString(description, desPtr + name.length + 1);
+        writeString(`/${name} - ${description}`, desPtr2);
+
+        this.commands.set(name, { name, description, ptr });
+      })()
+    } else {
+      this.commands.set(name, {ptr: 0, name, description});
+    }
+  }
+  getAllCommands() {
+    return this.commands;
+  }
+}
+
+export const commandManager = new CommandManager();
 
 export const dispatcher = new CommandDispatcher();
 
+
+if (window.electron) {
+  (async () => {
+    while (!window.Module?.HEAPU8) {
+      await new Promise(r => setTimeout(r, 50));
+    }
+
+    let firstPtr = 0;
+    window.Module.asm.registerHookPre(440);
+
+    window.ayuHooks.pre_79 = (arg0: number, arg1: number) => {
+      if (!firstPtr) {
+        firstPtr = arg1;
+      }
+      if (arg1 == firstPtr) {
+        for (const [, { ptr }] of commandManager.getAllCommands()) {
+          window.Module.asm[79]?.(arg0, ptr);
+        }
+      }
+    };
+  })();
+}
+
+
 document.addEventListener("keydown", async (e) => {
   if (e.key !== "Enter") {
     return;

src/commands/dispatcher.ts

@@ -6,7 +6,7 @@ interface Cp6 {
   simulateContextLoss: () => void;
 }
 
-type ModuleType = {
+type ModuleType = ({
   HEAP8: Int8Array;
   HEAP16: Int16Array;
   HEAP32: Int32Array;
@@ -15,7 +15,6 @@ type ModuleType = {
   HEAPU32: Uint32Array;
   HEAPF32: Float32Array;
   HEAPF64: Float64Array;
-  _malloc: (number) => number;
 } | {
   HEAP8: undefined;
   HEAP16: undefined;
@@ -25,6 +24,16 @@ type ModuleType = {
   HEAPU32: undefined;
   HEAPF32: undefined;
   HEAPF64: undefined;
+}) & {
+  _malloc: (number) => number;
+  asm: {
+    registerHookPre: (number) => void,
+    registerHookPost: (number) => void,
+    unregisterHookPre: (number) => void,
+    unregisterHookPost: (number) => void,
+    [key: string]: Function,
+    [key: number]: Function,
+  }
 }
 
 interface Florrio {

src/florr.d.ts

@@ -19,7 +19,7 @@ function utf8Length(str: string) {
   return utf8.length;
 }
 
-function canvasTextWidth(text: string) {
+export function canvasTextWidth(text: string) {
   const canvas = document.createElement('canvas');
   const ctx = canvas.getContext('2d');
   ctx!.font = '14px Game';

src/gameChat/gameChatUtility.ts

@@ -7,9 +7,14 @@ export interface AyuScriptAPI {
   }
 }
 
+export interface AyuHooks {
+  [key: string]: Function;
+}
+
 declare global {
   export interface Window {
     ayuScriptApi: AyuScriptAPI;
+    ayuHooks: AyuHooks;
 
     electron: true | undefined;
 

src/index.d.ts

@@ -11,13 +11,23 @@ export function getString(ptr: number) {
   return decoder.decode(stringBytes);
 }
 
+export function writeString(str: string, offset: number) {
+  if (!window.Module.HEAPU8) {
+    return;
+  }
+  const encoder = new TextEncoder();
+  const strBytes = encoder.encode(str);
+  window.Module.HEAPU8.set(strBytes, offset);
+  window.Module.HEAPU8[offset + strBytes.length] = 0;
+}
+
 export function readLeb128(data: Uint8Array, idx: number): { value: number; nextIndex: number } {
   let result = 0, shift = 0, byte;
   do {
     byte   = data[idx++];
-    result |= (byte & 0x7F) << shift;
+    result |= (byte! & 0x7F) << shift;
     shift  += 7;
-  } while (byte & 0x80);
+  } while (byte! & 0x80);
   return { value: result, nextIndex: idx };
 }