Java Code Examples for com.squareup.javapoet.CodeBlock#of()

@Override
public CodeBlock generateFactoryMethodsComponentInstance(SpecModel specModel) {
  return CodeBlock.of(
      "$L instance = new $L(context.getAndroidContext());\n",
      specModel.getComponentName(),
      specModel.getComponentName());
}
 

private static CodeBlock createValidateOnClickCodeBlock(ValidationClass validationClass) {
    Element button = validationClass.getValidateButton();
    if(button != null) {
        return CodeBlock.builder()
                .add("\n")
                .add(
                        "validateOnClickListener(target.$N, target);",
                        button.getSimpleName().toString()
                )
                .build();
    }
    return CodeBlock.of("");
}
 

private CodeBlock globalRegion(Partition partition) {
    ClassName region = ClassName.get(regionBasePackage, "Region");
    Service service = partition.getServices().get(this.serviceEndpointPrefix);
    boolean hasGlobalRegionForPartition = service.isRegionalized() != null &&
                                          !service.isRegionalized() &&
                                          service.isPartitionWideEndpointAvailable();
    String globalRegionForPartition = hasGlobalRegionForPartition ? service.getPartitionEndpoint() : null;
    return globalRegionForPartition == null
                                ? CodeBlock.of("null")
                                : CodeBlock.of("$T.of($S)", region, globalRegionForPartition);
}
 

@Test
public void codeBlocksJoined() {
    CodeBlock a = CodeBlock.of("a");
    CodeBlock b = CodeBlock.of("b");
    CodeBlock ab = CodeBlock.builder().add(a).add(b).build();

    CodeBlock result = Stream.of(a, b).collect(PoetCollectors.toCodeBlock());
    assertThat(result).isEqualTo(ab);
}
 

private CodeBlock enumGetterStatement(MemberModel member) {
    String fieldName = member.getVariable().getVariableName();
    if (member.isList() || member.isMap()) {
        Optional<ClassName> copier = serviceModelCopiers.copierClassFor(member);
        if (!copier.isPresent()) {
            throw new IllegalStateException("Don't know how to copy " + fieldName + " with enum elements!");
        }
        return CodeBlock.of("return $T.$N($N);", copier.get(), serviceModelCopiers.stringToEnumCopyMethodName(), fieldName);
    } else {
        ClassName enumClass = poetExtensions.getModelClass(member.getEnumType());
        return CodeBlock.of("return $T.fromValue($N);", enumClass, fieldName);
    }
}
 

public CodeBlock toStringValue(MemberModel member) {
    if (!member.isSensitive()) {
        return CodeBlock.of("$L()", member.getFluentGetterMethodName());
    }

    return CodeBlock.of("$L() == null ? null : $S",
                        member.getFluentGetterMethodName(),
                        "*** Sensitive Data Redacted ***");
}
 

private CodeBlock mapKeyValCopyExpr(MemberModel keyValModel, String getterName, EnumTransform enumTransform) {
    Optional<ClassName> keyCopier = serviceModelCopiers.copierClassFor(keyValModel);
    boolean hasCopier = keyCopier.isPresent();
    switch (enumTransform) {
        case STRING_TO_ENUM:
            if (hasCopier) {
                return CodeBlock.of("$T.$N(e.$N())",
                                    keyCopier.get(),
                                    serviceModelCopiers.stringToEnumCopyMethodName(),
                                    getterName);
            } else {
                return CodeBlock.of("$T.fromValue(e.$N())",
                                    poetExtensions.getModelClass(keyValModel.getEnumType()),
                                    getterName);
            }
        case ENUM_TO_STRING:
            if (hasCopier) {
                return CodeBlock.of("$T.$N(e.$N())",
                                    keyCopier.get(),
                                    serviceModelCopiers.enumToStringCopyMethodName(),
                                    getterName);
            } else {
                return CodeBlock.of("e.$N().toString()", getterName);
            }
        case NONE:
            if (hasCopier) {
                return CodeBlock.of("$T.$N(e.$N())",
                                    keyCopier.get(),
                                    serviceModelCopiers.copyMethodName(),
                                    getterName);
            } else {
                return CodeBlock.of("e.$N()", getterName);
            }
        default:
            throw new IllegalArgumentException("Unknown enum transform: " + enumTransform);
    }
}
 

private CodeBlock sdkBytesMapValueSetter() {
    return CodeBlock.of("$1N($2N == null ? null : " +
                        "$2N.entrySet().stream()" +
                        ".collect($4T.toMap(e -> e.getKey(), e -> $3T.fromByteBuffer(e.getValue()))));",
                        memberModel.getFluentSetterMethodName(), fieldName(),
                        SdkBytes.class, Collectors.class);
}
 

@Test
public void delimitedCodeBlocksJoined() {
    CodeBlock a = CodeBlock.of("a");
    CodeBlock b = CodeBlock.of("b");
    CodeBlock delimeter = CodeBlock.of(",");
    CodeBlock ab = CodeBlock.builder().add(a).add(delimeter).add(b).build();

    CodeBlock result = Stream.of(a, b).collect(PoetCollectors.toDelimitedCodeBlock(","));
    assertThat(result).isEqualTo(ab);
}
 

private CodeBlock getJavadoc() {
    return CodeBlock.of("This includes configuration specific to $L that is supported by both {@link $T} and {@link $T}.",
                        model.getMetadata().getDescriptiveServiceName(),
                        ClassName.get(basePackage, model.getMetadata().getSyncBuilderInterface()),
                        ClassName.get(basePackage, model.getMetadata().getAsyncBuilderInterface()));
}
 

private CodeBlock getJavadoc() {
    return CodeBlock.of("A builder for creating an instance of {@link $1T}. This can be created with the static "
                        + "{@link $1T#builder()} method.", clientInterfaceName);
}
 

private CodeBlock s3SignerDefinitionMethodBody() {
    return CodeBlock.of("return $T.create();\n",
                        ClassName.get("software.amazon.awssdk.auth.signer", "AwsS3V4Signer"));
}
 

private CodeBlock consumerBuilderVarargSetterBody() {
    return CodeBlock.of("$1L($3T.of($2L).map(c -> $4T.builder().applyMutation(c).build()).collect($5T.toList()));",
                        memberModel().getFluentSetterMethodName(), fieldName(),
                        Stream.class, listElementType(), Collectors.class);
}
 

private CodeBlock varargToListSetterBody() {
    return CodeBlock.of("$1L($2T.asList($3L));",
                        memberModel().getFluentSetterMethodName(), Arrays.class, fieldName());
}
 

private CodeBlock sdkBytesSetter() {
    return CodeBlock.of("$1N($2N == null ? null : $3T.fromByteBuffer($2N));",
                        memberModel.getFluentSetterMethodName(), fieldName(),
                        SdkBytes.class);
}
 

private CodeBlock getterStatement(MemberModel model) {
    VariableModel modelVariable = model.getVariable();
    return CodeBlock.of("return $N;", modelVariable.getVariableName());
}
 

private CodeBlock v4SignerDefinitionMethodBody() {
    return CodeBlock.of("return $T.create();", Aws4Signer.class);
}
 

public Id(int value, ClassName className, String resourceName) {
    this.value = value;
    this.code = className.topLevelClassName().equals(ANDROID_R)
            ? CodeBlock.of("$L.$N", className, resourceName)
            : CodeBlock.of("$T.$N", className, resourceName);
}
 

public Id(int value) {
    this.value = value;
    this.code = CodeBlock.of("$L", value);
}
 

private CodeBlock getJavadoc() {
    return CodeBlock.of("A builder for creating an instance of {@link $1T}. This can be created with the static "
                        + "{@link $1T#builder()} method.", clientInterfaceName);
}