feat: support for sandbox page customization (#42)

vuejs · Jun 14, 2023 · a22b969 · a22b969
1 parent 29263d8
commit a22b969
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 9 deletions.
diff --git a/src/Repl.vue b/src/Repl.vue
@@ -14,6 +14,14 @@ export interface Props {
   sfcOptions?: SFCOptions
   layout?: string
   ssr?: boolean
+  previewOptions?: {
+    headHTML?: string
+    bodyHTML?: string
+    customCode?: {
+      importCode?: string
+      useCode?: string
+    }
+  }
 }
 
 const props = withDefaults(defineProps<Props>(), {
@@ -22,7 +30,15 @@ const props = withDefaults(defineProps<Props>(), {
   showCompileOutput: true,
   showImportMap: true,
   clearConsole: true,
-  ssr: false
+  ssr: false,
+  previewOptions: () => ({
+    headHTML: '',
+    bodyHTML: '',
+    customCode: {
+      importCode: '',
+      useCode: '',
+    },
+  }),
 })
 
 const { store } = props
@@ -48,6 +64,7 @@ provide('store', store)
 provide('autoresize', props.autoResize)
 provide('import-map', toRef(props, 'showImportMap'))
 provide('clear-console', toRef(props, 'clearConsole'))
+provide('preview-options', props.previewOptions)
 </script>
 
 <template>

diff --git a/src/output/Preview.vue b/src/output/Preview.vue
@@ -14,11 +14,15 @@ import srcdoc from './srcdoc.html?raw'
 import { PreviewProxy } from './PreviewProxy'
 import { compileModulesForPreview } from './moduleCompiler'
 import { Store } from '../store'
+import { Props } from '../Repl.vue'
 
 const props = defineProps<{ show: boolean; ssr: boolean }>()
 
 const store = inject('store') as Store
 const clearConsole = inject('clear-console') as Ref<boolean>
+
+const previewOptions = inject('preview-options') as Props['previewOptions']
+
 const container = ref()
 const runtimeError = ref()
 const runtimeWarning = ref()
@@ -85,10 +89,12 @@ function createSandbox() {
   if (!importMap.imports.vue) {
     importMap.imports.vue = store.state.vueRuntimeURL
   }
-  const sandboxSrc = srcdoc.replace(
-    /<!--IMPORT_MAP-->/,
-    JSON.stringify(importMap)
-  )
+  const sandboxSrc = srcdoc
+    .replace(/<!--IMPORT_MAP-->/, JSON.stringify(importMap))
+    .replace(
+      /<!-- PREVIEW-OPTIONS-HEAD-HTML -->/,
+      previewOptions?.headHTML || ''
+    )
   sandbox.srcdoc = sandboxSrc
   container.value.appendChild(sandbox)
 
@@ -196,7 +202,9 @@ async function updatePreview() {
          }
          app.config.warnHandler = () => {}
          window.__ssr_promise__ = _renderToString(app).then(html => {
-           document.body.innerHTML = '<div id="app">' + html + '</div>'
+           document.body.innerHTML = '<div id="app">' + html + '</div>' + \`${
+             previewOptions?.bodyHTML || ''
+           }\`
          }).catch(err => {
            console.error("SSR Error", err)
          })
@@ -213,9 +221,13 @@ async function updatePreview() {
     )
 
     const codeToEval = [
-      `window.__modules__ = {}\nwindow.__css__ = ''\n` +
-        `if (window.__app__) window.__app__.unmount()\n` +
-        (isSSR ? `` : `document.body.innerHTML = '<div id="app"></div>'`),
+      `window.__modules__ = {};window.__css__ = '';` +
+        `if (window.__app__) window.__app__.unmount();` +
+        (isSSR
+          ? ``
+          : `document.body.innerHTML = '<div id="app"></div>' + \`${
+              previewOptions?.bodyHTML || ''
+            }\``),
       ...modules,
       `document.getElementById('__sfc-styles').innerHTML = window.__css__`
     ]
@@ -226,6 +238,7 @@ async function updatePreview() {
         `import { ${
           isSSR ? `createSSRApp` : `createApp`
         } as _createApp } from "vue"
+        ${previewOptions?.customCode?.importCode || ''}
         const _mount = () => {
           const AppComponent = __modules__["${mainFile}"].default
           AppComponent.name = 'Repl'
@@ -234,6 +247,7 @@ async function updatePreview() {
             app.config.unwrapInjectedRef = true
           }
           app.config.errorHandler = e => console.error(e)
+          ${previewOptions?.customCode?.useCode || ''}
           app.mount('#app')
         }
         if (window.__ssr_promise__) {

diff --git a/src/output/srcdoc.html b/src/output/srcdoc.html
@@ -7,6 +7,7 @@
 				Oxygen, Ubuntu, Cantarell, "Open Sans", "Helvetica Neue", sans-serif;
 			}
 		</style>
+		<!-- PREVIEW-OPTIONS-HEAD-HTML -->
 		<style id="__sfc-styles"></style>
 		<script>
 			(() => {