@yorha2b-lab/autodev 4.1.0 → 4.2.0

package/src/core/context.js

@@ -1,42 +1,45 @@
 const fs = require('fs')
 const path = require('path')
 const chalk = require('chalk')
+const OpenAI = require('openai')
 const Handlebars = require('handlebars')
 const stringify = require('json-stringify-pretty-compact')
 
-Handlebars.registerHelper('raw', options => options.fn())
-Handlebars.registerHelper('stringify', (context, maxLength = 200) => context ? new Handlebars.SafeString(stringify.default(context, { indent: 4, maxLength })) : '[]')
+const contextStringify = ({ context, indent = 4, maxLength = 200 }) => stringify.default(context, { indent, maxLength })
 
-let instance = null // 💡 物理封存的全局实例
+Handlebars.registerHelper('raw', opt => opt.fn())
+Handlebars.registerHelper('stringify', (context, maxLength = 200) => context ? new Handlebars.SafeString(contextStringify({ context, maxLength })) : '[]')
+
+let instance = null
 
 module.exports = {
-    /**
-     * @description 执行中枢神经元初始化
-     */
+
     init: options => {
 
-        const { apiLinker, recognizePage, generateMock, alignResponseFields } = require('../services/llm')
-        const { language, getConfig, unwrapSignal, isQuerySignal, getExistingMenus } = require('../utils/utils')
+        const template = options.template
+        const uxCtx = require('../utils/ux')
+        const coreCtx = require('../utils/core')
+        const infrastructureCtx = require('../utils/infrastructure')
+
+        const { language } = uxCtx
+        const { getConfig, getExistingMenus } = infrastructureCtx
 
         const config = getConfig()
+        const { hbsDir } = config
+
         const menus = getExistingMenus()
-        const template = options.template
         const apiHandler = require('../commands/handlers/api-handler')
         const pageHandler = require('../commands/handlers/page-handler')
         const partHandler = require('../commands/handlers/part-handler')
-        const tplDir = config.hbsDir !== '' ? path.join(process.cwd(), config.hbsDir) : path.join(__dirname, `../../templates/${template}`)
+        const tplDir = hbsDir !== '' ? path.join(process.cwd(), hbsDir) : path.join(__dirname, `../../templates/${template}`)
 
         const compilerPath = path.join(__dirname, `./${template}-compiler.js`)
         if (!fs.existsSync(compilerPath)) {
             console.error(chalk.red(language(`❌ 暂不支持 [${template}] 框架。欢迎提交 Pull Request 贡献。`, `❌ [${template}] framework not supported. Welcome to contribute a Pull Request to help.`)))
             return
         }
-        const { index, resource } = require(compilerPath)
 
-        // 💡 物理扫描：把目录下所有的 .hbs 文件全部识别并自动注册
-        const partialsDir = config.hbsDir !== ''
-            ? path.join(process.cwd(), config.hbsDir, 'handlebars')
-            : path.join(__dirname, `../../templates/${template}/handlebars`)
+        const partialsDir = hbsDir !== '' ? path.join(process.cwd(), hbsDir, 'handlebars') : path.join(__dirname, `../../templates/${template}/handlebars`)
         fs.readdirSync(partialsDir).forEach(file => {
             if (file.endsWith('.hbs')) {
                 const name = path.basename(file, '.hbs') // 去掉后缀拿名字
@@ -45,62 +48,46 @@ module.exports = {
             }
         })
 
-        const cliCtx = {
-            options,
-            template,
-        }
-
-        const llmCtx = {
-            apiLinker,
-            generateMock,
-            recognizePage,
-            alignResponseFields
-        }
-
-        const utilsCtx = {
-            menus,
-            config,
-            language,
-            unwrapSignal,
-            isQuerySignal,
-        }
-
-        const handlerCtx = {
-            apiHandler,
-            pageHandler,
-            partHandler
-        }
-
-        const compileCtx = {
-            index,
-            resource
-        }
-
         const promptCtx = {
+            mockPrompt: require(`../prompts/mock.js`),
+            sysPrompt: require(`../prompts/system.js`),
+            apiPrompt: require(`../prompts/${template}/watch-api.js`),
             pagePrompt: require(`../prompts/${template}/watch-page.js`),
             partPrompt: require(`../prompts/${template}/watch-part.js`),
+            linkerPrompt: require(`../prompts/${template}/api-linker.js`),
         }
 
-        const templateCtx = {
+        const hbsTplCtx = {
             indexTpl: Handlebars.compile(fs.readFileSync(path.join(tplDir, 'index.hbs'), 'utf-8')),
             resourceTpl: Handlebars.compile(fs.readFileSync(path.join(tplDir, 'resource.hbs'), 'utf-8')),
         }
 
+        const pkgEnv = path.resolve(__dirname, '../.env')
+        const localEnv = path.resolve(process.cwd(), '.env')
+        require('dotenv').config({ path: fs.existsSync(localEnv) ? localEnv : pkgEnv })
+        const openAI = new OpenAI({ apiKey: process.env.API_KEY, baseURL: process.env.BASE_URL })
+
+        const cliCtx = { options }
+        const handlerCtx = { apiHandler, pageHandler, partHandler }
+        const finalInfrastructureCtx = { menus, config, ...infrastructureCtx }
+        const llmCtx = require('../services/llm')({ config, openAI, template, ...promptCtx })
+        const compileCtx = require(compilerPath)({ config, contextStringify, ...coreCtx, ...hbsTplCtx })
 
         instance = {
+            openAI,
+            ...uxCtx,
             ...cliCtx,
             ...llmCtx,
-            ...utilsCtx,
+            ...coreCtx,
             ...promptCtx,
+            ...hbsTplCtx,
             ...handlerCtx,
             ...compileCtx,
-            ...templateCtx,
+            contextStringify,
+            ...finalInfrastructureCtx,
         }
         return instance
     },
 
-    /**
-     * @description 全频道信号接入：获取全局上下文
-     */
     get: () => instance
 }

package/src/core/react-compiler.js

@@ -1,98 +1,82 @@
-const stringify = require('json-stringify-pretty-compact')
-const { cleanCode, generateSmartImports, formatFormItemAndColumns } = require('../utils/utils.js')
+module.exports = ({ config, contextStringify, indexTpl, resourceTpl, formatFormItemAndColumns, cleanCode, generateSmartImports }) => {
 
-/**
- * 生成 resource.js 文件（表格列配置、表单配置、字典等）
- * @param {Object} param0 - 函数参数
- * @param {Object} param0.pageConfig - 页面配置对象（包含 tabs、formItems、table 等）
- * @param {Object} param0.resourceTpl - Handlebars 模板函数
- * @returns {string} 生成的 resource.js 代码
- */
-const resource = ({ pageConfig, resourceTpl }) => {
+    const { needMock, responseSuccess } = config
 
-    const { formItems, processedColumns, dictBlocks } = formatFormItemAndColumns({ pageConfig })
+    return {
+        index: ({ fileName, pageConfig }) => {
+            const hasTabs = pageConfig.tabs?.length > 0
+            const hasPagination = pageConfig.table.pagination
+            const hasFormItems = pageConfig.formItems?.length > 0
+            const hasOperate = pageConfig.table?.operation?.length > 0
+            const hasImageColumn = pageConfig.table?.columns?.some(item => item.type === 'image')
+            const functionButtons = pageConfig.functionButton?.filter(item => !['查询', '重置', 'query', 'search', 'reset'].includes(item.btn.toLowerCase().replaceAll(' ', ''))) || []
+            const needRenderAction = hasImageColumn
 
-    const hasTabs = pageConfig.tabs?.length > 0
-    const hasFormItems = pageConfig.formItems?.length > 0
-    const tabKeys = pageConfig.tabs.map(tab => tab.key).sort((a, b) => a.length - b.length)
-    const optionDict = pageConfig.optionList?.reduce((acc, item) => ({ ...acc, [item.name]: item.options }), {})
+            const pageStruct = ['stateBlock', 'handleBlock', 'hookBlock', hasOperate ? 'operateBlock' : '', 'renderBlock'].filter(Boolean)
 
-    const columnsData = processedColumns.map(item => {
-        delete item.type
-        return item
-    })
+            let columnsValue = hasTabs ? 'columns[activeKey]' : 'columns'
+            if (hasOperate) {
+                columnsValue = `${columnsValue}.concat(operate)`
+            }
 
-    const viewData = {
-        hasTabs,
-        columnsData,
-        hasFormItems,
-        tabs: pageConfig.tabs,
-        dictBlocks: dictBlocks.map(item => ({ name: item, data: optionDict[item] ?? [] })),
-        tabColumns: stringify.default(Object.fromEntries(tabKeys.map(tab => [tab, '_CODE_commonColumns_CODE_'])), { indent: 4, maxLength: 100 }),
-        formItemsData: !hasTabs ?
-            `export const formItems = ${stringify.default(formItems, { indent: 4, maxLength: 120 })}` :
-            `const searchItems = ${stringify.default(formItems, { indent: 4, maxLength: 120 })}\n\nexport const formItems = ${stringify.default(Object.fromEntries(tabKeys.map(tab => [tab, '_CODE_searchItems_CODE_'])), { indent: 4, maxLength: 100 })}`,
-    }
+            const imgAction = Object.fromEntries(pageConfig.table?.columns?.filter(item => item.type === 'image')?.map(item => [item.dataIndex, `_CODE_(_, record) => <a onClick={() => showImg(record, '${item.dataIndex}')}>查看图片</a>_CODE_`]))
+            const renderAction = `{${contextStringify({ context: imgAction, maxLength: 120 })}}`.replace(/\n/g, '\n    ')
 
-    const bodyCode = resourceTpl(viewData)
-    const importsStr = generateSmartImports({ module: 'resource', hasTabs, bodyCode, hasFormItems })
-    return cleanCode(`${importsStr}\n\n${bodyCode}`)
-}
+            const viewData = {
+                hasTabs,
+                pageStruct,
+                columnsValue,
+                hasFormItems,
+                renderAction,
+                hasPagination,
+                hasImageColumn,
+                functionButtons,
+                responseSuccess,
+                needRenderAction,
+                tabs: pageConfig.tabs,
+                renderTree: pageConfig.renderTree,
+                hasExpandable: pageConfig.table.expandable,
+                staticInfoText: pageConfig.staticInfo?.text,
+                operations: pageConfig.table.operation || [],
+                hasRowSelection: pageConfig.table.rowSelection,
+                uri: needMock ? fileName : 'BUNKER_API_ANCHOR_pages',
+                formItems: hasFormItems ? (hasTabs ? 'formItems[activeKey]' : 'formItems') : '[]',
+                initParams: `{ ${hasTabs ? 'type: tabs[0].key ,' : ''}${hasPagination ? 'pageNo: 1 , pageSize: 10' : ''}}`,
+            }
 
-/**
- * 生成 index.js 文件（页面组件）
- * @param {Object} param0 - 函数参数
- * @param {string} param0.fileName - 页面文件名
- * @param {Object} param0.indexTpl - Handlebars 模板函数
- * @param {Object} param0.pageConfig - 页面配置对象
- * @returns {string} 生成的 index.js 代码
- */
-const index = ({ config, fileName, indexTpl, pageConfig }) => {
+            const bodyCode = indexTpl(viewData)
+            const importsStr = generateSmartImports({ module: 'index', hasTabs, bodyCode, hasFormItems })
+            return cleanCode(`${importsStr}\n\n${bodyCode}`)
+        },
+        resource: ({ pageConfig }) => {
 
-    const hasTabs = pageConfig.tabs?.length > 0
-    const hasPagination = pageConfig.table.pagination
-    const hasFormItems = pageConfig.formItems?.length > 0
-    const hasOperate = pageConfig.table?.operation?.length > 0
-    const hasImageColumn = pageConfig.table?.columns?.some(item => item.type === 'image')
-    const renderTree = pageConfig.renderTree?.filter(item => item.toLowerCase() !== 'tabs') || []
-    const functionButtons = pageConfig.functionButton?.filter(item => !['查询', '重置', 'query', 'search', 'reset'].includes(item.btn.toLowerCase().replaceAll(' ', ''))) || []
-    const needRenderAction = hasImageColumn
+            const { formItems, processedColumns, dictBlocks } = formatFormItemAndColumns({ pageConfig })
 
-    const pageStruct = ['stateBlock', 'handleBlock', 'hookBlock', hasOperate ? 'operateBlock' : '', 'renderBlock'].filter(Boolean)
+            const hasTabs = pageConfig.tabs?.length > 0
+            const hasFormItems = pageConfig.formItems?.length > 0
+            const tabKeys = pageConfig.tabs.map(tab => tab.key).sort((a, b) => a.length - b.length)
+            const optionDict = pageConfig.optionList?.reduce((acc, item) => ({ ...acc, [item.name]: item.options }), {})
 
-    let columnsValue = hasTabs ? 'columns[activeKey]' : 'columns'
-    if (hasOperate) {
-        columnsValue = `${columnsValue}.concat(operate)`
-    }
+            const columnsData = processedColumns.map(item => {
+                delete item.type
+                return item
+            })
 
-    const imgAction = Object.fromEntries(pageConfig.table?.columns?.filter(item => item.type === 'image')?.map(item => [item.dataIndex, `_CODE_(_, record) => <a onClick={() => showImg(record, '${item.dataIndex}')}>查看图片</a>_CODE_`]))
-    const renderAction = `{${stringify.default({ ...imgAction }, { indent: 4, maxLength: 120 })}}`.replace(/\n/g, '\n    ')
+            const viewData = {
+                hasTabs,
+                columnsData,
+                hasFormItems,
+                tabs: pageConfig.tabs,
+                dictBlocks: dictBlocks.map(item => ({ name: item, data: optionDict[item] ?? [] })),
+                tabColumns: contextStringify({ context: Object.fromEntries(tabKeys.map(tab => [tab, '_CODE_commonColumns_CODE_'])), maxLength: 100 }),
+                formItemsData: !hasTabs ?
+                    `export const formItems = ${contextStringify({ context: formItems, maxLength: 120 })}` :
+                    `const searchItems = ${contextStringify({ context: formItems, maxLength: 120 })}\n\nexport const formItems = ${contextStringify({ context: Object.fromEntries(tabKeys.map(tab => [tab, '_CODE_searchItems_CODE_'])), maxLength: 100 })}`,
+            }
 
-    const viewData = {
-        hasTabs,
-        renderTree,
-        pageStruct,
-        columnsValue,
-        hasFormItems,
-        renderAction,
-        hasPagination,
-        hasImageColumn,
-        functionButtons,
-        needRenderAction,
-        tabs: pageConfig.tabs,
-        responseSuccess: config.responseSuccess,
-        hasExpandable: pageConfig.table.expandable,
-        staticInfoText: pageConfig.staticInfo?.text,
-        operations: pageConfig.table.operation || [],
-        hasRowSelection: pageConfig.table.rowSelection,
-        uri: config.needMock ? fileName : 'BUNKER_API_ANCHOR_pages',
-        formItems: hasFormItems ? (hasTabs ? 'formItems[activeKey]' : 'formItems') : '[]',
-        initParams: `{ ${hasTabs ? 'type: tabs[0].key ,' : ''}${hasPagination ? 'pageNo: 1 , pageSize: 10' : ''}}`,
+            const bodyCode = resourceTpl(viewData)
+            const importsStr = generateSmartImports({ module: 'resource', hasTabs, bodyCode, hasFormItems })
+            return cleanCode(`${importsStr}\n\n${bodyCode}`)
+        }
     }
-
-    const bodyCode = indexTpl(viewData)
-    const importsStr = generateSmartImports({ module: 'index', hasTabs, bodyCode, hasFormItems })
-    return cleanCode(`${importsStr}\n\n${bodyCode}`)
-}
-
-module.exports = { index, resource }
+}

package/src/prompts/mock.js

@@ -1,4 +1,4 @@
-module.exports = (columns, fileName) => `
+module.exports = ({ columns, fileName }) => `
 根据columns的dataIndex字段生成一个简单的接口mock数据: ${JSON.stringify(columns)},
 格式为{POST /api/${fileName}:{data:2条真实mock数据]}}
 `

package/src/prompts/react/api-linker.js

@@ -1,4 +1,4 @@
-module.exports = (bunkerAnchors, realApis) => `
+module.exports = ({ bunkerAnchors, realApis }) => `
 # MISSION: Multi-Point API Alignment
 目标：为地堡生成的多个作战动作匹配真实的后端接口。
 

package/src/prompts/react/watch-api.js

@@ -1,4 +1,4 @@
-module.exports = (responseStr, resourceStr) => `
+module.exports = ({ responseStr, resourceStr }) => `
 目标：执行【前端虚拟字段】与【后端真实响应】的物理重组与对账。
 
 ## 1. 核心输入 (Signal Capture)

package/src/prompts/react/watch-page.js

@@ -34,6 +34,7 @@ module.exports = `
 ### 表格主体 (Table)
 - ⚠️ **操作锁定**: 严禁在 columns 数组中包含“操作”列。
 - **标准项**: {title:'列名', dataIndex:'列名英文名词',type:''}。
+- **注意**: type为text时必须省略type属性。
 - **配置项**: [pagination, expandable, rowSelection] 均为布尔值false,只有页面明确有对应功能时才为true。
 - **标准定义 (Columns)**:
     - enum: 枚举列

package/src/prompts/react/watch-part.js

@@ -10,6 +10,7 @@ module.exports = `
 ## 1. 表格 (Table)
 - 结构: { columns: [{ title: '列名', dataIndex: '列名英文名词', type: '' }] }
 - ⚠️ 字段白名单: [title, dataIndex, sorter, render, filters, onFilter]
+- **注意**: type为text时必须省略type属性。
 - 禁止输出:
     - 严禁在 columns 中包含“操作”列。
     - ***⚠️⚠️⚠️严禁输出formItems,只允许输出columns⚠️⚠️⚠️***

package/src/services/api-linker.js

@@ -1,65 +1,27 @@
-const fs = require('fs')
-const path = require('path')
-const chalk = require('chalk')
-const { Segment, useDefault } = require('segmentit')
-const segmentit = useDefault(new Segment())
-
-/**
- * @function getSemanticKeywords
- * @description [语义精炼厂 2.0] 使用 segmentit 引擎进行物理切片。
- */
-const getSemanticKeywords = text => {
-    // 💡 物理脱水：确保输入为纯净字符串
-    const content = Array.isArray(text) ? text.join(' ') : (text || '')
-    // 💡 执行分词
-    const result = segmentit.doSegment(content, {
-        simple: true, // 开启简易模式，提升吞吐效率
-        stripPunctuation: true // 物理过滤掉所有的标点符号
-    })
-    // 💡 提取词条并执行“基因筛选”
-    return Array.from(new Set(result.filter(word => word.length >= 2 && !/^[0-9]+$/.test(word))))
-}
-
-/**
- * @function getLocalScore
- * @description [战力评估协议] 计算语义重合度。
- */
-const getLocalScore = (api, pageKeywords, moduleName) => {
-    let score = 0
-    const path = api.path.toLowerCase()
-    const desc = api.desc.toLowerCase()
-    const mod = moduleName.toLowerCase()
-
-    // 💡 权重 A：路径直接包含模块名 (暴击加分 +50)
-    if (path.includes(mod)) score += 50
-
-    // 💡 权重 B：描述包含模块核心词 (中量加分 +10)
-    pageKeywords.forEach(word => {
-        if (desc.includes(word.toLowerCase())) score += 10
-        if (path.includes(word.toLowerCase())) score += 5
-    })
-
-    return score
-}
-
-/**
- * @module apiLinker
- * @description [地堡 4.0 核心] 全频道自动寻址协议。
- */
 module.exports = async () => {
 
-    const { get } = require('../core/context')
-    const { config, options, language, apiLinker } = get()
-    if (!config.apiDoc) return
+    const fs = require('fs')
+    const path = require('path')
+    const chalk = require('chalk')
+
+    const {
+        config,
+        language,
+        apiLinker,
+        getLocalScore, getSemanticKeywords,
+    } = require('../core/context').get()
+
+    const { apiDoc, pagesDir } = config
+    if (!apiDoc) return
 
     console.log(chalk.cyan(language(
         `\n🤖 Pod 153: 启动‘全频道自动寻址协议 [语义扫描版]’...`,
-        `\n🤖 Pod 153: Initiating "Full-Channel Autonomous Addressing Protocol [Semantic Scan]"...`
+        `\n🤖 Pod 153: Initiating 'Full-Channel Autonomous Addressing Protocol [Semantic Scan]'...`
     )))
 
     try {
         // 1. 抓取情报
-        const apiData = await fetch(config.apiDoc).then(res => res.json())
+        const apiData = await fetch(apiDoc).then(res => res.json())
         const refinedApis = Object.entries(apiData.paths).flatMap(([apiUrl, methods]) => {
             // 💡 适配多方法接口：有的路径下可能既有 GET 也有 POST
             return Object.entries(methods).map(([method, info]) => ({
@@ -70,12 +32,12 @@ module.exports = async () => {
         })
 
         // 2. 战场侦察
-        const pagesDir = path.join(process.cwd(), config.pagesDir)
-        const files = fs.readdirSync(pagesDir)
+        const pages = path.join(process.cwd(), pagesDir)
+        const files = fs.readdirSync(pages)
 
         for (const fileName of files) {
-            const indexPath = path.join(pagesDir, fileName, 'index.js')
-            const resourcePath = path.join(pagesDir, fileName, 'resource.js')
+            const indexPath = path.join(pages, fileName, 'index.js')
+            const resourcePath = path.join(pages, fileName, 'resource.js')
 
             if (fs.existsSync(indexPath) && fs.existsSync(resourcePath)) {
                 let indexCode = fs.readFileSync(indexPath, 'utf-8')
@@ -83,7 +45,7 @@ module.exports = async () => {
                 // 💡 锁定待通电锚点
                 if (indexCode.includes('BUNKER_API_ANCHOR')) {
                     const resourceCode = fs.readFileSync(resourcePath, 'utf-8')
-                    const bunkerAnchors = indexCode.match(/BUNKER_API_ANCHOR_\w+/g).join('\n')
+                    const bunkerAnchors = indexCode.match(/BUNKER_API_ANCHOR_\w+/g)?.join('\n') ?? ''
 
                     // 💡 执行语义提取流程
                     const returnMatch = indexCode.match(/return\s*\(([\s\S]*?)\)\s*}/)
@@ -109,7 +71,7 @@ module.exports = async () => {
                     )))
 
                     // 3. 驱动 9S 最终裁决
-                    const result = await apiLinker(options, bunkerAnchors, finalCandidates)
+                    const result = await apiLinker({ bunkerAnchors, realApis: finalCandidates })
 
                     if (result) {
                         // 💡 执行物理更替

package/src/services/llm.js

@@ -1,24 +1,8 @@
-const fs = require('fs')
-const path = require('path')
-const chalk = require('chalk')
-const { language, getConfig } = require('../utils/utils.js')
-
-let client = null
 
-const getOpenAI = () => {
-    if (!client) {
-        const OpenAI = require('openai')
-        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
-}
+const chalk = require('chalk')
+const { language } = require('../utils/ux')
 
-const askAI = async ({ model, messages, response_format = { type: 'json_object' }, retryCount = 0 }) => {
+const askAI = async ({ model, openAI, messages, response_format = { type: 'json_object' }, retryCount = 0 }) => {
 
     if (retryCount > 3) {
         throw new Error(language(
@@ -28,8 +12,7 @@ const askAI = async ({ model, messages, response_format = { type: 'json_object'
     }
 
     try {
-        const openai = getOpenAI()
-        const response = await openai.chat.completions.create({
+        const response = await openAI.chat.completions.create({
             model,
             messages,
             top_p: 0.1,
@@ -60,66 +43,66 @@ const askAI = async ({ model, messages, response_format = { type: 'json_object'
     }
 }
 
-module.exports = {
-    generateMock: async (columns, fileName) => {
-        const config = getConfig()
-        const mockPrompt = require('../prompts/mock.js')
-        const { MOCK_DESIGNER } = require('../prompts/system.js')
-        return askAI({
-            model: config.textModel,
-            messages: [
-                { role: 'system', content: MOCK_DESIGNER },
-                { role: 'user', content: mockPrompt(columns, fileName) }
-            ]
-        })
-    },
-    apiLinker: async (options, bunkerAnchors, realApis) => {
-        const config = getConfig()
-        const { API_DESIGNER } = require('../prompts/system.js')
-        const apiPrompt = require(`../prompts/${options.template}/api-linker.js`)
-        return askAI({
-            model: config.textModel,
-            messages: [
-                { role: 'system', content: API_DESIGNER },
-                { role: 'user', content: apiPrompt(bunkerAnchors, realApis) }
-            ]
-        })
-    },
-    recognizePage: async (prompt, filePath, taskType = 'page') => {
-        const config = getConfig()
-        const { UI_DESIGNER } = require('../prompts/system.js')
-        const sharp = require('sharp')
-        const compressedBuffer = await sharp(filePath)
-            .resize(1280, null, { withoutEnlargement: true })
-            .jpeg({ quality: 80 })
-            .toBuffer()
-        const base64Image = compressedBuffer.toString('base64')
+module.exports = ({ config, openAI, template, sysPrompt, apiPrompt, mockPrompt, linkerPrompt }) => {
 
-        return askAI({
-            model: config.visionModel,
-            messages: [
-                { role: 'system', content: UI_DESIGNER },
-                {
-                    role: 'user',
-                    content: [
-                        { type: 'text', text: prompt },
-                        { type: 'image_url', image_url: { url: `data:image/jpeg;base64,${base64Image}` } }
-                    ]
-                }
-            ],
-            response_format: taskType === 'page' ? { type: 'json_schema', strict: true, json_schema: require('../schema/page.json') } : { type: 'json_object' }
-        })
-    },
-    alignResponseFields: async (options, responseStr, resourceStr) => {
-        const config = getConfig()
-        const { API_DESIGNER } = require('../prompts/system.js')
-        const apiPrompt = require(`../prompts/${options.template}/watch-api.js`)
-        return askAI({
-            model: config.textModel,
-            messages: [
-                { role: 'system', content: API_DESIGNER },
-                { role: 'user', content: apiPrompt(responseStr, resourceStr) }
-            ]
-        })
+    const sharp = require('sharp')
+
+    const { textModel, visionModel } = config
+    const { UI_DESIGNER, API_DESIGNER, MOCK_DESIGNER } = sysPrompt
+
+    return {
+        generateMock: async ({ columns, fileName }) => {
+            return askAI({
+                openAI,
+                model: textModel,
+                messages: [
+                    { role: 'system', content: MOCK_DESIGNER },
+                    { role: 'user', content: mockPrompt({ columns, fileName }) }
+                ]
+            })
+        },
+        apiLinker: async ({ bunkerAnchors, realApis }) => {
+            return askAI({
+                openAI,
+                model: textModel,
+                messages: [
+                    { role: 'system', content: API_DESIGNER },
+                    { role: 'user', content: linkerPrompt({ bunkerAnchors, realApis }) }
+                ]
+            })
+        },
+        alignResponseFields: async ({ responseStr, resourceStr }) => {
+            return askAI({
+                openAI,
+                model: textModel,
+                messages: [
+                    { role: 'system', content: API_DESIGNER },
+                    { role: 'user', content: apiPrompt({ responseStr, resourceStr }) }
+                ]
+            })
+        },
+        recognizePage: async ({ prompt, filePath, taskType = 'page' }) => {
+            const compressedBuffer = await sharp(filePath)
+                .resize(1280, null, { withoutEnlargement: true })
+                .jpeg({ quality: 80 })
+                .toBuffer()
+            const base64Image = compressedBuffer.toString('base64')
+
+            return askAI({
+                openAI,
+                model: visionModel,
+                messages: [
+                    { role: 'system', content: UI_DESIGNER },
+                    {
+                        role: 'user',
+                        content: [
+                            { type: 'text', text: prompt },
+                            { type: 'image_url', image_url: { url: `data:image/jpeg;base64,${base64Image}` } }
+                        ]
+                    }
+                ],
+                response_format: taskType === 'page' ? { type: 'json_schema', strict: true, json_schema: require(`../schema/${template}/page.json`) } : { type: 'json_object' }
+            })
+        },
     }
 }