com.squareup.javapoet.AnnotationSpec#builder

Source File: JRepeatableCodeGenerator.java From jackdaw with Apache License 2.0

6 votes

private AnnotationSpec createAnnotation(final AnnotationMirror annotation) {
    final AnnotationSpec.Builder annotationBuilder = AnnotationSpec.builder(
        ClassName.get((TypeElement) annotation.getAnnotationType().asElement())
    );
    final Map<? extends ExecutableElement, ? extends AnnotationValue> params = annotation.getElementValues();

    for (Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> entry : params.entrySet()) {
        final ExecutableElement paramMethod = entry.getKey();
        final String methodName = TypeUtils.getName(paramMethod);

        final AnnotationValue annotationValue = entry.getValue();
        final Object value = annotationValue.getValue();
        final Pair<String, Object> format = calculateValue(value);

        annotationBuilder.addMember(methodName, format.getKey(), format.getValue());
    }

    return annotationBuilder.build();
}

Source File: CodeGeneration.java From requery with Apache License 2.0

6 votes

static void addGeneratedAnnotation(ProcessingEnvironment processingEnvironment,
                                   TypeSpec.Builder builder) {
    Elements elements = processingEnvironment.getElementUtils();
    SourceVersion sourceVersion = processingEnvironment.getSourceVersion();
    AnnotationSpec.Builder annotationBuilder = null;
    if (sourceVersion.compareTo(SourceVersion.RELEASE_8) > 0) {
        ClassName name = ClassName.bestGuess("javax.annotation.processing.Generated");
        annotationBuilder = AnnotationSpec.builder(name);
    } else {
        if (elements.getTypeElement(Generated.class.getCanonicalName()) != null) {
            annotationBuilder = AnnotationSpec.builder(Generated.class);
        }
    }

    if (annotationBuilder != null) {
        builder.addAnnotation(annotationBuilder
                .addMember("value", "$S",
                        EntityProcessor.class.getCanonicalName()).build());
    }
}

Source File: PsiMethodExtractorUtils.java From litho with Apache License 2.0

6 votes

private static List<AnnotationSpec> getExternalAnnotations(PsiParameter param) {
  PsiAnnotation[] annotationsOnParam = AnnotationUtil.getAllAnnotations(param, false, null);
  final List<AnnotationSpec> annotations = new ArrayList<>();

  for (PsiAnnotation annotationOnParam : annotationsOnParam) {
    if (annotationOnParam.getQualifiedName().startsWith(COMPONENTS_PACKAGE)) {
      continue;
    }

    final AnnotationSpec.Builder annotationSpec =
        AnnotationSpec.builder(PsiTypeUtils.guessClassName(annotationOnParam.getQualifiedName()));

    PsiNameValuePair[] paramAttributes = annotationOnParam.getParameterList().getAttributes();
    for (PsiNameValuePair attribute : paramAttributes) {
      annotationSpec.addMember(attribute.getName(), attribute.getDetachedValue().getText());
    }

    annotations.add(annotationSpec.build());
  }

  return annotations;
}

Source File: MethodExtractorUtils.java From litho with Apache License 2.0

6 votes

private static List<AnnotationSpec> getExternalAnnotations(VariableElement param) {
  final List<? extends AnnotationMirror> annotationMirrors = param.getAnnotationMirrors();
  final List<AnnotationSpec> annotations = new ArrayList<>();

  for (AnnotationMirror annotationMirror : annotationMirrors) {
    if (annotationMirror.getAnnotationType().toString().startsWith(COMPONENTS_PACKAGE)) {
      continue;
    }

    final AnnotationSpec.Builder annotationSpec =
        AnnotationSpec.builder(
            ClassName.bestGuess(annotationMirror.getAnnotationType().toString()));

    Map<? extends ExecutableElement, ? extends AnnotationValue> elementValues =
        annotationMirror.getElementValues();
    for (Map.Entry<? extends ExecutableElement, ? extends AnnotationValue> elementValue :
        elementValues.entrySet()) {
      annotationSpec.addMember(
          elementValue.getKey().getSimpleName().toString(), elementValue.getValue().toString());
    }

    annotations.add(annotationSpec.build());
  }

  return annotations;
}

Source File: JavaClassGenerator.java From web3j with Apache License 2.0

6 votes

@Override
public void writeClass() throws IOException {
    ClassName EVM_ANNOTATION = ClassName.get("org.web3j", "EVMTest");
    AnnotationSpec.Builder annotationSpec = AnnotationSpec.builder(EVM_ANNOTATION);
    if (JavaVersion.getJavaVersionAsDouble() < 11) {
        ClassName GethContainer = ClassName.get("org.web3j", "NodeType");
        annotationSpec.addMember("value", "type = $T.GETH", GethContainer);
    }
    TypeSpec testClass =
            TypeSpec.classBuilder(theContract.getSimpleName() + "Test")
                    .addMethods(MethodFilter.generateMethodSpecsForEachTest(theContract))
                    .addAnnotation((annotationSpec).build())
                    .addField(
                            theContract,
                            toCamelCase(theContract),
                            Modifier.PRIVATE,
                            Modifier.STATIC)
                    .build();
    JavaFile javaFile = JavaFile.builder(packageName, testClass).build();
    javaFile.writeTo(new File(writePath));
}

Source File: JavadocProcessor.java From drift with Apache License 2.0

5 votes

private static AnnotationSpec documentationAnnotation(List<String> lines)
{
    AnnotationSpec.Builder builder = AnnotationSpec.builder(ThriftDocumentation.class);
    for (String line : lines) {
        builder.addMember("value", "$S", line);
    }
    return builder.build();
}

Source File: AnnotationSpecUtil.java From kaif-android with Apache License 2.0

5 votes

public static AnnotationSpec generate(AnnotationMirror annotation) {
  TypeElement element = (TypeElement) annotation.getAnnotationType().asElement();
  AnnotationSpec.Builder builder = AnnotationSpec.builder(ClassName.get(element));
  Visitor visitor = new Visitor(builder);
  for (ExecutableElement executableElement : annotation.getElementValues().keySet()) {
    String name = executableElement.getSimpleName().toString();
    AnnotationValue value = annotation.getElementValues().get(executableElement);
    value.accept(visitor, new Entry(name, value));
  }
  return builder.build();
}

Source File: TransformationTemplateIntegrationTest.java From Akatsuki with Apache License 2.0

5 votes

protected AnnotationSpec createTransformationTemplate(Bound bound, Class<?> staticClass,
		Class<?>... constraints) {
	final String objectFqcn = staticClass.getCanonicalName();
	AnnotationSpec save = AnnotationSpec
			.builder(StatementTemplate.class).addMember("value",
					"\"{{bundle}}.putString({{keyName}}, $L.wrap({{fieldName}}))\"", objectFqcn)
			.build();

	AnnotationSpec restore = AnnotationSpec.builder(StatementTemplate.class)
			.addMember("type", "$T.$L", StatementTemplate.Type.class,
					StatementTemplate.Type.ASSIGNMENT)
			.addMember("value", "\"$L.unwrap({{bundle}}.getString({{keyName}}))\"", objectFqcn)
			.addMember("variable", "\"{{fieldName}}\"").build();

	final AnnotationSpec.Builder annotationSpec = AnnotationSpec
			.builder(TransformationTemplate.class).addMember("save", "$L", save)
			.addMember("restore", "$L", restore)
			.addMember("execution", "$T.$L", Execution.class, Execution.BEFORE);

	for (Class<?> constraint : constraints) {
		final AnnotationSpec constraintsSpec = AnnotationSpec.builder(TypeConstraint.class)
				.addMember("type", "$T.class", constraint)
				.addMember("bound", "$T.$L", Bound.class, bound).build();

		AnnotationSpec.Builder filterSpec = AnnotationSpec.builder(TypeFilter.class);
		filterSpec.addMember("type", "$L", constraintsSpec);

		annotationSpec.addMember("filters", "$L", filterSpec.build());

	}
	return annotationSpec.build();
}

Source File: ArgTestField.java From Akatsuki with Apache License 2.0

5 votes

@Override
protected Builder fieldSpecBuilder() {
	final Builder builder = super.fieldSpecBuilder();
	final AnnotationSpec.Builder annotationBuilder = AnnotationSpec.builder(Arg.class);
	if (skip) {
		annotationBuilder.addMember("skip", "$L", true);
	}

	if (optional) {
		annotationBuilder.addMember("optional", "$L", true);
	}
	builder.addAnnotation(annotationBuilder.build());
	return builder;
}

Source File: RetainedTestField.java From Akatsuki with Apache License 2.0

5 votes

@Override
protected Builder fieldSpecBuilder() {
	final Builder builder = super.fieldSpecBuilder();
	final AnnotationSpec.Builder annotationBuilder = AnnotationSpec.builder(Retained.class);
	if (skip) {
		annotationBuilder.addMember("skip", "$L", true);
	}
	if (policy != null) {
		annotationBuilder.addMember("restorePolicy", "$T.$L", RestorePolicy.class, policy);
	}
	builder.addAnnotation(annotationBuilder.build());
	return builder;
}

Source File: ResourceMethodExtensionPlugin.java From raml-module-builder with Apache License 2.0

5 votes

private AnnotationSpec buildAnnotation(Class<?> annotationClass, Object value, String type){

    AnnotationSpec.Builder annoBuilder = AnnotationSpec.builder(annotationClass);

    if(type.equalsIgnoreCase("Size")){
      //size can contain two values (min and max) so it is unlike the other potential annotations
      String []multipleAnnos = ((String)value).split(",");
      for (int j = 0; j < multipleAnnos.length; j++) {
        if(j == 0){
          annoBuilder.addMember("min", "$L", Integer.valueOf(multipleAnnos[j].trim()));
        }
        else{
          annoBuilder.addMember("max", "$L", Integer.valueOf(multipleAnnos[j].trim()));
        }
      }
    }
    else if(!type.equalsIgnoreCase("REQUIRED")){
      //a required annotation will add a @notnull annotation
      //THERE IS CURRENTLY NO SUPPORT TO MAKE A REQUIRED PARAM NOT REQUIRED
      if(value instanceof String){
        if(type.equalsIgnoreCase("PATTERN")){
          annoBuilder.addMember("regexp", "$S", (String)value);
        }
        else{
          annoBuilder.addMember(ANNOTATION_VALUE, "$S", (String)value);
        }
      }
      else if(value instanceof Boolean){
        annoBuilder.addMember(ANNOTATION_VALUE, "$L", (Boolean)value);
      }
      else if(value instanceof Integer){
        annoBuilder.addMember(ANNOTATION_VALUE, "$L", (Integer)value);
      }
    }

    return annoBuilder.build();
  }

Source File: ValueParser.java From dataenum with Apache License 2.0

5 votes

private static Iterable<AnnotationSpec> parseMethodAnnotations(
    ExecutableElement methodElement, Messager messager) {
  ArrayList<AnnotationSpec> annotations = new ArrayList<>();

  for (AnnotationMirror annotationMirror : methodElement.getAnnotationMirrors()) {
    TypeName annotationTypeName =
        ClassName.get(annotationMirror.getAnnotationType().asElement().asType());

    if (!(annotationTypeName instanceof ClassName)) {
      messager.printMessage(
          Kind.ERROR,
          "Annotation is not a class; this shouldn't happen",
          methodElement,
          annotationMirror);
      continue;
    }

    Builder builder = AnnotationSpec.builder(((ClassName) annotationTypeName));

    for (Entry<? extends ExecutableElement, ? extends AnnotationValue> entry :
        annotationMirror.getElementValues().entrySet()) {

      builder.addMember(entry.getKey().getSimpleName().toString(), entry.getValue().toString());
    }

    annotations.add(builder.build());
  }

  return annotations;
}

Source File: OpenApiClientGenerator.java From spring-openapi with MIT License

5 votes

private FieldSpec.Builder getStringBasedSchemaField(String fieldName, Schema innerSchema, TypeSpec.Builder typeSpecBuilder) {
	if (innerSchema.getFormat() == null) {
		ClassName stringClassName = ClassName.get(JAVA_LANG_PKG, "String");
		FieldSpec.Builder fieldBuilder = FieldSpec.builder(stringClassName, fieldName, Modifier.PRIVATE);
		if (innerSchema.getPattern() != null) {
			fieldBuilder.addAnnotation(AnnotationSpec.builder(Pattern.class)
					.addMember("regexp", "$S", innerSchema.getPattern())
					.build()
			);
		}
		if (innerSchema.getMinLength() != null || innerSchema.getMaxLength() != null) {
			AnnotationSpec.Builder sizeAnnotationBuilder = AnnotationSpec.builder(Size.class);
			if (innerSchema.getMinLength() != null) {
				sizeAnnotationBuilder.addMember("min", "$L", innerSchema.getMinLength());
			}
			if (innerSchema.getMaxLength() != null) {
				sizeAnnotationBuilder.addMember("max", "$L", innerSchema.getMaxLength());
			}
			fieldBuilder.addAnnotation(sizeAnnotationBuilder.build());
		}
		enrichWithGetSet(typeSpecBuilder, stringClassName, fieldName);
		return fieldBuilder;
	} else if (equalsIgnoreCase(innerSchema.getFormat(), "date")) {
		ClassName localDateClassName = ClassName.get(JAVA_TIME_PKG, "LocalDate");
		enrichWithGetSet(typeSpecBuilder, localDateClassName, fieldName);
		return FieldSpec.builder(localDateClassName, fieldName, Modifier.PRIVATE);
	} else if (equalsIgnoreCase(innerSchema.getFormat(), "date-time")) {
		ClassName localDateTimeClassName = ClassName.get(JAVA_TIME_PKG, "LocalDateTime");
		enrichWithGetSet(typeSpecBuilder, localDateTimeClassName, fieldName);
		return FieldSpec.builder(localDateTimeClassName, fieldName, Modifier.PRIVATE);
	}
	throw new IllegalArgumentException(String.format("Error parsing string based property [%s]", fieldName));
}

Source File: SimpleProcessorTests.java From spring-init with Apache License 2.0

5 votes

private static TypeSpec.Builder importAnnotation(TypeSpec.Builder type, String... fullyQualifiedImports) {
	ClassName[] array = new ClassName[fullyQualifiedImports.length];
	for (int i = 0; i < fullyQualifiedImports.length; i++) {
		array[i] = ClassName.bestGuess(fullyQualifiedImports[i]);
	}
	AnnotationSpec.Builder builder = AnnotationSpec.builder(IMPORT);
	builder.addMember("value", array.length > 1 ? ("{" + typeParams(array.length) + "}") : "$T.class",
			(Object[]) array);
	return type.addAnnotation(builder.build());
}

Source File: InterfaceMetaInfo.java From raptor with Apache License 2.0

5 votes

public AnnotationSpec generateInterfaceSpec() {
    AnnotationSpec.Builder builder = AnnotationSpec.builder(RaptorInterface.class);
    OptionUtil.setAnnotationMember(builder, "appId", "$S", protoFileMetaInfo.getAppId());
    OptionUtil.setAnnotationMember(builder, "appName", "$S", protoFileMetaInfo.getAppName());
    OptionUtil.setAnnotationMember(builder, "version", "$S", protoFileMetaInfo.getVersion());
    OptionUtil.setAnnotationMember(builder, "protoFile", "$S", protoFileMetaInfo.getProtoFile());
    OptionUtil.setAnnotationMember(builder, "crc32", "$S", protoFileMetaInfo.getCrc32());
    // TODO: 2018/5/16 先写死spring
    OptionUtil.setAnnotationMember(builder, "library", "$S", "spring");

    return builder.build();

}

Source File: MessageMetaInfo.java From raptor with Apache License 2.0

5 votes

public AnnotationSpec generateMessageSpec() {
    AnnotationSpec.Builder builder = AnnotationSpec.builder(RaptorMessage.class);
    OptionUtil.setAnnotationMember(builder, "version", "$S", protoFileMetaInfo.getVersion());
    OptionUtil.setAnnotationMember(builder, "protoFile", "$S", protoFileMetaInfo.getProtoFile());
    OptionUtil.setAnnotationMember(builder, "crc32", "$S", protoFileMetaInfo.getCrc32());
    return builder.build();
}

Source File: JacksonDiscriminatorInheritanceTypeExtension.java From raml-java-tools with Apache License 2.0

4 votes

@Override
public TypeSpec.Builder classCreated(ObjectPluginContext objectPluginContext, ObjectTypeDeclaration ramlType, TypeSpec.Builder typeSpec, EventType eventType) {

  if ( eventType == EventType.IMPLEMENTATION) {
    return typeSpec;
  }

  ObjectTypeDeclaration otr = ramlType;

  if (otr.discriminator() != null && objectPluginContext.childClasses(otr.name()).size() > 0) {

    typeSpec.addAnnotation(AnnotationSpec.builder(JsonTypeInfo.class)
            .addMember("use", "$T.Id.NAME", JsonTypeInfo.class)
            .addMember("include", "$T.As.EXISTING_PROPERTY", JsonTypeInfo.class)
            .addMember("property", "$S", otr.discriminator()).build());

    AnnotationSpec.Builder subTypes = AnnotationSpec.builder(JsonSubTypes.class);
    for (CreationResult result : objectPluginContext.childClasses(ramlType.name())) {

      subTypes.addMember(
              "value",
              "$L",
              AnnotationSpec
                      .builder(JsonSubTypes.Type.class)
                      .addMember("value", "$L",
                              result.getJavaName(EventType.INTERFACE) + ".class").build());
    }

    subTypes.addMember(
            "value",
            "$L",
            AnnotationSpec
                    .builder(JsonSubTypes.Type.class)
                    .addMember("value", "$L",
                            objectPluginContext.creationResult().getJavaName(EventType.INTERFACE) + ".class").build());

    typeSpec.addAnnotation(subTypes.build());

  }

  if (otr.discriminatorValue() != null) {

    typeSpec.addAnnotation(AnnotationSpec.builder(JsonTypeName.class)
            .addMember("value", "$S", otr.discriminatorValue()).build());
  }


  if (!Annotations.ABSTRACT.get(otr)) {

    typeSpec.addAnnotation(AnnotationSpec.builder(JsonDeserialize.class)
            .addMember("as", "$T.class", objectPluginContext.creationResult().getJavaName(EventType.IMPLEMENTATION))
            .build());
  }


  return typeSpec;
}

Source File: JacksonDiscriminatorInheritanceTypeExtension.java From raml-java-tools with Apache License 2.0

4 votes

@Override
public TypeSpec.Builder classCreated(ObjectPluginContext objectPluginContext, ObjectTypeDeclaration ramlType, TypeSpec.Builder typeSpec, EventType eventType) {

  if ( eventType == EventType.IMPLEMENTATION) {
    return typeSpec;
  }

  ObjectTypeDeclaration otr = ramlType;

  if (otr.discriminator() != null && objectPluginContext.childClasses(otr.name()).size() > 0) {

    typeSpec.addAnnotation(AnnotationSpec.builder(JsonTypeInfo.class)
            .addMember("use", "$T.Id.NAME", JsonTypeInfo.class)
            .addMember("include", "$T.As.EXISTING_PROPERTY", JsonTypeInfo.class)
            .addMember("property", "$S", otr.discriminator()).build());

    AnnotationSpec.Builder subTypes = AnnotationSpec.builder(JsonSubTypes.class);
    for (CreationResult result : objectPluginContext.childClasses(ramlType.name())) {

      subTypes.addMember(
              "value",
              "$L",
              AnnotationSpec
                      .builder(JsonSubTypes.Type.class)
                      .addMember("value", "$L",
                              result.getJavaName(EventType.INTERFACE) + ".class").build());
    }

    subTypes.addMember(
            "value",
            "$L",
            AnnotationSpec
                    .builder(JsonSubTypes.Type.class)
                    .addMember("value", "$L",
                            objectPluginContext.creationResult().getJavaName(EventType.INTERFACE) + ".class").build());

    typeSpec.addAnnotation(subTypes.build());

  }

  if (otr.discriminatorValue() != null) {

    typeSpec.addAnnotation(AnnotationSpec.builder(JsonTypeName.class)
            .addMember("value", "$S", otr.discriminatorValue()).build());
  }


  if (!Annotations.ABSTRACT.get(otr)) {

    typeSpec.addAnnotation(AnnotationSpec.builder(JsonDeserialize.class)
            .addMember("as", "$T.class", objectPluginContext.creationResult().getJavaName(EventType.IMPLEMENTATION))
            .build());
  }


  return typeSpec;
}

Source File: InterfaceMetaInfo.java From raptor with Apache License 2.0

4 votes

public AnnotationSpec generateRequestMappingSpec() {
    AnnotationSpec.Builder builder = AnnotationSpec.builder(RequestMapping.class);
     OptionUtil.setAnnotationMember(builder, "path", "$S", servicePath);
     return builder.build();
}

Source File: MethodMetaInfo.java From raptor with Apache License 2.0

4 votes

public AnnotationSpec generateRaptorMethod() {
    AnnotationSpec.Builder builder = AnnotationSpec.builder(RaptorMethod.class);
    return builder.build();

}

Java Code Examples for com.squareup.javapoet.AnnotationSpec#builder()