thank ChatGPT

Author: heyzec

src/commands/showPanel.tsx

@@ -15,6 +15,7 @@ import { FRONTEND_ELEMENT_ID } from "../constants";
 import { client, SOURCE_ACADEMY_ICON_URI } from "../extension";
 import _ from "lodash";
 import { treeDataProvider } from "../treeview";
+import { getValue, mirror, setValue } from "../utils/store/test";
 
 let panel: vscode.WebviewPanel | null = null;
 // This needs to be a reference to active
@@ -52,18 +53,23 @@ async function handleMessage(
           message.initialCode,
         );
         activeEditor.uri;
-        const info = context.globalState.get("info") ?? {};
+        // const info = context.globalState.get("info") ?? {};
+        const info = getValue(mirror.info);
         if (activeEditor.uri) {
           // TODO: Write our own wrapper to set nested keys easily, removing lodash
           // @ts-ignore
-          _.set(info, `["${activeEditor.uri}"].chapter`, message.chapter ?? 1);
+          // _.set(info, `["${activeEditor.uri}"].chapter`, message.chapter ?? 1);
+          info[activeEditor.uri].chapter = message.chapter ?? 1;
+
           // TODO: message.prepend can be undefined in runtime, investigate
           const nPrependLines =
             message.prepend && message.prepend !== ""
               ? message.prepend.split("\n").length + 2 // account for start/end markers
               : 0;
-          _.set(info, `["${activeEditor.uri}"].prepend`, nPrependLines);
-          context.globalState.update("info", info);
+          // _.set(info, `["${activeEditor.uri}"].prepend`, nPrependLines);
+          info[activeEditor.uri].prepend = nPrependLines;
+          // context.globalState.update("info", info);
+          setValue(mirror.info, info);
           client.sendRequest("source/publishInfo", info);
         }
 
@@ -96,8 +102,10 @@ async function handleMessage(
       //   break;
       case MessageTypeNames.NotifyAssessmentsOverview:
         const { assessmentOverviews, courseId } = message;
-        context.globalState.update("assessmentOverviews", assessmentOverviews);
-        context.globalState.update("courseId", courseId);
+        // context.globalState.update("assessmentOverviews", assessmentOverviews);
+        setValue(mirror.assessmentOverviews, assessmentOverviews);
+        // context.globalState.update("courseId", courseId);
+        setValue(mirror.courseId, courseId);
         treeDataProvider.refresh();
         break;
     }

src/extension.ts

@@ -6,15 +6,20 @@ import { setupStatusBar } from "./statusbar/status";
 import { registerAllCommands } from "./commands";
 import { activateLspClient, deactivateLspClient } from "./lsp/client";
 import { LanguageClient } from "vscode-languageclient/node";
+import { getValue, mirror } from "./utils/store/test";
 
 // TODO: Don't expose this object directly, create an interface via a wrapper class
 export let client: LanguageClient;
 
 export let SOURCE_ACADEMY_ICON_URI: vscode.Uri;
 
+export let context_: vscode.ExtensionContext;
+
 // This method is called when your extension is activated
 // Your extension is activated the very first time the command is executed
 export function activate(context: vscode.ExtensionContext) {
+  console.log("Activain");
+  context_ = context;
   setupTreeView(context);
   registerAllCommands(context);
 
@@ -25,7 +30,8 @@ export function activate(context: vscode.ExtensionContext) {
   // const info = {
   //   "file:///home/heyzec/.sourceacademy/playground_1.js": { chapter: 4 },
   // };
-  const info = context.globalState.get("info") ?? {};
+  // const info = context.globalState.get("info") ?? {};
+  const info = getValue(mirror.info) ?? {};
 
   client.sendRequest("source/publishInfo", info);
 
@@ -34,6 +40,7 @@ export function activate(context: vscode.ExtensionContext) {
     "assets",
     "icon.svg",
   );
+  vscode.window.showInformationMessage(`mirror is ${JSON.stringify(mirror)}`);
 }
 
 // This method is called when your extension is deactivated

src/treeview/index.ts

@@ -1,14 +1,16 @@
 import * as vscode from "vscode";
 import { VscAssessmentOverview } from "../utils/messages";
 import { SOURCE_ACADEMY_ICON_URI } from "../extension";
+import { getValue, mirror } from "../utils/store/test";
 
 export let treeDataProvider: AssessmentsSidebarProvider;
 
 // This will be a source of bug on first extension loads!!
 let courseId: number;
 
 export function setupTreeView(context: vscode.ExtensionContext) {
-  courseId = context.globalState.get("courseId") as number;
+  // courseId = context.globalState.get("courseId") as number;
+  courseId = getValue(mirror.courseId);
 
   treeDataProvider = new AssessmentsSidebarProvider(context);
   vscode.window.createTreeView("assessments", {
@@ -33,9 +35,11 @@ export class AssessmentsSidebarProvider
   }
 
   getChildren(element?: BaseTreeItem): Thenable<BaseTreeItem[]> {
-    // @ts-ignore
-    const assessmentOverviews: VscAssessmentOverview[] =
-      this.context.globalState.get("assessmentOverviews");
+    // // @ts-ignore
+    // const assessmentOverviews: VscAssessmentOverview[] =
+    //   this.context.globalState.get("assessmentOverviews");
+    const assessmentOverviews = getValue(mirror.assessmentOverviews);
+
     if (!assessmentOverviews) {
       return Promise.resolve([]);
     }

src/utils/store/test.ts

@@ -0,0 +1,106 @@
+import { context_ } from "../../extension";
+import { VscAssessmentOverview } from "../messages";
+
+/**
+ * There is a lot of casting in this file but in theory it should all work...
+ */
+
+// 1. Leaf wrapper for types
+type Leaf<T> = { __type: T };
+
+type Entry = {
+  chapter: number;
+  prepend: number;
+};
+
+// 2. Schema with types
+const schema = {
+  courseId: {} as Leaf<number>,
+  assessmentOverviews: {} as Leaf<VscAssessmentOverview[]>,
+  info: {} as Leaf<Record<string, Entry>>,
+  A: {
+    B: {} as Leaf<string>,
+    ao: {} as Leaf<VscAssessmentOverview[]>,
+  },
+  C: {
+    D: {} as Leaf<number>,
+    E: {} as number,
+  },
+};
+
+// Purpose: Creates a nested object that mirrors the shape of T, but replaces leaves with the string path to that leaf.
+type PathTree<T, Prefix extends string = ""> = {
+  [K in keyof T]: T[K] extends Leaf<any>
+    ? `${Prefix}${K & string}`
+    : T[K] extends object
+      ? PathTree<T[K], `${Prefix}${K & string}.`>
+      : never;
+};
+
+function generateMirrorTree(obj: object, prefix = ""): object {
+  const result: any = {};
+  for (const key in obj) {
+    const path = prefix ? `${prefix}.${key}` : key;
+    // @ts-ignore
+    const val = obj[key];
+
+    if (val && typeof val === "object" && Object.keys(val).length > 0) {
+      result[key] = generateMirrorTree(val, path);
+    } else {
+      result[key] = path;
+    }
+  }
+  return result;
+}
+
+export const mirror = generateMirrorTree(schema) as PathTree<typeof schema>;
+
+const mockStore = {
+  "A.B": "a string",
+  "C.D": 1,
+} as any;
+
+// Purpose: Creates a union of objects, each mapping a full dot-path string to the type stored in the leaf.
+// prettier-ignore
+type LeafPathsToTypes<T, Prefix extends string = ""> = {
+  [K in keyof T]:
+    // Ask: Is this is a Leaf<...>?
+    T[K] extends Leaf<infer U>
+    // Yes, extract its type
+    ? { [P in `${Prefix}${K & string}`]: U }
+    // Else, ask: Is this a nested object?
+      // Assume yes, recurse
+    : T[K] extends object
+      ? LeafPathsToTypes<T[K], `${Prefix}${K & string}.`>
+      // Else, this never occurs
+      : never;
+}[keyof T];
+
+type UnionToIntersection<U> = (U extends any ? (k: U) => void : never) extends (
+  k: infer I,
+) => void
+  ? I
+  : never;
+
+type PathTypeMap = UnionToIntersection<LeafPathsToTypes<typeof schema>>;
+
+type Test1 = PathTree<typeof schema>;
+type Test2 = LeafPathsToTypes<typeof schema>;
+
+export function getValue<K extends keyof PathTypeMap>(key: K): PathTypeMap[K] {
+  console.log(`Getting vlaue of ${key}`);
+
+  // return mockStore[k];
+  if (!context_) {
+    console.log("OHNO w");
+  }
+  const x = context_.globalState.get(key);
+  return x as any;
+}
+
+export function setValue<K extends keyof PathTypeMap>(
+  key: K,
+  value: PathTypeMap[K],
+) {
+  context_.globalState.update(key, value);
+}