io.swagger.v3.parser.ResolverCache Maven / Gradle / Ivy
package io.swagger.v3.parser;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import io.swagger.v3.oas.models.OpenAPI;
import io.swagger.v3.oas.models.media.Schema;
import io.swagger.v3.oas.models.responses.ApiResponse;
import io.swagger.v3.parser.core.models.AuthorizationValue;
import io.swagger.v3.parser.models.RefFormat;
import io.swagger.v3.parser.models.RefType;
import io.swagger.v3.parser.util.DeserializationUtils;
import io.swagger.v3.parser.util.PathUtils;
import io.swagger.v3.parser.util.RefUtils;
import io.swagger.v3.parser.util.OpenAPIDeserializer;
import org.apache.commons.lang3.StringUtils;
import java.io.File;
import java.nio.file.Path;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
* A class that caches values that have been loaded so we don't have to repeat
* expensive operations like:
* 1) reading a remote URL with authorization (e.g. using RemoteURL.java)
* 2) reading the contents of a file into memory
* 3) extracting a sub object from a json/yaml tree
* 4) de-serializing json strings into objects
*/
public class ResolverCache {
private static final Pattern SCHEMAS_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "schemas/(?.+)");
private static final Pattern RESPONSES_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "responses/(?.+)");
private static final Pattern PARAMETERS_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "parameters/(?.+)");
private static final Pattern REQUEST_BODIES_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "requestBodies/(?.+)");
private static final Pattern EXAMPLES_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "examples/(?.+)");
private static final Pattern LINKS_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "links/(?.+)");
private static final Pattern CALLBACKS_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "callbacks/(?.+)");
private static final Pattern HEADERS_PATTERN = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "headers/(?.+)");
private static final Pattern SECURITY_SCHEMES = Pattern.compile("^" + RefType.COMPONENTS.getInternalPrefix() + "securitySchemes/(?.+)");
private static final Pattern PATHS_PATTERN = Pattern.compile("^" + RefType.PATH.getInternalPrefix() + "(?.+)");
private final OpenAPI openApi;
private final List auths;
private final Path parentDirectory;
private final String rootPath;
private Map resolutionCache = new HashMap<>();
private Map externalFileCache = new HashMap<>();
private Set referencedModelKeys = new HashSet<>();
/*
a map that stores original external references, and their associated renamed references
*/
private Map renameCache = new HashMap<>();
public ResolverCache(OpenAPI openApi, List auths, String parentFileLocation) {
this.openApi = openApi;
this.auths = auths;
this.rootPath = parentFileLocation;
if(parentFileLocation != null) {
if(parentFileLocation.startsWith("http")) {
parentDirectory = null;
} else {
parentDirectory = PathUtils.getParentDirectoryOfFile(parentFileLocation);
}
} else {
File file = new File(".");
parentDirectory = file.toPath();
}
}
public T loadRef(String ref, RefFormat refFormat, Class expectedType) {
if (refFormat == RefFormat.INTERNAL) {
//we don't need to go get anything for internal refs
Object loadedRef = loadInternalRef(ref);
try{
return expectedType.cast(loadedRef);
}
catch (Exception e) {
return null;
}
}
final String[] refParts = ref.split("#/");
if (refParts.length > 2) {
throw new RuntimeException("Invalid ref format: " + ref);
}
final String file = refParts[0];
final String definitionPath = refParts.length == 2 ? refParts[1] : null;
//we might have already resolved this ref, so check the resolutionCache
Object previouslyResolvedEntity = resolutionCache.get(ref);
if (previouslyResolvedEntity != null) {
return expectedType.cast(previouslyResolvedEntity);
}
//we have not resolved this particular ref
//but we may have already loaded the file or url in question
String contents = externalFileCache.get(file);
if (contents == null) {
if(parentDirectory != null) {
contents = RefUtils.readExternalRef(file, refFormat, auths, parentDirectory);
}
else if(rootPath != null) {
contents = RefUtils.readExternalUrlRef(file, refFormat, auths, rootPath);
}
externalFileCache.put(file, contents);
}
if (definitionPath == null) {
T result = DeserializationUtils.deserialize(contents, file, expectedType);
resolutionCache.put(ref, result);
return result;
}
//a definition path is defined, meaning we need to "dig down" through the JSON tree and get the desired entity
JsonNode tree = DeserializationUtils.deserializeIntoTree(contents, file);
String[] jsonPathElements = definitionPath.split("/");
for (String jsonPathElement : jsonPathElements) {
tree = tree.get(unescapePointer(jsonPathElement));
//if at any point we do find an element we expect, print and error and abort
if (tree == null) {
throw new RuntimeException("Could not find " + definitionPath + " in contents of " + file);
}
}
T result;
if (expectedType.equals(Schema.class)) {
OpenAPIDeserializer deserializer = new OpenAPIDeserializer();
result = (T) deserializer.getSchema((ObjectNode) tree, definitionPath.replace("/", "."), null);
} else {
result = DeserializationUtils.deserialize(tree, file, expectedType);
}
updateLocalRefs(file, result);
resolutionCache.put(ref, result);
return result;
}
protected void updateLocalRefs(String file, T result) {
if(result instanceof ApiResponse) {
ApiResponse response = (ApiResponse) result;
for(String mediaType : response.getContent().keySet()) {
updateLocalRefs(file, response.getContent().get(mediaType).getSchema());
}
}
if(result instanceof Schema && ((Schema)(result)).get$ref() != null) {
Schema prop = (Schema) result;
updateLocalRefs(file, prop);
}
else if(result instanceof Schema) {
Schema model = (Schema) result;
updateLocalRefs(file, model);
}
}
protected void updateLocalRefs(String file, Schema schema) {
if(schema.get$ref() != null) {
String updatedLocation = merge(file, schema.get$ref());
schema.set$ref(updatedLocation);
}
else if(schema.getProperties() != null) {
Map properties = schema.getProperties();
for(Schema property : properties.values()) {
updateLocalRefs(file, property);
}
}
}
protected String merge(String host, String ref) {
if(StringUtils.isBlank(host)) {
return ref;
}
if(ref.startsWith("http:") || ref.startsWith("https:")) {
// already an absolute ref
return ref;
}
if(!host.startsWith("http:") && !host.startsWith("https:")) {
return ref;
}
if(ref.startsWith(".")) {
// relative ref, leave alone
return ref;
}
if(host.endsWith("/") && ref.startsWith("/")) {
return host + ref.substring(1);
}
return host + ref;
}
private Object loadInternalRef(String ref) {
Object result = null;
if (ref.startsWith("#/paths")) {
result = getFromMap(ref, openApi.getPaths(), PATHS_PATTERN);
} else if (openApi.getComponents() != null){
if(ref.startsWith("#/components/schemas")) {
result = getFromMap(ref, openApi.getComponents().getSchemas(), SCHEMAS_PATTERN);
}
else if(ref.startsWith("#/components/requestBodies")) {
result = getFromMap(ref, openApi.getComponents().getRequestBodies(), REQUEST_BODIES_PATTERN);
}
else if(ref.startsWith("#/components/examples")) {
result = getFromMap(ref, openApi.getComponents().getExamples(), EXAMPLES_PATTERN);
}
else if(ref.startsWith("#/components/responses")) {
result = getFromMap(ref, openApi.getComponents().getResponses(), RESPONSES_PATTERN);
}
else if(ref.startsWith("#/components/parameters")) {
result = getFromMap(ref, openApi.getComponents().getParameters(), PARAMETERS_PATTERN);
}
else if(ref.startsWith("#/components/links")) {
result = getFromMap(ref, openApi.getComponents().getLinks(), LINKS_PATTERN);
}
else if(ref.startsWith("#/components/headers")) {
result = getFromMap(ref, openApi.getComponents().getHeaders(), HEADERS_PATTERN);
}
else if(ref.startsWith("#/components/callbacks")) {
result = getFromMap(ref, openApi.getComponents().getCallbacks(), CALLBACKS_PATTERN);
}
else if(ref.startsWith("#/components/securitySchemes")) {
result = getFromMap(ref, openApi.getComponents().getSecuritySchemes(), SECURITY_SCHEMES);
}
}
return result;
}
private String unescapePointer(String jsonPathElement) {
// Unescape the JSON Pointer segment using the algorithm described in RFC 6901, section 4:
// https://tools.ietf.org/html/rfc6901#section-4
// First transform any occurrence of the sequence '~1' to '/'
jsonPathElement = jsonPathElement.replaceAll("~1", "/");
// Then transforming any occurrence of the sequence '~0' to '~'.
return jsonPathElement.replaceAll("~0", "~");
}
private Object getFromMap(String ref, Map map, Pattern pattern) {
final Matcher parameterMatcher = pattern.matcher(ref);
if (parameterMatcher.matches()) {
final String paramName = unescapePointer(parameterMatcher.group("name"));
if (map != null) {
return map.get(paramName);
}
}
return null;
}
public boolean hasReferencedKey(String modelKey) {
if(referencedModelKeys == null) {
return false;
}
return referencedModelKeys.contains(modelKey);
}
public void addReferencedKey(String modelKey) {
referencedModelKeys.add(modelKey);
}
public String getRenamedRef(String originalRef) {
return renameCache.get(originalRef);
}
public void putRenamedRef(String originalRef, String newRef) {
renameCache.put(originalRef, newRef);
}
public Map getResolutionCache() {
return Collections.unmodifiableMap(resolutionCache);
}
public Map getExternalFileCache() {
return Collections.unmodifiableMap(externalFileCache);
}
public Map getRenameCache() {
return Collections.unmodifiableMap(renameCache);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy