organizeOpenApi function

Map<String, dynamic> organizeOpenApi(Map<String, dynamic> document)

Implementation

Map<String, dynamic> organizeOpenApi(Map<String, dynamic> document) {
  final schemas = <String, dynamic>{};

  Map<String, dynamic> resolveSchema(Map<String, dynamic> schema) {
    final properties = schema['properties'] as Map<String, dynamic>? ?? const {};
    final items = schema['items'] as Map<String, dynamic>?;
    schema = {
      ...schema,
      if (properties.isNotEmpty)
        'properties': properties.map((key, value) {
          return MapEntry(key, resolveSchema(value as Map<String, dynamic>));
        }),
      if (items != null) 'items': resolveSchema(items),
    };

    final title = schema['title'] as String?;
    if (title != null) {
      schemas[title] = schema;
      return {r'$ref': '#/components/schemas/$title'};
    }
    return schema;
  }

  Map<String, dynamic> resolveMedia(Map<String, dynamic> data) {
    return data.map((key, content) {
      if (key == 'content') {
        content as Map<String, dynamic>?;
        return MapEntry(
          key,
          content?.map((type, media) {
            media as Map<String, dynamic>?;
            return MapEntry(
              type,
              media?.map((key, value) {
                if (key == 'schema') {
                  return MapEntry(key, resolveSchema(value as Map<String, dynamic>));
                }
                return MapEntry(key, value);
              }),
            );
          }),
        );
      }
      return MapEntry(key, content);
    });
  }

  return {
    ...document,
    'paths': (document['paths'] as Map<String, dynamic>).map((path, endPoint) {
      final newEndpoint = (endPoint as Map<String, dynamic>).map((operationId, operation) {
        operation as Map<String, dynamic>;
        final requestBody = operation['requestBody'] as Map<String, dynamic>?;
        final responses = operation['responses'] as Map<String, dynamic>?;

        return MapEntry(operationId, {
          ...operation,
          if (requestBody != null) 'requestBody': resolveMedia(requestBody),
          if (responses != null)
            'responses': responses.map((code, media) {
              return MapEntry(code, resolveMedia(media));
            }),
        });
      });
      return MapEntry(path, newEndpoint);
    }),
    if (schemas.isNotEmpty) 'components': {'schemas': schemas},
  };
}