From 31b8c56ff74d84d336c9dc328408af61d1d180cc Mon Sep 17 00:00:00 2001
From: Benjamin Canac <canacb1@gmail.com>
Date: Mon, 9 Dec 2024 18:19:18 +0100
Subject: [PATCH] docs(ComponentCode): handle `pro` prop

---
 docs/app/components/content/ComponentCode.vue | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/docs/app/components/content/ComponentCode.vue b/docs/app/components/content/ComponentCode.vue
index 1b58cd33..bf58ab50 100644
--- a/docs/app/components/content/ComponentCode.vue
+++ b/docs/app/components/content/ComponentCode.vue
@@ -40,6 +40,8 @@ const castMap: Record<string, Cast> = {
 }
 
 const props = defineProps<{
+  pro?: boolean
+  prefix?: string
   /** Override the slug taken from the route */
   slug?: string
   class?: any
@@ -78,7 +80,17 @@ const { $prettier } = useNuxtApp()
 
 const camelName = camelCase(props.slug ?? route.params.slug?.[route.params.slug.length - 1] ?? '')
 const name = `U${upperFirst(camelName)}`
-const component = defineAsyncComponent(() => import(`#ui/components/${upperFirst(camelName)}.vue`))
+const component = defineAsyncComponent(() => {
+  if (props.pro) {
+    if (props.prefix) {
+      return import(`#ui-pro/components/${props.prefix}/${upperFirst(camelName)}.vue`)
+    }
+
+    return import(`#ui-pro/components/${upperFirst(camelName)}.vue`)
+  }
+
+  return import(`#ui/components/${upperFirst(camelName)}.vue`)
+})
 
 const componentProps = reactive({
   ...Object.fromEntries(Object.entries(props.props || {}).map(([key, value]) => {