如何将千余Mock JSON Fixture文件批量转为.mock()方法？

面对1000+个Fixture JSON文件，手动转换完全不现实，基于__typename的映射关系可以实现自动化处理，下面是具体可行的方案：

核心逻辑 #

每个JSON文件里的__typename字段直接对应Swift中的结构体/类名，我们可以通过批量读取JSON文件提取类型名和字段值，再结合Swift结构体的属性定义，自动生成对应的mockXXX()静态方法。

具体实现步骤 #

1. 批量读取并解析JSON文件 #

用脚本（Python/Swift均可）遍历项目中所有Fixture JSON文件，提取每个文件的__typename和所有键值对。例如Python核心实现代码：

import os
import json

fixture_dir = "/path/to/your/fixtures"
mock_data = {}

for filename in os.listdir(fixture_dir):
    if filename.endswith(".json"):
        with open(os.path.join(fixture_dir, filename), "r") as f:
            data = json.load(f)
            typename = data.pop("__typename")
            mock_data[typename] = data

2. 获取Swift结构体的属性信息 #

要生成准确的mock方法，需要对应结构体的属性类型和可选性。可以用SwiftSyntax解析项目中的Swift结构体代码，自动提取每个类型的属性列表、类型、是否可选。

示例Swift脚本片段（依赖SwiftSyntax）：

import SwiftSyntax

func parseStructProperties(from fileUrl: URL) -> [String: (type: String, isOptional: Bool)] {
    let sourceFile = try! SyntaxParser.parse(fileUrl)
    var properties = [String: (type: String, isOptional: Bool)]
    
    for node in sourceFile.children {
        if let structDecl = node.as(StructDeclSyntax.self) {
            for member in structDecl.memberBlock.members {
                if let varDecl = member.decl.as(VariableDeclSyntax.self), 
                   let binding = varDecl.bindings.first,
                   let identifier = binding.pattern.as(IdentifierPatternSyntax.self)?.identifier.text {
                    let typeAnnotation = binding.typeAnnotation?.type.description.trimmingCharacters(in: .whitespaces)
                    let isOptional = typeAnnotation?.hasSuffix("?") ?? false
                    let type = isOptional ? String(typeAnnotation!.dropLast()) : typeAnnotation!
                    properties[identifier] = (type, isOptional)
                }
            }
        }
    }
    return properties
}

3. 自动生成Mock方法代码 #

结合第一步的JSON字段值和第二步的结构体属性信息，生成对应的Swift静态方法代码。示例Python实现：

# 承接第一步的mock_data，遍历生成代码
output_code = ""

for typename, fields in mock_data.items():
    method_name = f"mock{typename}"
    output_code += f"extension {typename} {{\n"
    output_code += f"    static func {method_name}() -> {typename}? {{\n"
    output_code += "        .init(\n"
    
    for field_name, value in fields.items():
        # 处理值的类型转换，字符串加引号，数字直接保留，布尔值转Swift语法
        if isinstance(value, str):
            formatted_value = f"\"{value}\""
        elif isinstance(value, (int, float)):
            formatted_value = str(value)
        elif isinstance(value, bool):
            formatted_value = "true" if value else "false"
        else:
            formatted_value = "nil"  # 复杂类型可按需扩展处理逻辑
        
        output_code += f"            {field_name}: {formatted_value},\n"
    
    output_code += "        )\n"
    output_code += "    }\n"
    output_code += "}\n\n"

# 将生成的代码写入目标文件
with open("/path/to/your/MockExtensions.swift", "w") as f:
    f.write(output_code)

4. 验证与调整 #

生成代码后需要做少量验证：

• 检查嵌套结构体、数组等复杂类型的处理是否正确，按需扩展脚本的类型转换逻辑
• 确认可选属性的赋值符合结构体定义
• 处理JSON中不存在但结构体要求的必填字段，可在脚本中添加默认值填充逻辑

额外优化点 #

• 如果项目使用Codegen工具（比如阿波罗GraphQL的代码生成），可以直接在代码生成阶段加入mock方法的生成逻辑，一步到位
• 用Swift Package封装转换工具，方便团队复用

内容的提问来源于stack exchange，提问作者sarthak taneja