ai-unit-test-generator 2.0.6 → 2.0.8

package/CHANGELOG.md

@@ -5,6 +5,66 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [2.0.8] - 2025-01-11
+
+### ✨ Feature: Enhanced Framework Detection
+
+**Problem**: `context-builder.mjs` framework detection was too simplistic, only detecting basic React/Vue/Angular.
+
+**Solution**: Completely rewritten framework detection with comprehensive support:
+
+**Supported Frameworks & Platforms**:
+- ✅ **React Native** (including Expo)
+- ✅ **Next.js** (SSR/SSG)
+- ✅ **React Native + Next.js** (Monorepo detection)
+- ✅ **Taro** (Mini-Program + H5 + RN)
+- ✅ **Vue / Nuxt.js**
+- ✅ **Angular**
+- ✅ **Svelte / SvelteKit**
+- ✅ **Electron** (Desktop apps)
+- ✅ **Node.js** (Backend)
+
+**UI Library Detection**:
+- Ant Design, Material-UI, Chakra UI, Radix UI, Tailwind CSS
+- styled-components, Emotion, Bootstrap
+- React Native Paper, NativeBase
+
+**State Management Detection**:
+- Jotai, Zustand, Redux, MobX, Recoil, XState, Valtio
+- TanStack Query, SWR, Apollo Client
+
+**Testing Tools Detection**:
+- Jest, Vitest
+- Testing Library (React/RN), Enzyme
+- Cypress, Playwright
+
+**Enhanced AI Analysis Output**:
+```bash
+ai-test analyze
+
+📦 Step 4: Reading project context...
+   Framework: React Native
+   Platforms: iOS, Android
+   State: Jotai, SWR
+   Testing: Jest
+```
+
+**Benefits**:
+- AI 分析时可以获得更精准的项目上下文
+- 支持跨端、混合项目的识别
+- 为测试生成提供更准确的技术栈信息
+
+---
+
+## [2.0.7] - 2025-01-11
+
+### 🐛 Hotfix
+- **Fixed**: `ai-test generate` workflow - `batch.mjs` now properly captures `prompt-builder.mjs` stdout and writes to `prompt.txt`
+- Modified `sh()` helper to support stdout capture with `captureStdout` option
+- Resolves missing `prompt.txt` file issue in generation workflow
+
+---
+
 ## [2.0.6] - 2025-01-11
 
 ### 🐛 Hotfix

package/lib/ai/context-builder.mjs

@@ -10,21 +10,46 @@ import { existsSync, readFileSync } from 'node:fs'
 export async function buildProjectContext() {
   const context = {
     framework: 'Unknown',
+    platforms: [],
+    uiLibraries: [],
+    stateManagement: [],
     criticalDeps: [],
-    devDeps: []
+    devDeps: [],
+    testingTools: []
   }
   
   // 读取 package.json
   if (existsSync('package.json')) {
     try {
       const pkg = JSON.parse(readFileSync('package.json', 'utf-8'))
+      const allDeps = { ...(pkg.dependencies || {}), ...(pkg.devDependencies || {}) }
       
       context.name = pkg.name
-      context.framework = detectFramework(pkg.dependencies || {})
       
-      // 识别关键依赖
+      // 检测框架和平台
+      const frameworkInfo = detectFramework(allDeps)
+      context.framework = frameworkInfo.framework
+      context.platforms = frameworkInfo.platforms
+      
+      // 检测 UI 库
+      context.uiLibraries = detectUILibraries(allDeps)
+      
+      // 检测状态管理
+      context.stateManagement = detectStateManagement(allDeps)
+      
+      // 检测测试工具
+      context.testingTools = detectTestingTools(allDeps)
+      
+      // 识别关键业务依赖
       const deps = Object.keys(pkg.dependencies || {})
-      const criticalKeywords = ['stripe', 'payment', 'auth', 'prisma', 'db', 'axios', 'fetch', 'jotai', 'zustand', 'redux']
+      const criticalKeywords = [
+        'stripe', 'payment', 'paypal', 'checkout',
+        'auth', 'jwt', 'passport', 'oauth',
+        'prisma', 'typeorm', 'sequelize', 'mongoose', 'db', 'sql',
+        'axios', 'fetch', 'request', 'graphql', 'apollo',
+        'socket', 'websocket', 'pusher', 'firebase',
+        'sentry', 'datadog', 'analytics'
+      ]
       
       context.criticalDeps = deps.filter(dep => 
         criticalKeywords.some(kw => dep.toLowerCase().includes(kw))
@@ -40,13 +65,141 @@ export async function buildProjectContext() {
 }
 
 /**
- * 检测项目框架
+ * 检测项目框架和平台
  */
 function detectFramework(deps) {
-  if (deps['next']) return 'Next.js'
-  if (deps['react']) return 'React'
-  if (deps['vue']) return 'Vue'
-  if (deps['@angular/core']) return 'Angular'
-  return 'Node.js'
+  const result = {
+    framework: 'Unknown',
+    platforms: []
+  }
+  
+  // 检测 React Native
+  const hasRN = deps['react-native'] || deps['@react-native'] || deps['expo']
+  
+  // 检测 Next.js
+  const hasNext = deps['next']
+  
+  // 检测 React
+  const hasReact = deps['react']
+  
+  // 检测 Vue
+  const hasVue = deps['vue'] || deps['@vue/core']
+  const hasNuxt = deps['nuxt']
+  
+  // 检测 Angular
+  const hasAngular = deps['@angular/core']
+  
+  // 检测 Svelte
+  const hasSvelte = deps['svelte']
+  const hasSvelteKit = deps['@sveltejs/kit']
+  
+  // 检测 Taro (跨端框架)
+  const hasTaro = deps['@tarojs/taro'] || deps['@tarojs/runtime']
+  
+  // 检测 Electron
+  const hasElectron = deps['electron']
+  
+  // 组合判断
+  if (hasTaro) {
+    result.framework = 'Taro'
+    result.platforms.push('Mini-Program', 'H5', 'RN')
+    if (hasReact) result.framework = 'Taro (React)'
+    if (hasVue) result.framework = 'Taro (Vue)'
+  } else if (hasRN && hasNext) {
+    result.framework = 'React Native + Next.js (Monorepo)'
+    result.platforms.push('iOS', 'Android', 'Web')
+  } else if (hasRN) {
+    result.framework = 'React Native'
+    result.platforms.push('iOS', 'Android')
+    if (deps['expo']) result.framework = 'React Native (Expo)'
+  } else if (hasNext) {
+    result.framework = 'Next.js'
+    result.platforms.push('Web', 'SSR')
+  } else if (hasSvelteKit) {
+    result.framework = 'SvelteKit'
+    result.platforms.push('Web', 'SSR')
+  } else if (hasNuxt) {
+    result.framework = 'Nuxt.js'
+    result.platforms.push('Web', 'SSR')
+  } else if (hasAngular) {
+    result.framework = 'Angular'
+    result.platforms.push('Web')
+  } else if (hasSvelte) {
+    result.framework = 'Svelte'
+    result.platforms.push('Web')
+  } else if (hasVue) {
+    result.framework = 'Vue'
+    result.platforms.push('Web')
+  } else if (hasReact) {
+    result.framework = 'React'
+    result.platforms.push('Web')
+  } else if (hasElectron) {
+    result.framework = 'Electron'
+    result.platforms.push('Desktop')
+  } else if (Object.keys(deps).length > 0) {
+    result.framework = 'Node.js'
+    result.platforms.push('Backend')
+  }
+  
+  return result
+}
+
+/**
+ * 检测 UI 库
+ */
+function detectUILibraries(deps) {
+  const uiLibs = []
+  
+  if (deps['antd']) uiLibs.push('Ant Design')
+  if (deps['@mui/material'] || deps['@material-ui/core']) uiLibs.push('Material-UI')
+  if (deps['@chakra-ui/react']) uiLibs.push('Chakra UI')
+  if (deps['@radix-ui/react-dialog']) uiLibs.push('Radix UI')
+  if (deps['tailwindcss']) uiLibs.push('Tailwind CSS')
+  if (deps['styled-components']) uiLibs.push('styled-components')
+  if (deps['@emotion/react']) uiLibs.push('Emotion')
+  if (deps['bootstrap']) uiLibs.push('Bootstrap')
+  if (deps['semantic-ui-react']) uiLibs.push('Semantic UI')
+  if (deps['react-native-paper']) uiLibs.push('React Native Paper')
+  if (deps['native-base']) uiLibs.push('NativeBase')
+  
+  return uiLibs
+}
+
+/**
+ * 检测状态管理
+ */
+function detectStateManagement(deps) {
+  const stateLibs = []
+  
+  if (deps['jotai']) stateLibs.push('Jotai')
+  if (deps['zustand']) stateLibs.push('Zustand')
+  if (deps['redux'] || deps['@reduxjs/toolkit']) stateLibs.push('Redux')
+  if (deps['mobx'] || deps['mobx-react']) stateLibs.push('MobX')
+  if (deps['recoil']) stateLibs.push('Recoil')
+  if (deps['xstate'] || deps['@xstate/react']) stateLibs.push('XState')
+  if (deps['valtio']) stateLibs.push('Valtio')
+  if (deps['@tanstack/react-query']) stateLibs.push('TanStack Query')
+  if (deps['swr']) stateLibs.push('SWR')
+  if (deps['apollo-client'] || deps['@apollo/client']) stateLibs.push('Apollo Client')
+  
+  return stateLibs
+}
+
+/**
+ * 检测测试工具
+ */
+function detectTestingTools(deps) {
+  const testTools = []
+  
+  if (deps['jest']) testTools.push('Jest')
+  if (deps['vitest']) testTools.push('Vitest')
+  if (deps['@testing-library/react']) testTools.push('@testing-library/react')
+  if (deps['@testing-library/react-native']) testTools.push('@testing-library/react-native')
+  if (deps['enzyme']) testTools.push('Enzyme')
+  if (deps['cypress']) testTools.push('Cypress')
+  if (deps['playwright']) testTools.push('Playwright')
+  if (deps['@playwright/test']) testTools.push('Playwright')
+  
+  return testTools
 }
 

package/lib/workflows/analyze.mjs

@@ -45,7 +45,22 @@ export async function analyze(options) {
   console.log('📦 Step 4: Reading project context...')
   const projectCtx = await buildProjectContext()
   console.log(`   Framework: ${projectCtx.framework}`)
-  console.log(`   Critical deps: ${projectCtx.criticalDeps.length > 0 ? projectCtx.criticalDeps.join(', ') : 'None detected'}\n`)
+  if (projectCtx.platforms.length > 0) {
+    console.log(`   Platforms: ${projectCtx.platforms.join(', ')}`)
+  }
+  if (projectCtx.uiLibraries.length > 0) {
+    console.log(`   UI Libraries: ${projectCtx.uiLibraries.join(', ')}`)
+  }
+  if (projectCtx.stateManagement.length > 0) {
+    console.log(`   State: ${projectCtx.stateManagement.join(', ')}`)
+  }
+  if (projectCtx.testingTools.length > 0) {
+    console.log(`   Testing: ${projectCtx.testingTools.join(', ')}`)
+  }
+  if (projectCtx.criticalDeps.length > 0) {
+    console.log(`   Critical deps: ${projectCtx.criticalDeps.join(', ')}`)
+  }
+  console.log()
   
   // 5. 构建 AI Prompt
   console.log('✍️  Step 5: Building AI analysis prompt...')

package/lib/workflows/batch.mjs

@@ -12,10 +12,24 @@ const __filename = fileURLToPath(import.meta.url)
 const __dirname = dirname(__filename)
 const pkgRoot = join(__dirname, '../..')
 
-function sh(cmd, args = []) {
+function sh(cmd, args = [], options = {}) {
   return new Promise((resolve, reject) => {
-    const child = spawn(cmd, args, { stdio: 'inherit', cwd: process.cwd() })
-    child.on('close', code => code === 0 ? resolve(0) : reject(new Error(`${cmd} exited ${code}`)))
+    const stdio = options.captureStdout ? ['inherit', 'pipe', 'inherit'] : 'inherit'
+    const child = spawn(cmd, args, { stdio, cwd: process.cwd() })
+    
+    const chunks = []
+    if (options.captureStdout) {
+      child.stdout.on('data', d => chunks.push(Buffer.from(d)))
+    }
+    
+    child.on('close', code => {
+      if (code === 0) {
+        const output = options.captureStdout ? Buffer.concat(chunks).toString('utf8') : null
+        resolve(output)
+      } else {
+        reject(new Error(`${cmd} exited ${code}`))
+      }
+    })
     child.on('error', reject)
   })
 }
@@ -112,11 +126,16 @@ async function main(argv = process.argv) {
     '--skip', String(skip),
     '--only-todo' // 新增：只处理 TODO 状态
   ]
+  
+  let promptText
   try {
-    await sh('node', [...promptArgs, '--hints-file', 'reports/hints.txt'])
+    promptText = await sh('node', [...promptArgs, '--hints-file', 'reports/hints.txt'], { captureStdout: true })
   } catch {
-    await sh('node', promptArgs)
+    promptText = await sh('node', promptArgs, { captureStdout: true })
   }
+  
+  // 写入 prompt.txt
+  writeFileSync('prompt.txt', promptText, 'utf-8')
 
   // 2) 调用 AI
   console.log('\n🤖 Calling AI...')

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-unit-test-generator",
-  "version": "2.0.6",
+  "version": "2.0.8",
   "description": "AI-powered unit test generator with smart priority scoring",
   "keywords": [
     "unit-test",