/*
* Copyright (c) 2016. Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file is distributed
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
package com.amazonaws.codegen;
import com.amazonaws.codegen.model.intermediate.OperationModel;
import com.amazonaws.codegen.model.intermediate.ShapeModel;
import com.amazonaws.codegen.model.intermediate.ShapeType;
import com.amazonaws.codegen.model.intermediate.ShapeUnmarshaller;
import com.amazonaws.codegen.model.service.Operation;
import com.amazonaws.codegen.model.service.Output;
import com.amazonaws.codegen.model.service.Shape;
import java.util.HashMap;
import java.util.Map;
/**
* Constructs the result shapes for the intermediate model. Analyzes the operations in the service
* model to identify the result shapes that are to be generated.
*/
final class AddOutputShapes extends AddShapes implements IntermediateModelShapeProcessor {
AddOutputShapes(IntermediateModelBuilder builder) {
super(builder);
}
@Override
public Map<String, ShapeModel> process(Map<String, OperationModel> currentOperations,
Map<String, ShapeModel> currentShapes) {
return constructOutputShapes();
}
private Map<String, ShapeModel> constructOutputShapes() {
// C2j model components
final Map<String, Operation> operations = getServiceModel().getOperations();
final Map<String, Shape> c2jShapes = getServiceModel().getShapes();
// Java shape models, to be constructed
final Map<String, ShapeModel> javaShapes = new HashMap<String, ShapeModel>();
for (Map.Entry<String, Operation> entry : operations.entrySet()) {
String operationName = entry.getKey();
Operation operation = entry.getValue();
Output output = operation.getOutput();
if (output != null) {
String javaResponseClassName = getNamingStrategy()
.getResponseClassName(operationName);
ShapeModel outputShape = generateOutputShapeModel(operation, javaResponseClassName,
c2jShapes);
if (outputShape.getDocumentation() == null) {
outputShape.setDocumentation(output.getDocumentation());
}
javaShapes.put(javaResponseClassName, outputShape);
}
}
return javaShapes;
}
private final ShapeModel generateOutputShapeModel(Operation c2jOperationModel,
String javaOutputShapeNameOverride,
Map<String, Shape> c2jShapes) {
final Output c2jOutputModel = c2jOperationModel.getOutput();
final String c2jOutputShapeName = c2jOutputModel.getShape();
ShapeModel shapeModel = generateShapeModel(javaOutputShapeNameOverride, c2jOutputShapeName);
shapeModel.setType(ShapeType.Response.getValue());
// Set up unmarshaller metadata
ShapeUnmarshaller shapeUnmarshaller = new ShapeUnmarshaller()
.withFlattened(c2jShapes.get(c2jOutputShapeName).isFlattened())
.withResultWrapper(c2jOutputModel.getResultWrapper());
shapeModel.setUnmarshaller(shapeUnmarshaller);
return shapeModel;
}
}