chore: add register script

Author: Ohh-889

packages/ui/package.json

@@ -26,7 +26,9 @@
   "files": ["dist"],
   "scripts": {
     "build": "tsdown",
-    "dev": "tsdown --watch  ./src/index.ts"
+    "build:registry": "shadcn build -o ../../playground/public/r",
+    "dev": "tsdown --watch ./src/index.ts",
+    "register": "tsx script/registry.ts"
   },
   "dependencies": {
     "@radix-ui/react-accordion": "^1.2.11",
@@ -56,7 +58,9 @@
     "@types/react": "19.1.0",
     "@types/react-dom": "19.1.1",
     "fast-glob": "^3.3.3",
+    "shadcn": "^2.9.0",
     "tsdown": "^0.12.9",
+    "tsx": "^4.20.3",
     "typescript": "5.8.2"
   }
 }

packages/ui/script/constants.ts

@@ -0,0 +1,49 @@
+export const registryComponentsDependencies: Record<
+  string,
+  { dependencies: string[]; registryDependencies?: string[] }
+> = {
+  accordion: {
+    dependencies: ['@radix-ui/react-accordion', '@radix-ui/react-slot']
+  },
+  alert: {
+    dependencies: ['@radix-ui/react-slot']
+  },
+  'alert-dialog': {
+    dependencies: ['@radix-ui/react-alert-dialog', '@radix-ui/react-slot'],
+    registryDependencies: ['button']
+  },
+  'aspect-ratio': {
+    dependencies: ['@radix-ui/react-aspect-ratio']
+  },
+  avatar: {
+    dependencies: ['@radix-ui/react-avatar']
+  },
+  breadcrumb: {
+    dependencies: ['@radix-ui/react-slot']
+  },
+  button: {
+    dependencies: ['@radix-ui/react-slot']
+  },
+  carousel: {
+    dependencies: ['embla-carousel-react'],
+    registryDependencies: ['button']
+  },
+  checkbox: {
+    dependencies: ['@radix-ui/react-checkbox']
+  },
+  divider: {
+    dependencies: ['@radix-ui/react-separator']
+  },
+  label: {
+    dependencies: ['@radix-ui/react-label']
+  },
+  'scroll-area': {
+    dependencies: ['@radix-ui/react-scroll-area']
+  },
+  sonner: {
+    dependencies: ['sonner', 'next-themes']
+  },
+  tabs: {
+    dependencies: ['@radix-ui/react-tabs', '@radix-ui/react-compose-refs']
+  }
+} as const;

packages/ui/script/registry-type.ts

@@ -0,0 +1,14 @@
+import { type Registry } from 'shadcn/registry';
+
+export const registerType: Registry['items'][number] = {
+  files: [
+    {
+      path: 'src/types/other.ts',
+      target: 'types/other.ts',
+      type: 'registry:file'
+    }
+  ],
+  name: 'types',
+  title: 'Types',
+  type: 'registry:file'
+};

packages/ui/script/registry-ui.ts

@@ -0,0 +1,68 @@
+import path from 'node:path';
+
+import fg from 'fast-glob';
+import { type Registry } from 'shadcn/registry';
+
+import { registryComponentsDependencies } from './constants';
+
+const COMPONENTS_DIR = 'src/components';
+
+function formatComponentName(str: string): string {
+  return str
+    .split('-')
+    .map(part => part.charAt(0).toUpperCase() + part.slice(1))
+    .join(' ');
+}
+
+function getGroupedComponents(): Record<string, string[]> {
+  const components = fg.sync(`${COMPONENTS_DIR}/**/*`);
+  const grouped: Record<string, string[]> = {};
+
+  for (const fullPath of components) {
+    const segments = fullPath.split(path.sep);
+    const group = segments[2]; // 如 'accordion'
+    const name = segments[3]; // 文件名
+
+    // eslint-disable-next-line no-continue
+    if (!group || !name) continue;
+
+    if (!grouped[group]) grouped[group] = [];
+    grouped[group].push(name);
+  }
+
+  return grouped;
+}
+
+function generateRegistryItems(grouped: Record<string, string[]>): Registry['items'] {
+  const ui = Object.entries(grouped).map(([group, files]) => {
+    const item = {
+      files: files.map(file => ({
+        path: `${COMPONENTS_DIR}/${group}/${file}`,
+        target: `components/${group}/${file}`,
+        type: 'registry:ui'
+      })),
+      name: group,
+      title: formatComponentName(group),
+      type: 'registry:block'
+    };
+
+    if (registryComponentsDependencies[group]) {
+      Object.assign(item, registryComponentsDependencies[group]);
+    }
+
+    return item;
+  }) as Registry['items'];
+
+  console.log('🎨 UI component conversion completed.');
+
+  return ui;
+}
+
+export function getRegistryUi(): Registry['items'] {
+  console.log('📦 Starting to collect components for registry...');
+  const grouped = getGroupedComponents();
+
+  console.log('✅ Component collection completed.');
+
+  return generateRegistryItems(grouped);
+}

packages/ui/script/registry-utils.ts

@@ -0,0 +1,18 @@
+import { type Registry } from 'shadcn/registry';
+
+export const registryUtils: Registry['items'][number] = {
+  dependencies: ['clsx', 'tailwind-merge'],
+  files: [
+    {
+      path: 'src/lib/utils.ts',
+      type: 'registry:lib'
+    },
+    {
+      path: 'src/lib/typed.ts',
+      type: 'registry:lib'
+    }
+  ],
+  name: 'utils',
+  title: 'Utils',
+  type: 'registry:lib'
+};

packages/ui/script/registry.ts

@@ -0,0 +1,65 @@
+import { exec } from 'node:child_process';
+import fs from 'node:fs/promises';
+import path from 'node:path';
+
+import { registerType } from './registry-type';
+import { getRegistryUi } from './registry-ui';
+import { registryUtils } from './registry-utils';
+
+const registry = {
+  homepage: 'https://ui-playground.ohh-889.com/',
+  items: [
+    {
+      cssVars: {},
+      dependencies: ['tailwind-variants', 'lucide-react'],
+      files: [],
+      name: 'index',
+      registryDependencies: ['utils'],
+      type: 'registry:style'
+    },
+    ...getRegistryUi(),
+    registerType,
+    registryUtils
+  ],
+  name: 'soybean-react-ui'
+};
+
+async function writeRegistry() {
+  const registryJson = JSON.stringify(registry, null, 2);
+  // eslint-disable-next-line n/prefer-global/process
+  await fs.writeFile(path.join(process.cwd(), `registry.json`), registryJson);
+}
+
+async function buildRegistry() {
+  return new Promise((resolve, reject) => {
+    const process = exec(`pnpm build:registry`);
+
+    process.on('exit', code => {
+      if (code === 0) {
+        resolve(undefined);
+      } else {
+        reject(new Error(`Process exited with code ${code}`));
+      }
+    });
+  });
+}
+
+async function main() {
+  try {
+    console.log('🔨 Starting registry build...');
+    await writeRegistry();
+
+    await buildRegistry();
+    console.log('✅ Registry build completed');
+  } catch (error) {
+    console.error('❌ Build failed with error:');
+    console.error(error);
+    if (error instanceof Error) {
+      console.error('Error stack:', error.stack);
+    }
+    // eslint-disable-next-line n/prefer-global/process
+    process.exit(1);
+  }
+}
+
+main();

playground/public/r/index.json

@@ -0,0 +1,14 @@
+{
+  "$schema": "https://ui.shadcn.com/schema/registry-item.json",
+  "name": "index",
+  "type": "registry:style",
+  "dependencies": [
+    "tailwind-variants",
+    "lucide-react"
+  ],
+  "registryDependencies": [
+    "utils"
+  ],
+  "files": [],
+  "cssVars": {}
+}