Clean up of cheatsheet test

packages/lib-node-common/test/Cheatsheet.test.ts → packages/lib-node-common/src/cheatsheet/Cheatsheet.test.ts

@@ -1,7 +1,7 @@
 import assert from "node:assert/strict";
 import { parse } from "node-html-parser";
 import type { CheatsheetInfo } from "@cursorless/lib-common";
-import { injectCheatsheetInfo } from "../src/injectCheatsheetInfo";
+import { injectCheatsheetInfo } from "./injectCheatsheetInfo";
 
 suite("Cheatsheet", () => {
   test("injects cheatsheet info as executable script text", () => {

packages/lib-node-common/src/Cheatsheet.ts → packages/lib-node-common/src/cheatsheet/Cheatsheet.ts

@@ -7,7 +7,7 @@ import {
   getErrorMessage,
   showWarning,
 } from "@cursorless/lib-common";
-import type { FileSystemTalonSpokenForms } from "./FileSystemTalonSpokenForms";
+import type { FileSystemTalonSpokenForms } from "../FileSystemTalonSpokenForms";
 import { injectCheatsheetInfo } from "./injectCheatsheetInfo";
 
 interface CheatSheetCommandArgV0 {

packages/lib-node-common/src/injectCheatsheetInfo.ts → packages/lib-node-common/src/cheatsheet/injectCheatsheetInfo.ts

@@ -1,4 +1,4 @@
-import { parse, TextNode } from "node-html-parser";
+import { parse } from "node-html-parser";
 import type { CheatsheetInfo } from "@cursorless/lib-common";
 
 export function injectCheatsheetInfo(
@@ -11,11 +11,15 @@ export function injectCheatsheetInfo(
     String.raw`\u003c`,
   );
 
-  root
-    .getElementById("cheatsheet-data")!
-    .set_content(
-      new TextNode(`document.cheatsheetInfo = ${serializedCheatsheetInfo};`),
-    );
+  const dataElement = root.getElementById("cheatsheet-data");
+
+  if (dataElement == null) {
+    throw new Error("Cheatsheet cheatsheet-data element not found");
+  }
+
+  dataElement.set_content(
+    `document.cheatsheetInfo = ${serializedCheatsheetInfo};`,
+  );
 
   return root.toString();
 }

packages/lib-node-common/src/index.ts

@@ -1,4 +1,4 @@
-export * from "./Cheatsheet";
+export * from "./cheatsheet/Cheatsheet";
 export * from "./FileSystemCommandHistoryStorage";
 export * from "./FileSystemRawTreeSitterQueryProvider";
 export * from "./FileSystemTalonSpokenForms";