|
4 | 4 |
|
5 | 5 | namespace Mthole\OpenApiMerge; |
6 | 6 |
|
| 7 | +use cebe\openapi\spec\Components; |
7 | 8 | use Mthole\OpenApiMerge\FileHandling\File; |
8 | 9 | use Mthole\OpenApiMerge\FileHandling\SpecificationFile; |
9 | 10 | use Mthole\OpenApiMerge\Merge\PathMergerInterface; |
| 11 | +use Mthole\OpenApiMerge\Merge\ReferenceNormalizer; |
10 | 12 | use Mthole\OpenApiMerge\Reader\FileReader; |
11 | 13 |
|
| 14 | +use function array_merge; |
| 15 | +use function array_push; |
| 16 | +use function count; |
| 17 | + |
12 | 18 | class OpenApiMerge implements OpenApiMergeInterface |
13 | 19 | { |
14 | 20 | private FileReader $openApiReader; |
15 | 21 |
|
16 | 22 | private PathMergerInterface $pathMerger; |
| 23 | + private ReferenceNormalizer $referenceNormalizer; |
17 | 24 |
|
18 | 25 | public function __construct( |
19 | 26 | FileReader $openApiReader, |
20 | | - PathMergerInterface $pathMerger |
| 27 | + PathMergerInterface $pathMerger, |
| 28 | + ReferenceNormalizer $referenceResolver |
21 | 29 | ) { |
22 | | - $this->openApiReader = $openApiReader; |
23 | | - $this->pathMerger = $pathMerger; |
| 30 | + $this->openApiReader = $openApiReader; |
| 31 | + $this->pathMerger = $pathMerger; |
| 32 | + $this->referenceNormalizer = $referenceResolver; |
24 | 33 | } |
25 | 34 |
|
26 | | - public function mergeFiles(File $baseFile, File ...$additionalFiles): SpecificationFile |
| 35 | + /** @param array<int, File> $additionalFiles */ |
| 36 | + public function mergeFiles(File $baseFile, array $additionalFiles, bool $resolveReference = true): SpecificationFile |
27 | 37 | { |
28 | | - $mergedOpenApiDefinition = $this->openApiReader->readFile($baseFile)->getOpenApi(); |
| 38 | + $mergedOpenApiDefinition = $this->openApiReader->readFile($baseFile, $resolveReference)->getOpenApi(); |
| 39 | + |
| 40 | + // use "for" instead of "foreach" to iterate over new added files |
| 41 | + for ($i = 0; $i < count($additionalFiles); $i++) { |
| 42 | + $additionalFile = $additionalFiles[$i]; |
| 43 | + $additionalDefinition = $this->openApiReader->readFile($additionalFile, $resolveReference)->getOpenApi(); |
| 44 | + if (! $resolveReference) { |
| 45 | + $resolvedReferenceResult = $this->referenceNormalizer->normalizeInlineReferences( |
| 46 | + $additionalFile, |
| 47 | + $additionalDefinition |
| 48 | + ); |
| 49 | + array_push($additionalFiles, ...$resolvedReferenceResult->getFoundReferenceFiles()); |
| 50 | + $additionalDefinition = $resolvedReferenceResult->getNormalizedDefinition(); |
| 51 | + } |
29 | 52 |
|
30 | | - foreach ($additionalFiles as $additionalFile) { |
31 | | - $additionalDefinition = $this->openApiReader->readFile($additionalFile)->getOpenApi(); |
32 | 53 | $mergedOpenApiDefinition->paths = $this->pathMerger->mergePaths( |
33 | 54 | $mergedOpenApiDefinition->paths, |
34 | 55 | $additionalDefinition->paths |
35 | 56 | ); |
| 57 | + |
| 58 | + if ($additionalDefinition->components === null) { |
| 59 | + continue; |
| 60 | + } |
| 61 | + |
| 62 | + if ($mergedOpenApiDefinition->components === null) { |
| 63 | + $mergedOpenApiDefinition->components = new Components([]); |
| 64 | + } |
| 65 | + |
| 66 | + $mergedOpenApiDefinition->components->schemas = array_merge( |
| 67 | + $mergedOpenApiDefinition->components->schemas ?? [], |
| 68 | + $additionalDefinition->components->schemas ?? [] |
| 69 | + ); |
36 | 70 | } |
37 | 71 |
|
38 | | - if ($mergedOpenApiDefinition->components !== null) { |
| 72 | + if ($resolveReference && $mergedOpenApiDefinition->components !== null) { |
39 | 73 | $mergedOpenApiDefinition->components->schemas = []; |
40 | 74 | } |
41 | 75 |
|
|
0 commit comments