generateForAnnotatedElement method

  1. @override
String generateForAnnotatedElement(
  1. Element element,
  2. ConstantReader annotation,
  3. BuildStep buildStep
)

Implement to return source code to generate for element.

This method is invoked based on finding elements annotated with an instance of T. The annotation is provided as a ConstantReader.

Supported return values include a single String or multiple String instances within an Iterable or Stream. It is also valid to return a Future of String, Iterable, or Stream.

Implementations should return null when no content is generated. Empty or whitespace-only String instances are also ignored.

Implementation

@override
String generateForAnnotatedElement(
  Element element,
  ConstantReader annotation,
  BuildStep buildStep,
) {
  if (element is! ClassElement) {
    throw InvalidGenerationSourceError(
      '@DtoLog can only be applied to classes.',
      element: element,
    );
  }
  if (element.typeParameters.isNotEmpty) {
    throw InvalidGenerationSourceError(
      '@DtoLog does not support generic classes.',
      element: element,
    );
  }

  final className = element.name;
  final generateFromJson = annotation.read('generateFromJson').boolValue;
  final generateToJson = annotation.read('generateToJson').boolValue;
  final generateCopyWith = annotation.read('generateCopyWith').boolValue;
  final generateEquality = annotation.read('generateEquality').boolValue;
  final enableLogging = annotation.read('enableLogging').boolValue;

  final fields = _getFields(element);
  final buffer = StringBuffer();

  if (generateFromJson) {
    buffer.writeln(_generateFromJson(element, fields, enableLogging));
  }
  if (generateToJson) {
    final toJsonFields = fields.where((f) => !f.ignoreToJson).toList();
    buffer.writeln(_generateToJson(className, toJsonFields));
  }
  if (generateCopyWith) {
    buffer.writeln(_generateCopyWith(element, fields));
  }
  if (generateEquality) {
    buffer.writeln(_generateEquality(className, fields));
  }

  return buffer.toString();
}