refactor code storage (#2144)

* refactor code storage Signed-off-by: Jess Frazelle <github@jessfraz.com> * typo Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * for now dont do onupdate its lagging the editor Signed-off-by: Jess Frazelle <github@jessfraz.com> * way smaller delay Signed-off-by: Jess Frazelle <github@jessfraz.com> * turn abck on on update Signed-off-by: Jess Frazelle <github@jessfraz.com> * dont be fancy Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix linter Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * empty * empty * good things Signed-off-by: Jess Frazelle <github@jessfraz.com> * empty * empty * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * fixes Signed-off-by: Jess Frazelle <github@jessfraz.com> * make less flakey Signed-off-by: Jess Frazelle <github@jessfraz.com> * go abck to errors for now Signed-off-by: Jess Frazelle <github@jessfraz.com> --------- Signed-off-by: Jess Frazelle <github@jessfraz.com>
2024-04-17 20:18:07 -07:00
parent 168fed038d
commit 3f8c4e7b5a
32 changed files with 314 additions and 370 deletions
--- a/src/lang/codeManager.ts
+++ b/src/lang/codeManager.ts
@ -0,0 +1,115 @@
+// A little class for updating the code state when we need to and explicitly
+// NOT updating the code state when we don't need to.
+// This prevents re-renders of the codemirror editor, when typing.
+import { bracket } from 'lib/exampleKcl'
+import { isTauri } from 'lib/isTauri'
+import { writeTextFile } from '@tauri-apps/plugin-fs'
+import toast from 'react-hot-toast'
+import { Params } from 'react-router-dom'
+
+const PERSIST_CODE_TOKEN = 'persistCode'
+
+export default class CodeManager {
+  private _code: string = bracket
+  private _updateState: (arg: string) => void = () => {}
+  private _updateEditor: (arg: string) => void = () => {}
+  private _params: Params<string> = {}
+
+  constructor() {
+    if (isTauri()) {
+      this.code = ''
+      return
+    }
+
+    const storedCode = safeLSGetItem(PERSIST_CODE_TOKEN) || ''
+    // TODO #819 remove zustand persistence logic in a few months
+    // short term migration, shouldn't make a difference for tauri app users
+    // anyway since that's filesystem based.
+    const zustandStore = JSON.parse(safeLSGetItem('store') || '{}')
+    if (storedCode === null && zustandStore?.state?.code) {
+      this.code = zustandStore.state.code
+      zustandStore.state.code = ''
+      safeLSSetItem('store', JSON.stringify(zustandStore))
+    } else if (storedCode === null) {
+      this.code = bracket
+    } else {
+      this.code = storedCode
+    }
+  }
+
+  set code(code: string) {
+    this._code = code
+  }
+
+  get code(): string {
+    return this._code
+  }
+
+  registerCallBacks({
+    setCode,
+    setEditorCode,
+  }: {
+    setCode: (arg: string) => void
+    setEditorCode: (arg: string) => void
+  }) {
+    this._updateState = setCode
+    this._updateEditor = setEditorCode
+  }
+
+  setParams(params: Params<string>) {
+    this._params = params
+  }
+
+  // This updates the code state and calls the updateState function.
+  updateCodeState(code: string): void {
+    if (this._code !== code) {
+      this.code = code
+      this._updateState(code)
+    }
+  }
+
+  // Update the code in the editor.
+  updateCodeEditor(code: string): void {
+    if (this._code !== code) {
+      this.code = code
+      this._updateEditor(code)
+    }
+    this._updateEditor(code)
+  }
+
+  // Update the code, state, and the code the code mirror editor sees.
+  updateCodeStateEditor(code: string): void {
+    if (this._code !== code) {
+      this.code = code
+      this._updateState(code)
+      this._updateEditor(code)
+    }
+  }
+
+  async writeToFile() {
+    if (isTauri()) {
+      setTimeout(() => {
+        // Wait one event loop to give a chance for params to be set
+        // Save the file to disk
+        this._params.id &&
+          writeTextFile(this._params.id, this.code).catch((err) => {
+            // TODO: add tracing per GH issue #254 (https://github.com/KittyCAD/modeling-app/issues/254)
+            console.error('error saving file', err)
+            toast.error('Error saving file, please check file permissions')
+          })
+      })
+    } else {
+      safeLSSetItem(PERSIST_CODE_TOKEN, this.code)
+    }
+  }
+}
+
+function safeLSGetItem(key: string) {
+  if (typeof window === 'undefined') return null
+  return localStorage?.getItem(key)
+}
+
+function safeLSSetItem(key: string, value: string) {
+  if (typeof window === 'undefined') return
+  localStorage?.setItem(key, value)
+}