iFlyCode 逆向分析文档

Appearance

3. 子包间交互关系 ​

                    +-------------------+
                    |  TemplateGenerator|  (入口单例)
                    +--------+----------+
                             |
              +--------------+---------------+
              |                              |
    +---------v----------+       +-----------v-----------+
    | CreateTestFileTask |       | CreateTestMethodTask  |
    | (批量测试生成)      |       | (单方法测试生成)       |
    +----+------+--------+       +-----------+-----------+
         |      |                            |
         |      +-------+                    |
         v              v                    v
  +------+------+  +----+-------+    +-------+--------+
  | fileloader  |  | request    |    | generator      |
  | 子包        |  | 子包       |    | 子包(配置/定位) |
  +------+------+  +----+-------+    +-------+--------+
         |              |                     |
         v              v                     v
  +------+------+  +----+-------------------------+
  | TestTemplate|  | TemplateRequestService       |
  | ContextBuilder  | (AI 请求/响应处理)            |
  +----+------+--+  +----+------+---------+-------+
       |      |          |         |         |
       v      v          v         v         v
  +----+---+ +--+---+  +-+----+ +-+-----+ +-+------+
  |builder | |context| |agent | |test   | |settings|
  |子包    | |子包   | |包    | |包     | |包      |
  +--------+ +--+---+ +------+ +-------+ +--------+
                |
       +--------+--------+
       |        |        |
       v        v        v
  +----+--+ +---+---+ +--+----+
  |domain | |resolved| |service|
  |子包   | |子包    | |子包   |
  +-------+ +--------+ +-------+

交互关系详述 ​

  1. TemplateGenerator -> generator: 入口类创建 CreateTestFileTask/CreateTestMethodTask
  2. generator -> fileloader: 加载 Velocity 模板，获取 FileTemplateContext
  3. generator -> request: 通过 TemplateRequestService 请求 AI 生成用例
  4. generator -> builder: 使用 MockBuilderFactory 创建 mock 构建器
  5. TestTemplateContextBuilder -> builder: 调用 MockBuilderFactory 和 MethodReferencesBuilder
  6. TestTemplateContextBuilder -> context/domain: 构建 Type、Method、Field 域对象
  7. context/service -> context/domain: JavaTestBuilderImpl 使用 Type、Method、Param 生成代码
  8. context/service -> request/dto: 使用 CaseResult、CaseParam 渲染 AI 生成的用例
  9. request -> agent: TemplateRequestService 通过 PluginWebsocketClient 发送 AI 请求
  10. builder -> context/domain: MockitoMockBuilder 检查 Field、Type 的 mockability

4. 与 agent/service 的调用链 ​

4.1 AI 请求调用链 ​

TemplateGenerator.createTestClass()
  -> CreateTestMethodTask.run()
    -> TemplateRequestService.requestAI()
      -> appendTypeBody()              // 构建类结构信息
      -> getBodyContent()              // 获取方法体
      -> PluginWebsocketClient         // 发送 WebSocket 请求
      -> 等待 AI 响应
    -> TemplateRequestService.handleAgentAction()
      -> caseHandle()                  // 解析 <test-case> 标签
      -> caseMocks()                   // 解析 <case-mock> 标签
      -> convertInput()                // 转换输入参数
      -> convertOutput()               // 转换输出参数
      -> addMock()                     // 添加 mock 方法
      -> addBranches()                 // 添加分支信息
    -> TestTemplateContextBuilder.build()
      -> 创建 Type/Method/Field 域对象
      -> resolveMethodReferences()
      -> resolveMethodCallFields()
    -> Velocity 模板合并
    -> CodeRefactorUtil.uncommentImports()
    -> SocketMessageHandleListener.send2Web()  // 发送结果到 WebView

4.2 外部包依赖 ​

外部包被调用类调用场景
com.aicode.agentPluginWebsocketClient, SocketMessageHandleListener, MessageDto, CommandEnumAI 请求发送与响应接收
com.aicode.action.batchGeneratorConfig, MethodGeneratorConfig, BatchUnitTestTemplateService批量测试配置
com.aicode.testUnitTestService, UnitTestDto, MethodUnitTestData测试结果回传
com.aicode.settingsAICodeSettingsState, BatchUnitTestSettingsState读取用户配置
com.aicode.enumsDuplicateRule, PluginSceneEnum, TestGenerationProcess, UnitTestBaseEnum, UnitTestMockEnum, WebViewDataTypeEnum枚举引用
com.aicode.utilStringUtils, TypeUtils, ClassNameUtils, FileUtils, JavaPsiUtils, PropertyUtils, PsiUtils, ReflectUtil, AICodeUtils, UnitTestCollectUtil工具方法
com.aicode.messageBasicActionsBundle国际化消息

4.3 WebSocket 通信协议 ​

// 请求方向：IDE -> Agent Server
TemplateRequestService.requestAI()
  -> PluginWebsocketClient.send()
    -> CommandEnum: REQUEST_COMPLETE / GENERATE_UNIT_TEST
    -> TemplateTestPromptDto: &#123; stream, content, unitTest &#125;

// 响应方向：Agent Server -> IDE
TemplateRequestService.handleAgentAction()
  <- CommandEnum: UNIT_TEST_CASE / UNIT_TEST_ERROR
  <- JsonObject: &#123; caseMethodName, input, output, mockMethods, branches, exception &#125;

5. 模板生成完整流程图 ​

用户触发测试生成
       |
       v
[TemplateGenerator]  (单例入口)
       |
       +-- 批量模式 --> [batchTestClass()]
       |                    |
       |                    v
       |              [CreateTestFileTask] (Backgroundable)
       |                    |
       +-- 单方法模式 -> [createTestClass()]
                            |
                            v
                      [CreateTestMethodTask]
                            |
       +-------+-------+---+---+-------+-------+
       |       |       |       |       |       |
       v       v       v       v       v       v
    [解析    [创建    [构建    [请求    [合并    [写入
     PSI]    TypeDict] 上下文]  AI]    模板]   文件]
       |       |       |       |       |       |
       v       v       v       v       v       v
    PsiClass  Type    Test    Template  Velocity  PsiDirectory
    PsiMethod Dictionary Template Request  合并    写入
    PsiField         Context Builder  Service  结果
                            |           |
                   +--------+--+     +--+--------+
                   |           |     |           |
                   v           v     v           v
              [MockBuilder [Method [构建      [解析
               Factory]    Ref    Prompt]    响应]
                           Builder]
                                        |
                                        v
                              +---------+---------+
                              |                   |
                              v                   v
                         [CaseResult]       [CaseBranch]
                         [CaseParam]        [ToMockMethod]
                              |                   |
                              +---------+---------+
                                        |
                                        v
                              [JavaTestBuilderImpl]
                              (渲染测试代码)
                                        |
                              +---------+---------+
                              |                   |
                              v                   v
                         [参数渲染]          [断言渲染]
                         renderJavaCallParams  renderJavaMethodAssert
                         renderReturnParam     AssertUtil.assertResult
                         renderInitType
                              |
                              v
                         [Velocity 模板合并]
                              |
                              v
                         [CodeRefactorUtil.uncommentImports()]
                              |
                              v
                         [SocketMessageHandleListener.send2Web()]
                              |
                              v
                         WebView 前端展示

流程步骤详解 ​

  1. 用户触发: 通过 Action 或 WebView 触发测试生成
  2. 任务检查: 检查是否有未完成的测试生成任务（"上一个单元测试任务还未完成"）
  3. PSI 解析: 在 ReadAction 中解析 PsiClass、PsiMethod、PsiField
  4. 类型字典: 创建 TypeDictionary，缓存类型解析结果
  5. 上下文构建: TestTemplateContextBuilder 组装所有模板参数
  6. AI 请求: TemplateRequestService 构建并发送 AI prompt
  7. 响应解析: 解析 AI 返回的 XML 格式用例数据
  8. 代码渲染: JavaTestBuilderImpl 根据 CaseResult 渲染测试代码
  9. 模板合并: Velocity 引擎合并模板和参数
  10. 文件写入: 在 WriteAction 中写入测试文件
  11. 后处理: 取消注释 import、格式化代码
  12. 结果回传: 通过 WebSocket 发送结果到 WebView

6. 关键发现 ​

6.1 无 H() 混淆调用 ​

template 包中 没有发现 com.aicode.util.H 的调用。所有字符串常量均为明文，包括中文提示信息。

6.2 AI 通信协议 ​

AI 请求使用 XML 标签格式：

  • <test-case>...</test-case> - 测试用例
  • <case-mock>...</case-mock> - mock 信息
  • &lt;branch&gt;...&lt;/branch&gt; - 分支信息

6.3 模板框架支持 ​

支持 7 种测试模板组合：

  1. JUnit4 + 无 Mock
  2. JUnit5 + 无 Mock
  3. JUnit4 + Mockito
  4. JUnit4 + PowerMock
  5. JUnit5 + Mockito
  6. TestNG + Mockito
  7. SpringBootTest + Mockito

6.4 核心复杂度分布 ​

类行数复杂度来源
JavaTestBuilderImpl7056参数渲染、类型解析、构造函数优化、递归构建
TemplateRequestService3348AI prompt 构建、XML 解析、用例转换
CreateTestFileTask2461批量文件生成流程、错误处理、进度管理
MethodFactory1844PSI 方法解析、调用链构建、泛型替换
Type1807类型依赖解析、字段/方法/接口解析
FTManager1656模板文件管理、编码/解码、排序
MockitoMockBuilder1355Mock 代码生成、版本适配、匹配器选择

6.5 缓存策略 ​

  • TypeDictionary: 使用 hutool Cache 缓存类型解析结果
  • MethodFactory: methodIdCaches 缓存方法 ID -> Method 映射
  • TestTemplateContextBuilder: typeCache 和 typeDictionaryCache 双层缓存
  • TemplateRequestService: classModelRenders 缓存文件级 AI 请求结果

6.6 中文常量清单 ​

常量所在类用途
期望返回值是空，不进行断言AssertUtil断言跳过日志
未知的类型AssertUtil类型判断失败
上一个单元测试任务还未完成TemplateGenerator任务冲突警告
生成单测中...CreateTestFileTask进度提示
生成完成CreateTestFileTask完成提示
生成结束CreateTestFileTask结束提示
生成测试文件CreateTestFileTask文件生成提示
生成单测文件名称异常CreateTestFileTask异常提示
生成文件失败CreateTestFileTask失败提示
生成上下文异常CreateTestFileTask上下文异常
生成模板上下文信息失败CreateTestFileTask模板异常
模板地址为空CreateTestFileTask模板缺失
模板生成失败CreateTestFileTask生成失败
收集上下文CreateTestFileTask收集阶段
请求AI模型CreateTestFileTaskAI 请求阶段
解析完成CreateTestFileTask解析完成
手动取消PSI解析CreateTestFileTask用户取消
初始化完成CreateTestMethodTask初始化完成
单测实现失败CreateTestMethodTask实现失败
信息已清除TemplateRequestService缓存清除
处理异常信息已清除TemplateRequestService异常清除
读取模型数据异常TemplateRequestServiceAI 数据异常
读取用例Mock信息异常TemplateRequestServiceMock 数据异常
读取用例输入异常TemplateRequestService输入数据异常
补充用例分支信息异常TemplateRequestService分支数据异常
转换用例数据失败TemplateRequestService数据转换失败
转换异常TemplateRequestService转换异常
转换时间异常TemplateRequestService时间转换异常
断言生成异常AssertUtil断言异常
注释异常方法失败CreateTestMethodTask注释失败
更新测试目录异常TargetDirectoryLocator目录异常
清理缓存报错TemplateRequestService缓存清理错误
当前加载groovy plugin插件失败CodeRefactorUtilGroovy 加载失败
import文件%d信息CodeRefactorUtilimport 信息
请填写需要使用的测试类名GeneratedClassNameResolver类名输入
使用其他名称ClassNameSelection用户选择
&取消 / &打开已有文件ClassNameSelection按钮文本

本项目仅供学习研究，逆向分析内容归原厂商所有。

Copyright © 2026 Vibe Coding Labs