generate method

@override Future<String> generate(LibraryReader library, BuildStep buildStep)

Generates Dart code for an input Dart library.

May create additional outputs through the buildStep, but the 'primary' output is Dart code returned through the Future. If there is nothing to generate for this library may return null, or a Future that resolves to null or the empty string.

Implementation

@override
Future<String> generate(LibraryReader library, BuildStep buildStep) async {
  final buffer = StringBuffer();

  const schemaChecker = TypeChecker.fromUrl(
    'package:phorm_annotations/src/annotations.dart#Schema',
  );
  var annotated = library.annotatedWith(schemaChecker);

  if (annotated.isEmpty) {
    // Fallback: manually check classes. Defensive — `annotatedWith` already
    // finds every @Schema class in practice, so this path is unreachable in
    // normal builds and excluded from coverage.
    // coverage:ignore-start
    final allAnnotated = <AnnotatedElement>[];
    for (final c in library.allElements.whereType<ClassElement>()) {
      final annotation = schemaChecker.firstAnnotationOf(c);
      if (annotation != null) {
        allAnnotated.add(AnnotatedElement(ConstantReader(annotation), c));
      }
    }
    annotated = allAnnotated;
    // coverage:ignore-end
  }

  if (annotated.isEmpty) {
    return '';
  }

  final fileName = p.basename(buildStep.inputId.path);
  buffer.writeln("part of '$fileName';\n");

  for (final annotatedElement in annotated) {
    final schemaResult = await _schemaGen.generateForAnnotatedElement(
      annotatedElement.element,
      annotatedElement.annotation,
      buildStep,
    );
    buffer.writeln(schemaResult);

    final mixinResult = _mixinGen.generateForAnnotatedElement(
      annotatedElement.element,
      annotatedElement.annotation,
      buildStep,
    );
    buffer.writeln(mixinResult);
  }

  final generatedContent = buffer.toString();
  final finalBuffer = StringBuffer()..write(generatedContent);

  // Only output _$PhormToJsonValue if it is referenced in the generated models
  if (generatedContent.contains(r'_$PhormToJsonValue')) {
    finalBuffer.writeln(r'''
dynamic _$PhormToJsonValue(dynamic value) {
if (value == null) return null;
if (value is DateTime) return value.toIso8601String();
if (value is bool) return value ? 1 : 0;
// Collections and Maps are stored as JSON strings in SQLite
if (value is List || value is Set || value is Map) {
  return jsonEncode(value is Set ? value.toList() : value);
}
return value;
}
''');
  }

  // Only output _$PhormDecodeJson if it is referenced in the generated models (e.g., collections deserialization)
  if (generatedContent.contains(r'_$PhormDecodeJson')) {
    finalBuffer.writeln(r'''
/// Decodes a value from SQLite storage.
/// JSON strings (from List/Set/Map fields) are decoded back to Dart objects.
dynamic _$PhormDecodeJson(dynamic value) {
if (value == null) return null;
if (value is String) {
  final trimmed = value.trimLeft();
  if (trimmed.startsWith('[') || trimmed.startsWith('{')) {
    try {
      return jsonDecode(value);
    } catch (_) {}
  }
}
return value;
}
''');
  }

  return finalBuffer.toString();
}