You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

1412 lines
45 KiB

const { readFileSync, writeFileSync, existsSync, readdirSync } = require('fs')
const { resolve } = require('path')
const parser = require('@babel/parser')
const traverse = require('@babel/traverse').default
const { parse } = require('@vue/compiler-sfc')
const AstAnalyzer = require('./ast-analyzer')
/**
* 触发器分析模块
* 负责分析组件中的按钮和事件触发器
*/
class TriggerAnalyzer {
constructor() {
this.astAnalyzer = new AstAnalyzer()
}
/**
* 查找API方法的触发器源 - 第一步:第一层触发源分析
* @param {Array} apiMappings - API映射数组
* @param {Array} routes - 路由配置数组
* @returns {Array} 增强的API映射数组
*/
findTriggerSourcesForApiMappings(apiMappings, routes) {
const enhancedApiMappings = []
// 创建组件到authType的映射
const componentAuthMap = this.createComponentAuthMap(routes)
apiMappings.forEach(moduleMapping => {
const enhancedModuleMapping = {
...moduleMapping,
apiMappings: []
}
moduleMapping.apiMappings.forEach(apiMapping => {
// 第一步:查找第一层触发源
const triggerSources = this.findFirstLevelTriggerSources(
moduleMapping.serviceName, // 使用第一层的serviceName
apiMapping.apiMethodName,
moduleMapping.module
)
// 过滤掉authType为public的组件的触发源
const filteredTriggerSources = this.filterPublicComponentTriggers(
triggerSources,
componentAuthMap
)
// 只有当有触发源时才添加API映射
if (filteredTriggerSources.length > 0) {
enhancedModuleMapping.apiMappings.push({
...apiMapping,
triggerSources: filteredTriggerSources
})
}
})
// 只有当模块有API映射时才添加模块映射
if (enhancedModuleMapping.apiMappings.length > 0) {
enhancedApiMappings.push(enhancedModuleMapping)
}
})
return enhancedApiMappings
}
/**
* 查找第一层触发源 - 使用Babel分析真正的调用关系
* @param {string} serviceName - 服务名称
* @param {string} apiMethodName - API方法名称
* @param {string} moduleName - 模块名称
* @returns {Array} 触发器源数组
*/
findFirstLevelTriggerSources(serviceName, apiMethodName, moduleName) {
const triggerSources = []
try {
// 1. 遍历模块中的所有Vue组件文件
const modulePath = resolve(__dirname, `../../src/renderer/modules/${moduleName}`)
if (existsSync(modulePath)) {
const files = readdirSync(modulePath, { recursive: true })
files.forEach(file => {
if (typeof file === 'string' && file.endsWith('.vue')) {
const filePath = resolve(modulePath, file)
const componentName = this.extractComponentNameFromPath(file)
// 2. 使用Babel分析Vue组件中的API调用
const componentTriggerSources = this.analyzeComponentForApiCalls(
filePath,
serviceName,
apiMethodName,
componentName
)
triggerSources.push(...componentTriggerSources)
}
})
}
} catch (error) {
console.warn(`分析模块 ${moduleName} 时出错:`, error.message)
}
// 去重
const uniqueTriggerSources = this.deduplicateTriggerSources(triggerSources)
return uniqueTriggerSources
}
/**
* 查找API方法的触发器源(保留原方法用于后续扩展)
* @param {string} serviceName - 服务名称
* @param {string} apiMethodName - API方法名称
* @param {string} moduleName - 模块名称
* @returns {Array} 触发器源数组
*/
findTriggerSourcesForApiMethod(serviceName, apiMethodName, moduleName) {
const triggerSources = []
// 1. 检查路由组件
const routeComponents = this.getRouteComponents(moduleName)
routeComponents.forEach(route => {
const triggerAnalysis = this.analyzeComponentWithAST(
route.component,
serviceName,
apiMethodName
)
if (triggerAnalysis.triggerSources.length > 0) {
triggerSources.push(...triggerAnalysis.triggerSources)
}
})
// 2. 检查模块中的所有组件
const componentsInModule = this.getModuleComponents(moduleName)
componentsInModule.forEach(componentName => {
const triggerAnalysis = this.analyzeComponentWithAST(
componentName,
serviceName,
apiMethodName
)
if (triggerAnalysis.triggerSources.length > 0) {
triggerSources.push(...triggerAnalysis.triggerSources)
}
})
// 去重
const uniqueTriggerSources = this.deduplicateTriggerSources(triggerSources)
return uniqueTriggerSources
}
/**
* 获取路由组件
* @param {string} moduleName - 模块名称
* @returns {Array} 路由组件数组
*/
getRouteComponents(moduleName) {
const routeComponents = []
try {
const routeConfigPath = resolve(__dirname, '../../src/renderer/router/index.js')
if (existsSync(routeConfigPath)) {
const routeContent = readFileSync(routeConfigPath, 'utf-8')
// 根据模块名称映射到对应的组件名称
const moduleToComponentMap = {
'role-management': 'RoleManagement',
'user-management': 'UserManagement',
'system-settings': 'Settings',
'user-profile': 'UserProfile'
}
const componentName = moduleToComponentMap[moduleName]
if (componentName) {
// 查找包含该组件的路由
const routeMatches = routeContent.match(new RegExp(`component:\\s*${componentName}`, 'g'))
if (routeMatches) {
routeComponents.push({
component: componentName
})
}
}
}
} catch (error) {
// 静默处理错误
}
return routeComponents
}
/**
* 获取模块中的所有组件
* @param {string} moduleName - 模块名称
* @returns {Array} 组件名称数组
*/
getModuleComponents(moduleName) {
const components = []
try {
const modulePath = resolve(__dirname, '../../src/renderer/modules', moduleName)
if (existsSync(modulePath)) {
// 查找views目录
const viewsPath = resolve(modulePath, 'views')
if (existsSync(viewsPath)) {
const viewFiles = readdirSync(viewsPath).filter(file => file.endsWith('.vue'))
viewFiles.forEach(file => {
components.push(file.replace('.vue', ''))
})
}
// 查找components目录
const componentsPath = resolve(modulePath, 'components')
if (existsSync(componentsPath)) {
const componentFiles = readdirSync(componentsPath).filter(file => file.endsWith('.vue'))
componentFiles.forEach(file => {
components.push(file.replace('.vue', ''))
})
}
}
} catch (error) {
// 静默处理错误
}
return components
}
/**
* 使用AST分析组件
* @param {string} componentName - 组件名称
* @param {string} serviceName - 服务名称
* @param {string} apiMethodName - API方法名称
* @returns {Object} 触发器分析结果
*/
analyzeComponentWithAST(componentName, serviceName, apiMethodName) {
const triggerSources = []
try {
const filePath = this.findComponentFile(componentName)
if (!filePath) {
return { triggerSources: [] }
}
const content = readFileSync(filePath, 'utf-8')
// 1. 检查组件的authType属性,如果是public则跳过
const authTypeMatch = content.match(/authType\s*[=:]\s*["']([^"']+)["']/)
if (authTypeMatch && authTypeMatch[1] === 'public') {
return { triggerSources: [] }
}
// 2. 检查组件是否包含目标API调用
if (!content.includes(`${serviceName}.${apiMethodName}`)) {
return { triggerSources: [] }
}
// 3. 使用Babel AST分析找到调用该API的方法
const callingMethods = this.astAnalyzer.findMethodsCallingServiceWithBabel(content, serviceName, apiMethodName)
if (callingMethods.length > 0) {
// 4. 分析每个调用方法的触发源
const methodTriggers = this.analyzeMethodTriggerSources(content, componentName, callingMethods, filePath)
triggerSources.push(...methodTriggers)
} else {
// 如果没有找到按钮触发器,记录页面级触发器
triggerSources.push({
component: componentName,
triggerName: null,
triggerType: 'page'
})
}
} catch (error) {
// 静默处理错误
}
return { triggerSources }
}
/**
* 查找组件文件
* @param {string} componentName - 组件名称
* @returns {string|null} 组件文件路径
*/
findComponentFile(componentName) {
const possiblePaths = [
resolve(__dirname, '../../src/renderer/modules', componentName, 'views', `${componentName}.vue`),
resolve(__dirname, '../../src/renderer/modules', componentName, 'components', `${componentName}.vue`),
resolve(__dirname, '../../src/renderer/components', `${componentName}.vue`),
resolve(__dirname, '../../src/renderer/views', `${componentName}.vue`)
]
for (const path of possiblePaths) {
if (existsSync(path)) {
return path
}
}
return null
}
/**
* 分析方法的触发源
* @param {string} content - 组件内容
* @param {string} componentName - 组件名称
* @param {Array} methodNames - 方法名数组
* @param {string} filePath - 文件路径
* @returns {Array} 触发器源数组
*/
analyzeMethodTriggerSources(content, componentName, methodNames, filePath = null) {
const triggerSources = []
// 提取模板部分
const templateMatch = content.match(/<template>([\s\S]*?)<\/template>/)
if (!templateMatch) {
return triggerSources
}
const templateContent = templateMatch[1]
// 使用正则表达式查找调用这些方法的按钮
methodNames.forEach(methodName => {
// 1. 查找@click="methodName"或@click="methodName("
const clickPattern = new RegExp(`@click\\s*=\\s*["']${methodName}\\s*\\([^)]*\\)["']`, 'g')
const clickMatches = templateContent.match(clickPattern)
if (clickMatches) {
clickMatches.forEach((match, index) => {
// 查找包含这个@click的button标签
const buttonMatch = this.findButtonContainingClick(templateContent, match)
if (buttonMatch) {
const buttonName = this.extractButtonName(buttonMatch)
const triggerType = buttonName ? 'button' : 'button'
triggerSources.push({
component: componentName,
triggerName: buttonName || this.astAnalyzer.generateUniqueButtonName(componentName, methodName),
triggerType: triggerType
})
}
})
}
// 2. 查找@submit.prevent="methodName"的表单提交按钮
const submitPattern = new RegExp(`@submit\\.prevent\\s*=\\s*["']${methodName}["']`, 'g')
const submitMatches = templateContent.match(submitPattern)
if (submitMatches) {
submitMatches.forEach((match, index) => {
// 查找包含这个@submit.prevent的form标签
const formMatch = this.findFormContainingSubmit(templateContent, match)
if (formMatch) {
// 查找表单中的type="submit"按钮
const submitButtonMatch = this.findSubmitButtonInForm(formMatch)
if (submitButtonMatch) {
const buttonName = this.extractButtonName(submitButtonMatch)
const triggerType = buttonName ? 'button' : 'button'
triggerSources.push({
component: componentName,
triggerName: buttonName || this.astAnalyzer.generateUniqueButtonName(componentName, methodName),
triggerType: triggerType
})
} else {
// 如果没有找到具体的提交按钮,记录为form类型
triggerSources.push({
component: componentName,
triggerName: null,
triggerType: 'form'
})
}
}
})
}
})
return triggerSources
}
/**
* 查找包含指定@click的button标签
* @param {string} templateContent - 模板内容
* @param {string} clickMatch - @click匹配
* @returns {string|null} 按钮HTML
*/
findButtonContainingClick(templateContent, clickMatch) {
const clickIndex = templateContent.indexOf(clickMatch)
if (clickIndex === -1) return null
// 从@click位置往前查找最近的<button标签
const beforeClick = templateContent.substring(0, clickIndex)
const lastButtonIndex = beforeClick.lastIndexOf('<button')
if (lastButtonIndex === -1) return null
// 从<button开始往后查找对应的</button>
const afterButton = templateContent.substring(lastButtonIndex)
const buttonEndIndex = afterButton.indexOf('</button>')
if (buttonEndIndex === -1) return null
return afterButton.substring(0, buttonEndIndex + 9) // 包含</button>
}
/**
* 提取按钮名称
* @param {string} buttonHtml - 按钮HTML
* @returns {string|null} 按钮名称
*/
extractButtonName(buttonHtml) {
const nameMatch = buttonHtml.match(/name\s*=\s*["']([^"']+)["']/)
return nameMatch ? nameMatch[1] : null
}
/**
* 查找包含指定@submit.prevent的form标签
* @param {string} templateContent - 模板内容
* @param {string} submitMatch - @submit.prevent匹配
* @returns {string|null} 表单HTML
*/
findFormContainingSubmit(templateContent, submitMatch) {
const submitIndex = templateContent.indexOf(submitMatch)
if (submitIndex === -1) return null
// 从@submit.prevent位置往前查找最近的<form标签
const beforeSubmit = templateContent.substring(0, submitIndex)
const lastFormIndex = beforeSubmit.lastIndexOf('<form')
if (lastFormIndex === -1) return null
// 从<form开始往后查找对应的</form>
const afterForm = templateContent.substring(lastFormIndex)
const formEndIndex = afterForm.indexOf('</form>')
if (formEndIndex === -1) return null
return afterForm.substring(0, formEndIndex + 7) // 包含</form>
}
/**
* 查找表单中的type="submit"按钮
* @param {string} formHtml - 表单HTML
* @returns {string|null} 提交按钮HTML
*/
findSubmitButtonInForm(formHtml) {
const submitButtonPattern = /<button[^>]*type\s*=\s*["']submit["'][^>]*>[\s\S]*?<\/button>/g
const match = submitButtonPattern.exec(formHtml)
return match ? match[0] : null
}
/**
* 去重触发器源
* @param {Array} triggerSources - 触发器源数组
* @returns {Array} 去重后的触发器源数组
*/
deduplicateTriggerSources(triggerSources) {
const seen = new Set()
return triggerSources.filter(trigger => {
const key = `${trigger.component}-${trigger.triggerName}-${trigger.triggerType}`
if (seen.has(key)) {
return false
}
seen.add(key)
return true
})
}
/**
* 创建组件到authType的映射
* @param {Array} routes - 路由配置数组
* @returns {Map} 组件到authType的映射
*/
createComponentAuthMap(routes) {
const componentAuthMap = new Map()
// 从Vue组件文件中读取authType属性
routes.forEach(route => {
if (route.component) {
const authType = this.getComponentAuthType(route.component)
if (authType) {
componentAuthMap.set(route.component, authType)
}
}
})
// 扫描所有模块目录中的组件文件,查找authType属性
const moduleDirs = ['core', 'user-management', 'role-management', 'system-settings']
moduleDirs.forEach(moduleDir => {
this.scanComponentsForAuthType(moduleDir, componentAuthMap)
})
return componentAuthMap
}
/**
* 扫描指定模块目录中的组件文件,查找authType属性
* @param {string} moduleDir - 模块目录名
* @param {Map} componentAuthMap - 组件到authType的映射
*/
scanComponentsForAuthType(moduleDir, componentAuthMap) {
try {
const modulePath = resolve(__dirname, `../../src/renderer/modules/${moduleDir}`)
if (existsSync(modulePath)) {
const files = readdirSync(modulePath, { recursive: true })
files.forEach(file => {
if (typeof file === 'string' && file.endsWith('.vue')) {
const filePath = resolve(modulePath, file)
const componentName = this.extractComponentNameFromPath(file)
// 如果还没有这个组件的authType信息,则尝试获取
if (!componentAuthMap.has(componentName)) {
const authType = this.getComponentAuthType(componentName)
if (authType) {
componentAuthMap.set(componentName, authType)
}
}
}
})
}
} catch (error) {
console.warn(`扫描模块 ${moduleDir} 的组件时出错:`, error.message)
}
}
/**
* 从Vue组件文件中获取authType属性
* @param {string} componentName - 组件名称
* @returns {string|null} authType值
*/
getComponentAuthType(componentName) {
try {
// 在modules目录中查找组件文件
const moduleDirs = ['core', 'user-management', 'role-management', 'system-settings']
for (const moduleDir of moduleDirs) {
const componentPath = resolve(__dirname, `../../src/renderer/modules/${moduleDir}/components/${componentName}.vue`)
if (existsSync(componentPath)) {
const content = readFileSync(componentPath, 'utf-8')
const authTypeMatch = content.match(/<template\s+authType\s*=\s*["']([^"']+)["']/)
if (authTypeMatch) {
return authTypeMatch[1]
}
}
// 也检查views目录
const viewPath = resolve(__dirname, `../../src/renderer/modules/${moduleDir}/views/${componentName}.vue`)
if (existsSync(viewPath)) {
const content = readFileSync(viewPath, 'utf-8')
const authTypeMatch = content.match(/<template\s+authType\s*=\s*["']([^"']+)["']/)
if (authTypeMatch) {
return authTypeMatch[1]
}
}
}
} catch (error) {
console.warn(`获取组件 ${componentName} 的authType时出错:`, error.message)
}
return null
}
/**
* 过滤掉authType为public的组件的触发源
* @param {Array} triggerSources - 触发源数组
* @param {Map} componentAuthMap - 组件到authType的映射
* @returns {Array} 过滤后的触发源数组
*/
filterPublicComponentTriggers(triggerSources, componentAuthMap) {
return triggerSources.filter(trigger => {
const authType = componentAuthMap.get(trigger.component)
// 如果组件没有定义authType,或者authType不是public,则保留
return !authType || authType !== 'public'
})
}
/**
* 从文件路径提取组件名称
* @param {string} filePath - 文件路径
* @returns {string} 组件名称
*/
extractComponentNameFromPath(filePath) {
const fileName = filePath.split('/').pop().split('\\').pop()
return fileName.replace('.vue', '')
}
/**
* 分析Vue组件中的API调用
* @param {string} filePath - 组件文件路径
* @param {string} serviceName - 服务名称
* @param {string} apiMethodName - API方法名称
* @param {string} componentName - 组件名称
* @returns {Array} 触发源数组
*/
analyzeComponentForApiCalls(filePath, serviceName, apiMethodName, componentName) {
const triggerSources = []
try {
const content = readFileSync(filePath, 'utf-8')
// 使用Babel解析Vue组件
const ast = this.parseVueComponent(content)
if (!ast) return triggerSources
// 查找API调用和函数调用关系,传递文件路径用于追溯
const apiCalls = this.findApiCallsWithContext(ast, serviceName, apiMethodName, filePath, componentName)
// 为每个API调用创建触发源
apiCalls.forEach(call => {
triggerSources.push({
component: componentName,
triggerName: call.triggerName || '',
triggerType: call.triggerType || 'function'
})
})
} catch (error) {
console.warn(`分析组件 ${filePath} 时出错:`, error.message)
}
return triggerSources
}
/**
* 解析Vue组件内容
* @param {string} content - 组件内容
* @returns {Object|null} AST对象
*/
parseVueComponent(content) {
try {
// 提取<script>部分
const scriptMatch = content.match(/<script[^>]*>([\s\S]*?)<\/script>/)
if (!scriptMatch) return null
const scriptContent = scriptMatch[1]
// 使用Babel解析JavaScript代码
const ast = parser.parse(scriptContent, {
sourceType: 'module',
plugins: ['jsx', 'decorators-legacy', 'classProperties', 'objectRestSpread']
})
return ast
} catch (error) {
console.warn('解析Vue组件脚本时出错:', error.message)
return null
}
}
/**
* 在AST中查找API调用和函数调用关系
* @param {Object} ast - AST对象
* @param {string} serviceName - 服务名称
* @param {string} apiMethodName - API方法名称
* @param {string} filePath - 组件文件路径
* @param {string} componentName - 组件名称
* @returns {Array} API调用信息数组
*/
findApiCallsWithContext(ast, serviceName, apiMethodName, filePath, componentName) {
const apiCalls = []
const self = this
const functionCallMap = new Map() // 存储函数调用关系
// 第一遍:收集所有函数定义和调用关系
traverse(ast, {
// 收集函数定义
FunctionDeclaration(path) {
if (path.node.id && path.node.id.name) {
functionCallMap.set(path.node.id.name, {
type: 'function',
path: path
})
}
},
// 收集箭头函数和const函数
VariableDeclarator(path) {
if (path.node.id && path.node.id.type === 'Identifier' &&
path.node.init && path.node.init.type === 'ArrowFunctionExpression') {
functionCallMap.set(path.node.id.name, {
type: 'arrowFunction',
path: path
})
}
},
// 收集函数调用关系
CallExpression(path) {
const { node } = path
// 记录函数调用
if (node.callee.type === 'Identifier') {
const functionName = node.callee.name
const parentFunction = self.findParentFunction(path)
if (parentFunction) {
if (!functionCallMap.has(functionName)) {
functionCallMap.set(functionName, { callers: [] })
}
const funcInfo = functionCallMap.get(functionName)
if (!funcInfo.callers) funcInfo.callers = []
funcInfo.callers.push(parentFunction)
}
}
}
})
// 第二遍:查找API调用并分析调用链
traverse(ast, {
CallExpression(path) {
const { node } = path
// 查找 serviceName.apiMethodName() 调用
if (node.callee.type === 'MemberExpression' &&
node.callee.object.name === serviceName &&
node.callee.property.name === apiMethodName) {
// 查找调用上下文
const context = self.findCallContextWithChain(path, functionCallMap, filePath, componentName)
apiCalls.push({
triggerName: context.triggerName,
triggerType: context.triggerType
})
}
}
})
return apiCalls
}
/**
* 查找父级函数
* @param {Object} path - Babel路径对象
* @returns {string|null} 父级函数名
*/
findParentFunction(path) {
let currentPath = path.parentPath
while (currentPath) {
const { node } = currentPath
if (node.type === 'FunctionDeclaration' && node.id) {
return node.id.name
}
if (node.type === 'VariableDeclarator' &&
node.id && node.id.type === 'Identifier' &&
node.init && node.init.type === 'ArrowFunctionExpression') {
return node.id.name
}
if (node.type === 'ObjectMethod' && node.key) {
return node.key.name
}
currentPath = currentPath.parentPath
}
return null
}
/**
* 查找API调用的上下文(包含调用链分析)
* @param {Object} path - Babel路径对象
* @param {Map} functionCallMap - 函数调用关系映射
* @param {string} filePath - 组件文件路径
* @param {string} componentName - 组件名称
* @returns {Object} 上下文信息
*/
findCallContextWithChain(path, functionCallMap, filePath, componentName) {
let currentPath = path
let triggerName = ''
let triggerType = 'function'
// 向上遍历AST,查找触发源
while (currentPath) {
const { node } = currentPath
// 检查是否在Vue 3 Composition API生命周期钩子中
if (node.type === 'CallExpression' &&
node.callee.type === 'Identifier' &&
['onMounted', 'onCreated', 'onBeforeMount', 'onBeforeCreate', 'onUpdated', 'onBeforeUpdate', 'onUnmounted', 'onBeforeUnmount'].includes(node.callee.name)) {
triggerName = node.callee.name
triggerType = 'lifecycle'
break
}
// 检查是否在Vue 3 Composition API的setup方法中
if (node.type === 'ObjectMethod' && node.key && node.key.name === 'setup') {
triggerName = 'setup'
triggerType = 'lifecycle' // setup是Vue 3的生命周期钩子
break
}
// 检查是否在方法定义中
if (node.type === 'FunctionDeclaration' || node.type === 'FunctionExpression') {
if (node.id && node.id.name) {
triggerName = node.id.name
triggerType = 'function'
break
}
}
// 检查是否在对象方法中(非setup)
if (node.type === 'ObjectMethod') {
if (node.key && node.key.name && node.key.name !== 'setup') {
triggerName = node.key.name
triggerType = 'method'
break
}
}
// 检查是否在箭头函数中
if (node.type === 'ArrowFunctionExpression') {
// 查找父级的属性名
const parent = currentPath.parent
if (parent && parent.type === 'ObjectProperty' && parent.key) {
const methodName = parent.key.name || ''
if (methodName === 'setup') {
triggerName = 'setup'
triggerType = 'lifecycle'
} else {
triggerName = methodName
triggerType = 'method'
}
break
}
// 检查是否在const声明的箭头函数中
if (parent && parent.type === 'VariableDeclarator' && parent.id) {
const functionName = parent.id.name || ''
triggerName = functionName
triggerType = 'method'
break
}
}
// 检查是否在Vue 2生命周期钩子中
if (node.type === 'CallExpression' &&
node.callee.type === 'MemberExpression' &&
node.callee.object.name === 'this' &&
['mounted', 'created', 'beforeMount', 'beforeCreate'].includes(node.callee.property.name)) {
triggerName = node.callee.property.name
triggerType = 'lifecycle'
break
}
currentPath = currentPath.parentPath
}
// 如果找到的是method类型,继续向上追溯找到最终的可视组件
if (triggerType === 'method') {
const finalContext = this.traceToVisualComponent(path, triggerName, functionCallMap, filePath, componentName)
if (finalContext) {
return finalContext
}
}
return { triggerName, triggerType }
}
/**
* 追溯method类型的触发源到最终的可视组件
* @param {Object} path - Babel路径对象
* @param {string} methodName - 方法名称
* @param {Map} functionCallMap - 函数调用关系映射
* @param {string} filePath - 组件文件路径
* @returns {Object|null} 最终的可视组件上下文
*/
traceToVisualComponent(path, methodName, functionCallMap, filePath, componentName) {
try {
// 使用传入的文件路径
const componentPath = filePath
if (!componentPath) return null
// 读取组件文件内容
const content = readFileSync(componentPath, 'utf-8')
// 分析模板中的事件绑定
const visualContext = this.analyzeTemplateForMethod(content, methodName, componentName, filePath)
if (visualContext) {
return visualContext
}
// 如果模板中没有找到,检查是否有其他方法调用了这个方法
const callerContext = this.findMethodCaller(methodName, functionCallMap)
if (callerContext) {
// 检查调用者是否是生命周期钩子
const callerInfo = functionCallMap.get(callerContext)
if (callerInfo) {
// 检查调用者是否是生命周期钩子
if (this.isLifecycleHook(callerContext)) {
return {
triggerName: '',
triggerType: 'page'
}
}
}
// 继续递归追溯
return this.traceToVisualComponent(path, callerContext, functionCallMap, filePath, componentName)
}
} catch (error) {
console.warn(`追溯可视组件时出错:`, error.message)
}
return null
}
/**
* 检查方法名是否是生命周期钩子
* @param {string} methodName - 方法名称
* @returns {boolean} 是否是生命周期钩子
*/
isLifecycleHook(methodName) {
const lifecycleHooks = [
'setup',
'onMounted',
'onCreated',
'onBeforeMount',
'onBeforeCreate',
'onUpdated',
'onBeforeUpdate',
'onUnmounted',
'onBeforeUnmount',
'mounted',
'created',
'beforeMount',
'beforeCreate',
'updated',
'beforeUpdate',
'unmounted',
'beforeUnmount',
'watch'
]
return lifecycleHooks.includes(methodName)
}
/**
* 分析模板中与指定方法相关的事件绑定
* @param {string} content - 组件文件内容
* @param {string} methodName - 方法名称
* @param {string} componentName - 组件名称
* @param {string} filePath - 组件文件路径
* @returns {Object|null} 可视组件上下文
*/
analyzeTemplateForMethod(content, methodName, componentName, filePath) {
try {
// 使用Vue编译器解析单文件组件
const { descriptor } = parse(content)
if (!descriptor.template) {
return null
}
// 解析模板AST
const templateAst = descriptor.template.ast
if (!templateAst) {
return null
}
// 遍历模板AST,查找事件绑定
return this.findEventBindingInTemplate(templateAst, methodName, componentName, filePath)
} catch (error) {
console.warn(`解析Vue模板时出错:`, error.message)
return null
}
}
/**
* 在模板AST中查找事件绑定
* @param {Object} node - AST节点
* @param {string} methodName - 方法名称
* @param {string} componentName - 组件名称
* @param {string} filePath - 组件文件路径
* @returns {Object|null} 可视组件上下文
*/
findEventBindingInTemplate(node, methodName, componentName, filePath) {
if (!node) return null
// 检查当前节点是否有事件绑定
if (node.props) {
for (const prop of node.props) {
if (prop.type === 7) { // DIRECTIVE类型
const eventName = prop.name
if (eventName === 'on' && prop.arg) {
const eventType = prop.arg.content
if (['click', 'submit', 'change', 'input'].includes(eventType)) {
// 检查事件处理函数
if (prop.exp && this.isMethodCall(prop.exp, methodName)) {
const context = this.createVisualContext(node, eventType, componentName, methodName, filePath)
// 如果找到的是form,需要查找其中的submit按钮
if (context.triggerType === 'form') {
const submitButton = this.findSubmitButtonInForm(node, componentName, methodName, filePath)
if (submitButton) {
return submitButton
}
}
return context
}
}
}
}
}
}
// 递归检查子节点
if (node.children) {
for (const child of node.children) {
if (child.type === 1) { // ELEMENT类型
const result = this.findEventBindingInTemplate(child, methodName, componentName, filePath)
if (result) return result
}
}
}
return null
}
/**
* 在form中查找submit按钮
* @param {Object} formNode - form AST节点
* @param {string} componentName - 组件名称
* @param {string} methodName - 方法名称
* @returns {Object|null} submit按钮上下文
*/
findSubmitButtonInForm(formNode, componentName, methodName, filePath) {
if (!formNode || !formNode.children) return null
// 递归查找submit按钮
for (const child of formNode.children) {
if (child.type === 1) { // ELEMENT类型
if (child.tag === 'button') {
// 检查是否是submit按钮
const props = child.props || []
const attributes = {}
props.forEach(prop => {
if (prop.type === 6) { // ATTRIBUTE
attributes[prop.name] = prop.value ? prop.value.content : ''
}
})
if (attributes.type === 'submit') {
let triggerName = attributes.name
if (!triggerName) {
// 生成唯一的name属性:组件名+唯一尾缀
triggerName = `${componentName.toLowerCase()}-${this.generateUniqueSuffix()}`
// 自动为Vue文件添加name属性
this.addNameAttributeToVueFile(filePath, child, triggerName)
}
return {
triggerName: triggerName,
triggerType: 'button'
}
}
}
// 递归查找子元素
const result = this.findSubmitButtonInForm(child, componentName, methodName, filePath)
if (result) return result
}
}
return null
}
/**
* 检查表达式是否是方法调用
* @param {Object} exp - 表达式节点
* @param {string} methodName - 方法名称
* @returns {boolean} 是否是方法调用
*/
isMethodCall(exp, methodName) {
if (!exp) return false
// 简单的方法调用:methodName
if (exp.type === 4 && exp.content === methodName) { // SIMPLE_EXPRESSION
return true
}
// 带参数的方法调用:methodName(...) - 检查content是否以methodName开头
if (exp.type === 4 && exp.content && exp.content.startsWith(methodName + '(')) {
return true
}
// 带参数的方法调用:methodName(...)
if (exp.type === 8) { // COMPOUND_EXPRESSION
const children = exp.children
if (children && children.length >= 1) {
const firstChild = children[0]
if (firstChild.type === 4 && firstChild.content === methodName) {
return true
}
}
}
// 检查AST中的方法调用
if (exp.ast) {
return this.checkAstMethodCall(exp.ast, methodName)
}
return false
}
/**
* 检查AST中的方法调用
* @param {Object} ast - AST节点
* @param {string} methodName - 方法名称
* @returns {boolean} 是否是方法调用
*/
checkAstMethodCall(ast, methodName) {
if (!ast) return false
// 简单标识符:methodName
if (ast.type === 'Identifier' && ast.name === methodName) {
return true
}
// 方法调用:methodName(...)
if (ast.type === 'CallExpression') {
if (ast.callee && ast.callee.type === 'Identifier' && ast.callee.name === methodName) {
return true
}
}
return false
}
/**
* 创建可视组件上下文
* @param {Object} node - AST节点
* @param {string} eventType - 事件类型
* @param {string} componentName - 组件名称
* @param {string} methodName - 方法名称
* @param {string} filePath - 组件文件路径
* @returns {Object} 可视组件上下文
*/
createVisualContext(node, eventType, componentName, methodName, filePath) {
const tag = node.tag
const props = node.props || []
// 提取元素属性
const attributes = {}
props.forEach(prop => {
if (prop.type === 6) { // ATTRIBUTE
attributes[prop.name] = prop.value ? prop.value.content : ''
}
})
// 根据标签类型确定组件类型和名称
if (tag === 'button') {
let triggerName = attributes.name
if (!triggerName) {
// 生成唯一的name属性:组件名+唯一尾缀
triggerName = `${componentName.toLowerCase()}-${this.generateUniqueSuffix()}`
// 自动为Vue文件添加name属性
this.addNameAttributeToVueFile(filePath, node, triggerName)
}
return {
triggerName: triggerName,
triggerType: 'button'
}
} else if (tag === 'form') {
let triggerName = attributes.name
if (!triggerName) {
// 生成唯一的name属性:组件名+唯一尾缀
triggerName = `${componentName.toLowerCase()}-${this.generateUniqueSuffix()}`
}
return {
triggerName: triggerName,
triggerType: 'form'
}
} else if (tag === 'input') {
let triggerName = attributes.name
if (!triggerName) {
// 生成唯一的name属性:组件名+唯一尾缀
triggerName = `${componentName.toLowerCase()}-${this.generateUniqueSuffix()}`
}
return {
triggerName: triggerName,
triggerType: 'input'
}
} else if (tag === 'a') {
let triggerName = attributes.name
if (!triggerName) {
// 生成唯一的name属性:组件名+唯一尾缀
triggerName = `${componentName.toLowerCase()}-${this.generateUniqueSuffix()}`
}
return {
triggerName: triggerName,
triggerType: 'link'
}
} else if (tag === 'i') {
let triggerName = attributes.name
if (!triggerName) {
// 生成唯一的name属性:组件名+唯一尾缀
triggerName = `${componentName.toLowerCase()}-${this.generateUniqueSuffix()}`
}
return {
triggerName: triggerName,
triggerType: 'icon'
}
} else {
let triggerName = attributes.name
if (!triggerName) {
// 生成唯一的name属性:组件名+唯一尾缀
triggerName = `${componentName.toLowerCase()}-${this.generateUniqueSuffix()}`
}
return {
triggerName: triggerName,
triggerType: 'element'
}
}
}
/**
* 生成唯一的尾缀
* @returns {string} 唯一尾缀
*/
generateUniqueSuffix() {
// 生成6位随机字符串,类似 "2m2snp"
const chars = 'abcdefghijklmnopqrstuvwxyz0123456789'
let result = ''
for (let i = 0; i < 6; i++) {
result += chars.charAt(Math.floor(Math.random() * chars.length))
}
return result
}
/**
* 为Vue文件中的button添加name属性
* @param {string} filePath - Vue文件路径
* @param {Object} node - AST节点
* @param {string} nameValue - name属性值
*/
addNameAttributeToVueFile(filePath, node, nameValue) {
try {
// 读取Vue文件内容
const content = readFileSync(filePath, 'utf-8')
// 获取节点的位置信息
const loc = node.loc
if (!loc) {
console.warn(`节点没有位置信息,无法添加name属性`)
return
}
// 找到button标签的开始位置
const startLine = loc.start.line
const startColumn = loc.start.column
console.log(`尝试为第${startLine}行第${startColumn}列的button添加name属性: ${nameValue}`)
// 将内容按行分割
const lines = content.split('\n')
// 找到对应的行
if (startLine > 0 && startLine <= lines.length) {
const lineIndex = startLine - 1
let line = lines[lineIndex]
console.log(`当前行内容: ${line}`)
// 在button标签中添加name属性
// 查找button标签的开始位置
const buttonStartMatch = line.match(/<button[^>]*>/)
if (buttonStartMatch) {
// 如果button标签在同一行结束
const buttonEndIndex = buttonStartMatch.index + buttonStartMatch[0].length
const beforeButton = line.substring(0, buttonEndIndex - 1) // 去掉最后的>
const afterButton = line.substring(buttonEndIndex - 1)
// 添加name属性
const newLine = `${beforeButton} name="${nameValue}">${afterButton}`
lines[lineIndex] = newLine
// 写回文件
const newContent = lines.join('\n')
writeFileSync(filePath, newContent, 'utf-8')
console.log(`已为 ${filePath} 中的button添加name属性: ${nameValue}`)
} else {
// 如果button标签跨行,查找button开始标签
const buttonStartMatch = line.match(/<button[^>]*$/)
if (buttonStartMatch) {
// 在button开始标签后添加name属性
const beforeButton = line.substring(0, buttonStartMatch.index + buttonStartMatch[0].length)
const afterButton = line.substring(buttonStartMatch.index + buttonStartMatch[0].length)
// 添加name属性
const newLine = `${beforeButton} name="${nameValue}"${afterButton}`
lines[lineIndex] = newLine
// 写回文件
const newContent = lines.join('\n')
writeFileSync(filePath, newContent, 'utf-8')
console.log(`已为 ${filePath} 中的跨行button添加name属性: ${nameValue}`)
} else {
console.warn(`未找到button标签`)
}
}
} else {
console.warn(`行号超出范围: ${startLine}`)
}
} catch (error) {
console.warn(`为Vue文件添加name属性时出错:`, error.message)
}
}
/**
* 查找方法的调用者
*/
findMethodCaller(methodName, functionCallMap) {
const funcInfo = functionCallMap.get(methodName)
if (funcInfo && funcInfo.callers && funcInfo.callers.length > 0) {
return funcInfo.callers[0] // 返回第一个调用者
}
return null
}
/**
* 查找API调用的上下文
* @param {Object} path - Babel路径对象
* @returns {Object} 上下文信息
*/
findCallContext(path) {
let currentPath = path
let triggerName = ''
let triggerType = 'function'
// 向上遍历AST,查找触发源
while (currentPath) {
const { node } = currentPath
// 检查是否在Vue 3 Composition API生命周期钩子中
if (node.type === 'CallExpression' &&
node.callee.type === 'Identifier' &&
['onMounted', 'onCreated', 'onBeforeMount', 'onBeforeCreate', 'onUpdated', 'onBeforeUpdate', 'onUnmounted', 'onBeforeUnmount'].includes(node.callee.name)) {
triggerName = node.callee.name
triggerType = 'lifecycle'
break
}
// 检查是否在Vue 3 Composition API的setup方法中
if (node.type === 'ObjectMethod' && node.key && node.key.name === 'setup') {
triggerName = 'setup'
triggerType = 'lifecycle' // setup是Vue 3的生命周期钩子
break
}
// 检查是否在方法定义中
if (node.type === 'FunctionDeclaration' || node.type === 'FunctionExpression') {
if (node.id && node.id.name) {
triggerName = node.id.name
triggerType = 'function'
break
}
}
// 检查是否在对象方法中(非setup)
if (node.type === 'ObjectMethod') {
if (node.key && node.key.name && node.key.name !== 'setup') {
triggerName = node.key.name
triggerType = 'method'
break
}
}
// 检查是否在箭头函数中
if (node.type === 'ArrowFunctionExpression') {
// 查找父级的属性名
const parent = currentPath.parent
if (parent && parent.type === 'ObjectProperty' && parent.key) {
const methodName = parent.key.name || ''
if (methodName === 'setup') {
triggerName = 'setup'
triggerType = 'lifecycle'
} else {
triggerName = methodName
triggerType = 'method'
}
break
}
}
// 检查是否在Vue 2生命周期钩子中
if (node.type === 'CallExpression' &&
node.callee.type === 'MemberExpression' &&
node.callee.object.name === 'this' &&
['mounted', 'created', 'beforeMount', 'beforeCreate'].includes(node.callee.property.name)) {
triggerName = node.callee.property.name
triggerType = 'lifecycle'
break
}
currentPath = currentPath.parentPath
}
return { triggerName, triggerType }
}
}
module.exports = TriggerAnalyzer