@yorha2b-lab/autodev 1.0.39 → 1.0.41

package/README.md

@@ -143,6 +143,11 @@ module.exports = {
     visionModel: 'qwen3.5-plus',
     textModel: 'qwen-turbo',
 
+    // 自定义模板目录
+    // 如果为空，将使用内置模板
+    // 自定义模板需要在hbsdir目录创建自己的index.hbs和resource.hbs文件
+    hbsDir: '',
+
     // 目标项目的目录路径
     hooksDir: 'src/hooks',
     pagesDir: 'src/pages',

package/README_EN.md

@@ -143,6 +143,11 @@ module.exports = {
     visionModel: 'qwen3.5-plus',
     textModel: 'qwen-turbo',
 
+    // Custom template directory path
+    // If empty, will use built-in templates
+    // Custom templates need to create their own index.hbs and resource.hbs files in the hbsdir directory
+    hbsDir: '',
+
     // Target project directory paths
     hooksDir: 'src/hooks',
     pagesDir: 'src/pages',

package/config.js

@@ -16,6 +16,11 @@ module.exports = {
 
     textModel: 'qwen-turbo',
 
+    // 自定义模板目录
+    // 如果为空，将使用内置模板
+    // 自定义模板需要在hbsdir目录创建自己的index.hbs和resource.hbs文件
+    hbsDir: '',
+
     // 目标项目的 hooks 目录路径
     hooksDir: 'src/hooks',
 

package/package.json

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

package/src/commands/watch-page.js

@@ -42,9 +42,12 @@ const watchPage = options => {
     const { index, resource } = require(compilerPath)
 
     try {
-        copyTemplateDir(options, 'hooks', config.hooksDir)
-        copyTemplateDir(options, 'utils', config.utilsDir)
-        copyTemplateDir(options, 'components', config.componentsDir)
+        //默认使用内置模板
+        if (config.hbsDir === '') {
+            copyTemplateDir(options, 'hooks', config.hooksDir)
+            copyTemplateDir(options, 'utils', config.utilsDir)
+            copyTemplateDir(options, 'components', config.componentsDir)
+        }
     } catch (error) {
         console.error(language('❌ 程序运行出错:', '❌ Runtime error:'), error)
     }
@@ -52,7 +55,7 @@ const watchPage = options => {
     const queue = createTaskQueue(2)
     const menus = getExistingMenus()
 
-    const tplDir = path.join(__dirname, `../../templates/${options.template}`)
+    const tplDir = config.hbsDir !== '' ? path.join(process.cwd(), config.hbsDir) : path.join(__dirname, `../../templates/${options.template}`)
     const indexTpl = Handlebars.compile(fs.readFileSync(path.join(tplDir, 'index.hbs'), 'utf-8'))
     const resourceTpl = Handlebars.compile(fs.readFileSync(path.join(tplDir, 'resource.hbs'), 'utf-8'))
 
@@ -119,10 +122,10 @@ const watchPage = options => {
                         `🧑‍💻 9S: Leave it to me, 2B! Hacking and forging Mock data for [${fileName}]...`
                     )))
                     const rawContent = await generateMock(pageConfig.table.columns, fileName)
-                    fs.writeFileSync(path.join(mockDir, `${fileName}.js`), `export default ${stringify.default(rawContent, { indent: 4, maxLength: 200 })}`.replaceAll(`"'`, `"`).replaceAll(`'"`, `"`))
+                    fs.writeFileSync(path.join(mockDir, `${fileName}.js`), `export default ${stringify.default(rawContent, { indent: 4, maxLength: 200 })}`)
                     console.log(chalk.green(language(
-                        `🤖 Pod 042: [报告] Mock 数据注入完成，耗时: ${(Date.now() - startTime) / 1000} 秒`,
-                        `🤖 Pod 042: [Report] Mock data injection complete. Elapsed time: ${(Date.now() - startTime) / 1000}s`
+                        `\n🤖 Pod 042: [报告] Mock 数据注入完成，耗时: ${(Date.now() - startTime) / 1000} 秒`,
+                        `\n🤖 Pod 042: [Report] Mock data injection complete. Elapsed time: ${(Date.now() - startTime) / 1000}s`
                     )))
                 }
 

package/src/prompts/mock.js

@@ -1,5 +1,6 @@
 module.exports = (columns, fileName) => `
-根据这些字段生成一个简单的get接口mock数据: ${JSON.stringify(columns)},
-格式如下：
-'POST /api/${fileName}': {data:模块对象列表},
+根据这些字段生成一个简单的接口mock数据: ${JSON.stringify(columns)},
+格式为{key:value}:
+key: POST /api/${fileName}
+value: {data:模块对象列表},
 `