valitydev/openapi-generator
14
0
Fork 0
mirror of https://github.com/valitydev/openapi-generator.git synced 2024-11-08 19:33:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added inline resolver

Browse Source
This commit is contained in:
Tony Tam 2015-10-16 18:21:06 -07:00
parent a5a6ae72c3
commit 9508579eed
2 changed files with 280 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

183
modules/swagger-codegen/src/main/java/io/swagger/codegen/InlineModelResolver.java Normal file
View File

@ -0,0 +1,183 @@
package io.swagger.codegen;
import io.swagger.models.*;
import io.swagger.models.parameters.BodyParameter;
import io.swagger.models.parameters.Parameter;
import io.swagger.models.properties.ObjectProperty;
import io.swagger.models.properties.Property;
import io.swagger.models.properties.RefProperty;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class InlineModelResolver {
private Swagger swagger = null;
Map<String, Model> addedModels = new HashMap<String, Model>();
public void flatten(Swagger swagger) {
this.swagger = swagger;
// operations
Map<String, Path> paths = swagger.getPaths();
Map<String, Model> models = swagger.getDefinitions();
if(paths != null) {
for(String pathname : paths.keySet()) {
Path path = paths.get(pathname);
for(Operation operation: path.getOperations()) {
List<Parameter> parameters = operation.getParameters();
if(parameters != null) {
for(Parameter parameter : parameters) {
if(parameter instanceof BodyParameter) {
BodyParameter bp = (BodyParameter) parameter;
if(bp.getSchema() != null) {
Model model = bp.getSchema();
if(model instanceof ModelImpl) {
String name = bp.getName();
if(models == null) {
models = new HashMap<String, Model>();
swagger.setDefinitions(models);
}
else {
if (swagger.getDefinitions().containsKey(bp.getName())) {
name += "_" + "inline";
}
}
swagger.addDefinition(name, model);
bp.setSchema(new RefModel(name));
}
}
}
}
}
Map<String, Response> responses = operation.getResponses();
if(responses != null) {
for(String key : responses.keySet()) {
Response response = responses.get(key);
if(response.getSchema() != null) {
Property property = response.getSchema();
if(property instanceof ObjectProperty) {
String modelName = uniqueName("inline_response_" + key);
ObjectProperty op = (ObjectProperty) property;
Model model = modelFromProperty(op, modelName);
response.setSchema(new RefProperty(modelName));
swagger.addDefinition(modelName, model);
}
}
}
}
}
}
}
// definitions
if(models != null) {
List<String> modelNames = new ArrayList<String>(models.keySet());
for(String modelName : modelNames) {
Model model = models.get(modelName);
if(model instanceof ModelImpl) {
ModelImpl m = (ModelImpl) model;
Map<String, Property> properties = m.getProperties();
flattenProperties(properties, modelName);
}
else if (model instanceof ArrayModel) {
ArrayModel m = (ArrayModel) model;
Property inner = m.getItems();
if(inner instanceof ObjectProperty) {
String innerModelName = uniqueName(modelName + "_" + inner);
Model innerModel = modelFromProperty((ObjectProperty)inner, modelName);
swagger.addDefinition(innerModelName, innerModel);
m.setItems(new RefProperty(innerModelName));
}
}
else if (model instanceof ComposedModel) {
ComposedModel m = (ComposedModel) model;
}
}
}
}
public String uniqueName(String key) {
int count = 0;
boolean done = false;
while(!done) {
String name = key;
if(count > 0) {
name = key + "_" + count;
}
if(swagger.getDefinitions() == null) {
return name;
}
else if (!swagger.getDefinitions().containsKey(name)) {
return name;
}
count += 1;
}
return key;
}
public void flattenProperties(Map<String, Property> properties, String path) {
if(properties == null) {
return;
}
Map<String, Property> propsToUpdate = new HashMap<String, Property>();
Map<String, Model> modelsToAdd = new HashMap<String, Model>();
for(String key : properties.keySet()) {
Property property = properties.get(key);
if(property instanceof ObjectProperty) {
String modelName = uniqueName(path + "_" + key);
ObjectProperty op = (ObjectProperty) property;
Model model = modelFromProperty(op, modelName);
modelsToAdd.put(modelName, model);
propsToUpdate.put(key, new RefProperty(modelName));
}
}
if(propsToUpdate.size() > 0) {
for(String key : propsToUpdate.keySet()) {
properties.put(key, propsToUpdate.get(key));
}
}
for(String key : modelsToAdd.keySet()) {
swagger.addDefinition(key, modelsToAdd.get(key));
this.addedModels.put(key, modelsToAdd.get(key));
}
}
public Model modelFromProperty(ObjectProperty object, String path) {
String access = object.getAccess();
String description = object.getDescription();
String example = object.getExample();
String name = object.getName();
Integer position = object.getPosition();
Boolean readOnly = object.getReadOnly();
Boolean required = object.getRequired();
String title = object.getTitle();
Map<String, Object> extensions = object.getVendorExtensions();
Xml xml = object.getXml();
Map<String, Property> properties = object.getProperties();
ModelImpl model = new ModelImpl();
model.setDescription(description);
model.setExample(example);
model.setName(name);
model.setXml(xml);
if(properties != null) {
flattenProperties(properties, path);
model.setProperties(properties);
}
return model;
}
}

97
modules/swagger-codegen/src/test/java/io/swagger/codegen/InlineModelResolverTest.java Normal file
View File

@ -0,0 +1,97 @@
package io.swagger.codegen;
import io.swagger.models.*;
import io.swagger.models.parameters.BodyParameter;
import io.swagger.models.parameters.Parameter;
import io.swagger.models.properties.ObjectProperty;
import io.swagger.models.properties.StringProperty;
import io.swagger.util.Json;
import org.junit.Test;
public class InlineModelResolverTest {
@Test
public void resolveInlineModelTest() throws Exception {
Swagger swagger = new Swagger();
swagger.addDefinition("User", new ModelImpl()
.name("user")
.description("a common user")
.property("name", new StringProperty())
.property("address", new ObjectProperty()
.title("title")
._default("default")
.access("access")
.readOnly(false)
.required(true)
.description("description")
.name("name")
.property("street", new StringProperty())
.property("city", new StringProperty())));
new InlineModelResolver().flatten(swagger);
Json.prettyPrint(swagger);
}
@Test
public void testInlineResponseModel() throws Exception {
Swagger swagger = new Swagger();
swagger.path("/foo/bar", new Path()
.get(new Operation()
.response(200, new Response()
.description("it works!")
.schema(new ObjectProperty()
.property("name", new StringProperty())))))
.path("/foo/baz", new Path()
.get(new Operation()
.response(200, new Response()
.vendorExtension("x-foo", "bar")
.description("it works!")
.schema(new ObjectProperty()
.vendorExtension("x-baz", "boo")
.property("name", new StringProperty()
.vendorExtension("x-bars", "bleh"))))));
new InlineModelResolver().flatten(swagger);
Json.prettyPrint(swagger);
}
@Test
public void resolveInlineArrayModel() throws Exception {
Swagger swagger = new Swagger();
swagger.addDefinition("User", new ArrayModel()
.items(new ObjectProperty()
.title("title")
._default("default")
.access("access")
.readOnly(false)
.required(true)
.description("description")
.name("name")
.property("street", new StringProperty())
.property("city", new StringProperty())));
new InlineModelResolver().flatten(swagger);
Json.prettyPrint(swagger);
}
@Test
public void resolveInlineBodyParameter() throws Exception {
Swagger swagger = new Swagger();
swagger.path("/hello", new Path()
.get(new Operation()
.parameter(new BodyParameter()
.name("body")
.schema(new ModelImpl()
.property("name", new StringProperty())))));
new InlineModelResolver().flatten(swagger);
Json.prettyPrint(swagger);
}
}