@yorha2b-lab/autodev 1.0.6 → 1.0.8

package/bin/autodev.js

@@ -3,6 +3,7 @@ const fs = require('fs')
 const path = require('path')
 const chalk = require('chalk')
 const figlet = require('figlet')
+const { program } = require('commander')
 
 console.log(chalk.cyan(figlet.textSync('AutoDev', { horizontalLayout: 'full' })))
 console.log(chalk.gray('--------------------------------------------------'))
@@ -10,20 +11,7 @@ console.log(chalk.white(' [System] ') + chalk.green('YoRHa No.2 Type B Unit: ')
 console.log(chalk.white(' [Mission] ') + chalk.yellow('Generate Frontend CRUD: ') + chalk.cyan('Awaiting Command'))
 console.log(chalk.gray('--------------------------------------------------\n'))
 
-const localEnv = path.resolve(process.cwd(), '.env')
-const pkgEnv = path.resolve(__dirname, '../.env')
-
-require("dotenv").config({
-    path: fs.existsSync(localEnv) ? localEnv : pkgEnv
-})
-
-const { program } = require('commander')
-const watchApi = require('../src/commands/watch-api')
-const watchPage = require('../src/commands/watch-page')
-const watchPart = require('../src/commands/watch-part')
-
 program.version('1.0.0').description('AI驱动的前端CRUD代码生成器')
-
 program.option('-t, --template <type>', '指定前端框架模板', 'react')
 
 program
@@ -41,16 +29,25 @@ program
 program
     .command('watch:page')
     .description('监听截图，自动生成完整的增删改查页面')
-    .action(() => watchPage(program.opts()))
+    .action(() => {
+        const watchPage = require('../src/commands/watch-page')
+        watchPage(program.opts())
+    })
 
 program
     .command('watch:part')
     .description('监听截图，自动生成局部 UI 组件')
-    .action(() => watchPart(program.opts()))
+    .action(() => {
+        const watchPart = require('../src/commands/watch-part')
+        watchPart(program.opts())
+    })
 
 program
     .command('watch:api')
     .description('监听 Swagger，自动对齐真实接口字段')
-    .action(() => watchApi())
+    .action(() => {
+        const watchApi = require('../src/commands/watch-api')
+        watchApi()
+    })
 
 program.parse(process.argv)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yorha2b-lab/autodev",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "description": "基于视觉大模型的前端(Umi+Antd)全自动 CRUD 代码生成器",
   "bin": {
     "autodev": "bin/autodev.js"

package/src/commands/watch-api.js

@@ -1,11 +1,12 @@
 const fs = require('fs')
 const path = require('path')
 const chokidar = require('chokidar')
-const { getConfig } = require('../utils/utils.js')
-const { alignSwaggerFields } = require('../services/llm.js')
 
 const watchApi = () => {
 
+    const { getConfig } = require('../utils/utils.js')
+    const { alignSwaggerFields } = require('../services/llm.js')
+
     const config = getConfig()
 
     const watcher = chokidar.watch('./swagger', {

package/src/commands/watch-page.js

@@ -1,16 +1,17 @@
 const fs = require('fs')
 const ora = require('ora')
 const path = require('path')
-const chokidar = require('chokidar')
-const Handlebars = require('handlebars')
-const pagePrompt = require('../prompts/watch-page.js')
-const { createTaskQueue } = require('../core/task-queue')
-const stringify = require('json-stringify-pretty-compact')
-const { recognizePage, generateMock } = require('../services/llm.js')
-const { getConfig, copyHooks, copyComponents, getExistingMenus } = require('../utils/utils.js')
 
 const watchPage = options => {
 
+    const chokidar = require('chokidar')
+    const Handlebars = require('handlebars')
+    const pagePrompt = require('../prompts/watch-page.js')
+    const { createTaskQueue } = require('../core/task-queue')
+    const stringify = require('json-stringify-pretty-compact')
+    const { recognizePage, generateMock } = require('../services/llm.js')
+    const { getConfig, copyHooks, copyComponents, getExistingMenus } = require('../utils/utils.js')
+
     const config = getConfig()
     const compilerPath = path.join(__dirname, `../core/${options.template}-compiler.js`)
 

package/src/commands/watch-part.js

@@ -1,10 +1,11 @@
 const fs = require('fs')
 const chokidar = require('chokidar')
-const partPrompt = require('../prompts/watch-part.js')
-const { recognizePage } = require('../services/llm.js')
 const stringify = require('json-stringify-pretty-compact')
 
 const watchPart = () => {
+
+    const { recognizePage } = require('../services/llm.js')
+
     const watcher = chokidar.watch('./screenPart', {
         persistent: true,
         ignored: /(^|[\/\\])\../,
@@ -13,10 +14,12 @@ const watchPart = () => {
             pollInterval: 100
         }
     })
+
     watcher.on('add', async filePath => {
         try {
             const startTime = Date.now()
             console.log(`🚀 检测到新图片: 开始识别...`)
+            const partPrompt = require('../prompts/watch-part.js')
             const pageConfig = await recognizePage(partPrompt, filePath)
             const optionDict = pageConfig.optionDict || {}
             delete pageConfig.optionDict

package/src/services/llm.js

@@ -1,17 +1,19 @@
+const fs = require('fs')
+const path = require('path')
 const JSON5 = require('json5')
 const sharp = require('sharp')
 const OpenAI = require('openai')
-const mockPrompt = require('../prompts/mock.js')
 const { getConfig } = require('../utils/utils.js')
-const apiPrompt = require('../prompts/watch-api.js')
-const { UI_DESIGNER, API_DESIGNER, MOCK_DESIGNER } = require('../prompts/system.js')
-
-const config = getConfig()
 
 let client = null
 
 const getOpenAI = () => {
     if (!client) {
+        const localEnv = path.resolve(process.cwd(), '.env')
+        const pkgEnv = path.resolve(__dirname, '../.env')
+        require("dotenv").config({
+            path: fs.existsSync(localEnv) ? localEnv : pkgEnv
+        })
         client = new OpenAI({ apiKey: process.env.API_KEY, baseURL: process.env.BASE_URL })
     }
     return client
@@ -36,6 +38,9 @@ const askAI = async (model, messages, retryCount = 0) => {
 
 module.exports = {
     generateMock: async (columns, fileName) => {
+        const config = getConfig()
+        const mockPrompt = require('../prompts/mock.js')
+        const { MOCK_DESIGNER } = require('../prompts/system.js')
         return askAI(
             config.textModel,
             [
@@ -45,11 +50,15 @@ module.exports = {
         )
     },
     recognizePage: async (prompt, filePath) => {
+        const config = getConfig()
+        const { UI_DESIGNER } = require('../prompts/system.js')
+
         const compressedBuffer = await sharp(filePath)
             .resize(1280, null, { withoutEnlargement: true })
             .jpeg({ quality: 80 })
             .toBuffer()
         const base64Image = compressedBuffer.toString('base64')
+
         return askAI(
             config.visionModel,
             [
@@ -65,6 +74,9 @@ module.exports = {
         )
     },
     alignSwaggerFields: async (swaggerStr, resourceStr) => {
+        const config = getConfig()
+        const apiPrompt = require('../prompts/watch-api.js')
+        const { API_DESIGNER } = require('../prompts/system.js')
         return askAI(
             config.textModel,
             [