import yaml
from openapi_spec_validator import validate_spec

def extract_endpoints(input_file, output_file, endpoint_paths):
    with open(input_file, 'r') as f:
        spec = yaml.safe_load(f)
    
    # 验证原始规范
    validate_spec(spec)
    
    # 创建新的规范基础结构
    new_spec = {
        'openapi': spec['openapi'],
        'info': spec['info'],
        'paths': {},
        'components': spec.get('components', {})
    }
    
    # 添加选定的路径
    for path in endpoint_paths:
        if path in spec['paths']:
            new_spec['paths'][path] = spec['paths'][path]
    
    # 写入新文件
    with open(output_file, 'w') as f:
        yaml.dump(new_spec, f, sort_keys=False)
    
    print(f"成功提取端点到 {output_file}")

# 使用示例
extract_endpoints(
    input_file='full_api.yaml',
    output_file='partial_api.yaml',
    endpoint_paths=['/users', '/users/{id}']
)

const fs = require('fs');
const YAML = require('yaml');
const SwaggerParser = require('swagger-parser');

async function extractEndpoints(inputFile, outputFile, endpointPaths) {
    try {
        // 解析原始API文档
        const api = await SwaggerParser.validate(inputFile);
        
        // 创建新的API文档结构
        const newApi = {
            openapi: api.openapi,
            info: api.info,
            paths: {},
            components: api.components || {}
        };
        
        // 添加选定的路径
        endpointPaths.forEach(path => {
            if (api.paths[path]) {
                newApi.paths[path] = api.paths[path];
            }
        });
        
        // 写入新文件
        const yamlStr = YAML.stringify(newApi);
        fs.writeFileSync(outputFile, yamlStr);
        
        console.log(`成功提取端点到 ${outputFile}`);
    } catch (err) {
        console.error('处理过程中出错:', err);
    }
}

// 使用示例
extractEndpoints(
    'full_api.yaml',
    'partial_api.yaml',
    ['/products', '/products/{id}']
);