Skip to content

Commit

Permalink
format-only
Browse files Browse the repository at this point in the history
  • Loading branch information
@mweirauch
mweirauch committed Aug 3, 2017
1 parent 5c45409 commit e6e0618
Show file tree
Hide file tree
Showing 2 changed files with 225 additions and 211 deletions.
364 changes: 184 additions & 180 deletions src/main/java/com/mercateo/common/rest/schemagen/link/LinkCreator.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -35,185 +35,189 @@
import com.mercateo.common.rest.schemagen.link.relation.Relation;

public class LinkCreator {
public static final String TARGET_SCHEMA_PARAM_KEY = "targetSchema";

public static final String SCHEMA_PARAM_KEY = "schema";

public static final String METHOD_PARAM_KEY = "method";

private final JsonSchemaGenerator jsonSchemaGenerator;

private final LinkFactoryContext linkFactoryContext;

/**
* @param jsonSchemaGenerator
* @param linkFactoryContext
*
*/
LinkCreator(JsonSchemaGenerator jsonSchemaGenerator, LinkFactoryContext linkFactoryContext) {
this.jsonSchemaGenerator = requireNonNull(jsonSchemaGenerator);
this.linkFactoryContext = linkFactoryContext;
}

public static Builder setRelation(Relation relation, URI uri) {
requireNonNull(relation);
requireNonNull(uri);
Builder builder = Link.fromUri(uri).rel(relation.getName());
if (requireNonNull(relation).getType().isShouldBeSerialized()) {
builder.param("relType", relation.getType().getName());
builder.param("target", relation.getType().getSerializedName());
}
return builder;
}

/**
* create a link for a resource method
*
* @param scopes
* list of Scope objects for every scope level
* @param relation
* relation of method
* @return link with schema if applicable
*/
public Link createFor(List<Scope> scopes, Relation relation) {
return createFor(scopes, relation, requireNonNull(linkFactoryContext));
}

/**
* create a link for a resource method
*
* @param scopes
* list of Scope objects for every scope level
* @param relation
* relation of method
* @param linkFactoryContext
* the base URI for resolution of relative URIs and method and
* property checkers
* @return link with schema if applicable
*/
public Link createFor(List<Scope> scopes, Relation relation, LinkFactoryContext linkFactoryContext) {
final Class<?> resourceClass = scopes.get(0).getInvokedClass();
UriBuilder uriBuilder = UriBuilder.fromResource(resourceClass);

Map<String, Object> pathParameters = new HashMap<>();
for (Scope scope : scopes) {
final Method method = scope.getInvokedMethod();
final Object[] parameters = scope.getParams();
if (method.isAnnotationPresent(Path.class)) {
uriBuilder.path(method.getDeclaringClass(), method.getName());
}
pathParameters.putAll(collectPathParameters(scope, parameters));
setQueryParameters(uriBuilder, scope, parameters);
}

URI uri = mergeUri(linkFactoryContext.getBaseUri(), uriBuilder, pathParameters);

Builder builder = setRelation(relation, uri);

addLinkProperties(scopes, builder);

detectMediaType(scopes, builder);

final Scope lastScopedMethod = Iterables.getLast(scopes);
addHttpMethod(builder, lastScopedMethod);
addSchemaIfNeeded(builder, lastScopedMethod, linkFactoryContext);
return builder.build();
}

private URI mergeUri(URI baseUri, UriBuilder uriBuilder, Map<String, Object> pathParameters) {
URI uri = uriBuilder.buildFromMap(pathParameters);

if (baseUri != null) {
UriBuilder mergedUriBuilder = UriBuilder.fromUri(baseUri);
mergedUriBuilder.path(uri.getPath());
mergedUriBuilder.replaceQuery(uri.getQuery());
return mergedUriBuilder.buildFromMap(pathParameters);
}

return uri;
}

private void addLinkProperties(List<Scope> scopes, Builder builder) {
final LinkProperties properties = Iterables.getLast(scopes).getInvokedMethod()
.getAnnotation(LinkProperties.class);
if (properties != null) {
Stream.of(properties.value()).forEach(x -> builder.param(x.key(), x.value()));
}
}

private void detectMediaType(Collection<Scope> scopes, Builder builder) {
detectMediaType(Iterables.getLast(scopes).getInvokedMethod())
.ifPresent(mediatype -> builder.param("mediaType", mediatype));
}

private Optional<String> detectMediaType(Method method) {
return Optional.ofNullable(method.getAnnotation(Produces.class)).map(produces -> {
final String[] values = produces.value();
if (values.length > 0) {
return values[0];
}
return null;
});

}

private Map<String, Object> collectPathParameters(Scope scope, Object[] parameters) {
final Map<String, Object> pathParameters = new HashMap<>();
visitAnnotations((parameter, parameterIndex, annotation) -> {
if (annotation instanceof PathParam) {
PathParam pathParamAnnotation = (PathParam) annotation;
pathParameters.put(pathParamAnnotation.value(), parameter);
} else if (annotation instanceof BeanParam) {
BeanParamExtractor beanParamExtractor = new BeanParamExtractor();
pathParameters.putAll(beanParamExtractor.getPathParameters(parameter));
}
}, scope.getInvokedMethod(), parameters);

return pathParameters;
}

private void setQueryParameters(final UriBuilder uriBuilder, Scope scope, Object[] parameters) {
Type[] realParamTypes = GenericTypeReflector.getExactParameterTypes(scope.getInvokedMethod(),
scope.getInvokedClass());
visitAnnotations((parameter, parameterIndex, annotation) -> {
if (annotation instanceof QueryParam && parameter != null) {
uriBuilder.queryParam(((QueryParam) annotation).value(), parameter.toString());
} else if (annotation instanceof BeanParam && parameter != null) {
if (realParamTypes[parameterIndex] instanceof Class<?>) {
BeanParamExtractor beanParamExtractor = new BeanParamExtractor();
Map<String, Object[]> queryParameter = beanParamExtractor.getQueryParameters(parameter);
queryParameter.forEach((uriBuilder::queryParam));
}
}
}, scope.getInvokedMethod(), parameters);
}

private void addHttpMethod(Builder builder, Scope scope) {
final List<Class<? extends Annotation>> httpMethodAnnotations = Arrays.asList(GET.class, POST.class, PUT.class,
DELETE.class);
final Method invokedMethod = scope.getInvokedMethod();
final Optional<Class<? extends Annotation>> httpMethod = httpMethodAnnotations.stream()
.filter(invokedMethod::isAnnotationPresent).findFirst();

if (httpMethod.isPresent()) {
builder.param(METHOD_PARAM_KEY, httpMethod.get().getSimpleName());
} else {
throw new IllegalArgumentException("LinkCreator: The method has to be annotated with one of: "
+ String.join(", ", (Iterable<String>) httpMethodAnnotations.stream().map(Class::getSimpleName)
.map(m -> '@' + m)::iterator));
}
}

private void addSchemaIfNeeded(Builder builder, Scope method, LinkFactoryContext linkFactoryContext) {
Optional<String> optionalInputSchema = jsonSchemaGenerator.createInputSchema(method,
linkFactoryContext.getFieldCheckerForSchema());
optionalInputSchema.ifPresent(s -> builder.param(SCHEMA_PARAM_KEY, s));
Optional<String> mt = detectMediaType(method.getInvokedMethod());
if (mt.isPresent() && MediaType.APPLICATION_JSON.equals(mt.get())) {
Optional<String> optionalOutputSchema = jsonSchemaGenerator.createOutputSchema(method,
linkFactoryContext.getFieldCheckerForSchema());
optionalOutputSchema.ifPresent(s -> builder.param(TARGET_SCHEMA_PARAM_KEY, s));
}
}
public static final String TARGET_SCHEMA_PARAM_KEY = "targetSchema";

public static final String SCHEMA_PARAM_KEY = "schema";

public static final String METHOD_PARAM_KEY = "method";

private final JsonSchemaGenerator jsonSchemaGenerator;

private final LinkFactoryContext linkFactoryContext;

/**
* @param jsonSchemaGenerator
* @param linkFactoryContext
*
*/
LinkCreator(JsonSchemaGenerator jsonSchemaGenerator, LinkFactoryContext linkFactoryContext) {
this.jsonSchemaGenerator = requireNonNull(jsonSchemaGenerator);
this.linkFactoryContext = linkFactoryContext;
}

public static Builder setRelation(Relation relation, URI uri) {
requireNonNull(relation);
requireNonNull(uri);
Builder builder = Link.fromUri(uri).rel(relation.getName());
if (requireNonNull(relation).getType().isShouldBeSerialized()) {
builder.param("relType", relation.getType().getName());
builder.param("target", relation.getType().getSerializedName());
}
return builder;
}

/**
* create a link for a resource method
*
* @param scopes
* list of Scope objects for every scope level
* @param relation
* relation of method
* @return link with schema if applicable
*/
public Link createFor(List<Scope> scopes, Relation relation) {
return createFor(scopes, relation, requireNonNull(linkFactoryContext));
}

/**
* create a link for a resource method
*
* @param scopes
* list of Scope objects for every scope level
* @param relation
* relation of method
* @param linkFactoryContext
* the base URI for resolution of relative URIs and method and
* property checkers
* @return link with schema if applicable
*/
public Link createFor(List<Scope> scopes, Relation relation,
LinkFactoryContext linkFactoryContext) {
final Class<?> resourceClass = scopes.get(0).getInvokedClass();
UriBuilder uriBuilder = UriBuilder.fromResource(resourceClass);

Map<String, Object> pathParameters = new HashMap<>();
for (Scope scope : scopes) {
final Method method = scope.getInvokedMethod();
final Object[] parameters = scope.getParams();
if (method.isAnnotationPresent(Path.class)) {
uriBuilder.path(method.getDeclaringClass(), method.getName());
}
pathParameters.putAll(collectPathParameters(scope, parameters));
setQueryParameters(uriBuilder, scope, parameters);
}

URI uri = mergeUri(linkFactoryContext.getBaseUri(), uriBuilder, pathParameters);

Builder builder = setRelation(relation, uri);

addLinkProperties(scopes, builder);

detectMediaType(scopes, builder);

final Scope lastScopedMethod = Iterables.getLast(scopes);
addHttpMethod(builder, lastScopedMethod);
addSchemaIfNeeded(builder, lastScopedMethod, linkFactoryContext);
return builder.build();
}

private URI mergeUri(URI baseUri, UriBuilder uriBuilder, Map<String, Object> pathParameters) {
URI uri = uriBuilder.buildFromMap(pathParameters);

if (baseUri != null) {
UriBuilder mergedUriBuilder = UriBuilder.fromUri(baseUri);
mergedUriBuilder.path(uri.getPath());
mergedUriBuilder.replaceQuery(uri.getQuery());
return mergedUriBuilder.buildFromMap(pathParameters);
}

return uri;
}

private void addLinkProperties(List<Scope> scopes, Builder builder) {
final LinkProperties properties = Iterables.getLast(scopes).getInvokedMethod()
.getAnnotation(LinkProperties.class);
if (properties != null) {
Stream.of(properties.value()).forEach(x -> builder.param(x.key(), x.value()));
}
}

private void detectMediaType(Collection<Scope> scopes, Builder builder) {
detectMediaType(Iterables.getLast(scopes).getInvokedMethod()).ifPresent(mediatype -> builder
.param("mediaType", mediatype));
}

private Optional<String> detectMediaType(Method method) {
return Optional.ofNullable(method.getAnnotation(Produces.class)).map(produces -> {
final String[] values = produces.value();
if (values.length > 0) {
return values[0];
}
return null;
});

}

private Map<String, Object> collectPathParameters(Scope scope, Object[] parameters) {
final Map<String, Object> pathParameters = new HashMap<>();
visitAnnotations((parameter, parameterIndex, annotation) -> {
if (annotation instanceof PathParam) {
PathParam pathParamAnnotation = (PathParam) annotation;
pathParameters.put(pathParamAnnotation.value(), parameter);
} else if (annotation instanceof BeanParam) {
BeanParamExtractor beanParamExtractor = new BeanParamExtractor();
pathParameters.putAll(beanParamExtractor.getPathParameters(parameter));
}
}, scope.getInvokedMethod(), parameters);

return pathParameters;
}

private void setQueryParameters(final UriBuilder uriBuilder, Scope scope, Object[] parameters) {
Type[] realParamTypes = GenericTypeReflector.getExactParameterTypes(scope
.getInvokedMethod(), scope.getInvokedClass());
visitAnnotations((parameter, parameterIndex, annotation) -> {
if (annotation instanceof QueryParam && parameter != null) {
uriBuilder.queryParam(((QueryParam) annotation).value(), parameter.toString());
} else if (annotation instanceof BeanParam && parameter != null) {
if (realParamTypes[parameterIndex] instanceof Class<?>) {
BeanParamExtractor beanParamExtractor = new BeanParamExtractor();
Map<String, Object[]> queryParameter = beanParamExtractor.getQueryParameters(
parameter);
queryParameter.forEach((uriBuilder::queryParam));
}
}
}, scope.getInvokedMethod(), parameters);
}

private void addHttpMethod(Builder builder, Scope scope) {
final List<Class<? extends Annotation>> httpMethodAnnotations = Arrays.asList(GET.class,
POST.class, PUT.class, DELETE.class);
final Method invokedMethod = scope.getInvokedMethod();
final Optional<Class<? extends Annotation>> httpMethod = httpMethodAnnotations.stream()
.filter(invokedMethod::isAnnotationPresent).findFirst();

if (httpMethod.isPresent()) {
builder.param(METHOD_PARAM_KEY, httpMethod.get().getSimpleName());
} else {
throw new IllegalArgumentException(
"LinkCreator: The method has to be annotated with one of: " + String.join(", ",
(Iterable<String>) httpMethodAnnotations.stream().map(
Class::getSimpleName).map(m -> '@' + m)::iterator));
}
}

private void addSchemaIfNeeded(Builder builder, Scope method,
LinkFactoryContext linkFactoryContext) {
Optional<String> optionalInputSchema = jsonSchemaGenerator.createInputSchema(method,
linkFactoryContext.getFieldCheckerForSchema());
optionalInputSchema.ifPresent(s -> builder.param(SCHEMA_PARAM_KEY, s));
Optional<String> mt = detectMediaType(method.getInvokedMethod());
if (mt.isPresent() && MediaType.APPLICATION_JSON.equals(mt.get())) {
Optional<String> optionalOutputSchema = jsonSchemaGenerator.createOutputSchema(method,
linkFactoryContext.getFieldCheckerForSchema());
optionalOutputSchema.ifPresent(s -> builder.param(TARGET_SCHEMA_PARAM_KEY, s));
}
}

}
Loading

0 comments on commit e6e0618

Please sign in to comment.