From 8ab8b7a2311d623c9383c7d41a194ec0184c9c03 Mon Sep 17 00:00:00 2001
From: Dario Ferderber <dario.ferderber@broj42.com>
Date: Wed, 12 Jul 2023 18:33:49 +0200
Subject: [PATCH] fix(vite): support iframe blocking (#88)

Resolves https://github.com/dargmuesli/nuxt-cookie-control/issues/79

Co-authored-by: Jonas Thelemann <e-mail@jonas-thelemann.de>
---
 playground/app.vue                       | 12 +++++-
 playground/nuxt.config.ts                |  1 +
 src/module.ts                            | 21 +++++++++-
 src/replace.ts                           | 50 ++++++++++++++++++++++++
 src/runtime/components/CookieControl.vue | 15 +++++++
 src/runtime/components/CookieIframe.vue  | 25 ++++++------
 src/runtime/plugin.ts                    |  4 +-
 src/runtime/types.ts                     |  1 +
 8 files changed, 113 insertions(+), 16 deletions(-)
 create mode 100644 src/replace.ts

diff --git a/playground/app.vue b/playground/app.vue
index f97dc1a1..d67d039a 100644
--- a/playground/app.vue
+++ b/playground/app.vue
@@ -1,7 +1,17 @@
 <template>
   <div>
     <CookieControl :locale="locale" />
-    <CookieIframe :locale="locale" />
+
+    <iframe
+      src="https://player.vimeo.com/video/259622029?h=8dd8029ca1"
+      width="640"
+      height="480"
+      frameborder="0"
+      allow="autoplay; fullscreen; picture-in-picture"
+      allowfullscreen
+    >
+    </iframe>
+
     <fieldset>
       <legend>Choose a language</legend>
       <div>
diff --git a/playground/nuxt.config.ts b/playground/nuxt.config.ts
index deee483b..6fcbc512 100644
--- a/playground/nuxt.config.ts
+++ b/playground/nuxt.config.ts
@@ -42,6 +42,7 @@ export default defineNuxtConfig({
       ],
     },
     isCookieIdVisible: true,
+    isIframeBlocked: true,
     locales: ['en', 'de'],
     localeTexts: {
       de: {
diff --git a/src/module.ts b/src/module.ts
index 5855c994..2376e892 100644
--- a/src/module.ts
+++ b/src/module.ts
@@ -5,6 +5,7 @@ import {
   defineNuxtModule,
   addPlugin,
   extendWebpackConfig,
+  extendViteConfig,
   addTemplate,
   addImports,
   createResolver,
@@ -15,6 +16,7 @@ import { Nuxt } from '@nuxt/schema'
 
 import { name, version } from '../package.json'
 import { DEFAULTS, ModuleOptions } from './runtime/types'
+import { replaceCodePlugin } from './replace'
 
 const resolver = createResolver(import.meta.url)
 const runtimeDir = resolver.resolve('./runtime')
@@ -94,7 +96,24 @@ const blockIframes = (moduleOptions: ModuleOptions) => {
         },
       })
     })
-    // extendViteConfig(callback, options?)
+
+    extendViteConfig((config) => {
+      config?.plugins?.push(
+        replaceCodePlugin({
+          replacements: [
+            {
+              from: /<iframe[^>]*.*|<\/iframe>/g,
+              to: (match: string) =>
+                match.includes('cookie-enabled')
+                  ? match
+                  : match
+                      .replace(/<iframe/g, '<CookieIframe')
+                      .replace(/iframe>/g, 'CookieIframe>'),
+            },
+          ],
+        }),
+      )
+    })
   }
 }
 
diff --git a/src/replace.ts b/src/replace.ts
new file mode 100644
index 00000000..c79e6255
--- /dev/null
+++ b/src/replace.ts
@@ -0,0 +1,50 @@
+// TODO: move to own library (https://github.com/leanupjs/vite-plugin-replace/pull/2)
+
+const execSrcReplacements = (src: any, replacements: any[]) => {
+  for (const replacement of replacements) {
+    if (
+      (typeof replacement.from === 'string' ||
+        replacement.from instanceof RegExp) === false
+    ) {
+      throw new TypeError(
+        "[vite-plugin-replace]: The replacement option 'from' is not of type 'string' or 'RegExp'.",
+      )
+    } else if (
+      (typeof replacement.to === 'string' ||
+        replacement.to instanceof Function) === false
+    ) {
+      throw new TypeError(
+        "[vite-plugin-replace]: The replacement option 'to' is not of type 'string' or 'Function'",
+      )
+    } else src = src.replace(replacement.from, replacement.to)
+  }
+
+  return src
+}
+
+export const replaceCodePlugin = (config: any) => {
+  if (config === undefined) {
+    config = {
+      replacements: [],
+    }
+  } else if ((typeof config === 'object' || config !== null) === false) {
+    throw new TypeError(
+      "[vite-plugin-replace]: The configuration is not of type 'Object'.",
+    )
+  } else if (Array.isArray(config.replacements) === false) {
+    throw new TypeError(
+      "[vite-plugin-replace]: The configuration option 'replacement' is not of type 'Array'.",
+    )
+  }
+
+  return {
+    name: 'transform-file',
+    enforce: 'pre',
+    transform: function (src: any) {
+      return {
+        code: execSrcReplacements(src, config.replacements),
+        map: null,
+      }
+    },
+  } as any
+}
diff --git a/src/runtime/components/CookieControl.vue b/src/runtime/components/CookieControl.vue
index 2dc43df5..dfb0d25d 100644
--- a/src/runtime/components/CookieControl.vue
+++ b/src/runtime/components/CookieControl.vue
@@ -212,6 +212,7 @@ const {
   isModalActive,
   moduleOptions,
 } = useCookieControl()
+const nuxtApp = useNuxtApp()
 
 // data
 const expires = new Date(Date.now() + moduleOptions.cookieExpiryOffsetMs)
@@ -278,6 +279,9 @@ const getName = (name: Translatable) => {
     ? localeStrings.value?.cookiesFunctional
     : resolveTranslatable(name, props.locale)
 }
+const init = () => {
+  nuxtApp.$cookies.locale.value = props.locale
+}
 const onModalClick = () => {
   if (moduleOptions.closeModalOnClickOutside) {
     isModalActive.value = false
@@ -392,6 +396,7 @@ watch(
   },
   { deep: true },
 )
+
 watch(isConsentGiven, (current, _previous) => {
   if (current === undefined) {
     cookieIsConsentGiven.value = undefined
@@ -400,6 +405,16 @@ watch(isConsentGiven, (current, _previous) => {
   }
 })
 
+watch(
+  () => props.locale,
+  (locale) => {
+    nuxtApp.$cookies.locale.value = locale
+  },
+)
+
+// initialization
+init()
+
 defineExpose({
   accept,
   acceptPartial,
diff --git a/src/runtime/components/CookieIframe.vue b/src/runtime/components/CookieIframe.vue
index 23d54cda..b8bf4f2c 100644
--- a/src/runtime/components/CookieIframe.vue
+++ b/src/runtime/components/CookieIframe.vue
@@ -1,6 +1,10 @@
 <template>
-  <client-only>
-    <iframe v-if="isCookieFunctionalEnabled" />
+  <ClientOnly>
+    <iframe
+      v-if="isCookieFunctionalEnabled"
+      :cookie-enabled="null"
+      v-bind="$attrs"
+    />
     <div v-else class="cookieControl__BlockedIframe">
       <p>
         {{ localeStrings?.iframeBlocked }}
@@ -11,25 +15,18 @@
         />
       </p>
     </div>
-  </client-only>
+  </ClientOnly>
 </template>
 
 <script setup lang="ts">
 import { computed } from 'vue'
 
-import { LOCALE_DEFAULT } from '../constants'
-import { Cookie, Locale } from '../types'
+import { Cookie } from '../types'
 
 import { useCookieControl } from '#imports'
 
-export interface Props {
-  locale?: Locale
-}
-const props = withDefaults(defineProps<Props>(), {
-  locale: LOCALE_DEFAULT,
-})
-
 const { cookiesEnabled, isModalActive, moduleOptions } = useCookieControl()
+const nuxtApp = useNuxtApp()
 
 // computations
 const isCookieFunctionalEnabled = computed(
@@ -38,5 +35,7 @@ const isCookieFunctionalEnabled = computed(
       (cookieEnabled: Cookie) => cookieEnabled.name === 'functional',
     ).length > 0,
 )
-const localeStrings = computed(() => moduleOptions.localeTexts[props.locale])
+const localeStrings = computed(
+  () => moduleOptions.localeTexts[nuxtApp.$cookies.locale.value],
+)
 </script>
diff --git a/src/runtime/plugin.ts b/src/runtime/plugin.ts
index 0b711018..854d8b23 100644
--- a/src/runtime/plugin.ts
+++ b/src/runtime/plugin.ts
@@ -35,12 +35,14 @@ export default defineNuxtPlugin((_nuxtApp) => {
   )
   const cookiesEnabledIds = ref<string[] | undefined>(cookieCookiesEnabledIds)
   const isModalActive = ref<boolean>()
+  const locale = ref<string>()
 
   const state = {
-    isConsentGiven,
     cookiesEnabled,
     cookiesEnabledIds,
+    isConsentGiven,
     isModalActive,
+    locale,
     moduleOptions,
   } as State
 
diff --git a/src/runtime/types.ts b/src/runtime/types.ts
index 09406b26..113fc2a3 100644
--- a/src/runtime/types.ts
+++ b/src/runtime/types.ts
@@ -157,5 +157,6 @@ export interface State {
   cookiesEnabledIds: Ref<string[] | undefined>
   isConsentGiven: Ref<boolean | undefined>
   isModalActive: Ref<boolean>
+  locale: Ref<Locale>
   moduleOptions: ModuleOptions
 }