com.amazonaws.codegen.emitters.tasks.MarshallerGeneratorTasks Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2011-2024 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.emitters.tasks;

import static com.amazonaws.codegen.utils.FunctionalUtils.safeFunction;

import com.amazonaws.codegen.emitters.FreemarkerGeneratorTask;
import com.amazonaws.codegen.emitters.GeneratorTask;
import com.amazonaws.codegen.emitters.GeneratorTaskParams;
import com.amazonaws.codegen.internal.TypeUtils;
import com.amazonaws.codegen.model.intermediate.Metadata;
import com.amazonaws.codegen.model.intermediate.Protocol;
import com.amazonaws.codegen.model.intermediate.ShapeModel;
import com.amazonaws.codegen.model.intermediate.ShapeType;
import com.amazonaws.util.ImmutableMapParameter;
import freemarker.template.Template;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class MarshallerGeneratorTasks extends BaseGeneratorTasks {

    private final String transformClassDir;
    private final Metadata metadata;
    private final Map shapes;

    public MarshallerGeneratorTasks(GeneratorTaskParams dependencies) {
        super(dependencies);
        this.transformClassDir = dependencies.getPathProvider().getTransformDirectory();
        this.metadata = model.getMetadata();
        this.shapes = model.getShapes();
    }

    @Override
    protected List createTasks() throws Exception {
        info("Emitting marshaller classes");
        return model.getShapes().entrySet().stream()
                .filter(e -> shouldGenerate(e.getValue()))
                .flatMap(safeFunction(e -> createTask(e.getKey(), e.getValue())))
                .collect(Collectors.toList());
    }

    private boolean shouldGenerate(ShapeModel shapeModel) {
        if (shapeModel.getCustomization().isSkipGeneratingMarshaller()) {
            System.out.println("Skip generating marshaller class for " + shapeModel.getShapeName());
            return false;
        }
        return shouldGenerate(shapeModel.getShapeType());
    }

    private boolean shouldGenerate(ShapeType shapeType) {
        return ShapeType.Request == shapeType || (ShapeType.Model == shapeType && metadata.isJsonProtocol());
    }

    private Stream createTask(String javaShapeName, ShapeModel shapeModel) throws Exception {
        String shapeFqcn = javaShapeName;
        if (TypeUtils.isReserved(javaShapeName)) {
            shapeFqcn = shapeModel.getFullyQualifiedName();
        }

        if (shapeModel.getShapeType() == ShapeType.Request && metadata.isJsonProtocol()) {
            return Stream.of(
                    createMarshallerTask(javaShapeName, shapeFqcn,
                                         freemarker.getRequestMarshallerTemplate(),
                                         javaShapeName + "ProtocolMarshaller"),
                    createMarshallerTask(javaShapeName, shapeFqcn,
                                         freemarker.getModelMarshallerTemplate(),
                                         javaShapeName + "Marshaller"));
        } else {
            return Stream.of(
                    createMarshallerTask(javaShapeName, shapeFqcn,
                                         freemarker.getModelMarshallerTemplate(),
                                         javaShapeName + "Marshaller"));
        }
    }

    private GeneratorTask createMarshallerTask(String javaShapeName, String shapeFullyQualifiedName, Template template, String marshallerClassName) throws
                                                                                                                    IOException {
        Map marshallerDataModel = ImmutableMapParameter.builder()
                .put("fileHeader", model.getFileHeader())
                .put("shapeName", javaShapeName)
                .put("shapeFqcn", shapeFullyQualifiedName)
                .put("shapes", shapes)
                .put("metadata", metadata)
                .put("transformPackage", model.getTransformPackage())
                .put("customConfig", model.getCustomizationConfig())
                .put("className", marshallerClassName)
                .put("protocolEnum", getProtocolEnumName())
                .build();

        return new FreemarkerGeneratorTask(transformClassDir,
                                           marshallerClassName,
                                           template,
                                           marshallerDataModel);
    }

    private String getProtocolEnumName() {
        switch (metadata.getProtocol()) {
            case CBOR:
            case AWS_JSON:
                return Protocol.AWS_JSON.name();
            default:
                return metadata.getProtocol().name();
        }
    }

}