private void emitResolvedOptionalSequence(TypeSpec.Builder intentBuilderTypeBuilder) { // find superclass final ClassName optionalSequence = get(target.classPackage, builderClassName(), OPTIONAL_SEQUENCE_CLASS); final ClassName resolvedOptional = get(target.classPackage, builderClassName(), RESOLVED_OPTIONAL_SEQUENCE_CLASS); TypeSpec.Builder resolvedOptionalSequenceBuilder = TypeSpec.classBuilder(RESOLVED_OPTIONAL_SEQUENCE_CLASS) .superclass(ParameterizedTypeName.get(optionalSequence, resolvedOptional)) .addModifiers(Modifier.PUBLIC) .addModifiers(Modifier.STATIC); MethodSpec.Builder constructorBuilder = MethodSpec.constructorBuilder() .addModifiers(Modifier.PUBLIC) .addParameter(Bundler.class, "bundler") .addParameter(get("android.content", "Intent"), "intent") .addStatement("super(bundler, intent)"); resolvedOptionalSequenceBuilder.addMethod(constructorBuilder.build()); intentBuilderTypeBuilder.addType(resolvedOptionalSequenceBuilder.build()); }
private void addUsedTypeConverterMethods(TypeSpec.Builder builder) { Set<TypeName> usedTypeConverters = new HashSet<>(); for (JsonFieldHolder holder : mJsonObjectHolder.fieldMap.values()) { usedTypeConverters.addAll(holder.type.getUsedTypeConverters()); } for (TypeName usedTypeConverter : usedTypeConverters) { final String variableName = getTypeConverterVariableName(usedTypeConverter); builder.addField(FieldSpec.builder(ParameterizedTypeName.get(ClassName.get(TypeConverter.class), usedTypeConverter), variableName) .addModifiers(Modifier.PRIVATE, Modifier.STATIC) .build() ); builder.addMethod(MethodSpec.methodBuilder(getTypeConverterGetter(usedTypeConverter)) .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL) .returns(ParameterizedTypeName.get(ClassName.get(TypeConverter.class), usedTypeConverter)) .beginControlFlow("if ($L == null)", variableName) .addStatement("$L = $T.typeConverterFor($T.class)", variableName, LoganSquare.class, usedTypeConverter) .endControlFlow() .addStatement("return $L", variableName) .build() ); } }
private void createConfigClass(@NonNull List<Element> elements) throws IOException { final TypeSpec.Builder classBuilder = TypeSpec.classBuilder(configName) .addSuperinterface(Serializable.class) .addSuperinterface(org.acra.config.Configuration.class) .addModifiers(Modifier.PUBLIC, Modifier.FINAL); Strings.addClassJavadoc(classBuilder, ClassName.get(baseAnnotation.asType())); final MethodSpec.Builder constructor = MethodSpec.constructorBuilder().addModifiers(Modifier.PUBLIC) .addParameter(ParameterSpec.builder(ClassName.get(PACKAGE, builderName), PARAM_0).addAnnotation(Types.NON_NULL).build()); elements.stream().filter(ConfigElement.class::isInstance).map(ConfigElement.class::cast).forEach(element -> element.addToConfig(classBuilder, constructor)); classBuilder.addMethod(constructor.build()); Strings.writeClass(processingEnv.getFiler(), classBuilder.build()); }
private void onDraw(TypeSpec.Builder typeBuilder) { MethodSpec methodSpec = MethodSpec.methodBuilder("onDraw") .addAnnotation(Override.class) .addModifiers(Modifier.PUBLIC) .addParameter(ClassName.get("android.graphics", "Canvas"), "canvas") .addStatement("super.onDraw(canvas)") .addStatement("mBadgeViewHelper.drawBadge(canvas)") .build(); typeBuilder.addMethod(methodSpec); }
private TypeSpec enumAdapter(ClassName javaType, ClassName adapterJavaType) { return TypeSpec.classBuilder(adapterJavaType.simpleName()) .superclass(enumAdapterOf(javaType)) .addModifiers(PRIVATE, STATIC, FINAL) .addMethod(MethodSpec.constructorBuilder() .addStatement("super($T.class)", javaType) .build()) .addMethod(MethodSpec.methodBuilder("fromValue") .addAnnotation(Override.class) .addModifiers(PROTECTED) .returns(javaType) .addParameter(int.class, "value") .addStatement("return $T.fromValue(value)", javaType) .build()) .build(); }
Strings.lowercaseFirstLetter(functionName) + "EventFlowable"; ParameterizedTypeName parameterizedTypeName = ParameterizedTypeName.get(ClassName.get(Flowable.class), ClassName.get("", responseClassName)); MethodSpec.methodBuilder(generatedFunctionName) .addModifiers(Modifier.PUBLIC) .addParameter(EthFilter.class, FILTER) .returns(parameterizedTypeName); TypeSpec converter = TypeSpec.anonymousClassBuilder("") .addSuperinterface(ParameterizedTypeName.get( ClassName.get(io.reactivex.functions.Function.class), ClassName.get(Log.class), ClassName.get("", responseClassName))) .addMethod(MethodSpec.methodBuilder("apply") .addAnnotation(Override.class) .addModifiers(Modifier.PUBLIC) .addStatement("return typedResponse") .build()) .build();
@Override public TypeSpec poetSpec() { MethodSpec.Builder constructorBuilder = MethodSpec.constructorBuilder() .addModifiers(Modifier.PROTECTED) .addParameter(className().nestedClass("Builder"), "builder") .addStatement("super(builder)"); TypeSpec.Builder classBuilder = TypeSpec.classBuilder(className()) .addAnnotation(PoetUtils.generatedAnnotation()) .addModifiers(Modifier.PUBLIC, Modifier.ABSTRACT) .superclass(ClassName.get(AwsResponse.class)) .addType(builderInterfaceSpec()) .addType(builderImplSpec()); addResponseMetadata(classBuilder, constructorBuilder); classBuilder.addMethod(constructorBuilder.build()); return classBuilder.build(); }
@Override public void enterRule(Rule rule) { // generates a new ephemeral unique class name for each generated rule. Only valid for the runtime of the jvm classFile = TypeSpec.classBuilder("rule$" + rule.id()) .addSuperinterface(GeneratedRule.class) .addModifiers(Modifier.FINAL, Modifier.PUBLIC) .addAnnotation(AnnotationSpec.builder(SuppressWarnings.class) .addMember("value", "$S", "unchecked") .build() ) .addMethod(MethodSpec.methodBuilder("name") .returns(String.class) .addAnnotation(Override.class) .addModifiers(Modifier.PUBLIC) .addStatement("return $S", rule.name()) .build() ); constructorBuilder = MethodSpec.constructorBuilder() .addModifiers(Modifier.PUBLIC) .addParameter(FunctionRegistry.class, "functionRegistry"); lateConstructorBlock = CodeBlock.builder(); hoistedConstantExpressions = CodeBlock.builder(); }
@Override protected TypeSpec.Builder createClassBuilder() { ClassName parentClassName = getClassDetail().getGeneratedClassName(); ClassName generatedClassName = ClassName.get( getClassDetail().getPackageName(), getClassDetail().getTypeName() + BindingSuffix.CLASS_LOAD_MORE_VIEW_BINDER_SUFFIX); ClassName loadMoreCallbackClassName = ClassName.get( NameStore.Package.PLACE_HOLDER_VIEW, NameStore.Class.LOAD_MORE_CALLBACK_BINDER); return TypeSpec.classBuilder(generatedClassName) .superclass(parentClassName) .addSuperinterface(ParameterizedTypeName.get(loadMoreCallbackClassName, TypeVariableName.get(getClassDetail().getTypeElement().asType()))) .addModifiers(Modifier.PUBLIC) .addAnnotation(Keep.class); }
static TypeSpec generateStateContainer(SpecModel specModel) { final TypeSpec.Builder stateContainerClassBuilder = TypeSpec.classBuilder(getStateContainerClassName(specModel)) .addSuperinterface(specModel.getStateContainerClass()) .addAnnotation( AnnotationSpec.builder(VisibleForTesting.class) .addMember("otherwise", "$L", VisibleForTesting.PRIVATE) .build()) .addModifiers(Modifier.STATIC) .addTypeVariables(specModel.getTypeVariables()); stateContainerClassBuilder.addSuperinterface(specModel.getTransitionContainerClass()); stateContainerClassBuilder.addField( FieldSpec.builder(stateValue.getTypeName(), stateValue.getName()) .addAnnotation(State.class) stateContainerClassBuilder.addField( FieldSpec.builder( ParameterizedTypeName.get(ClassNames.LIST, specModel.getTransitionClass().box()), GeneratorConstants.STATE_TRANSITIONS_FIELD_NAME) .initializer("new $T<>()", ClassNames.ARRAY_LIST) stateContainerClassBuilder.addMethod( MethodSpec.methodBuilder("consumeTransitions") .addModifiers(Modifier.PUBLIC) .addAnnotation(Override.class) .returns( ParameterizedTypeName.get(ClassNames.LIST, specModel.getTransitionClass().box())) .addCode(
private void isDragging(TypeSpec.Builder typeBuilder) { MethodSpec methodSpec = MethodSpec.methodBuilder("isDragging") .addAnnotation(Override.class) .addModifiers(Modifier.PUBLIC) .addStatement("return mBadgeViewHelper.isDragging()") .returns(boolean.class) .build(); typeBuilder.addMethod(methodSpec); }
/** * Write a companion class for {@code type} that extends {@link StaticInjection}. */ private void generateStaticInjection(TypeElement type, List<Element> fields) throws IOException { ClassName typeName = ClassName.get(type); ClassName adapterClassName = adapterName(ClassName.get(type), STATIC_INJECTION_SUFFIX); TypeSpec.Builder result = TypeSpec.classBuilder(adapterClassName.simpleName()) .addOriginatingElement(type) .addJavadoc(AdapterJavadocs.STATIC_INJECTION_TYPE, type) .addModifiers(PUBLIC, FINAL) .superclass(StaticInjection.class); for (Element field : fields) { result.addField(memberBindingField(false, field)); } result.addMethod(attachMethod(null, fields, false, typeName, null, true)); result.addMethod(staticInjectMethod(fields, typeName)); String packageName = getPackage(type).getQualifiedName().toString(); JavaFile javaFile = JavaFile.builder(packageName, result.build()) .addFileComment(AdapterJavadocs.GENERATED_BY_DAGGER) .build(); javaFile.writeTo(processingEnv.getFiler()); }
TypeSpec.Builder builder = TypeSpec.enumBuilder(javaType.simpleName()) .addModifiers(PUBLIC) .addSuperinterface(WireEnum.class); builder.addJavadoc("$L\n", sanitizeJavadoc(type.documentation())); builder.addField(TypeName.INT, value, PRIVATE, FINAL); if (allOptionFieldsBuilder.add(protoMember)) { TypeName optionJavaType = typeName(optionField.type()); builder.addField(optionJavaType, optionField.name(), PUBLIC, FINAL); constructorBuilder.addParameter(optionJavaType, optionField.name()); builder.addMethod(constructorBuilder.build()); TypeSpec.Builder constantBuilder = TypeSpec.anonymousClassBuilder(enumArgsFormat, enumArgs); if (!constant.documentation().isEmpty()) { constantBuilder.addJavadoc("$L\n", sanitizeJavadoc(constant.documentation())); constantBuilder.addAnnotation(Deprecated.class); builder.addEnumConstant(constant.name(), constantBuilder.build()); builder.addMethod(fromValueBuilder.addStatement("default: return null") builder.addField(adapterBuilder.build()); builder.addField(options);
private void generate(Set<String> viewClassSet) throws IllegalAccessException, IOException { mMessager.printMessage(Diagnostic.Kind.NOTE, "生成 " + viewClassSet.size() + " 个"); for (String clazz : viewClassSet) { int lastDotIndex = clazz.lastIndexOf("."); String superPackageName = clazz.substring(0, lastDotIndex); String superClassName = clazz.substring(lastDotIndex + 1); String className = CLASS_PREFIX + superClassName; mMessager.printMessage(Diagnostic.Kind.NOTE, clazz + " ====> " + className); TypeSpec.Builder typeBuilder = TypeSpec.classBuilder(className) .addJavadoc(CLASS_JAVA_DOC) .addModifiers(Modifier.PUBLIC) .superclass(ClassName.get(superPackageName, superClassName)) .addSuperinterface(ClassName.get(PACKAGE_NAME, "BGABadgeable")) .addField(ClassName.get(PACKAGE_NAME, "BGABadgeViewHelper"), "mBadgeViewHelper", Modifier.PRIVATE); generateMethod(typeBuilder, clazz); JavaFile javaFile = JavaFile.builder(PACKAGE_NAME, typeBuilder.build()).build(); javaFile.writeTo(mFileUtils); } }
private static <T extends SpecModel & HasEnclosedSpecModel> TypeSpecDataHolder generateBuilder( final T specModel) { final TypeSpec.Builder propsBuilderClassBuilder = TypeSpec.classBuilder(BUILDER) .addModifiers(Modifier.PUBLIC, Modifier.STATIC) .superclass( ParameterizedTypeName.get(ClassNames.BASE_MATCHER, getMatcherType(specModel))) .addField( FieldSpec.builder( ClassNames.RESOURCE_RESOLVER, RESOURCE_RESOLVER, Modifier.PROTECTED) propsBuilderClassBuilder.addTypeVariables(specModel.getTypeVariables()); MethodSpec.constructorBuilder() .addParameter(specModel.getContextClass(), "c") .addStatement("$L = new $T(c)", RESOURCE_RESOLVER, ClassNames.RESOURCE_RESOLVER) .build(); propsBuilderClassBuilder.addMethod(constructor); .addMethod(generateBuildMethod(specModel)) .addMethod(generateGetThisMethod(specModel)); return TypeSpecDataHolder.newBuilder().addType(propsBuilderClassBuilder.build()).build();
static TypeSpec generatePreviousRenderDataContainerImpl(SpecModel specModel) { final String className = RenderDataGenerator.getRenderDataImplClassName(specModel); final TypeSpec.Builder renderInfoClassBuilder = TypeSpec.classBuilder(className).addSuperinterface(ClassNames.RENDER_DATA); renderInfoClassBuilder.addModifiers(Modifier.STATIC, Modifier.PRIVATE); renderInfoClassBuilder.addTypeVariables(specModel.getTypeVariables()); final MethodSpec.Builder copyBuilder = MethodSpec.methodBuilder("copy") .addParameter(ClassName.bestGuess(className), copyParamName); final MethodSpec.Builder recordBuilder = MethodSpec.methodBuilder("record") .addParameter(specModel.getComponentTypeName(), recordParamName); renderInfoClassBuilder.addField(FieldSpec.builder( modelToDiff.getTypeName(), name).addAnnotation(Prop.class).build()); } else { renderInfoClassBuilder.addField(FieldSpec.builder( modelToDiff.getTypeName(), modelToDiff.getName()).addAnnotation(State.class).build()); copyBuilder.addStatement("$L = $L.$L", name, copyParamName, name); recordBuilder.addStatement( "$L = $L.$L", .addMethod(copyBuilder.build()) .addMethod(recordBuilder.build()) .build();
private TypeSpec generateEnclosingType(EnclosingType type) { ClassName javaType = (ClassName) typeName(type.type()); TypeSpec.Builder builder = TypeSpec.classBuilder(javaType.simpleName()) .addModifiers(PUBLIC, FINAL); if (javaType.enclosingClassName() != null) { builder.addModifiers(STATIC); } String documentation = type.documentation(); if (!documentation.isEmpty()) { documentation += "\n\n<p>"; } documentation += "<b>NOTE:</b> This type only exists to maintain class structure" + " for its nested types and is not an actual message."; builder.addJavadoc("$L\n", documentation); builder.addMethod(MethodSpec.constructorBuilder() .addModifiers(PRIVATE) .addStatement("throw new $T()", AssertionError.class) .build()); for (Type nestedType : type.nestedTypes()) { builder.addType(generateType(nestedType)); } return builder.build(); }
private TypeSpec.Builder rotateBuilderState( TypeSpec.Builder builder, TypeSpec.Builder builderStateClass, String nextStateClassName) { if (builderStateClass != builder) { builder.addType(builderStateClass.build()); } //prepare next state class builderStateClass = TypeSpec.classBuilder(nextStateClassName).addModifiers(Modifier.PUBLIC); return builderStateClass; }
@Override protected TypeSpec.Builder createClassBuilder() { return TypeSpec.classBuilder(getClassDetail().getGeneratedClassName()) .superclass(ParameterizedTypeName.get( getClassDetail().getSuperClassName(), TypeVariableName.get(getClassDetail().getTypeElement().asType()), getClassDetail().getFrameViewClassName(), getClassDetail().getSwipeOptionClassName(), getClassDetail().getSwipeDecorClassName())) .addModifiers(Modifier.PUBLIC) .addAnnotation(Keep.class); }
private void addUsedJsonMapperVariables(TypeSpec.Builder builder) { Set<ClassNameObjectMapper> usedJsonObjectMappers = new HashSet<>(); for (JsonFieldHolder holder : mJsonObjectHolder.fieldMap.values()) { usedJsonObjectMappers.addAll(holder.type.getUsedJsonObjectMappers()); } for (ClassNameObjectMapper usedJsonObjectMapper : usedJsonObjectMappers) { builder.addField(FieldSpec.builder(ParameterizedTypeName.get(ClassName.get(JsonMapper.class), usedJsonObjectMapper.className), getMapperVariableName(usedJsonObjectMapper.objectMapper)) .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL) .initializer("$T.mapperFor($T.class)", LoganSquare.class, usedJsonObjectMapper.className) .build() ); } }