feat(runtime): custom inject function (#3028)

Co-authored-by: Anthony Fu <anthonyfu117@hotmail.com>
unocss · Aug 31, 2023 · 77ea190 · 77ea190
1 parent 4f4148e
commit 77ea190
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/runtime/src/index.ts b/packages/runtime/src/index.ts
@@ -38,6 +38,11 @@ export interface RuntimeOptions {
    * Callback to modify config
    */
   configResolved?: (config: UserConfig, defaults: UserConfigDefaults) => void
+  /**
+   * Optional function to control UnoCSS style element(s) injection into DOM.
+   * When provided, the default injection logic will be overridden.
+   */
+  inject?: (styleElement: HTMLStyleElement) => void
   /**
    * Callback when the runtime is ready. Returning false will prevent default extraction
    */
@@ -138,6 +143,7 @@ export default function init(inlineConfig: RuntimeOptions = {}) {
 
   runtimeOptions.configResolved?.(userConfig, userConfigDefaults)
   const uno = createGenerator(userConfig, userConfigDefaults)
+  const inject = (styleElement) => runtimeOptions.inject ? runtimeOptions.inject(styleElement) : html().prepend(styleElement)
   const styleElements = new Map<string, HTMLStyleElement>()
 
   let paused = true
@@ -177,15 +183,15 @@ export default function init(inlineConfig: RuntimeOptions = {}) {
       styleElements.set(layer, styleElement)
 
       if (previousLayer == null) {
-        html().prepend(styleElement)
+        inject(styleElement)
       }
       else {
         const previousStyle = getStyleElement(previousLayer)
         const parentNode = previousStyle.parentNode
         if (parentNode)
           parentNode.insertBefore(styleElement, previousStyle.nextSibling)
         else
-          html().prepend(styleElement)
+          inject(styleElement)
       }
     }