fix: module stability (#92)

lib/templates/color-scheme.js

@@ -9,11 +9,6 @@ export default {
     }
   },
   render (createElement, { parent, data, props, children }) {
-    const { $colorMode } = parent
-
-    if (!$colorMode.unknown) {
-      return children
-    }
     // transform props for <client-only>
     props = {
       placeholder: props.placeholder,

lib/templates/plugin.server.js

@@ -8,13 +8,26 @@ const script = {
   innerHTML: `<%= options.script %>`,
   pbody: true
 }
+const addScript = (head) => {
+  head.script = head.script || []
+  head.script.push(script)
+  const serializeProp = '__dangerouslyDisableSanitizersByTagID'
+  head[serializeProp] = head[serializeProp] || {}
+  head[serializeProp]['<%= options.hid %>'] = ['innerHTML']
+}
 
-export default function (ctx, inject) {
-  ctx.app.head.script.push(script)
 
-  const serializeProp = '__dangerouslyDisableSanitizersByTagID'
-  ctx.app.head[serializeProp] = ctx.app.head[serializeProp] || {}
-  ctx.app.head[serializeProp]['<%= options.hid %>'] = ['innerHTML']
+export default function (ctx, inject) {
+  if (typeof ctx.app.head === 'function') {
+    const originalHead = ctx.app.head
+    ctx.app.head = function () {
+      const head = originalHead.call(this) || {}
+      addScript(head)
+      return head
+    }
+  } else {
+    addScript(ctx.app.head)
+  }
 
   const preference = '<%= options.preference %>'
 

test/csr.test.js

@@ -0,0 +1,31 @@
+import { join } from 'path'
+import { setup, get } from '@nuxtjs/module-test-utils'
+import colorModeModule from '../'
+
+describe('module', () => {
+  let nuxt
+
+  beforeAll(async () => {
+    const rootDir = join(__dirname, '..', 'example')
+
+    const config = {
+      ssr: false,
+      rootDir,
+      modules: [
+        '@nuxtjs/svg',
+        colorModeModule
+      ]
+    }
+
+    nuxt = (await setup(config)).nuxt
+  }, 60000)
+
+  afterAll(async () => {
+    await nuxt.close()
+  })
+
+  test('render', async () => {
+    const html = await get('/')
+    expect(html).toContain('nuxt-color-mode')
+  })
+})

test/module.test.js → test/ssr.test.js

@@ -1,6 +1,6 @@
 import { join } from 'path'
 import { setup, get } from '@nuxtjs/module-test-utils'
-import colorModeModule from '..'
+import colorModeModule from '../'
 
 describe('module', () => {
   let nuxt