1 /*
2  *
3  *  * Copyright 2019-2020 the original author or authors.
4  *  *
5  *  * Licensed under the Apache License, Version 2.0 (the "License");
6  *  * you may not use this file except in compliance with the License.
7  *  * You may obtain a copy of the License at
8  *  *
9  *  *      https://www.apache.org/licenses/LICENSE-2.0
10  *  *
11  *  * Unless required by applicable law or agreed to in writing, software
12  *  * distributed under the License is distributed on an "AS IS" BASIS,
13  *  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  *  * See the License for the specific language governing permissions and
15  *  * limitations under the License.
16  *
17  */
18
19 package org.springdoc.core.converters;
20
21 import java.util.Iterator;
22 import java.util.List;
23 import java.util.Optional;
24
25 import io.swagger.v3.core.converter.AnnotatedType;
26 import io.swagger.v3.core.converter.ModelConverter;
27 import io.swagger.v3.core.converter.ModelConverterContext;
28 import io.swagger.v3.oas.models.media.Schema;
29 import org.springdoc.core.customizers.PropertyCustomizer;
30
31 public class PropertyCustomizingConverter implements ModelConverter {
32
33     private final Optional<List<PropertyCustomizer>> propertyCustomizers;
34
35     public PropertyCustomizingConverter(Optional<List<PropertyCustomizer>> customizers) {
36         this.propertyCustomizers = customizers;
37     }
38
39     @Override
40     public Schema resolve(AnnotatedType type, ModelConverterContext context, Iterator<ModelConverter> chain) {
41         if (chain.hasNext()) {
42             Schema<?> resolvedSchema = chain.next().resolve(type, context, chain);
43             if (type.isSchemaProperty() && propertyCustomizers.isPresent()) {
44                 List<PropertyCustomizer> propertyCustomizerList = propertyCustomizers.get();
45                 for (PropertyCustomizer propertyCustomizer : propertyCustomizerList)
46                     resolvedSchema = propertyCustomizer.customize(resolvedSchema, type);
47             }
48             return resolvedSchema;
49         }
50         else {
51             return null;
52         }
53     }
54 }
55