|
| 1 | +/* |
| 2 | + * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. |
| 3 | + * SPDX-License-Identifier: Apache-2.0 |
| 4 | + */ |
| 5 | + |
| 6 | +package software.amazon.smithy.docgen.core.interceptors; |
| 7 | + |
| 8 | +import java.util.Optional; |
| 9 | +import software.amazon.smithy.docgen.core.sections.ShapeDetailsSection; |
| 10 | +import software.amazon.smithy.docgen.core.writers.DocWriter; |
| 11 | +import software.amazon.smithy.docgen.core.writers.DocWriter.AdmonitionType; |
| 12 | +import software.amazon.smithy.model.traits.RequestCompressionTrait; |
| 13 | +import software.amazon.smithy.utils.CodeInterceptor; |
| 14 | +import software.amazon.smithy.utils.SmithyInternalApi; |
| 15 | + |
| 16 | +/** |
| 17 | + * Adds information about request compression to operations with the |
| 18 | + * <a href="https://smithy.io/2.0/spec/behavior-traits.html#requestcompression-trait"> |
| 19 | + * requestCompression trait</a>. |
| 20 | + */ |
| 21 | +@SmithyInternalApi |
| 22 | +public final class RequestCompressionInterceptor implements CodeInterceptor<ShapeDetailsSection, DocWriter> { |
| 23 | + @Override |
| 24 | + public Class<ShapeDetailsSection> sectionType() { |
| 25 | + return ShapeDetailsSection.class; |
| 26 | + } |
| 27 | + |
| 28 | + @Override |
| 29 | + public boolean isIntercepted(ShapeDetailsSection section) { |
| 30 | + return section.shape().hasTrait(RequestCompressionTrait.class); |
| 31 | + } |
| 32 | + |
| 33 | + @Override |
| 34 | + public void write(DocWriter writer, String previousText, ShapeDetailsSection section) { |
| 35 | + var trait = section.shape().expectTrait(RequestCompressionTrait.class); |
| 36 | + writer.openAdmonition(AdmonitionType.IMPORTANT); |
| 37 | + |
| 38 | + // Have particular support for single-element lists. |
| 39 | + writer.putContext("encoding", trait.getEncodings().size() == 1 |
| 40 | + ? Optional.of(trait.getEncodings().get(0)) |
| 41 | + : Optional.empty()); |
| 42 | + writer.putContext("encodings", trait.getEncodings()); |
| 43 | + |
| 44 | + writer.write(""" |
| 45 | + This operation supports optional request compression using \ |
| 46 | + ${?encoding}${encoding:L} encoding.${/encoding}\ |
| 47 | + ${^encoding}one of the following priority-ordered encodings: \ |
| 48 | + ${#encodings}${value:L}${^key.last}, ${/key.last}${/encodings}.${/encoding}"""); |
| 49 | + writer.closeAdmonition(); |
| 50 | + writer.writeWithNoFormatting(previousText); |
| 51 | + } |
| 52 | +} |
0 commit comments