cli: add --pro and --primitive options

cli/templates.mjs

@@ -1,12 +1,14 @@
 import { splitByCase, upperFirst, camelCase, kebabCase } from 'scule'
 
-const playground = ({ name }) => {
+const playground = ({ name, pro }) => {
   const upperName = splitByCase(name).map(p => upperFirst(p)).join('')
   const kebabName = kebabCase(name)
 
   return {
     filename: `playground/pages/${kebabName}.vue`,
-    contents: `
+    contents: pro
+      ? undefined
+      : `
 <template>
   <div>
     <U${upperName} />
@@ -16,24 +18,65 @@ const playground = ({ name }) => {
   }
 }
 
-const component = ({ name }) => {
+const component = ({ name, primitive, pro }) => {
   const upperName = splitByCase(name).map(p => upperFirst(p)).join('')
   const camelName = camelCase(name)
   const kebabName = kebabCase(name)
+  const key = pro ? 'uiPro' : 'ui'
+  const path = pro ? 'ui-pro' : 'ui'
 
   return {
     filename: `src/runtime/components/${upperName}.vue`,
-    contents: `
+    contents: primitive
+      ? `
+<script lang="ts">
+import { tv } from 'tailwind-variants'
+import type { PrimitiveProps } from 'radix-vue'
+import type { AppConfig } from '@nuxt/schema'
+import _appConfig from '#build/app.config'
+import theme from '#build/${path}/${kebabName}'
+
+const appConfig = _appConfig as AppConfig & { ${key}: { ${camelName}: Partial<typeof theme> } }
+
+const ${camelName} = tv({ extend: tv(theme), ...(appConfig.${key}?.${camelName} || {}) })
+
+export interface ${upperName}Props extends Omit<PrimitiveProps, 'asChild'> {
+  class?: any
+  ui?: Partial<typeof ${camelName}.slots>
+}
+
+export interface ${upperName}Slots {
+  default: any
+}
+</script>
+
+<script setup lang="ts">
+import { computed } from 'vue'
+import { Primitive } from 'radix-vue'
+
+const props = withDefaults(defineProps<${upperName}Props>(), { as: 'div' })
+defineSlots<${upperName}Slots>()
+
+const ui = computed(() => tv({ extend: ${camelName}, slots: props.ui })())
+</script>
+
+<template>
+  <Primitive :as="as" :class="ui.root({ class: props.class })">
+    <slot />
+  </Primitive>
+</template>
+`
+      : `
 <script lang="ts">
 import { tv, type VariantProps } from 'tailwind-variants'
 import type { ${upperName}RootProps, ${upperName}RootEmits } from 'radix-vue'
 import type { AppConfig } from '@nuxt/schema'
 import _appConfig from '#build/app.config'
-import theme from '#build/ui/${kebabName}'
+import theme from '#build/${path}/${kebabName}'
 
-const appConfig = _appConfig as AppConfig & { ui: { ${camelName}: Partial<typeof theme> } }
+const appConfig = _appConfig as AppConfig & { ${key}: { ${camelName}: Partial<typeof theme> } }
 
-const ${camelName} = tv({ extend: tv(theme), ...(appConfig.ui?.${camelName} || {}) })
+const ${camelName} = tv({ extend: tv(theme), ...(appConfig.${key}?.${camelName} || {}) })
 
 type ${upperName}Variants = VariantProps<typeof ${camelName}>
 
@@ -64,7 +107,7 @@ const ui = computed(() => tv({ extend: ${camelName}, slots: props.ui })())
 <template>
   <${upperName}Root v-bind="rootProps" :class="ui.root({ class: props.class })" />
 </template>
-    `
+`
   }
 }
 
@@ -74,17 +117,11 @@ const theme = ({ name }) => {
   return {
     filename: `src/theme/${kebabName}.ts`,
     contents: `
-export default (config: { colors: string[] }) => ({
+export default {
   slots: {
     root: ''
-  },
-  variants: {
-
-  },
-  defaultVariants: {
-
   }
-})
+}
     `
   }
 }
@@ -96,15 +133,18 @@ const test = ({ name }) => {
     filename: `test/components/${upperName}.spec.ts`,
     contents: `
 import { describe, it, expect } from 'vitest'
-import ${upperName}, { type ${upperName}Props } from '../../src/runtime/components/${upperName}.vue'
+import ${upperName}, { type ${upperName}Props, type ${upperName}Slots } from '../../src/runtime/components/${upperName}.vue'
 import ComponentRender from '../component-render'
 
 describe('${upperName}', () => {
   it.each([
     // Props
+    ['with as', { props: { as: 'div' } }],
     ['with class', { props: { class: '' } }],
-    ['with ui', { props: { ui: {} } }]
-  ])('renders %s correctly', async (nameOrHtml: string, options: { props?: ${upperName}Props, slots?: any }) => {
+    ['with ui', { props: { ui: {} } }],
+    // Slots
+    ['with default slot', { slots: { default: () => 'Default slot' } }]
+  ])('renders %s correctly', async (nameOrHtml: string, options: { props?: ${upperName}Props, slots?: Partial<${upperName}Slots> }) => {
     const html = await ComponentRender(nameOrHtml, options, ${upperName})
     expect(html).toMatchSnapshot()
   })