From 5f543da6815e30c76dc3a595b993a380043af54b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E5=92=B2=E6=99=BA=E5=AD=90=20Kevin=20Deng?=
 <sxzz@sxzz.moe>
Date: Sat, 24 Jun 2023 20:02:00 +0800
Subject: [PATCH] fix: don't dispose in-memory files

---
 src/editor/MonacoEditor.vue |  4 +--
 src/monaco/Monaco.vue       | 66 ++++---------------------------------
 src/monaco/env.ts           | 51 ++++++++++++++++++++++++++++
 3 files changed, 60 insertions(+), 61 deletions(-)

diff --git a/src/editor/MonacoEditor.vue b/src/editor/MonacoEditor.vue
index 12e95c43..b3fe7683 100644
--- a/src/editor/MonacoEditor.vue
+++ b/src/editor/MonacoEditor.vue
@@ -9,7 +9,7 @@ defineProps<{
   mode?: PreviewMode
 }>()
 
-const emits = defineEmits<{
+const emit = defineEmits<{
   (e: 'change', code: string): void
 }>()
 
@@ -18,7 +18,7 @@ defineOptions({
 })
 
 const onChange = (code: string) => {
-  emits('change', code)
+  emit('change', code)
 }
 </script>
 
diff --git a/src/monaco/Monaco.vue b/src/monaco/Monaco.vue
index 1544a48e..2b0e75c1 100644
--- a/src/monaco/Monaco.vue
+++ b/src/monaco/Monaco.vue
@@ -1,9 +1,3 @@
-<script lang="ts">
-import { loadMonacoEnv, loadWasm } from './env'
-
-let init = false
-</script>
-
 <script lang="ts" setup>
 import {
   onMounted,
@@ -11,12 +5,12 @@ import {
   ref,
   shallowRef,
   nextTick,
-  watchEffect,
   inject,
   watch,
   computed,
 } from 'vue'
 import * as monaco from 'monaco-editor-core'
+import { initMonaco } from './env'
 import { getOrCreateModel } from './utils'
 import { loadGrammars, loadTheme } from 'monaco-volar'
 import { Store } from '../store'
@@ -34,46 +28,16 @@ const props = withDefaults(
   }
 )
 
-const emits = defineEmits<{
+const emit = defineEmits<{
   (e: 'change', value: string): void
 }>()
 
-const containerRef = ref<HTMLDivElement | null>()
+const containerRef = ref<HTMLDivElement>()
 const ready = ref(false)
-const editor = shallowRef<monaco.editor.IStandaloneCodeEditor | undefined>(
-  undefined
-)
-const store = inject('store') as Store
+const editor = shallowRef<monaco.editor.IStandaloneCodeEditor>()
+const store = inject<Store>('store')!
 
-if (!init) {
-  init = true
-  loadMonacoEnv(store)
-  loadWasm()
-}
-
-if (!props.readonly) {
-  watchEffect(() => {
-    // create a model for each file in the store
-    for (const filename in store.state.files) {
-      const file = store.state.files[filename]
-      if (monaco.editor.getModel(monaco.Uri.parse(`file:///${filename}`)))
-        continue
-      getOrCreateModel(
-        monaco.Uri.parse(`file:///${filename}`),
-        file.language,
-        file.code
-      )
-    }
-
-    // dispose of any models that are not in the store
-    for (const model of monaco.editor.getModels()) {
-      if (store.state.files[model.uri.toString().substring('file:///'.length)])
-        continue
-      if (model.uri.toString().startsWith('file:///node_modules/')) continue
-      model.dispose()
-    }
-  })
-}
+initMonaco(store)
 
 const lang = computed(() => (props.mode === 'css' ? 'css' : 'javascript'))
 
@@ -106,22 +70,6 @@ onMounted(async () => {
   })
   editor.value = editorInstance
 
-  // Support for go to definition
-  monaco.editor.registerEditorOpener({
-    openCodeEditor(_, resource) {
-      const path = resource.path
-      if (/^\//.test(path) && !/^\/node_modules/.test(path)) {
-        const fileName = path.replace('/', '')
-        if (fileName !== store.state.activeFile.filename) {
-          store.setActive(fileName)
-          return true
-        }
-      }
-
-      return false
-    },
-  })
-
   // Support for semantic highlighting
   const t = (editorInstance as any)._themeService._theme
   t.getTokenStyleMetadata = (
@@ -182,7 +130,7 @@ onMounted(async () => {
   })
 
   editorInstance.onDidChangeModelContent(() => {
-    emits('change', editorInstance.getValue())
+    emit('change', editorInstance.getValue())
   })
 
   editorInstance.onDidChangeCursorSelection((e) => {
diff --git a/src/monaco/env.ts b/src/monaco/env.ts
index 0b248cf7..42dd633b 100644
--- a/src/monaco/env.ts
+++ b/src/monaco/env.ts
@@ -1,3 +1,5 @@
+import { watchEffect } from 'vue'
+import * as monaco from 'monaco-editor-core'
 import editorWorker from 'monaco-editor-core/esm/vs/editor/editor.worker?worker'
 import vueWorker from './vue.worker?worker'
 import * as onigasm from 'onigasm'
@@ -8,6 +10,55 @@ import { Store } from '../store'
 import { createJsDelivrDtsHost } from 'volar-service-typescript'
 import { getOrCreateModel } from './utils'
 
+let initted = false
+export function initMonaco(store: Store) {
+  if (initted) return
+  loadMonacoEnv(store)
+  loadWasm()
+
+  watchEffect(() => {
+    // create a model for each file in the store
+    for (const filename in store.state.files) {
+      const file = store.state.files[filename]
+      if (monaco.editor.getModel(monaco.Uri.parse(`file:///${filename}`)))
+        continue
+      getOrCreateModel(
+        monaco.Uri.parse(`file:///${filename}`),
+        file.language,
+        file.code
+      )
+    }
+
+    // dispose of any models that are not in the store
+    for (const model of monaco.editor.getModels()) {
+      const uri = model.uri.toString()
+      if (store.state.files[uri.substring('file:///'.length)]) continue
+      if (uri.startsWith('file:///node_modules/')) continue
+      if (uri.startsWith('inmemory://')) continue
+
+      model.dispose()
+    }
+  })
+
+  // Support for go to definition
+  monaco.editor.registerEditorOpener({
+    openCodeEditor(_, resource) {
+      const path = resource.path
+      if (/^\//.test(path) && !/^\/node_modules/.test(path)) {
+        const fileName = path.replace('/', '')
+        if (fileName !== store.state.activeFile.filename) {
+          store.setActive(fileName)
+          return true
+        }
+      }
+
+      return false
+    },
+  })
+
+  initted = true
+}
+
 export function loadWasm() {
   return onigasm.loadWASM(onigasmWasm)
 }