feat: chat memory and squad command demo

Author: LittleSwift

src/App.vue

@@ -14,6 +14,7 @@ import {injectAPI} from "@/memory/wasmExtraction.ts";
 import {petalCountLoggerInit} from "@/petalCountLogger.ts";
 import Api from "@/Api.vue";
 import '@/commands/builtin';
+import {gameChatUtilityInit} from "@/gameChat/gameChatUtility.ts";
 const { t } = useI18n();
 
 const webSocketServerAddress = import.meta.env.VITE_SERVER;
@@ -23,8 +24,9 @@ switch (import.meta.env.VITE_ENCRYPTION) {
     webSocketService.applyEncryptor(new NoneEncryptor());
     break;
 }
-injectAPI().then(({ inventory }) => {
+injectAPI().then(({ inventory, chatBase, chatOverflow }) => {
   petalCountLoggerInit(inventory);
+  gameChatUtilityInit(chatBase, chatOverflow);
 });
 patchWebsocket();
 webSocketService.subscribeOpen(() => {

src/commands/builtin/asquad.ts

@@ -0,0 +1,19 @@
+import {dispatcher} from "@/commands/dispatcher.ts";
+import {literal} from "@jsprismarine/brigadier";
+import {addChatMessage} from "@/gameChat/gameChatUtility.ts";
+
+dispatcher.register(
+  literal('aaccept').executes(
+    async () => {
+      addChatMessage([
+        {
+          text: 'Joining squad 3v7f7v',
+          r: 163,
+          g: 196,
+          b: 243
+        }
+      ]);
+      return '/squad-join 3v7f7v';
+    }
+  )
+)

src/commands/builtin/index.ts

@@ -1 +1,2 @@
-import './ahelp.ts';
+import './ahelp.ts';
+import './asquad.ts';

src/commands/dispatcher.ts

@@ -2,7 +2,7 @@ import {CommandDispatcher} from "@jsprismarine/brigadier";
 
 export const dispatcher = new CommandDispatcher();
 
-document.addEventListener("keydown", (e) => {
+document.addEventListener("keydown", async (e) => {
   if (e.key !== "Enter") {
     return;
   }
@@ -17,13 +17,16 @@ document.addEventListener("keydown", (e) => {
   }
 
   const value = target.value.trim();
-  console.log(value);
   if (!value.startsWith("/")) {
     return;
   }
 
   if (dispatcher.parse(value.slice(1), null)) {
-    dispatcher.execute(dispatcher.parse(value.slice(1), null));
-    target.value = "";
+    const result = await (dispatcher.execute(dispatcher.parse(value.slice(1), null))[0]);
+    if (typeof result === 'string') {
+      target.value = result;
+    } else {
+      target.value = '';
+    }
   }
 }, true);

src/gameChat/gameChatUtility.ts

@@ -1,7 +1,5 @@
-// TODO refactor this file
-// TODO find these two value in wasm
-const base = 19286288;
-const overflow = 3563868;
+let base!:number;
+let overflow!:number;
 
 function writeString(str: string) {
   if (!window.Module.HEAPU8) {
@@ -94,4 +92,9 @@ export function addChatMessage(segments: {text: string, r: number, g: number, b:
   }
 
   window.Module.HEAP32[(base - 8) / 4] = count + 1;
+}
+
+export function gameChatUtilityInit(chatBase: number, chatOverflow: number) {
+  base = chatBase;
+  overflow = chatOverflow;
 }

src/memory/wasmExtraction.ts

@@ -1,59 +1,54 @@
-import {matches, readLeb128} from "@/memory/utils.ts";
-
-function parseInventory(bytes: Uint8Array): number {
-  const patternPrefix = [0x41, 0x01, 0x3a, 0x00, 0x00, 0x41];
-  const patternSuffix = [0x41, 0x05, 0x36, 0x02, 0x00];
-
+import { matches, readLeb128 } from "@/memory/utils.ts";
+
+function parsePattern(
+  bytes: Uint8Array,
+  prefix: number[],
+  suffix: number[],
+  transform: (value: number) => number = v => v
+): number {
   for (let i = 0; i < bytes.length; i++) {
-    if (!matches(bytes, i, patternPrefix)) continue;
+    if (!matches(bytes, i, prefix)) continue;
 
-    const addrStart = i + patternPrefix.length;
-    const { value: addr, nextIndex } = readLeb128(bytes, addrStart);
+    const addrStart = i + prefix.length;
+    const { value, nextIndex } = readLeb128(bytes, addrStart);
 
-    if (!matches(bytes, nextIndex, patternSuffix)) continue;
+    if (!matches(bytes, nextIndex, suffix)) continue;
 
-    return addr >> 2;
+    return transform(value);
   }
 
-  throw new Error('Could not find the magic value for "inventory"');
+  throw new Error(`Could not find pattern.`);
 }
 
-async function streamToUint8Array(stream: ReadableStream<Uint8Array>): Promise<Uint8Array> {
-  const reader = stream.getReader();
-  const chunks: Uint8Array[] = [];
-
-  while (true) {
-    const { value, done } = await reader.read();
-    if (done) break;
-    if (value) chunks.push(value);
-  }
-
-  // Calculate total length
-  const totalLength = chunks.reduce((sum, chunk) => sum + chunk.length, 0);
-
-  // Concatenate all chunks
-  const result = new Uint8Array(totalLength);
-  let offset = 0;
-  for (const chunk of chunks) {
-    result.set(chunk, offset);
-    offset += chunk.length;
-  }
-
-  return result;
+async function fetchWasmBytes(): Promise<Uint8Array> {
+  const response = await fetch(`https://static.florr.io/${window.versionHash}/client.wasm`);
+  const buffer = await response.arrayBuffer();
+  return new Uint8Array(buffer);
 }
 
-
 export async function injectAPI(): Promise<{
-  inventory: number
-}>{
-  return new Promise((resolve => {
-    fetch(`https://static.florr.io/${window.versionHash}/client.wasm`)
-      .then(response => response.body)
-      .then(async stream => {
-      const bytes = await streamToUint8Array(stream!);
-      resolve({
-        inventory: parseInventory(bytes)
-      });
-    });
-  }));
-}
+  inventory: number;
+  chatBase: number;
+  chatOverflow: number;
+}> {
+  const bytes = await fetchWasmBytes();
+
+  return {
+    inventory: parsePattern(
+      bytes,
+      [0x41, 0x01, 0x3a, 0x00, 0x00, 0x41],
+      [0x41, 0x05, 0x36, 0x02, 0x00],
+      value => value >> 2
+    ),
+    chatBase: parsePattern(
+      bytes,
+      [0x41, 0x98, 0x03, 0x6C, 0x22, 0x00, 0x41],
+      [0x6A, 0x41, 0x00, 0x41, 0x98, 0x03]
+    ),
+    chatOverflow: parsePattern(
+      bytes,
+      [0x22, 0x05, 0x41],
+      [0x36, 0x02, 0x00, 0x20, 0x02, 0x41, 0x08, 0x6A]
+    )
+  };
+}