From 935ce5bf81cf0296d7d35d41bd612a34999eb3e3 Mon Sep 17 00:00:00 2001 From: George Fu Date: Wed, 27 Jul 2022 17:33:14 +0000 Subject: [PATCH] feat(codegen): xml factory --- .../HttpBindingProtocolGenerator.java | 95 ++++++++++--------- .../HttpProtocolGeneratorUtils.java | 77 ++++++++------- 2 files changed, 96 insertions(+), 76 deletions(-) diff --git a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java index bc5bea1874b..fdb05300c6f 100644 --- a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java +++ b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpBindingProtocolGenerator.java @@ -735,31 +735,30 @@ private void writeResolvedPath( // Handle any label bindings. if (!labelBindings.isEmpty()) { + writer.addImport("resolvedPath", "__resolvedPath", "@aws-sdk/smithy-client"); + Model model = context.getModel(); List uriLabels = trait.getUri().getLabels(); for (HttpBinding binding : labelBindings) { String memberName = symbolProvider.toMemberName(binding.getMember()); Shape target = model.expectShape(binding.getMember().getTarget()); - String labelValue = getInputValue(context, binding.getLocation(), "input." + memberName, - binding.getMember(), target); + + String labelValueProvider = "() => " + getInputValue( + context, + binding.getLocation(), + "input." + memberName + "!", + binding.getMember(), + target + ); + // Get the correct label to use. Segment uriLabel = uriLabels.stream().filter(s -> s.getContent().equals(memberName)).findFirst().get(); - writer.addImport("extendedEncodeURIComponent", "__extendedEncodeURIComponent", - "@aws-sdk/smithy-client"); - String encodedSegment = uriLabel.isGreedyLabel() - ? "labelValue.split(\"/\").map(segment => __extendedEncodeURIComponent(segment)).join(\"/\")" - : "__extendedEncodeURIComponent(labelValue)"; - - // Set the label's value and throw a clear error if empty or undefined. - writer.write("if (input.$L !== undefined) {", memberName).indent() - .write("const labelValue: string = $L;", labelValue) - .openBlock("if (labelValue.length <= 0) {", "}", () -> { - writer.write("throw new Error('Empty value provided for input HTTP label: $L.');", memberName); - }) - .write("resolvedPath = resolvedPath.replace($S, $L);", uriLabel.toString(), encodedSegment).dedent() - .write("} else {").indent() - .write("throw new Error('No value provided for input HTTP label: $L.');", memberName).dedent() - .write("}"); + writer.write("resolvedPath = __resolvedPath(resolvedPath, input, '$L', $L, '$L', $L)", + memberName, + labelValueProvider, + uriLabel.toString(), + uriLabel.isGreedyLabel() ? "true" : "false" + ); } } } @@ -1650,11 +1649,11 @@ private void generateOperationRequestDeserializer( .forEach((memberName, memberShape) -> writer.write( "$L: undefined,", memberName)); }); + readRequestHeaders(context, operation, bindingIndex, "output"); }); readQueryString(context, operation, bindingIndex); readPath(context, operation, bindingIndex, trait); readHost(context, operation); - readRequestHeaders(context, operation, bindingIndex, "output"); List documentBindings = readRequestBody(context, operation, bindingIndex); // Track all shapes bound to the document so their deserializers may be generated. documentBindings.forEach(binding -> { @@ -1703,7 +1702,7 @@ private void handleContentType( operation, getDocumentContentType()); writer.write("const contentTypeHeaderKey: string | undefined = Object.keys(output.headers)" + ".find(key => key.toLowerCase() === 'content-type');"); - writer.openBlock("if (contentTypeHeaderKey !== undefined && contentTypeHeaderKey !== null) {", "};", () -> { + writer.openBlock("if (contentTypeHeaderKey != null) {", "};", () -> { writer.write("const contentType = output.headers[contentTypeHeaderKey];"); if (optionalContentType.isPresent() || operation.getInput().isPresent()) { String contentType = optionalContentType.orElse(getDocumentContentType()); @@ -1758,7 +1757,7 @@ private void handleAccept( writer.addImport("acceptMatches", "__acceptMatches", "@aws-smithy/server-common"); writer.write("const acceptHeaderKey: string | undefined = Object.keys(output.headers)" + ".find(key => key.toLowerCase() === 'accept');"); - writer.openBlock("if (acceptHeaderKey !== undefined && acceptHeaderKey !== null) {", "};", () -> { + writer.openBlock("if (acceptHeaderKey != null) {", "};", () -> { writer.write("const accept = output.headers[acceptHeaderKey];"); String contentType = optionalContentType.orElse(getDocumentContentType()); // Validate that the content type matches the protocol default, or what's modeled if there's @@ -1793,7 +1792,7 @@ private void readQueryString( return; } writer.write("const query = output.query"); - writer.openBlock("if (query !== undefined && query !== null) {", "}", () -> { + writer.openBlock("if (query != null) {", "}", () -> { readDirectQueryBindings(context, directQueryBindings); if (!mappedQueryBindings.isEmpty()) { // There can only ever be one of these bindings on a given operation. @@ -2041,8 +2040,10 @@ private void generateErrorDeserializer(GenerationContext context, StructureShape + " context: __SerdeContext\n" + "): Promise<$T> => {", "};", errorDeserMethodName, outputName, errorSymbol, () -> { - writer.write("const contents: any = {};"); - readResponseHeaders(context, error, bindingIndex, outputName); + writer.openBlock("const contents: any = map({", "});", () -> { + readResponseHeaders(context, error, bindingIndex, outputName); + }); + List documentBindings = readErrorResponseBody(context, error, bindingIndex); // Track all shapes bound to the document so their deserializers may be generated. documentBindings.forEach(binding -> { @@ -2159,26 +2160,27 @@ private void readPrefixHeaders( TypeScriptWriter writer = context.getWriter(); // Run through the headers one time, matching any prefix groups. - writer.openBlock("...(Object.keys($L.headers).reduce((acc, header) => {", "}, {}));", outputName, () -> { - for (HttpBinding binding : prefixHeaderBindings) { - // Prepare a grab bag for these headers if necessary - String memberName = symbolProvider.toMemberName(binding.getMember()); - writer.write("acc.$L = [, {}];", memberName); - - // Generate a single block for each group of lower-cased prefix headers. + for (HttpBinding binding : prefixHeaderBindings) { + // Prepare a grab bag for these headers if necessary + String memberName = symbolProvider.toMemberName(binding.getMember()); + writer.openBlock("$L: [, ", "],", memberName, () -> { String headerLocation = binding.getLocationName().toLowerCase(Locale.US); - writer.openBlock("if (header.startsWith($S)) {", "}", headerLocation, () -> { + writer.write( + "Object.keys($L.headers).filter(header => header.startsWith('$L'))", + outputName, + headerLocation + ); + writer.indent().openBlock(".reduce((acc, header) => {", "}, {} as any)", () -> { MapShape prefixMap = model.expectShape(binding.getMember().getTarget()).asMapShape().get(); Shape target = model.expectShape(prefixMap.getValue().getTarget()); String headerValue = getOutputValue(context, binding.getLocation(), - outputName + ".headers[header]", binding.getMember(), target); - - // Extract the non-prefix portion as the key. - writer.write("acc.$L[header.substring($L)] = [, $L];", - memberName, headerLocation.length(), headerValue); + outputName + ".headers[header]", binding.getMember(), target); + writer.write("acc[header.substring($L)] = $L;", + headerLocation.length(), headerValue); + writer.write("return acc;"); }); - } - }); + }); + } } private List readRequestBody( @@ -2253,12 +2255,17 @@ private List readBody( // isn't set in the body. // These are only relevant when a payload is not present, as it cannot // coexist with a payload. - for (HttpBinding responseCodeBinding : responseCodeBindings) { - // The name of the member to get from the input shape. - String memberName = symbolProvider.toMemberName(responseCodeBinding.getMember()); - writer.openBlock("if (contents.$L === undefined) {", "}", memberName, () -> - writer.write("contents.$L = output.statusCode;", memberName)); + if (!responseCodeBindings.isEmpty()) { + writer.openBlock("map(contents, {", "});", () -> { + for (HttpBinding responseCodeBinding : responseCodeBindings) { + // The name of the member to get from the input shape. + String memberName = symbolProvider.toMemberName(responseCodeBinding.getMember()); + + writer.write("$L: [, output.statusCode]", memberName); + } + }); } + if (!documentBindings.isEmpty()) { return documentBindings; } diff --git a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpProtocolGeneratorUtils.java b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpProtocolGeneratorUtils.java index f44e3054601..c989388ea04 100644 --- a/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpProtocolGeneratorUtils.java +++ b/smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/integration/HttpProtocolGeneratorUtils.java @@ -348,28 +348,13 @@ static Set generateErrorDispatcher( // Error responses must be at least BaseException interface SymbolReference baseExceptionReference = getClientBaseException(context); - writer.write("let response: $T;", baseExceptionReference); errorCodeGenerator.accept(context); - writer.openBlock("switch (errorCode) {", "}", () -> { - // Generate the case statement for each error, invoking the specific deserializer. - new TreeSet<>(operationIndex.getErrors(operation, context.getService())).forEach(error -> { - final ShapeId errorId = error.getId(); - // Track errors bound to the operation so their deserializers may be generated. - errorShapes.add(error); - Symbol errorSymbol = symbolProvider.toSymbol(error); - String errorDeserMethodName = ProtocolGenerator.getDeserFunctionName(errorSymbol, - context.getProtocolName()) + "Response"; - // Dispatch to the error deserialization function. - String outputParam = shouldParseErrorBody ? "parsedOutput" : "output"; - writer.write("case $S:", errorId.getName()); - writer.write("case $S:", errorId.toString()); - writer.indent() - .write("throw await $L($L, context);", errorDeserMethodName, outputParam) - .dedent(); - }); - // Build a generic error the best we can for ones we don't know about. - writer.write("default:").indent(); + TreeSet structureShapes = new TreeSet<>( + operationIndex.getErrors(operation, context.getService()) + ); + + Runnable defaultErrorHandler = () -> { if (shouldParseErrorBody) { // Body is already parsed above writer.write("const parsedBody = parsedOutput.body;"); @@ -378,21 +363,49 @@ static Set generateErrorDispatcher( writer.write("const parsedBody = await parseBody(output.body, context);"); } + writer.addImport("throwDefaultError", "throwDefaultError", "@aws-sdk/smithy-client"); + // Get the protocol specific error location for retrieving contents. String errorLocation = bodyErrorLocationModifier.apply(context, "parsedBody"); - writer.write("const $$metadata = deserializeMetadata(output);"); - writer.write("const statusCode = $$metadata.httpStatusCode ? $$metadata.httpStatusCode" - + " + '' : undefined;"); - writer.openBlock("response = new $T({", "});", baseExceptionReference, () -> { - writer.write("name: $1L.code || $1L.Code || errorCode || statusCode || 'UnknowError',", - errorLocation); - writer.write("$$fault: \"client\","); - writer.write("$$metadata"); + writer.openBlock("throwDefaultError({", "})", () -> { + writer.write("output,"); + if (errorLocation.equals("parsedBody")) { + writer.write("parsedBody,"); + } else { + writer.write("parsedBody: $L,", errorLocation); + } + writer.write("exceptionCtor: $T,", baseExceptionReference); + writer.write("errorCode"); }); - writer.addImport("decorateServiceException", "__decorateServiceException", - TypeScriptDependency.AWS_SMITHY_CLIENT.packageName); - writer.write("throw __decorateServiceException(response, $L);", errorLocation); - }); + }; + + if (!structureShapes.isEmpty()) { + writer.openBlock("switch (errorCode) {", "}", () -> { + // Generate the case statement for each error, invoking the specific deserializer. + + structureShapes.forEach(error -> { + final ShapeId errorId = error.getId(); + // Track errors bound to the operation so their deserializers may be generated. + errorShapes.add(error); + Symbol errorSymbol = symbolProvider.toSymbol(error); + String errorDeserMethodName = ProtocolGenerator.getDeserFunctionName(errorSymbol, + context.getProtocolName()) + "Response"; + // Dispatch to the error deserialization function. + String outputParam = shouldParseErrorBody ? "parsedOutput" : "output"; + writer.write("case $S:", errorId.getName()); + writer.write("case $S:", errorId.toString()); + writer.indent() + .write("throw await $L($L, context);", errorDeserMethodName, outputParam) + .dedent(); + }); + + // Build a generic error the best we can for ones we don't know about. + writer.write("default:").indent(); + defaultErrorHandler.run(); + }); + } else { + defaultErrorHandler.run(); + } }); writer.write("");